pydantic-ai 1.9.0__tar.gz → 1.9.1__tar.gz

{pydantic_ai-1.9.0 → pydantic_ai-1.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 1.9.0
+Version: 1.9.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
@@ -26,25 +26,25 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
-Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,fastmcp,google,groq,huggingface,logfire,mcp,mistral,openai,retries,temporal,ui,vertexai]==1.9.0
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,fastmcp,google,groq,huggingface,logfire,mcp,mistral,openai,retries,temporal,ui,vertexai]==1.9.1
 Provides-Extra: a2a
 Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: dbos
-Requires-Dist: pydantic-ai-slim[dbos]==1.9.0; extra == 'dbos'
+Requires-Dist: pydantic-ai-slim[dbos]==1.9.1; extra == 'dbos'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==1.9.0; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==1.9.1; extra == 'examples'
 Provides-Extra: outlines-llamacpp
-Requires-Dist: pydantic-ai-slim[outlines-llamacpp]==1.9.0; extra == 'outlines-llamacpp'
+Requires-Dist: pydantic-ai-slim[outlines-llamacpp]==1.9.1; extra == 'outlines-llamacpp'
 Provides-Extra: outlines-mlxlm
-Requires-Dist: pydantic-ai-slim[outlines-mlxlm]==1.9.0; extra == 'outlines-mlxlm'
+Requires-Dist: pydantic-ai-slim[outlines-mlxlm]==1.9.1; extra == 'outlines-mlxlm'
 Provides-Extra: outlines-sglang
-Requires-Dist: pydantic-ai-slim[outlines-sglang]==1.9.0; extra == 'outlines-sglang'
+Requires-Dist: pydantic-ai-slim[outlines-sglang]==1.9.1; extra == 'outlines-sglang'
 Provides-Extra: outlines-transformers
-Requires-Dist: pydantic-ai-slim[outlines-transformers]==1.9.0; extra == 'outlines-transformers'
+Requires-Dist: pydantic-ai-slim[outlines-transformers]==1.9.1; extra == 'outlines-transformers'
 Provides-Extra: outlines-vllm-offline
-Requires-Dist: pydantic-ai-slim[outlines-vllm-offline]==1.9.0; extra == 'outlines-vllm-offline'
+Requires-Dist: pydantic-ai-slim[outlines-vllm-offline]==1.9.1; extra == 'outlines-vllm-offline'
 Provides-Extra: prefect
-Requires-Dist: pydantic-ai-slim[prefect]==1.9.0; extra == 'prefect'
+Requires-Dist: pydantic-ai-slim[prefect]==1.9.1; extra == 'prefect'
 Description-Content-Type: text/markdown
 
 <div align="center">

{pydantic_ai-1.9.0 → pydantic_ai-1.9.1}/tests/test_ag_ui.py

@@ -1154,15 +1154,21 @@ async def test_request_with_state() -> None:
         ),
     ]
 
-    deps = StateDeps(StateInt(value=0))
+    seen_deps_states: list[int] = []
 
     for run_input in run_inputs:
         events = list[dict[str, Any]]()
-        async for event in run_ag_ui(agent, run_input, deps=deps):
+        deps = StateDeps(StateInt(value=0))
+
+        async def on_complete(result: AgentRunResult[Any]):
+            seen_deps_states.append(deps.state.value)
+
+        async for event in run_ag_ui(agent, run_input, deps=deps, on_complete=on_complete):
             events.append(json.loads(event.removeprefix('data: ')))
 
         assert events == simple_result()
     assert seen_states == snapshot([41, 0, 0, 42])
+    assert seen_deps_states == snapshot([42, 1, 1, 43])
 
 
 async def test_request_with_state_without_handler() -> None:
@@ -1275,8 +1281,10 @@ async def test_concurrent_runs() -> None:
 async def test_to_ag_ui() -> None:
     """Test the agent.to_ag_ui method."""
 
-    agent = Agent(model=FunctionModel(stream_function=simple_stream))
-    app = agent.to_ag_ui()
+    agent = Agent(model=FunctionModel(stream_function=simple_stream), deps_type=StateDeps[StateInt])
+
+    deps = StateDeps(StateInt(value=0))
+    app = agent.to_ag_ui(deps=deps)
     async with LifespanManager(app):
         transport = httpx.ASGITransport(app)
         async with httpx.AsyncClient(transport=transport) as client:
@@ -1286,6 +1294,7 @@ async def test_to_ag_ui() -> None:
                     id='msg_1',
                     content='Hello, world!',
                 ),
+                state=StateInt(value=42),
             )
             async with client.stream(
                 'POST',
@@ -1301,6 +1310,9 @@ async def test_to_ag_ui() -> None:
 
             assert events == simple_result()
 
+    # Verify the state was not mutated by the run
+    assert deps.state.value == 0
+
 
 async def test_callback_sync() -> None:
     """Test that sync callbacks work correctly."""