pydantic-ai 0.7.6__tar.gz → 0.8.1__tar.gz

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.7.6
+Version: 0.8.1
 Summary: Agent Framework / shim to use Pydantic with LLMs
 Project-URL: Homepage, https://ai.pydantic.dev
 Project-URL: Source, https://github.com/pydantic/pydantic-ai
@@ -28,11 +28,11 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,temporal,vertexai]==0.7.6
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,temporal,vertexai]==0.8.1
 Provides-Extra: a2a
 Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.7.6; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.8.1; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.14.1; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/conftest.py

@@ -414,7 +414,7 @@ def bedrock_provider():
         pytest.skip('boto3 is not installed')
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture()
 def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
     # Locally, we authenticate via `gcloud` CLI, so we don't need to patch anything.
     if not os.getenv('CI', False):
@@ -423,7 +423,7 @@ def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
     try:
         from google.genai import _api_client
     except ImportError:
-        pytest.skip('google is not installed')
+        return  # do nothing if this isn't installed
 
     @dataclass
     class NoOpCredentials:
@@ -440,7 +440,7 @@ def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
 
 
 @pytest.fixture()
-async def vertex_provider():  # pragma: lax no cover
+async def vertex_provider(vertex_provider_auth: None):  # pragma: lax no cover
     # NOTE: You need to comment out this line to rewrite the cassettes locally.
     if not os.getenv('CI', False):
         pytest.skip('Requires properly configured local google vertex config to pass')

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/mcp_server.py

@@ -3,8 +3,7 @@ from pathlib import Path
 from typing import Any
 
 from mcp.server.fastmcp import Context, FastMCP, Image
