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

♻️ tidy up

Browse source
This commit is contained in:
StarHeartHunt 2025-02-15 18:17:07 +08:00
parent 40ff908a32
commit 29e966eb21
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
nonebot_plugin_llmchat/__init__.py
View file

@ -53,23 +53,23 @@ driver = get_driver()
tasks: set["asyncio.Task"] = set() tasks: set["asyncio.Task"] = set()
def filter_think(content: Optional[str]) -> tuple[Optional[str], Optional[str]]: def pop_reasoning_content(
content: Optional[str],
) -> tuple[Optional[str], Optional[str]]:
if content is None: if content is None:
return None, None return None, None
think_content: Optional[str] = None
# 匹配 <think> 标签和其中的内容 # 匹配 <think> 标签和其中的内容
think_content = re.findall(r"<think>(.*?)</think>", content, flags=re.DOTALL) if matched := re.match(r"<think>(.*?)</think>", content, flags=re.DOTALL):
think_content = matched.group(0)
# 去除 <think> 标签及其内容
filtered_content = re.sub(
r"<think>.*?</think>", "", content, flags=re.DOTALL
).strip()
# 如果找到了 <think> 标签内容，返回过滤后的文本和标签内的内容，否则只返回过滤后的文本和None # 如果找到了 <think> 标签内容，返回过滤后的文本和标签内的内容，否则只返回过滤后的文本和None
if think_content: if think_content:
return filtered_content, think_content[0].strip() filtered_content = content.replace(think_content, "").strip()
return filtered_content, think_content.strip()
else: else:
return filtered_content, None return content, None
# 初始化群组状态 # 初始化群组状态
@ -247,7 +247,7 @@ 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, matched_reasoning_content = filter_think( reply, matched_reasoning_content = pop_reasoning_content(
response.choices[0].message.content response.choices[0].message.content
) )
reasoning_content: Optional[str] = ( reasoning_content: Optional[str] = (