fix(anthropic): set max input tokens based on 1m context beta header (#35341)

2026-06-09 10:17:00 +00:00 · 2026-02-19 21:01:49 -05:00
parent 135a208919
commit 5362bf5666
2 changed files with 23 additions and 0 deletions
--- a/libs/partners/anthropic/langchain_anthropic/chat_models.py
+++ b/libs/partners/anthropic/langchain_anthropic/chat_models.py
@@ -1024,6 +1024,12 @@ class ChatAnthropic(BaseChatModel):
        """Set model profile if not overridden."""
        if self.profile is None:
            self.profile = _get_default_model_profile(self.model)
+        if (
+            self.profile is not None
+            and self.betas
+            and "context-1m-2025-08-07" in self.betas
+        ):
+            self.profile["max_input_tokens"] = 1_000_000
        return self

    @cached_property