kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-21 06:07:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: gemini tool call bug

Browse Source
This commit is contained in:
zhayujie
2026-03-11 02:04:09 +08:00
parent e8d4f7dc4f
commit eda82bac92
2 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
agent/protocol/agent_stream.py
View File

@@ -521,6 +521,7 @@ class AgentStreamExecutor:
# Streaming response # Streaming response
full_content = "" full_content = ""
tool_calls_buffer = {} # {index: {id, name, arguments}} tool_calls_buffer = {} # {index: {id, name, arguments}}
gemini_raw_parts = None # Preserve Gemini thoughtSignature for round-trip
stop_reason = None # Track why the stream stopped stop_reason = None # Track why the stream stopped
try: try:
@@ -612,6 +613,10 @@ class AgentStreamExecutor:
if "arguments" in func: if "arguments" in func:
tool_calls_buffer[index]["arguments"] += func["arguments"] tool_calls_buffer[index]["arguments"] += func["arguments"]
# Preserve _gemini_raw_parts for Gemini thoughtSignature round-trip
if "_gemini_raw_parts" in delta:
gemini_raw_parts = delta["_gemini_raw_parts"]
except Exception as e: except Exception as e:
error_str = str(e) error_str = str(e)
error_str_lower = error_str.lower() error_str_lower = error_str.lower()
@@ -793,6 +798,9 @@ class AgentStreamExecutor:
"input": tc.get("arguments", {}) "input": tc.get("arguments", {})
}) })
if gemini_raw_parts:
assistant_msg["_gemini_raw_parts"] = gemini_raw_parts
# Only append if content is not empty # Only append if content is not empty
if assistant_msg["content"]: if assistant_msg["content"]:
self.messages.append(assistant_msg) self.messages.append(assistant_msg)

3
models/openai_compatible_bot.py
View File

@@ -294,6 +294,9 @@ class OpenAICompatibleBot:
if tool_calls: if tool_calls:
openai_msg["tool_calls"] = tool_calls openai_msg["tool_calls"] = tool_calls
if msg.get("_gemini_raw_parts"):
openai_msg["_gemini_raw_parts"] = msg["_gemini_raw_parts"]
openai_messages.append(openai_msg) openai_messages.append(openai_msg)
else: else:
# Other list content, keep as is # Other list content, keep as is