update test

Author: alexander-alderman-webb

tests/integrations/openai_agents/test_openai_agents.py

@@ -1345,6 +1345,24 @@ def simple_test_tool(message: str) -> str:
     assert ai_client_span2["attributes"]["gen_ai.request.max_tokens"] == 100
     assert ai_client_span2["attributes"]["gen_ai.request.messages"] == safe_serialize(
         [
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": "Please use the simple test tool"}
+                ],
+            },
+            {
+                "role": "assistant",
+                "content": [
+                    {
+                        "arguments": '{"message": "hello"}',
+                        "call_id": "call_123",
+                        "name": "simple_test_tool",
+                        "type": "function_call",
+                        "id": "call_123",
+                    }
+                ],
+            },
             {
                 "role": "tool",
                 "content": [
@@ -2288,7 +2306,6 @@ def test_openai_agents_message_role_mapping(
 
     get_response_kwargs = {"input": [test_message]}
 
-    from sentry_sdk.integrations.openai_agents.utils import _set_input_data
     from sentry_sdk import start_span
 
     with start_span(op="test") as span:
@@ -3036,47 +3053,6 @@ def calculator(a: int, b: int) -> int:
     )
 
 
-def test_openai_agents_message_truncation(sentry_init, capture_items):
-    """Test that large messages are truncated properly in OpenAI Agents integration."""
-
-    large_content = (
-        "This is a very long message that will exceed our size limits. " * 1000
-    )
-
-    sentry_init(
-        integrations=[OpenAIAgentsIntegration()],
-        traces_sample_rate=1.0,
-        send_default_pii=True,
-    )
-
-    test_messages = [
-        {"role": "user", "content": large_content},
-        {"role": "assistant", "content": large_content},
-        {"role": "user", "content": "small message 4"},
-        {"role": "assistant", "content": "small message 5"},
-    ]
-
-    get_response_kwargs = {"input": test_messages}
-
-    with start_span(op="gen_ai.chat") as span:
-        scope = sentry_sdk.get_current_scope()
-        _set_input_data(span, get_response_kwargs)
-        if hasattr(scope, "_gen_ai_original_message_count"):
-            truncated_count = scope._gen_ai_original_message_count.get(span.span_id)
-            assert truncated_count == 4, (
-                f"Expected 4 original messages, got {truncated_count}"
-            )
-
-        assert SPANDATA.GEN_AI_REQUEST_MESSAGES in span._data
-        messages_data = span._data[SPANDATA.GEN_AI_REQUEST_MESSAGES]
-        assert isinstance(messages_data, str)
-
-        parsed_messages = json.loads(messages_data)
-        assert isinstance(parsed_messages, list)
-        assert len(parsed_messages) == 1
-        assert "small message 5" in str(parsed_messages[0])
-
-
 @pytest.mark.asyncio
 async def test_streaming_span_update_captures_response_data(
     sentry_init, test_agent, mock_usage