pydantic-ai 1.0.15__tar.gz → 1.0.16__tar.gz

{pydantic_ai-1.0.15 → pydantic_ai-1.0.16}/PKG-INFO

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

{pydantic_ai-1.0.15 → pydantic_ai-1.0.16}/tests/graph/test_persistence.py

@@ -2,6 +2,7 @@
 from __future__ import annotations as _annotations
 
 import json
+from copy import deepcopy
 from dataclasses import dataclass
 from datetime import datetime, timezone
 
@@ -345,3 +346,59 @@ async def test_record_lookup_error(persistence_cls: type[BaseStatePersistence]):
 def test_snapshot_type_adapter_error():
     with pytest.raises(RuntimeError, match='Unable to build a Pydantic schema for `BaseNode` without setting'):
         build_snapshot_list_type_adapter(int, int)
+
+
+async def test_full_state_persistence_snapshot_state_stability():
+    @dataclass
+    class CountDownState:
+        counter: int
+
+    @dataclass
+    class CountDown(BaseNode[CountDownState, None, int]):
+        async def run(self, ctx: GraphRunContext[CountDownState]) -> CountDown | End[int]:
+            if ctx.state.counter <= 0:
+                return End(ctx.state.counter)
+            ctx.state.counter -= 1
+            return CountDown()
+
+    persistence = FullStatePersistence()
+    state = CountDownState(counter=3)
+    count_down_graph = Graph(nodes=[CountDown])
+
+    await count_down_graph.initialize(CountDown(), state=state, persistence=persistence)
+
+    done = False
+    while not done:
+        history = deepcopy(persistence.history)
+        async with count_down_graph.iter_from_persistence(persistence) as run:
+            result = await run.next()
+            done = isinstance(result, End)
+
+        for i in range(len(history)):
+            assert history[i].id == persistence.history[i].id
+            assert history[i].state == persistence.history[i].state, 'State should not change'
+
+
+async def test_simple_state_persistence_snapshot_state_stability():
+    @dataclass
+    class CountDownState:
+        counter: int
+
+    @dataclass
+    class CountDown(BaseNode[CountDownState, None, int]):
+        async def run(self, ctx: GraphRunContext[CountDownState]) -> CountDown | End[int]:
+            ctx.state.counter -= 1
+            return CountDown()
+
+    persistence = SimpleStatePersistence()
+    state = CountDownState(counter=3)
+    count_down_graph = Graph(nodes=[CountDown])
+
+    await count_down_graph.initialize(CountDown(), state=state, persistence=persistence)
+
+    last_snapshot = persistence.last_snapshot
+    async with count_down_graph.iter_from_persistence(persistence) as run:
+        await run.next()
+
+    assert last_snapshot and last_snapshot.state.counter == 3
+    assert persistence.last_snapshot and persistence.last_snapshot.state.counter == 2