vision-agent 0.2.128__tar.gz → 0.2.130__tar.gz

{vision_agent-0.2.128 → vision_agent-0.2.130}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.128
+Version: 0.2.130
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.128 → vision_agent-0.2.130}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.128"
+version = "0.2.130"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.128 → vision_agent-0.2.130}/vision_agent/tools/tool_utils.py

@@ -1,3 +1,4 @@
+from base64 import b64encode
 import inspect
 import logging
 import os
@@ -27,6 +28,7 @@ class ToolCallTrace(BaseModel):
     request: MutableMapping[str, Any]
     response: MutableMapping[str, Any]
     error: Optional[Error]
+    files: Optional[List[tuple[str, str]]]
 
 
 def send_inference_request(
@@ -202,12 +204,16 @@ def _call_post(
     files: Optional[List[Tuple[Any, ...]]] = None,
     function_name: str = "unknown",
 ) -> Any:
+    files_in_b64 = None
+    if files:
+        files_in_b64 = [(file[0], b64encode(file[1]).decode("utf-8")) for file in files]
     try:
         tool_call_trace = ToolCallTrace(
             endpoint_url=url,
             request=payload,
             response={},
             error=None,
+            files=files_in_b64,
         )
 
         if files is not None:

{vision_agent-0.2.128 → vision_agent-0.2.130}/vision_agent/utils/execute.py

@@ -243,6 +243,9 @@ class Logs(BaseModel):
             f"----- stdout -----\n{stdout_str}\n----- stderr -----\n{stderr_str}"
         )
 
+    def to_json(self) -> dict[str, list[str]]:
+        return {"stdout": self.stdout, "stderr": self.stderr}
+
 
 class Error(BaseModel):
     """Represents an error that occurred during the execution of a cell. The error