fix: lint

This commit is contained in:
keenborder786 2025-07-11 06:05:56 +05:00
parent a04c1c76e6
commit 8c628fe0c1

View File

@ -1307,15 +1307,13 @@ class ChatAnthropic(BaseChatModel):
if values.get("max_tokens") is None and values.get("model"): if values.get("max_tokens") is None and values.get("model"):
if "claude-opus-4" in values.get("model", ""): if "claude-opus-4" in values.get("model", ""):
values["max_tokens"] = 32000 values["max_tokens"] = 32000
elif ( elif "claude-sonnet-4" in values.get(
"claude-sonnet-4" in values.get("model", "") "model", ""
or "claude-3-7-sonnet" in values.get("model", "") ) or "claude-3-7-sonnet" in values.get("model", ""):
):
values["max_tokens"] = 64000 values["max_tokens"] = 64000
elif ( elif "claude-3-5-sonnet" in values.get(
"claude-3-5-sonnet" in values.get("model", "") "model", ""
or "claude-3-5-haiku" in values.get("model", "") ) or "claude-3-5-haiku" in values.get("model", ""):
):
values["max_tokens"] = 8192 values["max_tokens"] = 8192
# leaves us with "claude-3-5-opus", "claude-3-haiku" # leaves us with "claude-3-5-opus", "claude-3-haiku"
else: else: