1
"""测试 OpenAI SDK 完整交互流程（含工具结果回传）。"""
2
import os
3
import json
4
from datetime import datetime
5
from pathlib import Path
6

7
from dotenv import load_dotenv
8
from openai import OpenAI
9

10
ROOT = Path(__file__).parent.parent
11
load_dotenv(ROOT / ".env")
12

13
LOG_DIR = ROOT / "logs" / "sdk"
14
LOG_DIR.mkdir(parents=True, exist_ok=True)
15

16
client = OpenAI(
17
    api_key=os.getenv("DEEPSEEK_API_KEY"),
18
    base_url="https://api.deepseek.com/v1",
19
)
20

21
MODEL = "deepseek-chat"
22

23
tools = [{
24
    "type": "function",
25
    "function": {
26
        "name": "get_weather",
27
        "description": "获取指定城市的天气",
28
        "parameters": {
29
            "type": "object",
30
            "properties": {"city": {"type": "string", "description": "城市名"}},
31
            "required": ["city"],
32
        },
33
    },
34
}]
35

36

37
def mock_execute_tool(name: str, args: dict) -> str:
38
    if name == "get_weather":
39
        return f"{args.get('city', '未知')}：晴天，25°C，微风"
40
    return f"未知工具: {name}"
41

42

43
results = {}
44

45
# ── 场景 A：纯文本 ──
46
r = client.chat.completions.create(
47
    model=MODEL,
48
    max_tokens=256,
49
    messages=[{"role": "user", "content": "用一句话介绍你自己"}],
50
)
51
results["A_纯文本"] = r.model_dump()
52

53
# ── 场景 B：工具调用（只看第一轮）──
54
r = client.chat.completions.create(
55
    model=MODEL,
56
    max_tokens=256,
57
    tools=tools,
58
    messages=[{"role": "user", "content": "北京今天天气怎么样"}],
59
)
60
results["B_工具调用"] = r.model_dump()
61

62
# ── 场景 C：完整 ReAct 两轮 ──
63
messages = [{"role": "user", "content": "北京今天天气怎么样"}]
64
r1 = client.chat.completions.create(model=MODEL, max_tokens=256, tools=tools, messages=messages)
65
msg1 = r1.choices[0].message
66

67
# 回传：先 append assistant 消息，再逐个 append tool 结果
68
messages.append(msg1.model_dump())
69
for tc in msg1.tool_calls:
70
    args = json.loads(tc.function.arguments)
71
    result = mock_execute_tool(tc.function.name, args)
72
    messages.append({
73
        "role": "tool",
74
        "tool_call_id": tc.id,
75
        "content": result,
76
    })
77

78
# 第 2 轮：把工具结果发给 LLM
79
r2 = client.chat.completions.create(model=MODEL, max_tokens=256, tools=tools, messages=messages)
80
results["C_完整两轮"] = {
81
    "第1轮_工具调用": r1.model_dump(),
82
    "回传的messages": messages,
83
    "第2轮_最终回复": r2.model_dump(),
84
}
85

86
# ── 写入日志 ──
87
date_str = datetime.now().strftime("%Y-%m-%d")
88
log_file = LOG_DIR / f"{date_str}-openai-response.json"
89
log_file.write_text(json.dumps(results, indent=2, ensure_ascii=False, default=str), encoding="utf-8")

1
"""测试 Anthropic SDK 完整交互流程（含工具结果回传）。"""
2
import os
3
import json
4
from datetime import datetime
5
from pathlib import Path
6

7
from dotenv import load_dotenv
8
from anthropic import Anthropic
9

10
ROOT = Path(__file__).parent.parent
11
load_dotenv(ROOT / ".env")
12

13
LOG_DIR = ROOT / "logs" / "sdk"
14
LOG_DIR.mkdir(parents=True, exist_ok=True)
15

16
client = Anthropic(
17
    api_key=os.getenv("DEEPSEEK_API_KEY"),
18
    base_url="https://api.deepseek.com/anthropic",
19
)
20

21
MODEL = "deepseek-chat"
22

23
tools = [{
24
    "name": "get_weather",
25
    "description": "获取指定城市的天气",
26
    "input_schema": {
27
        "type": "object",
28
        "properties": {"city": {"type": "string", "description": "城市名"}},
29
        "required": ["city"],
30
    },
31
}]
32

