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

🐛 fix output reasoning content control

Browse source
This commit is contained in:
StarHeartHunt 2025-02-15 18:06:31 +08:00
parent bb106c991b
commit f0a5dd8bb9
1 changed files with 16 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

33
nonebot_plugin_llmchat/__init__.py
View file

@ -53,9 +53,9 @@ driver = get_driver()
tasks: set["asyncio.Task"] = set()
def filter_think(content: Optional[str]) -> tuple[str, Optional[str]]:
def filter_think(content: Optional[str]) -> tuple[Optional[str], Optional[str]]:
if content is None:
return None
return None, None
# 匹配 <think> 标签和其中的内容
think_content = re.findall(r"<think>(.*?)</think>", content, flags=re.DOTALL)
@ -247,17 +247,16 @@ async def process_messages(group_id: int):
state.history.append({"role": "user", "content": content})
state.past_events.clear()
reply = ""
if state.output_reasoning_content:
reasoning_content = getattr(
response.choices[0].message, "reasoning_content", None
)
reply, think_content = filter_think(response.choices[0].message.content)
reply, matched_reasoning_content = filter_think(
response.choices[0].message.content
)
reasoning_content: Optional[str] = (
getattr(response.choices[0].message, "reasoning_content", None)
or matched_reasoning_content
)
if reasoning_content or think_content:
await handler.send(Message(reasoning_content or think_content))
else:
reply = response.choices[0].message.content
if state.output_reasoning_content and reasoning_content:
await handler.send(Message(reasoning_content))
assert reply is not None
logger.info(
@ -265,7 +264,7 @@ async def process_messages(group_id: int):
)
for r in reply.split("<botbr>"):
# 似乎会有空消息的情况导致string index out of range异常
if r.__len__() < 1:
if len(r) == 0 or r.isspace():
continue
# 删除前后多余的换行和空格
r = r.strip()
@ -336,7 +335,7 @@ async def handle_edit_preset(event: GroupMessageEvent, args: Message = CommandAr
group_states[group_id] = GroupState()
group_states[group_id].group_prompt = group_prompt
await preset_handler.finish("修改成功")
await edit_preset_handler.finish("修改成功")
reset_handler = on_command(
@ -356,7 +355,7 @@ async def handle_reset(event: GroupMessageEvent, args: Message = CommandArg()):
group_states[group_id].past_events.clear()
group_states[group_id].history.clear()
await preset_handler.finish("记忆已清空")
await reset_handler.finish("记忆已清空")
# 预设切换命令
@ -377,10 +376,10 @@ async def handle_think(event: GroupMessageEvent, args: Message = CommandArg()):
if group_states[group_id].output_reasoning_content:
group_states[group_id].output_reasoning_content = False
await preset_handler.finish("已关闭思维输出")
await think_handler.finish("已关闭思维输出")
else:
group_states[group_id].output_reasoning_content = True
await preset_handler.finish("已开启思维输出")
await think_handler.finish("已开启思维输出")
# region 持久化与定时任务