goose-py 0.9.0__tar.gz → 0.9.3__tar.gz

{goose_py-0.9.0 → goose_py-0.9.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.9.0
+Version: 0.9.3
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Author-email: Nash Taylor <nash@chelle.ai>, Joshua Cook <joshua@chelle.ai>, Michael Sankur <michael@chelle.ai>
 Requires-Python: >=3.12

{goose_py-0.9.0 → goose_py-0.9.3}/goose/_internal/flow.py

@@ -70,7 +70,7 @@ class Flow[FlowArgumentsT: FlowArguments]:
 
         run.start(flow_name=self.name, run_id=run_id, agent_logger=self._agent_logger)
         yield run
-        await self._store.save(run=run)
+        await self._store.save(run_id=run_id, run=run.dump())
         run.end()
 
         set_current_flow_run(old_run)

{goose_py-0.9.0 → goose_py-0.9.3}/goose/_internal/store.py

@@ -7,9 +7,9 @@ from goose._internal.state import FlowArguments, SerializedFlowRun
 
 
 class IFlowRunStore[FlowArgumentsT: FlowArguments](Protocol):
-    def __init__(self, *, flow_name: str) -> None: ...
+    def __init__(self, *, flow_name: str, flow_arguments_model: type[FlowArgumentsT]) -> None: ...
     async def get(self, *, run_id: str) -> FlowRun[FlowArgumentsT] | None: ...
-    async def save(self, *, run: FlowRun[FlowArgumentsT]) -> None: ...
+    async def save(self, *, run_id: str, run: SerializedFlowRun) -> None: ...
     async def delete(self, *, run_id: str) -> None: ...
 
 
@@ -26,8 +26,8 @@ class InMemoryFlowRunStore[FlowArgumentsT: FlowArguments](IFlowRunStore[FlowArgu
                 serialized_flow_run=serialized_flow_run, flow_arguments_model=self._flow_arguments_model
             )
 
-    async def save(self, *, run: FlowRun[FlowArgumentsT]) -> None:
-        self._runs[run.id] = run.dump()
+    async def save(self, *, run_id: str, run: SerializedFlowRun) -> None:
+        self._runs[run_id] = run
 
     async def delete(self, *, run_id: str) -> None:
         self._runs.pop(run_id, None)

goose_py-0.9.3/goose/flow.py

@@ -0,0 +1,3 @@
+from goose._internal.flow import Flow, FlowArguments, IGenerator
+
+__all__ = ["Flow", "FlowArguments", "IGenerator"]

{goose_py-0.9.0 → goose_py-0.9.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "goose-py"
-version = "0.9.0"
+version = "0.9.3"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 readme = "README.md"
 authors = [

{goose_py-0.9.0 → goose_py-0.9.3}/uv.lock

@@ -234,7 +234,7 @@ wheels = [
 
 [[package]]
 name = "goose-py"
-version = "0.9.0"
+version = "0.9.3"
 source = { editable = "." }
 dependencies = [
     { name = "jsonpath-ng" },

goose_py-0.9.0/goose/flow.py

@@ -1,3 +0,0 @@
-from goose._internal.flow import Flow
-
-__all__ = ["Flow"]