vision-agent 0.2.179__tar.gz → 0.2.181__tar.gz

{vision_agent-0.2.179 → vision_agent-0.2.181}/PKG-INFO

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

{vision_agent-0.2.179 → vision_agent-0.2.181}/pyproject.toml

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

{vision_agent-0.2.179 → vision_agent-0.2.181}/vision_agent/agent/vision_agent.py

@@ -492,29 +492,8 @@ class VisionAgent(Agent):
                 code_interpreter.upload_file(artifacts.local_save_path)
 
                 response = run_conversation(self.agent, int_chat)
-                code_action = use_extra_vision_agent_args(
-                    response.get("execute_python", None),
-                    test_multi_plan,
-                    custom_tool_names,
-                )
                 if self.verbosity >= 1:
                     _LOGGER.info(response)
-                int_chat.append(
-                    {
-                        "role": "assistant",
-                        "content": json.dumps(
-                            new_format_to_old_format(add_step_descriptions(response))
-                        ),
-                    }
-                )
-                orig_chat.append(
-                    {
-                        "role": "assistant",
-                        "content": json.dumps(
-                            new_format_to_old_format(add_step_descriptions(response))
-                        ),
-                    }
-                )
 
                 code_action = response.get("execute_python", None)
                 # sometimes it gets stuck in a loop, so we force it to exit
@@ -529,7 +508,7 @@ class VisionAgent(Agent):
                                 "value": "Agent is stuck in conversation loop, exited",
                                 "traceback_raw": [],
                             },
-                            "finished": code_action is None,
+                            "finished": True,
                         }
                     )
                 else:
@@ -544,6 +523,22 @@ class VisionAgent(Agent):
                         }
                     )
 
+                int_chat.append(
+                    {
+                        "role": "assistant",
+                        "content": json.dumps(
+                            new_format_to_old_format(add_step_descriptions(response))
+                        ),
+                    }
+                )
+                orig_chat.append(
+                    {
+                        "role": "assistant",
+                        "content": json.dumps(
+                            new_format_to_old_format(add_step_descriptions(response))
+                        ),
+                    }
+                )
                 finished = response.get("let_user_respond", False)
 
                 if code_action is not None:

{vision_agent-0.2.179 → vision_agent-0.2.181}/vision_agent/tools/tools.py

@@ -1970,8 +1970,14 @@ def overlay_bounding_boxes(
     medias_int: List[np.ndarray] = (
         [medias] if isinstance(medias, np.ndarray) else medias
     )
-    bbox_int = [bboxes] if isinstance(bboxes[0], dict) else bboxes
-    bbox_int = cast(List[List[Dict[str, Any]]], bbox_int)
+    if len(bboxes) == 0:
+        bbox_int: List[List[Dict[str, Any]]] = [[] for _ in medias_int]
+    else:
+        if isinstance(bboxes[0], dict):
+            bbox_int = [cast(List[Dict[str, Any]], bboxes)]
+        else:
+            bbox_int = cast(List[List[Dict[str, Any]]], bboxes)
+
     labels = set([bb["label"] for b in bbox_int for bb in b])
 
     if len(labels) > len(COLORS):