jianf/TrulyMEM-TrueHumanMEM
1
0
Fork 0
mirror of https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git synced 2026-10-15 12:15:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复多次 Tool Calls 消息累积问题

Browse Source
This commit is contained in:
root
2026-04-09 07:09:12 +08:00
parent 70953cd412
commit c30f4260ad
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
openclaw_neo4j_demo.py
View File

@ -593,16 +593,16 @@ class OpenClawClient:
last_assistant_msg = None last_assistant_msg = None
# 首次请求 # 首次请求
response = self.send_message(user_input, tool_results) response = self.send_message(user_input, [])
message = response.choices[0].message message = response.choices[0].message
# 处理工具调用循环 # 处理工具调用循环 - 持续处理直到没有新的 tool_calls
while message.tool_calls: while message.tool_calls:
# 打印模型响应（如果有） # 打印模型响应（如果有）
if message.content: if message.content:
print(f"\n[模型] {message.content}") print(f"\n[模型] {message.content}")
# 保存包含 tool_calls 的 assistant 消息 # 保存包含 tool_calls 的 assistant 消息（只保留最后一个）
last_assistant_msg = { last_assistant_msg = {
"role": "assistant", "role": "assistant",
"content": message.content, "content": message.content,
@ -610,7 +610,8 @@ class OpenClawClient:
"tool_calls": [{"id": tc.id, "type": "function", "function": {"name": tc.function.name, "arguments": tc.function.arguments}} for tc in message.tool_calls] "tool_calls": [{"id": tc.id, "type": "function", "function": {"name": tc.function.name, "arguments": tc.function.arguments}} for tc in message.tool_calls]
} }
# 执行所有工具调用 # 执行当前轮的所有工具调用
current_tool_results = []
for tool_call in message.tool_calls: for tool_call in message.tool_calls:
tool_name = tool_call.function.name tool_name = tool_call.function.name
tool_args = json.loads(tool_call.function.arguments) tool_args = json.loads(tool_call.function.arguments)
@ -619,18 +620,18 @@ class OpenClawClient:
result = execute_tool(self.graph, tool_name, tool_args) result = execute_tool(self.graph, tool_name, tool_args)
print(f"\n[工具结果] {result}") print(f"\n[工具结果] {result}")
# 添加工具结果 - 包含 tool_call_id current_tool_results.append({
tool_results.append({
"role": "tool", "role": "tool",
"tool_call_id": tool_id, "tool_call_id": tool_id,
"content": result "content": result
}) })
# 继续调用 - 包含 system + assistant消息(with tool_calls) + tool_results + user # 继续调用 - 只发送当前轮的 assistant 消息和工具结果
messages = [{"role": "system", "content": self.system_prompt}] messages = [
if last_assistant_msg: {"role": "system", "content": self.system_prompt},
messages.append(last_assistant_msg) last_assistant_msg,
messages.extend(tool_results) ]
messages.extend(current_tool_results)
messages.append({"role": "user", "content": user_input}) messages.append({"role": "user", "content": user_input})
response = self.client.chat.completions.create( response = self.client.chat.completions.create(