chore(llmo): run formatter

Author: Radu-Raicea

posthog/test/ai/anthropic/test_anthropic.py

@@ -442,7 +442,7 @@ def test_tool_definition(mock_client, mock_anthropic_response):
         return_value=mock_anthropic_response,
     ):
         client = Anthropic(api_key="test-key", posthog_client=mock_client)
-        
+
         tools = [
             {
                 "name": "get_weather",
@@ -452,14 +452,14 @@ def test_tool_definition(mock_client, mock_anthropic_response):
                     "properties": {
                         "location": {
                             "type": "string",
-                            "description": "The city or location name to get weather for"
+                            "description": "The city or location name to get weather for",
                         }
                     },
-                    "required": ["location"]
-                }
+                    "required": ["location"],
+                },
             }
         ]
-        
+
         response = client.messages.create(
             model="claude-3-5-sonnet-20241022",
             max_tokens=200,

posthog/test/ai/gemini/test_gemini.py

@@ -337,11 +337,11 @@ def test_tool_use_response(mock_client, mock_google_genai_client, mock_gemini_re
                 properties={
                     "location": MagicMock(
                         type="STRING",
-                        description="The city and state, e.g. San Francisco, CA"
+                        description="The city and state, e.g. San Francisco, CA",
                     )
                 },
-                required=["location"]
-            )
+                required=["location"],
+            ),
         )
     ]
 

posthog/test/ai/langchain/test_callbacks.py

@@ -1796,7 +1796,7 @@ def test_tool_definition(mock_client):
     """Test that tools defined in invocation parameters are captured in $ai_tools property"""
     callbacks = CallbackHandler(mock_client)
     run_id = uuid.uuid4()
-    
+
     # Define tools to be passed to the invocation parameters
     tools = [
         {
@@ -1809,15 +1809,15 @@ def test_tool_definition(mock_client):
                     "properties": {
                         "location": {
                             "type": "string",
-                            "description": "The city or location name to get weather for"
+                            "description": "The city or location name to get weather for",
                         }
                     },
-                    "required": ["location"]
-                }
-            }
+                    "required": ["location"],
+                },
+            },
         }
     ]
-    
+
     with patch("time.time", return_value=1234567890):
         callbacks._set_llm_metadata(
             {"kwargs": {"openai_api_base": "https://api.openai.com/v1"}},
@@ -1827,7 +1827,7 @@ def test_tool_definition(mock_client):
             metadata={"ls_model_name": "gpt-4o-mini", "ls_provider": "openai"},
             name="test",
         )
-    
+
     expected = GenerationMetadata(
         model="gpt-4o-mini",
         input=[{"role": "user", "content": "hey"}],
@@ -1840,29 +1840,29 @@ def test_tool_definition(mock_client):
         end_time=None,
     )
     assert callbacks._runs[run_id] == expected
-    
+
     with patch("time.time", return_value=1234567891):
         run = callbacks._pop_run_metadata(run_id)
     expected.end_time = 1234567891
     assert run == expected
     assert callbacks._runs == {}
-    
+
     # Now test that the tools are properly captured in the PostHog event
     mock_response = MagicMock()
     mock_response.generations = [[MagicMock()]]
-    
+
     callbacks._capture_generation(
         trace_id=run_id,
         run_id=run_id,
         run=run,
         output=mock_response,
         parent_run_id=None,
     )
-    
+
     assert mock_client.capture.call_count == 1
     call_args = mock_client.capture.call_args[1]
     props = call_args["properties"]
-    
+
     assert call_args["distinct_id"] == run_id
     assert call_args["event"] == "$ai_generation"
     assert props["$ai_provider"] == "openai"

posthog/test/ai/openai/test_openai.py

@@ -775,7 +775,7 @@ def test_tool_definition(mock_client, mock_openai_response):
         return_value=mock_openai_response,
     ):
         client = OpenAI(api_key="test-key", posthog_client=mock_client)
-        
+
         # Define tools to be passed to the create function
         tools = [
             {
@@ -788,15 +788,15 @@ def test_tool_definition(mock_client, mock_openai_response):
                         "properties": {
                             "location": {
                                 "type": "string",
-                                "description": "The city or location name to get weather for"
+                                "description": "The city or location name to get weather for",
                             }
                         },
-                        "required": ["location"]
-                    }
-                }
+                        "required": ["location"],
+                    },
+                },
             }
         ]
-        
+
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[{"role": "user", "content": "hey"}],