goose-py 0.3.3__tar.gz → 0.3.5__tar.gz

{goose_py-0.3.3 → goose_py-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goose-py
-Version: 0.3.3
+Version: 0.3.5
 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.3 → goose_py-0.3.5}/goose/agent.py

@@ -166,25 +166,26 @@ class AgentResponse[R: BaseModel](BaseModel):
         if self.system is None:
             minimized_system_message = ""
         else:
-            minimized_system_message = self.system.model_dump()
-            for part in minimized_system_message["parts"]:
+            minimized_system_message = self.system.render()
+            for part in minimized_system_message["content"]:
                 if part["type"] == "image_url":
-                    part["content"] = b"__MEDIA__"
+                    part["image_url"] = "__MEDIA__"
+            minimized_system_message = json.dumps(minimized_system_message)
 
-        minimized_input_messages = [
-            message.model_dump() for message in self.input_messages
-        ]
+        minimized_input_messages = [message.render() for message in self.input_messages]
         for message in minimized_input_messages:
-            if message["type"] == "image_url":
-                message["content"] = b"__MEDIA__"
+            for part in message["content"]:
+                if part["type"] == "image_url":
+                    part["image_url"] = "__MEDIA__"
+        minimized_input_messages = json.dumps(minimized_input_messages)
 
         return {
             "run_id": self.run_id,
             "flow_name": self.flow_name,
             "task_name": self.task_name,
             "model": self.model.value,
-            "system_message": json.dumps(minimized_system_message),
-            "input_messages": json.dumps(minimized_input_messages),
+            "system_message": minimized_system_message,
+            "input_messages": minimized_input_messages,
             "output_message": self.response.model_dump_json(),
             "input_tokens": self.input_tokens,
             "output_tokens": self.output_tokens,

{goose_py-0.3.3 → goose_py-0.3.5}/goose/flow.py

@@ -236,8 +236,8 @@ class Flow[**P]:
         fn: Callable[P, Awaitable[None]],
         /,
         *,
-        store: IFlowRunStore | None = None,
         name: str | None = None,
+        store: IFlowRunStore | None = None,
         agent_logger: IAgentLogger | None = None,
     ) -> None:
         self._fn = fn
@@ -389,20 +389,24 @@ def task[**P, R: Result](
 def flow[**P](fn: Callable[P, Awaitable[None]], /) -> Flow[P]: ...
 @overload
 def flow[**P](
-    *, name: str | None = None, agent_logger: IAgentLogger | None = None
+    *,
+    name: str | None = None,
+    store: IFlowRunStore | None = None,
+    agent_logger: IAgentLogger | None = None,
 ) -> Callable[[Callable[P, Awaitable[None]]], Flow[P]]: ...
 def flow[**P](
     fn: Callable[P, Awaitable[None]] | None = None,
     /,
     *,
     name: str | None = None,
+    store: IFlowRunStore | None = None,
     agent_logger: IAgentLogger | None = None,
 ) -> Flow[P] | Callable[[Callable[P, Awaitable[None]]], Flow[P]]:
     if fn is None:
 
         def decorator(fn: Callable[P, Awaitable[None]]) -> Flow[P]:
-            return Flow(fn, name=name, agent_logger=agent_logger)
+            return Flow(fn, name=name, store=store, agent_logger=agent_logger)
 
         return decorator
 
-    return Flow(fn, name=name, agent_logger=agent_logger)
+    return Flow(fn, name=name, store=store, agent_logger=agent_logger)

{goose_py-0.3.3 → goose_py-0.3.5}/pyproject.toml

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