From 9e503191d6dde74209ffb668a66867626726bdfc Mon Sep 17 00:00:00 2001 From: Tigrex Dai Date: Sat, 1 Mar 2025 17:27:01 +0100 Subject: [PATCH] Update anthropicmsgs.py --- pkg/provider/modelmgr/requesters/anthropicmsgs.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pkg/provider/modelmgr/requesters/anthropicmsgs.py b/pkg/provider/modelmgr/requesters/anthropicmsgs.py index 039836c9..9168f1bc 100644 --- a/pkg/provider/modelmgr/requesters/anthropicmsgs.py +++ b/pkg/provider/modelmgr/requesters/anthropicmsgs.py @@ -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'] = '' + block.thinking + '' if block.type == 'text': args['content'] += block.text elif block.type == 'tool_use':