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() 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: if content is None:
return None return None, None
# 匹配 <think> 标签和其中的内容 # 匹配 <think> 标签和其中的内容
think_content = re.findall(r"<think>(.*?)</think>", content, flags=re.DOTALL) 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.history.append({"role": "user", "content": content})
state.past_events.clear() state.past_events.clear()
reply = "" reply, matched_reasoning_content = filter_think(
if state.output_reasoning_content: response.choices[0].message.content
reasoning_content = getattr( )
response.choices[0].message, "reasoning_content", None reasoning_content: Optional[str] = (
) getattr(response.choices[0].message, "reasoning_content", None)
reply, think_content = filter_think(response.choices[0].message.content) or matched_reasoning_content
)
if reasoning_content or think_content: if state.output_reasoning_content and reasoning_content:
await handler.send(Message(reasoning_content or think_content)) await handler.send(Message(reasoning_content))
else:
reply = response.choices[0].message.content
assert reply is not None assert reply is not None
logger.info( logger.info(
@ -265,7 +264,7 @@ async def process_messages(group_id: int):
) )
for r in reply.split("<botbr>"): for r in reply.split("<botbr>"):
# 似乎会有空消息的情况导致string index out of range异常 # 似乎会有空消息的情况导致string index out of range异常
if r.__len__() < 1: if len(r) == 0 or r.isspace():
continue continue
# 删除前后多余的换行和空格 # 删除前后多余的换行和空格
r = r.strip() 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] = GroupState()
group_states[group_id].group_prompt = group_prompt group_states[group_id].group_prompt = group_prompt
await preset_handler.finish("修改成功") await edit_preset_handler.finish("修改成功")
reset_handler = on_command( 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].past_events.clear()
group_states[group_id].history.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: if group_states[group_id].output_reasoning_content:
group_states[group_id].output_reasoning_content = False group_states[group_id].output_reasoning_content = False
await preset_handler.finish("已关闭思维输出") await think_handler.finish("已关闭思维输出")
else: else:
group_states[group_id].output_reasoning_content = True group_states[group_id].output_reasoning_content = True
await preset_handler.finish("已开启思维输出") await think_handler.finish("已开启思维输出")
# region 持久化与定时任务 # region 持久化与定时任务