fuquan/nonebot-plugin-llmchat
1
0
Fork 0
mirror of https://github.com/FuQuan233/nonebot-plugin-llmchat.git synced 2025-09-06 11:30:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

🐛 Fix message history order and add error handling for reasoning content

Browse source
This commit is contained in:
FuQuan233 2025-02-21 18:07:28 +08:00
parent 04883cda00
commit 505fab406b
1 changed files with 16 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

21
nonebot_plugin_llmchat/__init__.py
View file

@ -260,10 +260,6 @@ async def process_messages(group_id: int):
if response.usage is not None: if response.usage is not None:
logger.debug(f"收到API响应 使用token数：{response.usage.total_tokens}") logger.debug(f"收到API响应 使用token数：{response.usage.total_tokens}")
# 请求成功后再保存历史记录，保证user和assistant穿插，防止R1模型报错
state.history.append({"role": "user", "content": content})
state.past_events.clear()
reply, matched_reasoning_content = pop_reasoning_content( reply, matched_reasoning_content = pop_reasoning_content(
response.choices[0].message.content response.choices[0].message.content
) )
@ -272,7 +268,14 @@ async def process_messages(group_id: int):
or matched_reasoning_content or matched_reasoning_content
) )
# 请求成功后再保存历史记录，保证user和assistant穿插，防止R1模型报错
state.history.append({"role": "user", "content": content})
# 添加助手回复到历史
state.history.append({"role": "assistant","content": reply})
state.past_events.clear()
if state.output_reasoning_content and reasoning_content: if state.output_reasoning_content and reasoning_content:
try:
bot = get_bot(str(event.self_id)) bot = get_bot(str(event.self_id))
await bot.send_group_forward_msg( await bot.send_group_forward_msg(
group_id=group_id, group_id=group_id,
@ -280,6 +283,8 @@ async def process_messages(group_id: int):
bot.self_id, reasoning_content bot.self_id, reasoning_content
), ),
) )
except Exception as e:
await handler.send(f"合并转发消息发送失败：\n{e!s}\n"+reasoning_content)
assert reply is not None assert reply is not None
logger.info( logger.info(
@ -297,14 +302,6 @@ async def process_messages(group_id: int):
) # 只记录前50个字符避免日志过大 ) # 只记录前50个字符避免日志过大
await handler.send(Message(r)) await handler.send(Message(r))
# 添加助手回复到历史
state.history.append(
{
"role": "assistant",
"content": reply,
}
)
except Exception as e: except Exception as e:
logger.opt(exception=e).error(f"API请求失败 群号：{group_id}") logger.opt(exception=e).error(f"API请求失败 群号：{group_id}")
await handler.send(Message(f"服务暂时不可用，请稍后再试\n{e!s}")) await handler.send(Message(f"服务暂时不可用，请稍后再试\n{e!s}"))