goose-py 0.3.4__py3-none-any.whl → 0.3.6__py3-none-any.whl

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.dist-info → goose_py-0.3.6.dist-info}/METADATA

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

@@ -1,9 +1,9 @@
 goose/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-goose/agent.py,sha256=uCqHl5BKUX5niPbuZBOqtwnmHf8IZeBcOMylVLUfQlY,7453
+goose/agent.py,sha256=5S9p_EclF97F_TTSP61KJFk5yxFqSJ5QxCsnbb-wNEI,7650
 goose/errors.py,sha256=-0OyZQJWYTRw5YgnCB2_uorVaUsL6Z0QYQO2FqzCiyg,32
 goose/flow.py,sha256=_kSNB9CnDoduOm0YLLq0q5kzkUrtNEtvAx7A9D0_YkU,12605
 goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/store.py,sha256=6IIGkIYt6CWiNENi2gaS9C0Z-62ia-Qdz29HhCg8njw,698
-goose_py-0.3.4.dist-info/METADATA,sha256=XvAPlMqFJIZ4TLmSwPllzxcylb8D3w8PTi0qO0Y8v4c,1106
-goose_py-0.3.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-goose_py-0.3.4.dist-info/RECORD,,
+goose_py-0.3.6.dist-info/METADATA,sha256=kAU3G6iWj09wiISWH5-ripnLwbiln6qNrlROqwPHM10,1106
+goose_py-0.3.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+goose_py-0.3.6.dist-info/RECORD,,