fix: always miss one

k11kirky · k11kirky · commit 0601945106f5 · 2025-01-17T12:04:26.000-08:00
diff --git a/posthog/test/ai/langchain/test_callbacks.py b/posthog/test/ai/langchain/test_callbacks.py
@@ -760,7 +760,7 @@ def test_tool_calls(mock_client):
 
     assert mock_client.capture.call_count == 1
     call = mock_client.capture.call_args[1]
-    assert call["properties"]["$ai_output"]["choices"][0]["tool_calls"] == [
+    assert call["properties"]["$ai_output_choices"][0]["tool_calls"] == [
         {
             "type": "function",
             "id": "123",

Original file line number	Diff line number	Diff line change
`@@ -760,7 +760,7 @@ def test_tool_calls(mock_client):`
`760`	`760`
`761`	`761`	`assert mock_client.capture.call_count == 1`
`762`	`762`	`call = mock_client.capture.call_args[1]`
`763`		`- assert call["properties"]["$ai_output"]["choices"][0]["tool_calls"] == [`
	`763`	`+ assert call["properties"]["$ai_output_choices"][0]["tool_calls"] == [`
`764`	`764`	`{`
`765`	`765`	`"type": "function",`
`766`	`766`	`"id": "123",`