vision-agent 0.2.139__tar.gz → 0.2.140__tar.gz

{vision_agent-0.2.139 → vision_agent-0.2.140}/PKG-INFO

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

{vision_agent-0.2.139 → vision_agent-0.2.140}/pyproject.toml

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

{vision_agent-0.2.139 → vision_agent-0.2.140}/vision_agent/tools/tool_utils.py

@@ -208,19 +208,25 @@ def _call_post(
     if files:
         files_in_b64 = [(file[0], b64encode(file[1]).decode("utf-8")) for file in files]
     try:
+        if files is not None:
+            response = session.post(url, data=payload, files=files)
+        else:
+            response = session.post(url, json=payload)
+
+        # make sure function_name is in the payload so we can display it
+        tool_call_trace_payload = (
+            payload
+            if "function_name" in payload
+            else {**payload, **{"function_name": function_name}}
+        )
         tool_call_trace = ToolCallTrace(
             endpoint_url=url,
-            request=payload,
+            request=tool_call_trace_payload,
             response={},
             error=None,
             files=files_in_b64,
         )
 
-        if files is not None:
-            response = session.post(url, data=payload, files=files)
-        else:
-            response = session.post(url, json=payload)
-
         if response.status_code != 200:
             tool_call_trace.error = Error(
                 name="RemoteToolCallFailed",