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

Merge pull request #10 from StarHeartHunt/fix/reasoning-tag

Browse source 
Fix: 移除普通内容里的 `<think>` 标签
This commit is contained in:
FuQuan233 2025-02-17 16:36:20 +08:00 • committed by GitHub
commit 54e4766dfd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
nonebot_plugin_llmchat/__init__.py
View file

@ -59,15 +59,13 @@ def pop_reasoning_content(
if content is None:
return None, None
think_content: Optional[str] = None
# 匹配 <think> 标签和其中的内容
if matched := re.match(r"<think>(.*?)</think>", content, flags=re.DOTALL):
think_content = matched.group(1)
# 如果找到了 <think> 标签内容，返回过滤后的文本和标签内的内容，否则只返回过滤后的文本和None
if think_content:
filtered_content = content.replace(think_content, "").strip()
return filtered_content, think_content.strip()
if matched := re.match(r"<think>(.*?)</think>", content, flags=re.DOTALL):
reasoning_element = matched.group(0)
reasoning_content = matched.group(1).strip()
filtered_content = content.replace(reasoning_element, "").strip()
return filtered_content, reasoning_content
else:
return content, None