From f7f52cab1240cc83f8e11424905fe9d87fed2a47 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Thu, 5 Jun 2025 16:15:03 -0400
Subject: [PATCH] anthropic[patch]: cache tokens nit (#31484)

if you pass in beta headers directly cache_creation is a dict
---
 .../anthropic/langchain_anthropic/chat_models.py      | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/libs/partners/anthropic/langchain_anthropic/chat_models.py b/libs/partners/anthropic/langchain_anthropic/chat_models.py
index 60a95747de5..581008d3f1b 100644
--- a/libs/partners/anthropic/langchain_anthropic/chat_models.py
+++ b/libs/partners/anthropic/langchain_anthropic/chat_models.py
@@ -2131,11 +2131,12 @@ def _create_usage_metadata(anthropic_usage: BaseModel) -> UsageMetadata:
     }
     # Add (beta) cache TTL information if available
     cache_creation = getattr(anthropic_usage, "cache_creation", None)
+    cache_creation_keys = ("ephemeral_1h_input_tokens", "ephemeral_5m_input_tokens")
     if cache_creation:
-        for k in ["ephemeral_1h_input_tokens", "ephemeral_5m_input_tokens"]:
-            v = getattr(cache_creation, k, None)
-            if v:
-                input_token_details[k] = v
+        if isinstance(cache_creation, BaseModel):
+            cache_creation = cache_creation.model_dump()
+        for k in cache_creation_keys:
+            input_token_details[k] = cache_creation.get(k)
 
     # Anthropic input_tokens exclude cached token counts.
     input_tokens = (
@@ -2149,6 +2150,6 @@ def _create_usage_metadata(anthropic_usage: BaseModel) -> UsageMetadata:
         output_tokens=output_tokens,
         total_tokens=input_tokens + output_tokens,
         input_token_details=InputTokenDetails(
-            **{k: v for k, v in input_token_details.items() if v is not None}
+            **{k: v for k, v in input_token_details.items() if v is not None},
         ),
     )