add a streaming test

Author: alexander-alderman-webb

tests/integrations/pydantic_ai/test_pydantic_ai.py

@@ -1256,6 +1256,65 @@ async def test_invoke_agent_with_instructions(
         assert SPANDATA.GEN_AI_SYSTEM_INSTRUCTIONS not in chat_span["data"]
 
 
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "send_default_pii, include_prompts",
+    [
+        (True, True),
+        (True, False),
+        (False, True),
+        (False, False),
+    ],
+)
+async def test_invoke_agent_streaming_with_instructions(
+    sentry_init, capture_events, send_default_pii, include_prompts
+):
+    """
+    Test that invoke_agent span handles instructions correctly.
+    """
+    from pydantic_ai import Agent
+
+    # Create agent with instructions (can be string or list)
+    agent = Agent(
+        "test",
+        name="test_instructions",
+    )
+
+    # Add instructions via _instructions attribute (internal API)
+    agent._instructions = ["Instruction 1", "Instruction 2"]
+    agent._system_prompts = ["System prompt"]
+
+    sentry_init(
+        integrations=[PydanticAIIntegration(include_prompts=include_prompts)],
+        traces_sample_rate=1.0,
+        send_default_pii=send_default_pii,
+    )
+
+    events = capture_events()
+
+    async with agent.run_stream("Test input") as result:
+        async for _ in result.stream_output():
+            pass
+
+    (transaction,) = events
+    spans = transaction["spans"]
+
+    # The transaction IS the invoke_agent span, check for messages in chat spans instead
+    chat_spans = [s for s in spans if s["op"] == "gen_ai.chat"]
+    assert len(chat_spans) >= 1
+
+    chat_span = chat_spans[0]
+
+    if send_default_pii and include_prompts:
+        system_instructions = chat_span["data"][SPANDATA.GEN_AI_SYSTEM_INSTRUCTIONS]
+        assert json.loads(system_instructions) == [
+            {"type": "text", "content": "System prompt"},
+            {"type": "text", "content": "Instruction 1\nInstruction 2"},
+        ]
+    else:
+        assert SPANDATA.GEN_AI_SYSTEM_INSTRUCTIONS not in chat_span["data"]
+
+
 @pytest.mark.asyncio
 async def test_model_name_extraction_with_callable(sentry_init, capture_events):
     """