zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

final_messages 移除 tool_call 和 tool_response内容

Browse Source
This commit is contained in:
朱潮 2025-12-01 16:18:53 +08:00
parent e467111859
commit 38d22ff92e
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/fastapi_utils.py
View File

@ -267,13 +267,14 @@ def process_messages(messages: List[Dict], language: Optional[str] = None) -> Li
assistant_msg = {"role": ASSISTANT} assistant_msg = {"role": ASSISTANT}
if assistant_content.strip(): if assistant_content.strip():
assistant_msg["content"] = assistant_content.strip() assistant_msg["content"] = assistant_content.strip()
if function_calls:
# 如果有多个 function_call，只取第一个（兼容原有逻辑） # 如果有多个 function_call，只取第一个（兼容原有逻辑）
assistant_msg["function_call"] = function_calls[0] # if function_calls:
# assistant_msg["function_call"] = function_calls[0]
final_messages.append(assistant_msg) final_messages.append(assistant_msg)
# 添加所有 tool_responses 作为 function 消息 # 添加所有 tool_responses 作为 function 消息
final_messages.extend(tool_responses) # final_messages.extend(tool_responses)
else: else:
# 非 assistant 消息或不包含 [TOOL_RESPONSE] 的消息直接添加 # 非 assistant 消息或不包含 [TOOL_RESPONSE] 的消息直接添加
final_messages.append(msg) final_messages.append(msg)