vision-agent 0.0.31__py3-none-any.whl → 0.0.32__py3-none-any.whl

vision_agent/agent/easytool.py

@@ -49,6 +49,33 @@ def format_tools(tools: Dict[int, Any]) -> str:
     return tool_str
 
 
+def topological_sort(tasks: List[Dict]) -> List[Dict]:
+    in_degree = {task["id"]: 0 for task in tasks}
+    for task in tasks:
+        for dep in task["dep"]:
+            if dep in in_degree:
+                in_degree[task["id"]] += 1
+
+    queue = [task for task in tasks if in_degree[task["id"]] == 0]
+    sorted_order = []
+
+    while queue:
+        current = queue.pop(0)
+        sorted_order.append(current)
+
+        for task in tasks:
+            if current["id"] in task["dep"]:
+                in_degree[task["id"]] -= 1
+                if in_degree[task["id"]] == 0:
+                    queue.append(task)
+
+    if len(sorted_order) != len(tasks):
+        completed_ids = set([task["id"] for task in sorted_order])
+        remaining_tasks = [task for task in tasks if task["id"] not in completed_ids]
+        sorted_order.extend(remaining_tasks)
+    return sorted_order
+
+
 def task_decompose(
     model: Union[LLM, LMM, Agent], question: str, tools: Dict[int, Any]
 ) -> Optional[Dict]:
@@ -265,6 +292,10 @@ class EasyTool(Agent):
         if tasks is not None:
             task_list = [{"task": task, "id": i + 1} for i, task in enumerate(tasks)]
             task_list = task_topology(self.task_model, question, task_list)
+            try:
+                task_list = topological_sort(task_list)
+            except Exception:
+                _LOGGER.error(f"Failed topological_sort on: {task_list}")
         else:
             task_list = []
 
@@ -274,7 +305,6 @@ class EasyTool(Agent):
         answers = []
         for task in task_list:
             task_depend[task["id"]] = {"task": task["task"], "answer": ""}  # type: ignore
-        # TODO topological sort task_list
         all_tool_results = []
         for task in task_list:
             task_str = task["task"]

{vision_agent-0.0.31.dist-info → vision_agent-0.0.32.dist-info}/METADATA

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

{vision_agent-0.0.31.dist-info → vision_agent-0.0.32.dist-info}/RECORD

@@ -1,7 +1,7 @@
 vision_agent/__init__.py,sha256=wD1cssVTAJ55uTViNfBGooqJUV0p9fmVAuTMHHrmUBU,229
 vision_agent/agent/__init__.py,sha256=WZrGKMglRlwy0xL77Kr7S0Lrk696gOYI5CeaiJNRmlw,89
 vision_agent/agent/agent.py,sha256=PRLItaPfMc94H6mAIPj_gBvJ8RezDEPanB6Cmu81A0M,306
-vision_agent/agent/easytool.py,sha256=tp0U--TjfN_pqHagl9Y_LTEqvk6iOcoEETmLNVE8Ras,10161
+vision_agent/agent/easytool.py,sha256=LhilXcYnhcayALE_EVPegzNWGNMnTKH0-9QFtc7xtVg,11175
 vision_agent/agent/easytool_prompts.py,sha256=tTuWwthYVELTVH3iLaZUbtX0MJsYG5TWC7Pim844cJ4,4505
 vision_agent/agent/reflexion.py,sha256=DXxFenwlZSPUKsA4vkoZ-xcdKF4YVReEZWcMKgFdfWg,9774
 vision_agent/agent/reflexion_prompts.py,sha256=UPGkt_qgHBMUY0VPVoF-BqhR0d_6WPjjrhbYLBYOtnQ,9342
@@ -17,7 +17,7 @@ vision_agent/lmm/lmm.py,sha256=ARcbgkcyP83TbVVoXI9B-gtG0gJuTaG_MjcUGbams4U,8052
 vision_agent/tools/__init__.py,sha256=aX0pU3pXU1V0Cj9FzYCvdsX76TAglFMHx59kNhXHbPs,131
 vision_agent/tools/prompts.py,sha256=9RBbyqlNlExsGKlJ89Jkph83DAEJ8PCVGaHoNbyN7TM,1416
 vision_agent/tools/tools.py,sha256=ZCak2hERXBR3SAy5zLe165PIoQQX_TykvO9ve8ZXvYY,15140
-vision_agent-0.0.31.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.0.31.dist-info/METADATA,sha256=YNf_AG8E1Ryb6lPtua8-EwFsa-LY3ESQSNtlenKQ9UY,4384
-vision_agent-0.0.31.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.0.31.dist-info/RECORD,,
+vision_agent-0.0.32.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vision_agent-0.0.32.dist-info/METADATA,sha256=oQdX4D0FzapLeEyRQF_-YmSxgDYrdITkVTI89KSlJu0,4384
+vision_agent-0.0.32.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+vision_agent-0.0.32.dist-info/RECORD,,