bug(llmo): run formatter

Author: Radu-Raicea

posthog/ai/utils.py

@@ -118,7 +118,12 @@ def format_response(response, provider: str):
 def format_response_anthropic(response):
     output = []
     for choice in response.content:
-        if hasattr(choice, "type") and choice.type == "text" and hasattr(choice, "text") and choice.text:
+        if (
+            hasattr(choice, "type")
+            and choice.type == "text"
+            and hasattr(choice, "text")
+            and choice.text
+        ):
             output.append(
                 {
                     "role": "assistant",
@@ -230,8 +235,14 @@ def format_tool_calls(response, provider: str):
 
             for content_item in response.content:
                 if hasattr(content_item, "type") and content_item.type == "tool_use":
-                    tool_call_dict = vars(content_item)
-                    tool_calls.append(tool_call_dict)
+                    tool_calls.append(
+                        {
+                            "type": content_item.type,
+                            "id": content_item.id,
+                            "name": content_item.name,
+                            "input": content_item.input,
+                        }
+                    )
 
             return tool_calls if tool_calls else None
     elif provider == "openai":

posthog/test/ai/anthropic/test_anthropic.py

@@ -96,7 +96,12 @@ def mock_anthropic_response_with_tool_use():
         role="assistant",
         content=[
             {"type": "text", "text": "I'll help you with that."},
-            {"type": "tool_use", "id": "tool_1", "name": "get_weather", "input": {"location": "New York"}}
+            {
+                "type": "tool_use",
+                "id": "tool_1",
+                "name": "get_weather",
+                "input": {"location": "New York"},
+            },
         ],
         model="claude-3-opus-20240229",
         usage=Usage(
@@ -479,7 +484,9 @@ def test_tool_use_response(mock_client, mock_anthropic_response_with_tool_use):
         assert call_args["event"] == "$ai_generation"
         assert props["$ai_provider"] == "anthropic"
         assert props["$ai_model"] == "claude-3-opus-20240229"
-        assert props["$ai_input"] == [{"role": "user", "content": "What's the weather like?"}]
+        assert props["$ai_input"] == [
+            {"role": "user", "content": "What's the weather like?"}
+        ]
         # Should only include text content, not tool_use content
         assert props["$ai_output_choices"] == [
             {"role": "assistant", "content": "I'll help you with that."}
@@ -490,4 +497,11 @@ def test_tool_use_response(mock_client, mock_anthropic_response_with_tool_use):
         assert props["foo"] == "bar"
         assert isinstance(props["$ai_latency"], float)
         # Verify that tools are captured separately
-        assert props["$ai_tools"] == [{"type": "tool_use", "id": "tool_1", "name": "get_weather", "input": {"location": "New York"}}]
+        assert props["$ai_tools"] == [
+            {
+                "type": "tool_use",
+                "id": "tool_1",
+                "name": "get_weather",
+                "input": {"location": "New York"},
+            }
+        ]