From d1e45afd0f4a3d6a71dd3f33d3f312357779ab92 Mon Sep 17 00:00:00 2001
From: mo <mosafariuk@gmail.com>
Date: Wed, 21 Oct 2026 20:04:10 +0100
Subject: [PATCH] fix(ai): capture OpenAI chat custom tool calls instead of
 raising

A custom tool call has `custom` instead of `function`, so formatting the
response raised AttributeError from the wrapped create(), after the provider
call had succeeded. Read the name and input from `custom`, matching
PostHog/posthog-js#5251.
---
 .sampo/changesets/openai-custom-tool-calls.md |  5 ++
 posthog/ai/openai/openai_converter.py         | 14 ++-
 posthog/test/ai/openai/test_openai.py         | 86 +++++++++++++++++++
 3 files changed, 103 insertions(+), 2 deletions(-)
 create mode 100644 .sampo/changesets/openai-custom-tool-calls.md

diff --git a/.sampo/changesets/openai-custom-tool-calls.md b/.sampo/changesets/openai-custom-tool-calls.md
new file mode 100644
index 000000000..49f5b66b6
--- /dev/null
+++ b/.sampo/changesets/openai-custom-tool-calls.md
@@ -0,0 +1,5 @@
+---
+pypi/posthog: patch
+---
+
+Capture Chat Completions responses that contain custom tool calls instead of raising `AttributeError` from the wrapped `create()` call.
diff --git a/posthog/ai/openai/openai_converter.py b/posthog/ai/openai/openai_converter.py
index 0f391fb9f..21c58478b 100644
--- a/posthog/ai/openai/openai_converter.py
+++ b/posthog/ai/openai/openai_converter.py
@@ -153,13 +153,23 @@ def format_openai_response(response: Any) -> List[FormattedMessage]:
 
                 if hasattr(choice.message, "tool_calls") and choice.message.tool_calls:
                     for tool_call in choice.message.tool_calls:
+                        # Custom tools carry their free-form text in `custom.input`,
+                        # with no `function` object.
+                        is_custom = getattr(tool_call, "type", None) == "custom"
+                        call = getattr(
+                            tool_call, "custom" if is_custom else "function", None
+                        )
+                        if call is None:
+                            continue
                         content.append(
                             {
                                 "type": "function",
                                 "id": tool_call.id,
                                 "function": {
-                                    "name": tool_call.function.name,
-                                    "arguments": tool_call.function.arguments,
+                                    "name": call.name,
+                                    "arguments": call.input
+                                    if is_custom
+                                    else call.arguments,
                                 },
                             }
                         )
diff --git a/posthog/test/ai/openai/test_openai.py b/posthog/test/ai/openai/test_openai.py
index 83304a4b1..a0a550191 100644
--- a/posthog/test/ai/openai/test_openai.py
+++ b/posthog/test/ai/openai/test_openai.py
@@ -736,6 +736,92 @@ def test_tool_calls_only_no_content(mock_client, mock_openai_response_tool_calls
         assert props["$ai_http_status"] == 200
 
 
+@pytest.mark.parametrize("is_async", [False, True])
+async def test_custom_tool_calls_are_captured(mock_client, is_async):
+    """A custom tool call has `custom` instead of `function`. The wrapped call
+    must return the completion and capture it, not raise AttributeError."""
+    completion = ChatCompletion.model_validate(
+        {
+            "id": "test",
+            "model": "gpt-5",
+            "object": "chat.completion",
+            "created": 0,
+            "choices": [
+                {
+                    "finish_reason": "tool_calls",
+                    "index": 0,
+                    "message": {
+                        "role": "assistant",
+                        "content": None,
+                        "tool_calls": [
+                            {
+                                "id": "call_fn",
+                                "type": "function",
+                                "function": {
+                                    "name": "get_weather",
+                                    "arguments": '{"location": "New York"}',
+                                },
+                            },
+                            {
+                                "id": "call_custom",
+                                "type": "custom",
+                                "custom": {"name": "code_exec", "input": "print(1)"},
+                            },
+                        ],
+                    },
+                }
+            ],
+            "usage": {"prompt_tokens": 25, "completion_tokens": 10, "total_tokens": 35},
+        }
+    )
+    kwargs = dict(
+        model="gpt-5",
+        messages=[{"role": "user", "content": "Run it"}],
+        tools=[{"type": "custom", "custom": {"name": "code_exec"}}],
+        posthog_distinct_id="test-id",
+    )
+
+    if is_async:
+        with patch(
+            "openai.resources.chat.completions.AsyncCompletions.create",
+            new_callable=AsyncMock,
+            return_value=completion,
+        ):
+            client = AsyncOpenAI(api_key="test-key", posthog_client=mock_client)
+            response = await client.chat.completions.create(**kwargs)
+    else:
+        with patch(
+            "openai.resources.chat.completions.Completions.create",
+            return_value=completion,
+        ):
+            client = OpenAI(api_key="test-key", posthog_client=mock_client)
+            response = client.chat.completions.create(**kwargs)
+
+    assert response == completion
+    assert mock_client.capture.call_count == 1
+    props = mock_client.capture.call_args[1]["properties"]
+    assert props["$ai_output_choices"] == [
+        {
+            "role": "assistant",
+            "content": [
+                {
+                    "type": "function",
+                    "id": "call_fn",
+                    "function": {
+                        "name": "get_weather",
+                        "arguments": '{"location": "New York"}',
+                    },
+                },
+                {
+                    "type": "function",
+                    "id": "call_custom",
+                    "function": {"name": "code_exec", "arguments": "print(1)"},
+                },
+            ],
+        }
+    ]
+
+
 def test_responses_api_tool_calls(mock_client, mock_responses_api_with_tool_calls):
     with patch(
         "openai.resources.responses.Responses.create",