fix: format

Author: skoob13

posthog/ai/langchain/callbacks.py

@@ -751,11 +751,16 @@ def _parse_usage_model(
         "reasoning": "reasoning_tokens",
     }
     normalized_usage = ModelUsage(
-        **{dataclass_key: parsed_usage.get(mapped_key) or 0 for mapped_key, dataclass_key in field_mapping.items()},
+        **{
+            dataclass_key: parsed_usage.get(mapped_key) or 0
+            for mapped_key, dataclass_key in field_mapping.items()
+        },
     )
     # input_tokens is the sum of input and cache read tokens.
     if normalized_usage.input_tokens and normalized_usage.cache_read_tokens:
-        normalized_usage.input_tokens = max(normalized_usage.input_tokens - normalized_usage.cache_read_tokens, 0)
+        normalized_usage.input_tokens = max(
+            normalized_usage.input_tokens - normalized_usage.cache_read_tokens, 0
+        )
     return normalized_usage
 
 

posthog/test/ai/langchain/test_callbacks.py

@@ -1969,7 +1969,9 @@ def test_no_cache_read_tokens_no_subtraction(mock_client):
 
     This tests the conditional check before the subtraction in callbacks.py line 757.
     """
-    prompt = ChatPromptTemplate.from_messages([("user", "Normal request without cache")])
+    prompt = ChatPromptTemplate.from_messages(
+        [("user", "Normal request without cache")]
+    )
 
     # No cache usage - input_tokens should remain as-is
     model = FakeMessagesListChatModel(