From 6335e9dd8bdc488967e2442f46712984824a6346 Mon Sep 17 00:00:00 2001 From: Tigrex Dai Date: Fri, 28 Feb 2025 13:02:06 +0100 Subject: [PATCH] Update anthropicmsgs.py --- pkg/provider/modelmgr/requesters/anthropicmsgs.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pkg/provider/modelmgr/requesters/anthropicmsgs.py b/pkg/provider/modelmgr/requesters/anthropicmsgs.py index 72301efd..1514753d 100644 --- a/pkg/provider/modelmgr/requesters/anthropicmsgs.py +++ b/pkg/provider/modelmgr/requesters/anthropicmsgs.py @@ -50,6 +50,12 @@ class AnthropicMessages(requester.LLMAPIRequester): args = self.ap.provider_cfg.data['requester']['anthropic-messages']['args'].copy() 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} + # 处理消息 # system