33

34
def mock_execute_tool(name: str, args: dict) -> str:
35
    if name == "get_weather":
36
        return f"{args.get('city', '未知')}：晴天，25°C，微风"
37
    return f"未知工具: {name}"
38

39

40
results = {}
41

42
# ── 场景 A：纯文本 ──
43
r = client.messages.create(
44
    model=MODEL,
45
    max_tokens=256,
46
    messages=[{"role": "user", "content": "用一句话介绍你自己"}],
47
)
48
results["A_纯文本"] = r.model_dump()
49

50
# ── 场景 B：工具调用（只看第一轮）──
51
r = client.messages.create(
52
    model=MODEL,
53
    max_tokens=256,
54
    tools=tools,
55
    messages=[{"role": "user", "content": "北京今天天气怎么样"}],
56
)
57
results["B_工具调用"] = r.model_dump()
58

59
# ── 场景 C：完整 ReAct 两轮 ──
60
messages = [{"role": "user", "content": "北京今天天气怎么样"}]
61
r1 = client.messages.create(model=MODEL, max_tokens=256, tools=tools, messages=messages)
62

63
# 回传：先 append assistant 消息（content 块数组），再打包所有 tool_result
64
messages.append({"role": "assistant", "content": r1.content})
65
tool_results = []
66
for block in r1.content:
67
    if block.type == "tool_use":
68
        result = mock_execute_tool(block.name, block.input)
69
        tool_results.append({
70
            "type": "tool_result",
71
            "tool_use_id": block.id,
72
            "content": result,
73
        })
74
messages.append({"role": "user", "content": tool_results})
75

76
# 第 2 轮：把工具结果发给 LLM
77
r2 = client.messages.create(model=MODEL, max_tokens=256, tools=tools, messages=messages)
78
results["C_完整两轮"] = {
79
    "第1轮_工具调用": r1.model_dump(),
80
    "回传的messages": str(messages),
81
    "第2轮_最终回复": r2.model_dump(),
82
}
83

84
# ── 写入日志 ──
85
date_str = datetime.now().strftime("%Y-%m-%d")
86
log_file = LOG_DIR / f"{date_str}-anthropic-response.json"
87
log_file.write_text(json.dumps(results, indent=2, ensure_ascii=False, default=str), encoding="utf-8")

1
{
2
  "A_纯文本": {
3
    "id": "f38b2f23-16a1-48b9-837e-4c2238a6432c",
4
    "choices": [{
5
      "finish_reason": "stop",
6
      "index": 0,
7
      "message": {
8
        "role": "assistant",
9
        "content": "我是一名人造智能助手，旨在通过实时信息检索和多语言支持，为用户提供高效、准确且友好的帮助。",
10
        "tool_calls": null
11
      }
12
    }],
13
    "model": "deepseek-v4-flash",
14
    "usage": {
15
      "prompt_tokens": 8,
16
      "completion_tokens": 25,
17
      "total_tokens": 33
18
    }
19
  },
20
  "B_工具调用": {
21
    "id": "279e0d46-ca7e-4495-806b-e3d7449aa465",
22
    "choices": [{
23
      "finish_reason": "tool_calls",
24
      "index": 0,
25
      "message": {
26
        "role": "assistant",
27
        "content": "让我查一下北京今天的天气情况。",
28
        "tool_calls": [{
29
          "id": "call_00_8WcIl14wEDOeacZ0n8xL9829",
30
          "type": "function",
31
          "function": {
32
            "name": "get_weather",
33
            "arguments": "{\"city\": \"北京\"}"
34
          }
35
        }]
36
      }
37
    }],
38
    "model": "deepseek-v4-flash",
39
    "usage": {
40
      "prompt_tokens": 282,
41
      "completion_tokens": 52,
42
      "total_tokens": 334
43
    }
44
  },
45
  "C_完整两轮": {
46
    "第1轮_工具调用": {
47
      "choices": [{
48
        "finish_reason": "tool_calls",
49
        "message": {
50
          "content": "让我查询一下北京今天的天气情况。",
51
          "tool_calls": [{
52
            "id": "call_00_Sh0T5t8PXiCWHA6Gcr3w7264",
53
            "function": {
54
              "name": "get_weather",
55
              "arguments": "{\"city\": \"北京\"}"
56
            }
57
          }]
58
        }
59
      }]
60
    },
61
    "回传的messages": [
62
      {"role": "user", "content": "北京今天天气怎么样"},
63
      {
64
        "role": "assistant",
65
        "tool_calls": [{
66
          "id": "call_00_Sh0T5t8PXiCWHA6Gcr3w7264",
67
          "function": {"name": "get_weather", "arguments": "{\"city\": \"北京\"}"}
68
        }]
69
      },
70
      {
71
        "role": "tool",
72
        "tool_call_id": "call_00_Sh0T5t8PXiCWHA6Gcr3w7264",
73
        "content": "北京：晴天，25°C，微风"
74
      }
75
    ],
76
    "第2轮_最终回复": {
77
      "choices": [{
78
        "finish_reason": "stop",
79
        "message": {
80
          "content": "北京今天天气很好！具体情况如下：\n\n- **天气状况**：晴天\n- **气温**：25°C\n- **风力**：微风",
81
          "tool_calls": null
82
        }
83
      }],
84
      "usage": {
85
        "prompt_tokens": 354,
86
        "completion_tokens": 69,
87
        "total_tokens": 423
88
      }
89
    }
90
  }
91
}

