xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-04 04:54:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update anthropicmsgs.py

Browse Source
This commit is contained in:
Tigrex Dai
2025-03-01 17:27:01 +01:00
committed by GitHub
parent 1fd23a0d8d
commit 9e503191d6
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -51,10 +51,7 @@ class AnthropicMessages(requester.LLMAPIRequester):
args["model"] = model.name if model.model_name is None else model.model_name
if model.thinking:
if args["max_tokens"] >= 20000:
args["thinking"] = {"type": "enabled", "budget_tokens": 16000}
elif args["max_tokens"] > 1:
args["thinking"] = {"type": "enabled", "budget_tokens": args["max_tokens"]-1}
args["thinking"] = {"type": "enabled", "budget_tokens": 16000}
# 处理消息
@@ -152,6 +149,8 @@ class AnthropicMessages(requester.LLMAPIRequester):
assert type(resp) is anthropic.types.message.Message
for block in resp.content:
if block.type == 'thinking':
args['content'] = '<think>' + block.thinking + '</think>'
if block.type == 'text':
args['content'] += block.text
elif block.type == 'tool_use':