goose-py 0.9.0__py3-none-any.whl → 0.9.3__py3-none-any.whl

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/_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/flow.py

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

{goose_py-0.9.0.dist-info → goose_py-0.9.3.dist-info}/METADATA

@@ -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.dist-info → goose_py-0.9.3.dist-info}/RECORD

@@ -1,18 +1,18 @@
 goose/__init__.py,sha256=_qg6tofbArI5KO7fWqaFNY5IcbVELTd7oS6uYijPmzg,185
 goose/agent.py,sha256=g2tPFqEhqBABEjmpNJ2ShfjHDGzmeUXIgOZCKDZ2-40,600
 goose/errors.py,sha256=-0OyZQJWYTRw5YgnCB2_uorVaUsL6Z0QYQO2FqzCiyg,32
-goose/flow.py,sha256=A1bzNIjnoVXRFm6LGhQglxVnKMP0vEVfvTubTol7Kww,58
+goose/flow.py,sha256=mRyGvgyGgy39P5ZDc1L-8_XqPf8HCcrqGrTBDSswn4M,116
 goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/runs.py,sha256=kAHw08ayGbwzSH_p-CfsQDvTcA4H09nUuq6zt3gIFxo,170
 goose/_internal/agent.py,sha256=l0pKfShovrs238sKAr-zubtcacYm82TGwQHcBWVJm2g,5875
 goose/_internal/conversation.py,sha256=1OZQ_N6QZE7L_ZpXG2bjoWkVQ-G7h0JvKkqswmQWG58,1202
-goose/_internal/flow.py,sha256=LX6fdDBBgGNPfpSIUFenJ8-EI6tnNRuGZfUpvSb1isg,4128
+goose/_internal/flow.py,sha256=ZrskFSw5R5rKdVzFGWiQv4-Jwha2ySYnHJd7u2CZ9Yo,4150
 goose/_internal/result.py,sha256=-eZJn-2sPo7rHZ38Sz6IAHXqiJ-Ss39esEoFGimJEBI,155
 goose/_internal/state.py,sha256=wJdXHGw_KPqc1Tuv8cHj1loqLct_KHWBNXzbjBv6GAo,6552
-goose/_internal/store.py,sha256=ZdQwYRe1PKkTUd58p0ja1KyyGhEYGVP4gWvYsqOEWuk,1356
+goose/_internal/store.py,sha256=-HSnvT9PscqyuDh-vekakGBGdhz9QuOzP9Kp1fvS0E8,1407
 goose/_internal/task.py,sha256=vfmWfxhr76nSF7STMZISFqP9ugt3BkHzraMbEZKuozI,5196
 goose/_internal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/_internal/types/agent.py,sha256=rNVt2gEr_m4_8tGFgcdichpPp8xhOS5GY0kN2C4tiE8,2153
-goose_py-0.9.0.dist-info/METADATA,sha256=RUdJVQbLFodUEjPNSTq7XTSx8zcI2W08F0lttsn0nM8,441
-goose_py-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-goose_py-0.9.0.dist-info/RECORD,,
+goose_py-0.9.3.dist-info/METADATA,sha256=4Dy7yIyNgBj_o7Y79lqzjAW49i4cdJP1mEBROFMjtbo,441
+goose_py-0.9.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+goose_py-0.9.3.dist-info/RECORD,,