1
{
2
  "A_纯文本": {
3
    "id": "b12f46f3-b005-4b75-9a3e-b3cc98001d2e",
4
    "role": "assistant",
5
    "stop_reason": "end_turn",
6
    "content": [
7
      {"type": "text", "text": "我是你身边的AI伙伴，随时准备用精准又温暖的语言为你答疑解惑、碰撞灵感。"}
8
    ],
9
    "model": "deepseek-v4-flash",
10
    "usage": {
11
      "input_tokens": 8,
12
      "output_tokens": 20
13
    }
14
  },
15
  "B_工具调用": {
16
    "id": "2e6c9b04-dea3-461a-b4c9-e68a7b8254da",
17
    "role": "assistant",
18
    "stop_reason": "tool_use",
19
    "content": [
20
      {"type": "text", "text": "让我查一下北京的天气情况。"},
21
      {
22
        "type": "tool_use",
23
        "id": "call_00_DnX2SEF9esVcv3HRBefj1699",
24
        "name": "get_weather",
25
        "input": {"city": "北京"}
26
      }
27
    ],
28
    "model": "deepseek-v4-flash",
29
    "usage": {
30
      "input_tokens": 26,
31
      "output_tokens": 51
32
    }
33
  },
34
  "C_完整两轮": {
35
    "第1轮_工具调用": {
36
      "role": "assistant",
37
      "stop_reason": "tool_use",
38
      "content": [
39
        {"type": "text", "text": "让我查一下北京的天气情况。"},
40
        {
41
          "type": "tool_use",
42
          "id": "call_00_ORUjAw43qGE5D9xUlpTI2193",
43
          "name": "get_weather",
44
          "input": {"city": "北京"}
45
        }
46
      ]
47
    },
48
    "回传的messages": [
49
      {"role": "user", "content": "北京今天天气怎么样"},
50
      {
51
        "role": "assistant",
52
        "content": [
53
          {"type": "text", "text": "让我查一下北京的天气情况。"},
54
          {"type": "tool_use", "id": "call_00_ORUjAw43qGE5D9xUlpTI2193", "name": "get_weather", "input": {"city": "北京"}}
55
        ]
56
      },
57
      {
58
        "role": "user",
59
        "content": [
60
          {"type": "tool_result", "tool_use_id": "call_00_ORUjAw43qGE5D9xUlpTI2193", "content": "北京：晴天，25°C，微风"}
61
        ]
62
      }
63
    ],
64
    "第2轮_最终回复": {
65
      "role": "assistant",
66
      "stop_reason": "end_turn",
67
      "content": [
68
        {"type": "text", "text": "北京今天天气不错！具体情况如下：\n- 天气状况：晴天\n- 温度：25°C\n- 风力：微风"}
69
      ],
70
      "usage": {
71
        "input_tokens": 97,
72
        "output_tokens": 53
73
      }
74
    }
75
  }
76
}