1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

import json
from openai import OpenAI

client = OpenAI(base_url="https://api.deepseek.com", api_key="你的DeepSeek API Key")  # 去 platform.deepseek.com 注册获取

# --- 两个工具函数 ---
def get_weather(city: str) -> str:
    """查询指定城市的当前天气"""
    db = {"北京": "晴 28°C", "上海": "多云 25°C", "深圳": "雷阵雨 31°C"}
    return db.get(city, f"暂无{city}的天气数据")

def get_aqi(city: str) -> str:
    """查询指定城市的空气质量指数（AQI）"""
    db = {"北京": "85（轻度污染）", "上海": "42（优）", "深圳": "30（优）"}
    return db.get(city, f"暂无{city}的 AQI 数据")

# --- 工具的 JSON Schema ---
tools = [
    {
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "查询指定城市的当前天气",
            "parameters": {
                "type": "object",
                "properties": {"city": {"type": "string", "description": "城市名"}},
                "required": ["city"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_aqi",
            "description": "查询指定城市的空气质量指数 AQI",
            "parameters": {
                "type": "object",
                "properties": {"city": {"type": "string", "description": "城市名"}},
                "required": ["city"],
            },
        },
    },
]

available_functions = {"get_weather": get_weather, "get_aqi": get_aqi}

# --- 对话历史：system 先定调，再上用户问题 ---
messages = [
    {"role": "system", "content": "你是一个帮用户做决策的助手。需要数据时自己调工具，信息够了就直接给结论，不要废话。"},
    {"role": "user", "content": "北京今天适合户外跑步吗？需要什么数据你自己查。"},
]

# --- 这就是 Agent 循环 ---
MAX_TURNS = 5  # 防止模型死循环，最多转 5 圈
for turn in range(MAX_TURNS):
    print(f"\n===== 第 {turn + 1} 轮 =====")

    resp = client.chat.completions.create(
        model="deepseek-v4-flash",
        messages=messages,
        tools=tools,
    )
    msg = resp.choices[0].message
    messages.append(msg.model_dump(exclude_none=True))

    # 模型不再要求调工具 = 任务完成
    if not msg.tool_calls:
        print(msg.content)
        break

    # 否则，把这一轮要求的工具全执行掉
    for call in msg.tool_calls:
        fn_name = call.function.name
        fn_args = json.loads(call.function.arguments)
        print(f"[tool] {fn_name}({fn_args})")
        try:
            result = available_functions[fn_name](**fn_args)
        except Exception as e:
            result = f"工具执行失败：{type(e).__name__}: {e}"
        messages.append({
            "role": "tool",
            "tool_call_id": call.id,
            "content": result,
        })
else:
    # for 循环正常结束 = 达到 MAX_TURNS 还没停
    print("（达到最大轮次，强制停止）")

1
2
3
4
5
6
7

===== 第 1 轮 =====
[tool] get_weather({'city': '北京'})
[tool] get_aqi({'city': '北京'})

===== 第 2 轮 =====
今天北京晴，气温 28°C，适合户外跑步；但 AQI 85 属于轻度污染，
敏感人群建议改为室内运动。