goose-py 0.3.4__tar.gz → 0.3.6__tar.gz

{goose_py-0.3.4 → goose_py-0.3.6}/PKG-INFO

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

@@ -104,7 +104,7 @@ class AgentResponseDump(TypedDict):
     task_name: str
     model: str
     system_message: str
-    input_messages: str
+    input_messages: list[str]
     output_message: str
     input_cost: float
     output_cost: float
@@ -166,25 +166,28 @@ 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.render() for message in self.input_messages]
+        for message in minimized_input_messages:
+            for part in message["content"]:
+                if part["type"] == "image_url":
+                    part["image_url"] = "__MEDIA__"
         minimized_input_messages = [
-            message.model_dump() for message in self.input_messages
+            json.dumps(message) for message in minimized_input_messages
         ]
-        for message in minimized_input_messages:
-            if message["type"] == "image_url":
-                message["content"] = b"__MEDIA__"
 
         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.4 → goose_py-0.3.6}/pyproject.toml

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