-from mcp.server.session import ServerSessionT
-from mcp.shared.context import LifespanContextT, RequestT
+from mcp.server.session import ServerSession
 from mcp.types import (
     BlobResourceContents,
     EmbeddedResource,
@@ -13,7 +12,7 @@ from mcp.types import (
     TextContent,
     TextResourceContents,
 )
-from pydantic import AnyUrl
+from pydantic import AnyUrl, BaseModel
 
 mcp = FastMCP('Pydantic AI MCP Server')
 log_level = 'unset'
@@ -170,7 +169,7 @@ async def get_log_level(ctx: Context) -> str:  # type: ignore
 
 
 @mcp.tool()
-async def echo_deps(ctx: Context[ServerSessionT, LifespanContextT, RequestT]) -> dict[str, Any]:
+async def echo_deps(ctx: Context[ServerSession, None]) -> dict[str, Any]:
     """Echo the run context.
 
     Args:
@@ -186,7 +185,7 @@ async def echo_deps(ctx: Context[ServerSessionT, LifespanContextT, RequestT]) ->
 
 
 @mcp.tool()
-async def use_sampling(ctx: Context, foo: str) -> str:  # type: ignore
+async def use_sampling(ctx: Context[ServerSession, None], foo: str) -> str:
     """Use sampling callback."""
 
     result = await ctx.session.create_message(
@@ -202,6 +201,22 @@ async def use_sampling(ctx: Context, foo: str) -> str:  # type: ignore
     return result.model_dump_json(indent=2)
 
 
+class UserResponse(BaseModel):
+    response: str
+
+
+@mcp.tool()
+async def use_elicitation(ctx: Context[ServerSession, None], question: str) -> str:
+    """Use elicitation callback to ask the user a question."""
+
+    result = await ctx.elicit(message=question, schema=UserResponse)
+
+    if result.action == 'accept' and result.data:
+        return f'User responded: {result.data.response}'
+    else:
+        return f'User {result.action}ed the elicitation'
+
+
 @mcp._mcp_server.set_logging_level()  # pyright: ignore[reportPrivateUsage]
 async def set_logging_level(level: str) -> None:
     global log_level

pydantic_ai-0.8.1/tests/models/mock_openai.py

@@ -0,0 +1,161 @@
+from __future__ import annotations as _annotations
+
+from collections.abc import Sequence
+from dataclasses import dataclass, field
+from functools import cached_property
+from typing import Any, Union, cast
+
+from ..conftest import raise_if_exception, try_import
+from .mock_async_stream import MockAsyncStream
+
+with try_import() as imports_successful:
+    from openai import NOT_GIVEN, AsyncOpenAI
+    from openai.types import chat, responses
+    from openai.types.chat.chat_completion import Choice, ChoiceLogprobs
+    from openai.types.chat.chat_completion_message import ChatCompletionMessage
+    from openai.types.completion_usage import CompletionUsage
+    from openai.types.responses.response import ResponseUsage
+    from openai.types.responses.response_output_item import ResponseOutputItem
+
+    # note: we use Union here so that casting works with Python 3.9
+    MockChatCompletion = Union[chat.ChatCompletion, Exception]
+    MockChatCompletionChunk = Union[chat.ChatCompletionChunk, Exception]
+    MockResponse = Union[responses.Response, Exception]
+    MockResponseStreamEvent = Union[responses.ResponseStreamEvent, Exception]
+
+
+@dataclass
+class MockOpenAI:
+    completions: MockChatCompletion | Sequence[MockChatCompletion] | None = None
+    stream: Sequence[MockChatCompletionChunk] | Sequence[Sequence[MockChatCompletionChunk]] | None = None
+    index: int = 0
+    chat_completion_kwargs: list[dict[str, Any]] = field(default_factory=list)
+
+    @cached_property
+    def chat(self) -> Any:
+        chat_completions = type('Completions', (), {'create': self.chat_completions_create})
+        return type('Chat', (), {'completions': chat_completions})
+
+    @classmethod
+    def create_mock(cls, completions: MockChatCompletion | Sequence[MockChatCompletion]) -> AsyncOpenAI:
+        return cast(AsyncOpenAI, cls(completions=completions))
+
+    @classmethod
+    def create_mock_stream(
+        cls,
+        stream: Sequence[MockChatCompletionChunk] | Sequence[Sequence[MockChatCompletionChunk]],
+    ) -> AsyncOpenAI:
+        return cast(AsyncOpenAI, cls(stream=stream))
+
+    async def chat_completions_create(  # pragma: lax no cover
+        self, *_args: Any, stream: bool = False, **kwargs: Any
+    ) -> chat.ChatCompletion | MockAsyncStream[MockChatCompletionChunk]:
+        self.chat_completion_kwargs.append({k: v for k, v in kwargs.items() if v is not NOT_GIVEN})
+
+        if stream:
+            assert self.stream is not None, 'you can only used `stream=True` if `stream` is provided'
+            if isinstance(self.stream[0], Sequence):
+                response = MockAsyncStream(iter(cast(list[MockChatCompletionChunk], self.stream[self.index])))
+            else:
+                response = MockAsyncStream(iter(cast(list[MockChatCompletionChunk], self.stream)))
+        else:
+            assert self.completions is not None, 'you can only used `stream=False` if `completions` are provided'
+            if isinstance(self.completions, Sequence):
+                raise_if_exception(self.completions[self.index])
+                response = cast(chat.ChatCompletion, self.completions[self.index])
+            else:
+                raise_if_exception(self.completions)
+                response = cast(chat.ChatCompletion, self.completions)
+        self.index += 1
+        return response
+
+
+def get_mock_chat_completion_kwargs(async_open_ai: AsyncOpenAI) -> list[dict[str, Any]]:
+    if isinstance(async_open_ai, MockOpenAI):
+        return async_open_ai.chat_completion_kwargs
+    else:  # pragma: no cover
+        raise RuntimeError('Not a MockOpenAI instance')
+
+
+def completion_message(
+    message: ChatCompletionMessage, *, usage: CompletionUsage | None = None, logprobs: ChoiceLogprobs | None = None
+) -> chat.ChatCompletion:
+    choices = [Choice(finish_reason='stop', index=0, message=message)]
+    if logprobs:
+        choices = [Choice(finish_reason='stop', index=0, message=message, logprobs=logprobs)]
+    return chat.ChatCompletion(
+        id='123',
+        choices=choices,
+        created=1704067200,  # 2024-01-01
+        model='gpt-4o-123',
+        object='chat.completion',
+        usage=usage,
+    )
+
+
+@dataclass
+class MockOpenAIResponses:
+    response: MockResponse | Sequence[MockResponse] | None = None
+    stream: Sequence[MockResponseStreamEvent] | Sequence[Sequence[MockResponseStreamEvent]] | None = None
+    index: int = 0
+    response_kwargs: list[dict[str, Any]] = field(default_factory=list)
+
+    @cached_property
+    def responses(self) -> Any:
+        return type('Responses', (), {'create': self.responses_create})
+
+    @classmethod
+    def create_mock(cls, responses: MockResponse | Sequence[MockResponse]) -> AsyncOpenAI:
+        return cast(AsyncOpenAI, cls(response=responses))
+
+    @classmethod
+    def create_mock_stream(
+        cls,
+        stream: Sequence[MockResponseStreamEvent] | Sequence[Sequence[MockResponseStreamEvent]],
+    ) -> AsyncOpenAI:
+        return cast(AsyncOpenAI, cls(stream=stream))  # pragma: lax no cover
+
+    async def responses_create(  # pragma: lax no cover
+        self, *_args: Any, stream: bool = False, **kwargs: Any
+    ) -> responses.Response | MockAsyncStream[MockResponseStreamEvent]:
+        self.response_kwargs.append({k: v for k, v in kwargs.items() if v is not NOT_GIVEN})
+
+        if stream:
+            assert self.stream is not None, 'you can only used `stream=True` if `stream` is provided'
+            if isinstance(self.stream[0], Sequence):
+                response = MockAsyncStream(iter(cast(list[MockResponseStreamEvent], self.stream[self.index])))
+            else:
+                response = MockAsyncStream(iter(cast(list[MockResponseStreamEvent], self.stream)))
+        else:
+            assert self.response is not None, 'you can only used `stream=False` if `response` are provided'
+            if isinstance(self.response, Sequence):
+                raise_if_exception(self.response[self.index])
+                response = cast(responses.Response, self.response[self.index])
+            else:
+                raise_if_exception(self.response)
+                response = cast(responses.Response, self.response)
+        self.index += 1
+        return response
+
+
+def get_mock_responses_kwargs(async_open_ai: AsyncOpenAI) -> list[dict[str, Any]]:
+    if isinstance(async_open_ai, MockOpenAIResponses):  # pragma: lax no cover
+        return async_open_ai.response_kwargs
+    else:  # pragma: no cover
+        raise RuntimeError('Not a MockOpenAIResponses instance')
+
+
+def response_message(
+    output_items: Sequence[ResponseOutputItem], *, usage: ResponseUsage | None = None
+) -> responses.Response:
+    return responses.Response(
+        id='123',
+        model='gpt-4o-123',
+        object='response',
+        created_at=1704067200,  # 2024-01-01
+        output=list(output_items),
+        parallel_tool_calls=True,
+        tool_choice='auto',
+        tools=[],
+        usage=usage,
+    )

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/models/test_anthropic.py

@@ -205,7 +205,7 @@ async def test_sync_request_text_response(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
             ModelRequest(parts=[UserPromptPart(content='hello', timestamp=IsNow(tz=timezone.utc))]),
             ModelResponse(
@@ -214,7 +214,7 @@ async def test_sync_request_text_response(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
         ]
     )
@@ -303,7 +303,7 @@ async def test_request_structured_response(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
             ModelRequest(
                 parts=[
@@ -368,7 +368,7 @@ async def test_request_tool_call(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
             ModelRequest(
                 parts=[
@@ -392,7 +392,7 @@ async def test_request_tool_call(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
             ModelRequest(
                 parts=[
@@ -410,7 +410,7 @@ async def test_request_tool_call(allow_model_requests: None):
                 model_name='claude-3-5-haiku-123',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='anthropic',
-                provider_request_id='123',
+                provider_response_id='123',
             ),
         ]
     )
@@ -652,7 +652,7 @@ async def test_stream_structured(allow_model_requests: None):
 
     async with agent.run_stream('') as result:
         assert not result.is_complete
-        chunks = [c async for c in result.stream(debounce_by=None)]
+        chunks = [c async for c in result.stream_output(debounce_by=None)]
 
         # The tool output doesn't echo any content to the stream, so we only get the final payload once when
         # the block starts and once when it ends.
@@ -757,7 +757,7 @@ async def test_image_as_binary_content_tool_response(
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01Kwjzggomz7bv9og51qGFuH',
+                provider_response_id='msg_01Kwjzggomz7bv9og51qGFuH',
             ),
             ModelRequest(
                 parts=[
@@ -795,7 +795,7 @@ async def test_image_as_binary_content_tool_response(
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_015btMBYLTuDnMP7zAeuHQGi',
+                provider_response_id='msg_015btMBYLTuDnMP7zAeuHQGi',
             ),
         ]
     )
@@ -917,7 +917,7 @@ async def test_anthropic_model_instructions(allow_model_requests: None, anthropi
                 model_name='claude-3-opus-20240229',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01Fg1JVgvCYUHWsxrj9GkpEv',
+                provider_response_id='msg_01Fg1JVgvCYUHWsxrj9GkpEv',
             ),
         ]
     )
@@ -965,7 +965,7 @@ I'll provide this information in a clear, helpful way, emphasizing safety withou
                 model_name='claude-3-7-sonnet-20250219',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01BnZvs3naGorn93wjjCDwbd',
+                provider_response_id='msg_01BnZvs3naGorn93wjjCDwbd',
             ),
         ]
     )
@@ -992,7 +992,7 @@ I'll provide this information in a clear, helpful way, emphasizing safety withou
                 model_name='claude-3-7-sonnet-20250219',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id=IsStr(),
+                provider_response_id=IsStr(),
             ),
             ModelRequest(
                 parts=[
@@ -1035,7 +1035,7 @@ I'll keep the format similar to my street-crossing response for consistency.\
                 model_name='claude-3-7-sonnet-20250219',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id=IsStr(),
+                provider_response_id=IsStr(),
             ),
         ]
     )
@@ -1465,7 +1465,7 @@ Several major events are happening today, including:
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01W2YfD2EF8BbAqLRr8ftH4W',
+                provider_response_id='msg_01W2YfD2EF8BbAqLRr8ftH4W',
             ),
         ]
     )
@@ -1521,7 +1521,7 @@ print(f"3 * 12390 = {result}")\
                 model_name='claude-sonnet-4-20250514',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01RJnbK7VMxvS2SyvtyJAQVU',
+                provider_response_id='msg_01RJnbK7VMxvS2SyvtyJAQVU',
             ),
         ]
     )
@@ -1571,7 +1571,7 @@ It's being celebrated as:
                 model_name='gpt-4.1-2025-04-14',
                 timestamp=IsDatetime(),
                 provider_name='openai',
-                provider_request_id='resp_689dc4abe31c81968ed493d15d8810fe0afe80ec3d42722e',
+                provider_response_id='resp_689dc4abe31c81968ed493d15d8810fe0afe80ec3d42722e',
             ),
         ]
     )
@@ -1687,7 +1687,7 @@ async def test_anthropic_tool_output(allow_model_requests: None, anthropic_api_k
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_012TXW181edhmR5JCsQRsBKx',
+                provider_response_id='msg_012TXW181edhmR5JCsQRsBKx',
             ),
             ModelRequest(
                 parts=[
@@ -1720,7 +1720,7 @@ async def test_anthropic_tool_output(allow_model_requests: None, anthropic_api_k
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01K4Fzcf1bhiyLzHpwLdrefj',
+                provider_response_id='msg_01K4Fzcf1bhiyLzHpwLdrefj',
             ),
             ModelRequest(
                 parts=[
@@ -1785,7 +1785,7 @@ async def test_anthropic_text_output_function(allow_model_requests: None, anthro
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01MsqUB7ZyhjGkvepS1tCXp3',
+                provider_response_id='msg_01MsqUB7ZyhjGkvepS1tCXp3',
             ),
             ModelRequest(
                 parts=[
@@ -1816,7 +1816,7 @@ async def test_anthropic_text_output_function(allow_model_requests: None, anthro
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_0142umg4diSckrDtV9vAmmPL',
+                provider_response_id='msg_0142umg4diSckrDtV9vAmmPL',
             ),
         ]
     )
@@ -1874,7 +1874,7 @@ Don't include any text or Markdown fencing before or after.\
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_018YiNXULHGpoKoHkTt6GivG',
+                provider_response_id='msg_018YiNXULHGpoKoHkTt6GivG',
             ),
             ModelRequest(
                 parts=[
@@ -1908,7 +1908,7 @@ Don't include any text or Markdown fencing before or after.\
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01WiRVmLhCrJbJZRqmAWKv3X',
+                provider_response_id='msg_01WiRVmLhCrJbJZRqmAWKv3X',
             ),
         ]
     )
@@ -1966,7 +1966,7 @@ Don't include any text or Markdown fencing before or after.\
                 model_name='claude-3-5-sonnet-20241022',
                 timestamp=IsDatetime(),
                 provider_name='anthropic',
-                provider_request_id='msg_01N2PwwVQo2aBtt6UFhMDtEX',
+                provider_response_id='msg_01N2PwwVQo2aBtt6UFhMDtEX',
             ),
         ]
     )

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/models/test_cohere.py

@@ -443,7 +443,7 @@ async def test_cohere_model_thinking_part(allow_model_requests: None, co_api_key
                 model_name='o3-mini-2025-01-31',
                 timestamp=IsDatetime(),
                 provider_name='openai',
-                provider_request_id='resp_680739f4ad748191bd11096967c37c8b048efc3f8b2a068e',
+                provider_response_id='resp_680739f4ad748191bd11096967c37c8b048efc3f8b2a068e',
             ),
         ]
     )
@@ -468,7 +468,7 @@ async def test_cohere_model_thinking_part(allow_model_requests: None, co_api_key
                 model_name='o3-mini-2025-01-31',
                 timestamp=IsDatetime(),
                 provider_name='openai',
-                provider_request_id='resp_680739f4ad748191bd11096967c37c8b048efc3f8b2a068e',
+                provider_response_id='resp_680739f4ad748191bd11096967c37c8b048efc3f8b2a068e',
             ),
             ModelRequest(
                 parts=[

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/models/test_deepseek.py

@@ -56,7 +56,7 @@ async def test_deepseek_model_thinking_part(allow_model_requests: None, deepseek
                 model_name='deepseek-reasoner',
                 timestamp=IsDatetime(),
                 provider_name='deepseek',
-                provider_request_id='181d9669-2b3a-445e-bd13-2ebff2c378f6',
+                provider_response_id='181d9669-2b3a-445e-bd13-2ebff2c378f6',
             ),
         ]
     )

{pydantic_ai-0.7.6 → pydantic_ai-0.8.1}/tests/models/test_fallback.py

@@ -171,7 +171,7 @@ async def test_first_failed_instrumented_stream(capfire: CaptureLogfire) -> None
     fallback_model = FallbackModel(failure_model_stream, success_model_stream)
     agent = Agent(model=fallback_model, instrument=True)
     async with agent.run_stream('input') as result:
-        assert [c async for c, _is_last in result.stream_structured(debounce_by=None)] == snapshot(
+        assert [c async for c, _is_last in result.stream_responses(debounce_by=None)] == snapshot(
             [
                 ModelResponse(
                     parts=[TextPart(content='hello ')],
@@ -356,7 +356,7 @@ async def test_first_success_streaming() -> None:
     fallback_model = FallbackModel(success_model_stream, failure_model_stream)
     agent = Agent(model=fallback_model)
     async with agent.run_stream('input') as result:
-        assert [c async for c, _is_last in result.stream_structured(debounce_by=None)] == snapshot(
+        assert [c async for c, _is_last in result.stream_responses(debounce_by=None)] == snapshot(
             [
                 ModelResponse(
                     parts=[TextPart(content='hello ')],
@@ -385,7 +385,7 @@ async def test_first_failed_streaming() -> None:
     fallback_model = FallbackModel(failure_model_stream, success_model_stream)
     agent = Agent(model=fallback_model)
     async with agent.run_stream('input') as result:
-        assert [c async for c, _is_last in result.stream_structured(debounce_by=None)] == snapshot(
+        assert [c async for c, _is_last in result.stream_responses(debounce_by=None)] == snapshot(
             [
                 ModelResponse(
                     parts=[TextPart(content='hello ')],
@@ -415,7 +415,7 @@ async def test_all_failed_streaming() -> None:
     agent = Agent(model=fallback_model)
     with pytest.raises(ExceptionGroup) as exc_info:
         async with agent.run_stream('hello') as result:
-            [c async for c, _is_last in result.stream_structured(debounce_by=None)]  # pragma: lax no cover
+            [c async for c, _is_last in result.stream_responses(debounce_by=None)]  # pragma: lax no cover
     assert 'All models from FallbackModel failed' in exc_info.value.args[0]
     exceptions = exc_info.value.exceptions
     assert len(exceptions) == 2