Commit d7fa394

benny-dou <60535774+benny-dou@users.noreply.github.com>
2025-02-10 14:37:57
chore(gpt): show reasoning model in reasoning text
1 parent 1f793c2
Changed files (2)
src/llm/gpt.py
@@ -80,7 +80,10 @@ async def gpt_response(client: Client, message: Message, **kwargs):
         res = await send2tg(client, message, texts=msg, **kwargs)
         kwargs["progress"] = res[0]
     response = await get_gpt_response(config, **kwargs)
-    media = [{"document": save_txt(reasoning, f"{DOWNLOAD_DIR}/GPT-Reasoning-{rand_number()}.txt")}] if (reasoning := response.get("reasoning")) else []
+    media = []
+    if reasoning := response.get("reasoning"):
+        reasoning_model = f"推理模型: {response['reasoning_model']}\n\n" if response.get("reasoning_model") else ""
+        media = [{"document": save_txt(f"{reasoning_model}{reasoning}", f"{DOWNLOAD_DIR}/GPT-Reasoning-{rand_number()}.txt")}]
     if content := response.get("content"):
         texts = f"{response['bot_msg_prefix']}\n\n{content}"
         logger.debug(texts)
src/llm/response.py
@@ -88,9 +88,9 @@ async def parse_tool_call(config: dict, response: dict, retry: int = 0, **kwargs
                 return await get_gpt_response(config, retry, **kwargs)
         content = glom(response, "choices.0.message.content", default="") or ""
         reasoning = glom(response, "choices.0.message.reasoning", default="") or ""
-        res = {"content": content.strip(), "reasoning": reasoning.strip(), "bot_msg_prefix": config["bot_msg_prefix"]}
         primary_model = glom(config, "completions.model", default="") or ""
         used_model = glom(response, "model", default="") or ""
+        res = {"content": content.strip(), "reasoning": reasoning.strip(), "reasoning_model": used_model, "bot_msg_prefix": config["bot_msg_prefix"]}
         if not (used_model in primary_model or primary_model in used_model):
             # do not use `!=` to compare. (deepseek/deepseek-r1:free != deepseek/deepseek-r1,  gpt-4o != gpt-4o-2024-07-18)
             used_model = used_model.split("/")[-1]