pydantic-ai 0.4.9__tar.gz → 0.4.11__tar.gz

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.4.9
+Version: 0.4.11
 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,vertexai]==0.4.9
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,vertexai]==0.4.11
 Provides-Extra: a2a
 Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.4.9; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.4.11; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/tests/test_ag_ui.py

@@ -58,7 +58,7 @@ with contextlib.suppress(ImportError):
     from pydantic_ai.ag_ui import (
         SSE_CONTENT_TYPE,
         StateDeps,
-        _Adapter,  # type: ignore[reportPrivateUsage]
+        run_ag_ui,
     )
 
     has_ag_ui = True
@@ -95,13 +95,12 @@ def simple_result() -> Any:
     )
 
 
-async def collect_events_from_adapter(
-    adapter: _Adapter[AgentDepsT, OutputDataT], *run_inputs: RunAgentInput, deps: AgentDepsT = None
+async def run_and_collect_events(
+    agent: Agent[AgentDepsT, OutputDataT], *run_inputs: RunAgentInput, deps: AgentDepsT = None
 ) -> list[dict[str, Any]]:
-    """Helper function to collect events from an AG-UI adapter run."""
     events = list[dict[str, Any]]()
     for run_input in run_inputs:
-        async for event in adapter.run(run_input, deps=deps):
+        async for event in run_ag_ui(agent, run_input, deps=deps):
             events.append(json.loads(event.removeprefix('data: ')))
     return events
 
@@ -202,7 +201,7 @@ async def test_basic_user_message() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=simple_stream),
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -210,7 +209,7 @@ async def test_basic_user_message() -> None:
         )
     )
 
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == simple_result()
 
@@ -227,9 +226,9 @@ async def test_empty_messages() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=stream_function),
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input()
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == snapshot(
         [
@@ -248,7 +247,7 @@ async def test_multiple_messages() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=simple_stream),
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -272,7 +271,7 @@ async def test_multiple_messages() -> None:
         ),
     )
 
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == simple_result()
 
@@ -282,7 +281,7 @@ async def test_messages_with_history() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=simple_stream),
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -294,7 +293,7 @@ async def test_messages_with_history() -> None:
         ),
     )
 
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == simple_result()
 
@@ -317,7 +316,7 @@ async def test_tool_ag_ui() -> None:
         model=FunctionModel(stream_function=stream_function),
         tools=[send_snapshot, send_custom, current_time],
     )
-    adapter = _Adapter(agent=agent)
+
     thread_id = uuid_str()
     run_inputs = [
         create_input(
@@ -355,7 +354,7 @@ async def test_tool_ag_ui() -> None:
         ),
     ]
 
-    events = await collect_events_from_adapter(adapter, *run_inputs)
+    events = await run_and_collect_events(agent, *run_inputs)
 
     assert events == snapshot(
         [
@@ -427,7 +426,7 @@ async def test_tool_ag_ui_multiple() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=stream_function),
     )
-    adapter = _Adapter(agent=agent)
+
     tool_call_id1 = uuid_str()
     tool_call_id2 = uuid_str()
     run_inputs = [
@@ -486,7 +485,7 @@ async def test_tool_ag_ui_multiple() -> None:
         ),
     ]
 
