AgentBench 使用与启动教程
1. 项目介绍
AgentBench 是由 THUDM 开发的一个综合性评测框架,旨在评估大型语言模型(LLM)作为智能体在不同环境下的性能。该项目包含多个不同的环境,以全面评估 LLM 的自主操作能力,这些环境包括操作系统、数据库、知识图谱、数字卡牌游戏以及横向思维谜题等。
2. 项目快速启动
以下是快速启动 AgentBench 的步骤:
步骤 1: 准备环境
首先,克隆仓库并安装所需的依赖项。
cd AgentBench
conda create -n agent-bench python=3.9
conda activate agent-bench
pip install -r requirements.txt
确保已经正确安装了 Docker。
docker ps
步骤 2: 配置智能体
在 configs/agents/openai-chat.yaml
文件中填写你的 OpenAI API Key。你可以使用以下命令来检查智能体是否配置正确。
python -m src.client.agent_test
默认情况下会启动 gpt-3.5-turbo-0613
。你可以通过修改参数来使用其他智能体。
python -m src.client.agent_test --config configs/agents/api_agents.yaml --agent gpt-3.5-turbo-0613
步骤 3: 启动任务服务器
启动任务服务器需要执行特定的脚本。假设端口 5000 到 5015 是可用的,可以使用以下命令来自动启动。
python -m src.start_task -a
执行此命令后,大约需要等待 1 分钟以完成任务设置。
步骤 4: 启动任务分配器
当任务服务器准备好后,可以在另一个终端中启动任务分配器。
python -m src.assigner
3. 应用案例和最佳实践
AgentBench 的使用案例涵盖了多种不同的任务,例如在数据库环境中执行查询,或在操作系统中执行命令。最佳实践包括为不同任务配置合适的智能体,以及根据任务需求调整参数。
4. 典型生态项目
AgentBench 的生态系统包括了多个相关的项目,例如:
- AvalonBench:一个多智能体框架的实现。
- VisualAgentBench:专为评估和训练基于大型多模态模型(LMMs)的视觉基础智能体而设计。
以上是 AgentBench 的基本使用和启动教程,你可以根据具体需求进行相应的配置和扩展。