python-durable 0.2.0__tar.gz → 0.2.1__tar.gz

{python_durable-0.2.0 → python_durable-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-durable
-Version: 0.2.0
+Version: 0.2.1
 Summary: Lightweight workflow durability for Python — make any async workflow resumable after crashes with just a decorator.
 Project-URL: Repository, https://github.com/WillemDeGroef/python-durable
 Author: Willem

{python_durable-0.2.0 → python_durable-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-durable"
-version = "0.2.0"
+version = "0.2.1"
 description = "Lightweight workflow durability for Python — make any async workflow resumable after crashes with just a decorator."
 readme = "README.md"
 license = "MIT"

{python_durable-0.2.0 → python_durable-0.2.1}/src/durable/pydantic_ai.py

@@ -69,6 +69,20 @@ def _deserialize_messages(data: list[dict]) -> list[Any]:
     return result
 
 
+def _serialize_run_result(result: Any) -> dict:
+    """Convert an agent RunResult to a JSON-serializable dict."""
+    output = result.output
+    # If output is a pydantic model, convert to dict
+    if hasattr(output, "model_dump"):
+        output = output.model_dump(mode="json")
+    data: dict[str, Any] = {"output": output}
+    try:
+        data["all_messages"] = _serialize_messages(result.all_messages())
+    except Exception:
+        data["all_messages"] = []
+    return data
+
+
 def _run_id_for_agent(agent_name: str, prompt: str, run_id: str | None) -> str:
     """Generate a deterministic run ID from the agent name and prompt."""
     if run_id:
@@ -207,7 +221,7 @@ class DurableAgent(Generic[AgentDepsT, OutputT]):
             step_id="agent-run",
             **kwargs,
         )
-        return result
+        return _AgentRunResult(result)
 
     async def _do_model_request(
         self,
@@ -225,7 +239,7 @@ class DurableAgent(Generic[AgentDepsT, OutputT]):
         run_kwargs.update(kwargs)
 
         result = await self.agent.run(prompt, **run_kwargs)
-        return _AgentRunResult(result)
+        return _serialize_run_result(result)
 
     async def _do_tool_call(
         self,

{python_durable-0.2.0 → python_durable-0.2.1}/tests/test_pydantic_ai.py

@@ -326,3 +326,19 @@ class TestIntegration:
         r2 = await durable.run("What is 6*7?", run_id="calc-1")
         assert r2.output == "42"
         assert agent.run.call_count == 1
+
+    async def test_durable_agent_with_sqlite_store(self, tmp_path):
+        """Ensure agent results are JSON-serializable for SQLiteStore."""
+        db_path = str(tmp_path / "test.db")
+        wf = Workflow("test-sqlite", db=db_path)
+
+        agent, _ = _mock_agent(output="serialized ok")
+        durable = DurableAgent(agent, wf, name="sqlite-agent")
+
+        r1 = await durable.run("Test prompt", run_id="sqlite-1")
+        assert r1.output == "serialized ok"
+        assert agent.run.call_count == 1
+
+        r2 = await durable.run("Test prompt", run_id="sqlite-1")
+        assert r2.output == "serialized ok"
+        assert agent.run.call_count == 1