-    events = await collect_events_from_adapter(adapter, *run_inputs)
+    events = await run_and_collect_events(agent, *run_inputs)
 
     assert events == snapshot(
         [
@@ -562,7 +561,7 @@ async def test_tool_ag_ui_parts() -> None:
             yield '{"get_weather": "Tool result"}'
 
     agent = Agent(model=FunctionModel(stream_function=stream_function))
-    adapter = _Adapter(agent=agent)
+
     run_inputs = [
         (
             first_input := create_input(
@@ -600,7 +599,7 @@ async def test_tool_ag_ui_parts() -> None:
             thread_id=first_input.thread_id,
         ),
     ]
-    events = await collect_events_from_adapter(adapter, *run_inputs)
+    events = await run_and_collect_events(agent, *run_inputs)
 
     assert events == snapshot(
         [
@@ -675,14 +674,14 @@ async def test_tool_local_single_event() -> None:
         model=FunctionModel(stream_function=stream_function),
         tools=[send_snapshot],
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
             content='Please call send_snapshot',
         ),
     )
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == snapshot(
         [
@@ -744,14 +743,14 @@ async def test_tool_local_multiple_events() -> None:
         model=FunctionModel(stream_function=stream_function),
         tools=[send_custom],
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
             content='Please call send_custom',
         ),
     )
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == snapshot(
         [
@@ -812,7 +811,6 @@ async def test_tool_local_parts() -> None:
         tools=[send_snapshot, send_custom, current_time],
     )
 
-    adapter = _Adapter(agent=agent)
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -820,7 +818,7 @@ async def test_tool_local_parts() -> None:
         ),
     )
 
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == snapshot(
         [
@@ -873,7 +871,7 @@ async def test_thinking() -> None:
     agent = Agent(
         model=FunctionModel(stream_function=stream_function),
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -881,7 +879,7 @@ async def test_thinking() -> None:
         ),
     )
 
-    events = await collect_events_from_adapter(adapter, run_input)
+    events = await run_and_collect_events(agent, run_input)
 
     assert events == snapshot(
         [
@@ -933,7 +931,7 @@ async def test_tool_local_then_ag_ui() -> None:
         model=FunctionModel(stream_function=stream_function),
         tools=[current_time],
     )
-    adapter = _Adapter(agent=agent)
+
     run_inputs = [
         (
             first_input := create_input(
@@ -989,7 +987,7 @@ async def test_tool_local_then_ag_ui() -> None:
             thread_id=first_input.thread_id,
         ),
     ]
-    events = await collect_events_from_adapter(adapter, *run_inputs)
+    events = await run_and_collect_events(agent, *run_inputs)
 
     assert events == snapshot(
         [
@@ -1068,7 +1066,7 @@ async def test_request_with_state() -> None:
         deps_type=StateDeps[StateInt],  # type: ignore[reportUnknownArgumentType]
         prepare_tools=store_state,
     )
-    adapter = _Adapter(agent=agent)
+
     run_inputs = [
         create_input(
             UserMessage(
@@ -1102,7 +1100,7 @@ async def test_request_with_state() -> None:
 
     for run_input in run_inputs:
         events = list[dict[str, Any]]()
-        async for event in adapter.run(run_input, deps=deps):
+        async for event in run_ag_ui(agent, run_input, deps=deps):
             events.append(json.loads(event.removeprefix('data: ')))
 
         assert events == simple_result()
@@ -1122,7 +1120,7 @@ async def test_request_with_state() -> None:
 
 async def test_request_with_state_without_handler() -> None:
     agent = Agent(model=FunctionModel(stream_function=simple_stream))
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -1135,7 +1133,7 @@ async def test_request_with_state_without_handler() -> None:
         UserError,
         match='AG-UI state is provided but `deps` of type `NoneType` does not implement the `StateHandler` protocol: it needs to be a dataclass with a non-optional `state` field.',
     ):
-        async for _ in adapter.run(run_input):
+        async for _ in run_ag_ui(agent, run_input):
             pass
 
 
@@ -1155,7 +1153,7 @@ async def test_request_with_state_with_custom_handler() -> None:
         deps_type=CustomStateDeps,
         prepare_tools=store_state,
     )
-    adapter = _Adapter(agent=agent)
+
     run_input = create_input(
         UserMessage(
             id='msg_1',
@@ -1164,7 +1162,7 @@ async def test_request_with_state_with_custom_handler() -> None:
         state={'value': 42},
     )
 
-    async for _ in adapter.run(run_input, deps=CustomStateDeps(state={'value': 0})):
+    async for _ in run_ag_ui(agent, run_input, deps=CustomStateDeps(state={'value': 0})):
         pass
 
     assert seen_states[-1] == {'value': 42}
@@ -1183,7 +1181,6 @@ async def test_concurrent_runs() -> None:
     async def get_state(ctx: RunContext[StateDeps[StateInt]]) -> int:
         return ctx.deps.state.value
 
-    adapter = _Adapter(agent=agent)
     concurrent_tasks: list[asyncio.Task[list[dict[str, Any]]]] = []
 
     for i in range(5):  # Test with 5 concurrent runs
@@ -1196,7 +1193,7 @@ async def test_concurrent_runs() -> None:
             thread_id=f'test_thread_{i}',
         )
 
-        task = asyncio.create_task(collect_events_from_adapter(adapter, run_input, deps=StateDeps(StateInt())))
+        task = asyncio.create_task(run_and_collect_events(agent, run_input, deps=StateDeps(StateInt())))
         concurrent_tasks.append(task)
 
     results = await asyncio.gather(*concurrent_tasks)

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/tests/test_agent.py

@@ -3866,7 +3866,7 @@ def test_prepare_output_tools():
     )
 
 
-async def test_context_manager():
+async def test_explicit_context_manager():
     try:
         from pydantic_ai.mcp import MCPServerStdio
     except ImportError:  # pragma: lax no cover
@@ -3886,6 +3886,47 @@ async def test_context_manager():
             assert server2.is_running
 
 
+async def test_implicit_context_manager():
+    try:
+        from pydantic_ai.mcp import MCPServerStdio
+    except ImportError:  # pragma: lax no cover
+        pytest.skip('mcp is not installed')
+
+    server1 = MCPServerStdio('python', ['-m', 'tests.mcp_server'])
+    server2 = MCPServerStdio('python', ['-m', 'tests.mcp_server'])
+    toolset = CombinedToolset([server1, PrefixedToolset(server2, 'prefix')])
+    agent = Agent('test', toolsets=[toolset])
+
+    async with agent.iter(
+        user_prompt='Hello',
+    ):
+        assert server1.is_running
+        assert server2.is_running
+
+
+def test_parallel_mcp_calls():
+    try:
+        from pydantic_ai.mcp import MCPServerStdio
+    except ImportError:  # pragma: lax no cover
+        pytest.skip('mcp is not installed')
+
+    async def call_tools_parallel(messages: list[ModelMessage], info: AgentInfo) -> ModelResponse:
+        if len(messages) == 1:
+            return ModelResponse(
+                parts=[
+                    ToolCallPart(tool_name='get_none'),
+                    ToolCallPart(tool_name='get_multiple_items'),
+                ]
+            )
+        else:
+            return ModelResponse(parts=[TextPart('finished')])
+
+    server = MCPServerStdio('python', ['-m', 'tests.mcp_server'])
+    agent = Agent(FunctionModel(call_tools_parallel), toolsets=[server])
+    result = agent.run_sync()
+    assert result.output == snapshot('finished')
+
+
 def test_set_mcp_sampling_model():
     try:
         from pydantic_ai.mcp import MCPServerStdio

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/tests/test_parts_manager.py

@@ -82,14 +82,15 @@ def test_handle_dovetailed_text_deltas():
 
 def test_handle_text_deltas_with_think_tags():
     manager = ModelResponsePartsManager()
+    thinking_tags = ('<think>', '</think>')
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='pre-', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='pre-', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartStartEvent(index=0, part=TextPart(content='pre-', part_kind='text'), event_kind='part_start')
     )
     assert manager.get_parts() == snapshot([TextPart(content='pre-', part_kind='text')])
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartDeltaEvent(
             index=0, delta=TextPartDelta(content_delta='thinking', part_delta_kind='text'), event_kind='part_delta'
@@ -97,7 +98,7 @@ def test_handle_text_deltas_with_think_tags():
     )
     assert manager.get_parts() == snapshot([TextPart(content='pre-thinking', part_kind='text')])
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='<think>', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='<think>', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartStartEvent(index=1, part=ThinkingPart(content='', part_kind='thinking'), event_kind='part_start')
     )
@@ -105,7 +106,7 @@ def test_handle_text_deltas_with_think_tags():
         [TextPart(content='pre-thinking', part_kind='text'), ThinkingPart(content='', part_kind='thinking')]
     )
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartDeltaEvent(
             index=1,
@@ -117,7 +118,7 @@ def test_handle_text_deltas_with_think_tags():
         [TextPart(content='pre-thinking', part_kind='text'), ThinkingPart(content='thinking', part_kind='thinking')]
     )
 
-    event = manager.handle_text_delta(vendor_part_id='content', content=' more', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content=' more', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartDeltaEvent(
             index=1, delta=ThinkingPartDelta(content_delta=' more', part_delta_kind='thinking'), event_kind='part_delta'
@@ -130,10 +131,10 @@ def test_handle_text_deltas_with_think_tags():
         ]
     )
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='</think>', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='</think>', thinking_tags=thinking_tags)
     assert event is None
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='post-', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='post-', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartStartEvent(index=2, part=TextPart(content='post-', part_kind='text'), event_kind='part_start')
     )
@@ -145,7 +146,7 @@ def test_handle_text_deltas_with_think_tags():
         ]
     )
 
