handle just in case

Author: mdrxy

libs/partners/groq/langchain_groq/chat_models.py

@@ -1463,15 +1463,26 @@ def _create_usage_metadata(groq_token_usage: dict) -> UsageMetadata:
         or 0
     )
     total_tokens = groq_token_usage.get("total_tokens") or input_tokens + output_tokens
+
+    # Support both formats for token details:
+    # Responses API uses "*_tokens_details", Chat Completions API might use
+    # "prompt_token_details"
+    input_details_dict = (
+        groq_token_usage.get("input_tokens_details")
+        or groq_token_usage.get("prompt_tokens_details")
+        or {}
+    )
+    output_details_dict = (
+        groq_token_usage.get("output_tokens_details")
+        or groq_token_usage.get("completion_tokens_details")
+        or {}
+    )
+
     input_token_details: dict = {
-        "cache_read": (groq_token_usage.get("input_tokens_details") or {}).get(
-            "cached_tokens"
-        ),
+        "cache_read": input_details_dict.get("cached_tokens"),
     }
     output_token_details: dict = {
-        "reasoning": (groq_token_usage.get("output_tokens_details") or {}).get(
-            "reasoning_tokens"
-        ),
+        "reasoning": output_details_dict.get("reasoning_tokens"),
     }
     usage_metadata: UsageMetadata = {
         "input_tokens": input_tokens,

libs/partners/groq/tests/unit_tests/test_chat_models.py

@@ -330,6 +330,26 @@ def test_create_usage_metadata_responses_api_format() -> None:
     assert "output_token_details" not in result
 
 
+def test_create_usage_metadata_chat_completions_with_details() -> None:
+    """Test usage metadata with hypothetical Chat Completions API format."""
+    token_usage = {
+        "prompt_tokens": 100,
+        "completion_tokens": 50,
+        "total_tokens": 150,
+        "prompt_tokens_details": {"cached_tokens": 80},
+        "completion_tokens_details": {"reasoning_tokens": 25},
+    }
+
+    result = _create_usage_metadata(token_usage)
+
+    assert isinstance(result, dict)
+    assert result["input_tokens"] == 100
+    assert result["output_tokens"] == 50
+    assert result["total_tokens"] == 150
+    assert result.get("input_token_details", {}).get("cache_read") == 80
+    assert result.get("output_token_details", {}).get("reasoning") == 25
+
+
 def test_create_usage_metadata_with_cached_tokens() -> None:
     """Test usage metadata with prompt caching."""
     token_usage = {