Commit d7fa394
Changed files (2)
src
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]