fix(llma): run ruff

Author: Radu-Raicea

posthog/ai/openai/openai_converter.py

@@ -273,7 +273,7 @@ def extract_openai_usage_from_chunk(
     """
 
     usage: StreamingUsageStats = {}
-    
+
     if provider_type == "chat":
         if not hasattr(chunk, "usage") or not chunk.usage:
             return usage
@@ -302,7 +302,11 @@ def extract_openai_usage_from_chunk(
     elif provider_type == "responses":
         # For Responses API, usage is only in chunk.response.usage for completed events
         if hasattr(chunk, "type") and chunk.type == "response.completed":
-            if hasattr(chunk, "response") and hasattr(chunk.response, "usage") and chunk.response.usage:
+            if (
+                hasattr(chunk, "response")
+                and hasattr(chunk.response, "usage")
+                and chunk.response.usage
+            ):
                 response_usage = chunk.response.usage
                 usage["input_tokens"] = getattr(response_usage, "input_tokens", 0)
                 usage["output_tokens"] = getattr(response_usage, "output_tokens", 0)

posthog/test/ai/openai/test_openai.py

@@ -1037,22 +1037,22 @@ def test_responses_api_streaming_with_tokens(mock_client):
     """Test that Responses API streaming properly captures token usage from response.usage."""
     from openai.types.responses import ResponseUsage
     from unittest.mock import MagicMock
-    
+
     # Create mock response chunks with usage data in the correct location
     chunks = []
-    
+
     # First chunk - just content, no usage
     chunk1 = MagicMock()
     chunk1.type = "response.text.delta"
     chunk1.text = "Test "
     chunks.append(chunk1)
-    
+
     # Second chunk - more content
     chunk2 = MagicMock()
     chunk2.type = "response.text.delta"
     chunk2.text = "response"
     chunks.append(chunk2)
-    
+
     # Final chunk - completed event with usage in response.usage
     chunk3 = MagicMock()
     chunk3.type = "response.completed"
@@ -1066,43 +1066,41 @@ def test_responses_api_streaming_with_tokens(mock_client):
     )
     chunk3.response.output = ["Test response"]
     chunks.append(chunk3)
-    
+
     captured_kwargs = {}
-    
+
     def mock_streaming_response(**kwargs):
         # Capture the kwargs to verify stream_options was NOT added
         captured_kwargs.update(kwargs)
         return iter(chunks)
-    
+
     with patch(
         "openai.resources.responses.Responses.create",
         side_effect=mock_streaming_response,
     ):
         client = OpenAI(api_key="test-key", posthog_client=mock_client)
-        
+
         # Consume the streaming response
         response = client.responses.create(
             model="gpt-4o-mini",
-            input=[
-                {"role": "user", "content": "Test message"}
-            ],
+            input=[{"role": "user", "content": "Test message"}],
             stream=True,
             posthog_distinct_id="test-id",
             posthog_properties={"test": "streaming"},
         )
-        
+
         # Consume all chunks
         list(response)
-        
+
         # Verify stream_options was NOT added (Responses API doesn't support it)
         assert "stream_options" not in captured_kwargs
-        
+
         # Verify capture was called
         assert mock_client.capture.call_count == 1
-        
+
         call_args = mock_client.capture.call_args[1]
         props = call_args["properties"]
-        
+
         # Verify tokens are captured correctly from response.usage (not 0)
         assert call_args["distinct_id"] == "test-id"
         assert call_args["event"] == "$ai_generation"