From 4412d5e1bf9d5f08779dd282e18aa8a617b761a1 Mon Sep 17 00:00:00 2001 From: Alexander Alderman Webb Date: Fri, 31 Oct 2025 13:54:30 +0100 Subject: [PATCH 1/2] tests(openai-agents): Remove MagicMock from ModelResponse --- tests/integrations/openai_agents/test_openai_agents.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/tests/integrations/openai_agents/test_openai_agents.py b/tests/integrations/openai_agents/test_openai_agents.py index bc1de4e95b..1e3975248f 100644 --- a/tests/integrations/openai_agents/test_openai_agents.py +++ b/tests/integrations/openai_agents/test_openai_agents.py @@ -294,9 +294,6 @@ async def test_handoff_span(sentry_init, capture_events, mock_usage): name="transfer_to_secondary_agent", type="function_call", arguments="{}", - function=MagicMock( - name="transfer_to_secondary_agent", arguments="{}" - ), ) ], usage=mock_usage, @@ -377,9 +374,6 @@ def simple_test_tool(message: str) -> str: name="simple_test_tool", type="function_call", arguments='{"message": "hello"}', - function=MagicMock( - name="simple_test_tool", arguments='{"message": "hello"}' - ), ) # First response with tool call @@ -1111,9 +1105,6 @@ def failing_tool(message: str) -> str: name="failing_tool", type="function_call", arguments='{"message": "test"}', - function=MagicMock( - name="failing_tool", arguments='{"message": "test"}' - ), ) # First response with tool call From a5c0591f43baee38ca6a17faf71885899ad54374 Mon Sep 17 00:00:00 2001 From: Alexander Alderman Webb Date: Fri, 31 Oct 2025 14:08:51 +0100 Subject: [PATCH 2/2] . --- .../openai_agents/test_openai_agents.py | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/tests/integrations/openai_agents/test_openai_agents.py b/tests/integrations/openai_agents/test_openai_agents.py index 1e3975248f..306fc08ac7 100644 --- a/tests/integrations/openai_agents/test_openai_agents.py +++ b/tests/integrations/openai_agents/test_openai_agents.py @@ -501,11 +501,8 @@ def simple_test_tool(message: str) -> str: assert ai_client_span1["data"]["gen_ai.usage.output_tokens"] == 5 assert ai_client_span1["data"]["gen_ai.usage.output_tokens.reasoning"] == 0 assert ai_client_span1["data"]["gen_ai.usage.total_tokens"] == 15 - assert re.sub( - r"SerializationIterator\(.*\)", - "NOT_CHECKED", - ai_client_span1["data"]["gen_ai.response.tool_calls"], - ) == safe_serialize( + + assert ai_client_span1["data"]["gen_ai.response.tool_calls"] == safe_serialize( [ { "arguments": '{"message": "hello"}', @@ -514,7 +511,6 @@ def simple_test_tool(message: str) -> str: "type": "function_call", "id": "call_123", "status": None, - "function": "NOT_CHECKED", } ] ) @@ -553,11 +549,7 @@ def simple_test_tool(message: str) -> str: == available_tools ) assert ai_client_span2["data"]["gen_ai.request.max_tokens"] == 100 - assert re.sub( - r"SerializationIterator\(.*\)", - "NOT_CHECKED", - ai_client_span2["data"]["gen_ai.request.messages"], - ) == safe_serialize( + assert ai_client_span2["data"]["gen_ai.request.messages"] == safe_serialize( [ { "role": "system", @@ -580,7 +572,6 @@ def simple_test_tool(message: str) -> str: "name": "simple_test_tool", "type": "function_call", "id": "call_123", - "function": "NOT_CHECKED", } ], },