BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

移除思考链输出

pull/1267/head
foxplaying 2025-09-26 14:04:23 +08:00 committed by GitHub
parent 5cc1e56904
commit 10aa3bc9fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/llm_models/model_client/gemini_client.py
View File

@ -183,13 +183,15 @@ def _process_delta(
if delta.text: if delta.text:
fc_delta_buffer.write(delta.text) fc_delta_buffer.write(delta.text)
# 处理 thought（Gemini 的特殊字段） # 处理 thought（Gemini 的特殊字段 -> 存到 reasoning，不写客户端 buffer）
for c in getattr(delta, "candidates", []): for c in getattr(delta, "candidates", []):
if c.content and getattr(c.content, "parts", None): if c.content and getattr(c.content, "parts", None):
for p in c.content.parts: for p in c.content.parts:
if getattr(p, "thought", False) and getattr(p, "text", None): if getattr(p, "thought", False) and getattr(p, "text", None):
# 把 thought 写入 buffer，避免 resp.content 永远为空 # 只存到临时属性 _reasoning_texts
fc_delta_buffer.write(p.text) if not hasattr(delta, "_reasoning_texts"):
delta._reasoning_texts = []
delta._reasoning_texts.append(p.text)
if delta.function_calls: # 为什么不用hasattr呢，是因为这个属性一定有，即使是个空的 if delta.function_calls: # 为什么不用hasattr呢，是因为这个属性一定有，即使是个空的
for call in delta.function_calls: for call in delta.function_calls:
@ -237,6 +239,10 @@ def _build_stream_api_resp(
resp.tool_calls.append(ToolCall(call_id, function_name, arguments)) resp.tool_calls.append(ToolCall(call_id, function_name, arguments))
# 思考链：放到 reasoning_content，不返回给客户端
if last_resp and hasattr(last_resp, "_reasoning_texts"):
resp.reasoning_content = "\n".join(last_resp._reasoning_texts)
# 检查是否因为 max_tokens 截断 # 检查是否因为 max_tokens 截断
reason = None reason = None
if last_resp and getattr(last_resp, "candidates", None): if last_resp and getattr(last_resp, "candidates", None):