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

🐛 remove tag in normal element

Browse source
This commit is contained in:
StarHeartHunt 2025-02-17 16:12:41 +08:00
parent f662d4ae00
commit 65ea3dc582
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