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

 1# 在线阅读
 2# https://datawhalechina.github.io/hello-agents/
 3
 4# 本地搭建
 5git clone https://github.com/datawhalechina/hello-agents.git
 6cd hello-agents
 7# 参考 Extra-Chapter/07 配置环境
 8
 9# 运行第4章 ReAct Agent
10python code/chapter4/react_agent.py

 1import openai
 2import json
 3
 4tools = [
 5    {
 6        "type": "function",
 7        "function": {
 8            "name": "search_web",
 9            "description": "搜索网络信息",
10            "parameters": {
11                "type": "object",
12                "properties": {"query": {"type": "string"}},
13                "required": ["query"]
14            }
15        }
16    },
17    {
18        "type": "function",
19        "function": {
20            "name": "calculate",
21            "description": "执行数学计算",
22            "parameters": {
23                "type": "object",
24                "properties": {"expression": {"type": "string"}},
25                "required": ["expression"]
26            }
27        }
28    }
29]
30
31def search_web(query): return f"搜索结果: {query}"
32def calculate(expr): return str(eval(expr))
33
34messages = [
35    {"role": "system", "content": "你是一个有用的助手，需要时使用工具。"},
36    {"role": "user", "content": "东京人口除以1000是多少？"}
37]
38
39for step in range(5):
40    response = openai.chat.completions.create(
41        model="gpt-4", messages=messages, tools=tools, tool_choice="auto"
42    )
43    message = response.choices[0].message
44    messages.append(message)
45    
46    if message.tool_calls:
47        for tc in message.tool_calls:
48            name = tc.function.name
49            args = json.loads(tc.function.arguments)
50            result = search_web(**args) if name == "search_web" else calculate(**args)
51            messages.append({"role": "tool", "tool_call_id": tc.id, "content": result})
52    else:
53        print("最终答案:", message.content)
54        break