Hello-Agents: Datawhale 开源 AI 智能体教程如何帮你从零构建生产级 Agent

# 在线阅读
# https://datawhalechina.github.io/hello-agents/

# 本地搭建
git clone https://github.com/datawhalechina/hello-agents.git
cd hello-agents
# 参考 Extra-Chapter/07 配置环境

# 运行第4章 ReAct Agent
python code/chapter4/react_agent.py

import openai
import json

tools = [
    {
        "type": "function",
        "function": {
            "name": "search_web",
            "description": "搜索网络信息",
            "parameters": {
                "type": "object",
                "properties": {"query": {"type": "string"}},
                "required": ["query"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "calculate",
            "description": "执行数学计算",
            "parameters": {
                "type": "object",
                "properties": {"expression": {"type": "string"}},
                "required": ["expression"]
            }
        }
    }
]

def search_web(query): return f"搜索结果: {query}"
def calculate(expr): return str(eval(expr))

messages = [
    {"role": "system", "content": "你是一个有用的助手，需要时使用工具。"},
    {"role": "user", "content": "东京人口除以1000是多少？"}
]

for step in range(5):
    response = openai.chat.completions.create(
        model="gpt-4", messages=messages, tools=tools, tool_choice="auto"
    )
    message = response.choices[0].message
    messages.append(message)
    
    if message.tool_calls:
        for tc in message.tool_calls:
            name = tc.function.name
            args = json.loads(tc.function.arguments)
            result = search_web(**args) if name == "search_web" else calculate(**args)
            messages.append({"role": "tool", "tool_call_id": tc.id, "content": result})
    else:
        print("最终答案:", message.content)
        break