injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 19:37:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix:fix remove_think logic, and end<think> fix </think>

Browse Source
This commit is contained in:
Dong_master
2025-08-10 00:28:55 +08:00
parent 8c5cb71812
commit 03b11481ed
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/provider/modelmgr/requesters/anthropicmsgs.py
View File

@@ -299,12 +299,12 @@ class AnthropicMessages(requester.ProviderAPIRequester):
async for chunk in await self.client.messages.create(**args):
# print(chunk)
if isinstance(chunk, anthropic.types.raw_content_block_start_event.RawContentBlockStartEvent): # 记录开始
if chunk.content_block.type == 'thinking' and not remove_think:
think_started = True
continue
elif chunk.content_block.type == 'text' and not remove_think:
think_ended = True
continue
if not remove_think:
if chunk.content_block.type == 'thinking':
think_started = True
elif chunk.content_block.type == 'text':
think_ended = True
continue
elif isinstance(chunk, anthropic.types.raw_content_block_delta_event.RawContentBlockDeltaEvent):
if chunk.delta.type == "thinking_delta":
if think_started:
@@ -317,7 +317,7 @@ class AnthropicMessages(requester.ProviderAPIRequester):
elif chunk.delta.type == "text_delta":
if think_ended:
think_ended = False
content = '\n<think>\n' + chunk.delta.text
content = '\n</think>\n' + chunk.delta.text
else:
content = chunk.delta.text
elif isinstance(chunk, anthropic.types.raw_content_block_stop_event.RawContentBlockStopEvent):