-    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', extract_think_tags=True)
+    event = manager.handle_text_delta(vendor_part_id='content', content='thinking', thinking_tags=thinking_tags)
     assert event == snapshot(
         PartDeltaEvent(
             index=2, delta=TextPartDelta(content_delta='thinking', part_delta_kind='text'), event_kind='part_delta'

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/tests/test_thinking_part.py

@@ -8,25 +8,72 @@ from pydantic_ai.messages import ModelResponsePart, TextPart, ThinkingPart, Thin
 
 
 @pytest.mark.parametrize(
-    ('content', 'parts'),
+    'thinking_tags, content, parts',
     [
-        ('foo bar', [TextPart(content='foo bar')]),
+        # default <think>…</think> cases
         (
+            ('<think>', '</think>'),
+            'foo bar',
+            [TextPart(content='foo bar')],
+        ),
+        (
+            ('<think>', '</think>'),
             'foo bar<think>thinking</think>',
             [TextPart(content='foo bar'), ThinkingPart(content='thinking')],
         ),
         (
+            ('<think>', '</think>'),
             'foo bar<think>thinking</think>baz',
-            [TextPart(content='foo bar'), ThinkingPart(content='thinking'), TextPart(content='baz')],
+            [
+                TextPart(content='foo bar'),
+                ThinkingPart(content='thinking'),
+                TextPart(content='baz'),
+            ],
         ),
         (
+            ('<think>', '</think>'),
             'foo bar<think>thinking',
             [TextPart(content='foo bar'), TextPart(content='thinking')],
         ),
+        (
+            ('<think>', '</think>'),
+            'foo bar<custom>thinking</custom>baz',
+            [TextPart(content='foo bar<custom>thinking</custom>baz')],
+        ),
+        # custom <custom>…</custom> cases
+        (
+            ('<custom>', '</custom>'),
+            'foo bar',
+            [TextPart(content='foo bar')],
+        ),
+        (
+            ('<custom>', '</custom>'),
+            'foo bar<custom>thinking</custom>',
+            [TextPart(content='foo bar'), ThinkingPart(content='thinking')],
+        ),
+        (
+            ('<custom>', '</custom>'),
+            'foo bar<custom>thinking</custom>baz',
+            [
+                TextPart(content='foo bar'),
+                ThinkingPart(content='thinking'),
+                TextPart(content='baz'),
+            ],
+        ),
+        (
+            ('<custom>', '</custom>'),
+            'foo bar<custom>thinking',
+            [TextPart(content='foo bar'), TextPart(content='thinking')],
+        ),
+        (
+            ('<custom>', '</custom>'),
+            'foo bar<think>thinking</think>baz',
+            [TextPart(content='foo bar<think>thinking</think>baz')],
+        ),
     ],
 )
-def test_split_content_into_text_and_thinking(content: str, parts: list[ModelResponsePart]):
-    assert split_content_into_text_and_thinking(content) == parts
+def test_split_content(thinking_tags: tuple[str, str], content: str, parts: list[ModelResponsePart]):
+    assert split_content_into_text_and_thinking(content, thinking_tags) == parts
 
 
 def test_thinking_part_delta_applies_both_content_and_signature():

{pydantic_ai-0.4.9 → pydantic_ai-0.4.11}/tests/test_tools.py

@@ -13,7 +13,16 @@ from typing_extensions import TypedDict
 
 from pydantic_ai import Agent, RunContext, Tool, UserError
 from pydantic_ai.exceptions import ModelRetry, UnexpectedModelBehavior
-from pydantic_ai.messages import ModelMessage, ModelRequest, ModelResponse, TextPart, ToolCallPart, ToolReturnPart
+from pydantic_ai.messages import (
+    ModelMessage,
+    ModelRequest,
+    ModelResponse,
+    TextPart,
+    ToolCallPart,
+    ToolReturn,
+    ToolReturnPart,
+    UserPromptPart,
+)
 from pydantic_ai.models.function import AgentInfo, FunctionModel
 from pydantic_ai.models.test import TestModel
 from pydantic_ai.output import DeferredToolCalls, ToolOutput
@@ -21,8 +30,9 @@ from pydantic_ai.tools import ToolDefinition
 from pydantic_ai.toolsets.deferred import DeferredToolset
 from pydantic_ai.toolsets.function import FunctionToolset
 from pydantic_ai.toolsets.prefixed import PrefixedToolset
+from pydantic_ai.usage import Usage
 
-from .conftest import IsStr
+from .conftest import IsDatetime, IsStr
 
 
 def test_tool_no_ctx():
@@ -1321,3 +1331,91 @@ def test_output_type_deferred_tool_calls_by_itself():
 def test_output_type_empty():
     with pytest.raises(UserError, match='At least one output type must be provided.'):
         Agent(TestModel(), output_type=[])
+
+
+def test_parallel_tool_return():
+    def llm(messages: list[ModelMessage], info: AgentInfo) -> ModelResponse:
+        if len(messages) == 1:
+            return ModelResponse(
+                parts=[ToolCallPart('get_price', {'fruit': 'apple'}), ToolCallPart('get_price', {'fruit': 'banana'})]
+            )
+        else:
+            return ModelResponse(
+                parts=[
+                    TextPart('Done!'),
+                ]
+            )
+
+    agent = Agent(FunctionModel(llm))
+
+    @agent.tool_plain
+    def get_price(fruit: str) -> ToolReturn:
+        return ToolReturn(
+            return_value=10.0,
+            content=f'The price of {fruit} is 10.0',
+            metadata={'foo': 'bar'},
+        )
+
+    result = agent.run_sync('What do an apple and a banana cost?')
+
+    assert result.all_messages() == snapshot(
+        [
+            ModelRequest(
+                parts=[
+                    UserPromptPart(
+                        content='What do an apple and a banana cost?',
+                        timestamp=IsDatetime(),
+                    )
+                ]
+            ),
+            ModelResponse(
+                parts=[
+                    ToolCallPart(
+                        tool_name='get_price',
+                        args={'fruit': 'apple'},
+                        tool_call_id=IsStr(),
+                    ),
+                    ToolCallPart(
+                        tool_name='get_price',
+                        args={'fruit': 'banana'},
+                        tool_call_id=IsStr(),
+                    ),
+                ],
+                usage=Usage(requests=1, request_tokens=58, response_tokens=10, total_tokens=68),
+                model_name='function:llm:',
+                timestamp=IsDatetime(),
+            ),
+            ModelRequest(
+                parts=[
+                    ToolReturnPart(
+                        tool_name='get_price',
+                        content=10.0,
+                        tool_call_id=IsStr(),
+                        metadata={'foo': 'bar'},
+                        timestamp=IsDatetime(),
+                    ),
+                    ToolReturnPart(
+                        tool_name='get_price',
+                        content=10.0,
+                        tool_call_id=IsStr(),
+                        metadata={'foo': 'bar'},
+                        timestamp=IsDatetime(),
+                    ),
+                    UserPromptPart(
+                        content='The price of apple is 10.0',
+                        timestamp=IsDatetime(),
+                    ),
+                    UserPromptPart(
+                        content='The price of banana is 10.0',
+                        timestamp=IsDatetime(),
+                    ),
+                ]
+            ),
+            ModelResponse(
+                parts=[TextPart(content='Done!')],
+                usage=Usage(requests=1, request_tokens=76, response_tokens=11, total_tokens=87),
+                model_name='function:llm:',
+                timestamp=IsDatetime(),
+            ),
+        ]
+    )