修复Agent在处理信息格式时的错误

This commit is contained in:
jackyzzy
2025-11-05 11:27:06 +08:00
parent 3afabec1a8
commit 1c0a0c22e1

View File

@@ -51,7 +51,24 @@ class Agent:
stream=False,
)
if response.choices[0].message.tool_calls:
self.messages.append({"role": "assistant", "content": response.choices[0].message.content})
# 将包含 tool_calls 的完整 assistant 消息添加到历史中
assistant_message = {
"role": "assistant",
"content": response.choices[0].message.content,
"tool_calls": [
{
"id": tool_call.id,
"type": "function",
"function": {
"name": tool_call.function.name,
"arguments": tool_call.function.arguments
}
}
for tool_call in response.choices[0].message.tool_calls
]
}
self.messages.append(assistant_message)
# 处理工具调用
tool_list = []
for tool_call in response.choices[0].message.tool_calls: