goose-py 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

goose/agent.py

@@ -112,7 +112,7 @@ class AgentResponse[R: BaseModel](BaseModel):
     }
 
     response: R
-    run_name: str
+    run_id: str
     flow_name: str
     task_name: str
     model: GeminiModel
@@ -149,11 +149,11 @@ class Agent:
         self,
         *,
         flow_name: str,
-        run_name: str,
+        run_id: str,
         logger: Callable[[AgentResponse[Any]], Awaitable[None]] | None = None,
     ) -> None:
         self.flow_name = flow_name
-        self.run_name = run_name
+        self.run_id = run_id
         self.logger = logger
 
     async def __call__[R: BaseModel](
@@ -189,7 +189,7 @@ class Agent:
         end_time = datetime.now()
         agent_response = AgentResponse(
             response=parsed_response,
-            run_name=self.run_name,
+            run_id=self.run_id,
             flow_name=self.flow_name,
             task_name=task_name,
             model=model,

goose/flow.py

@@ -122,12 +122,17 @@ class FlowRun:
     def __init__(self) -> None:
         self._node_states: dict[tuple[str, int], str] = {}
         self._last_requested_indices: dict[str, int] = {}
-        self._name = ""
+        self._flow_name = ""
+        self._id = ""
         self._agent: Agent | None = None
 
     @property
-    def name(self) -> str:
-        return self._name
+    def flow_name(self) -> str:
+        return self._flow_name
+
+    @property
+    def id(self) -> str:
+        return self._id
 
     @property
     def agent(self) -> Agent:
@@ -169,14 +174,16 @@ class FlowRun:
                 last_input_hash=0,
             )
 
-    def start(self, *, name: str) -> None:
+    def start(self, *, flow_name: str, run_id: str) -> None:
         self._last_requested_indices = {}
-        self._name = name
-        self._agent = Agent(flow_name=self.name, run_name=name)
+        self._flow_name = flow_name
+        self._id = run_id
+        self._agent = Agent(flow_name=self.flow_name, run_id=self.id)
 
     def end(self) -> None:
         self._last_requested_indices = {}
-        self._name = ""
+        self._flow_name = ""
+        self._id = ""
         self._agent = None
 
     def dump(self) -> SerializedFlowRun:
@@ -226,14 +233,18 @@ class Flow[**P]:
         return run
 
     @contextmanager
-    def start_run(self, *, name: str, run: FlowRun | None = None) -> Iterator[FlowRun]:
-        if run is None:
+    def start_run(
+        self, *, run_id: str, preload: FlowRun | None = None
+    ) -> Iterator[FlowRun]:
+        if preload is None:
             run = FlowRun()
+        else:
+            run = preload
 
         old_run = _current_flow_run.get()
         _current_flow_run.set(run)
 
-        run.start(name=name)
+        run.start(flow_name=self.name, run_id=run_id)
         yield run
         run.end()
 

{goose_py-0.3.0.dist-info → goose_py-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goose-py
-Version: 0.3.0
+Version: 0.3.1
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Home-page: https://github.com/chelle-ai/goose
 Keywords: ai,yaml,configuration,llm

goose_py-0.3.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+goose/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+goose/agent.py,sha256=cJO6Jlh8En_FGLzjelm0w1pM4rsVqfIOpfKUdpkord4,5650
+goose/errors.py,sha256=-0OyZQJWYTRw5YgnCB2_uorVaUsL6Z0QYQO2FqzCiyg,32
+goose/flow.py,sha256=_ws3Jrp96E5UG7D0dgT0_c1RDs07H-hgJ0mSAmJf_yE,11735
+goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+goose_py-0.3.1.dist-info/METADATA,sha256=wOP3QDcyVUJg-vjBw2KBb1xWdw_NA-50wgJtwg24WQc,1106
+goose_py-0.3.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+goose_py-0.3.1.dist-info/RECORD,,

goose_py-0.3.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-goose/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-goose/agent.py,sha256=j8r9p7PqIdXCX_Vl8pUTrF2SshJXUDitgJlCV_C6hdw,5662
-goose/errors.py,sha256=-0OyZQJWYTRw5YgnCB2_uorVaUsL6Z0QYQO2FqzCiyg,32
-goose/flow.py,sha256=5a9lIHTyW5hLeMdt1AfIxe5z-VlTti_3OJ1o4f_nuoo,11458
-goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-goose_py-0.3.0.dist-info/METADATA,sha256=dMNVLzZZ7KyEUacFGWqLeNJuv0ni5-GS_FP9ifaF4bM,1106
-goose_py-0.3.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-goose_py-0.3.0.dist-info/RECORD,,