goose-py 0.3.0__tar.gz → 0.3.1__tar.gz

{goose_py-0.3.0 → goose_py-0.3.1}/PKG-INFO

@@ -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.0 → goose_py-0.3.1}/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_py-0.3.0 → goose_py-0.3.1}/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 → goose_py-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "goose-py"
-version = "0.3.0"
+version = "0.3.1"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 authors = [
     "Nash Taylor <nash@chelle.ai>",