vision-agent 0.2.209__py3-none-any.whl → 0.2.211__py3-none-any.whl

vision_agent/tools/__init__.py

@@ -23,40 +23,27 @@ from .tools import (
     TOOLS_INFO,
     UTIL_TOOLS,
     UTILITIES_DOCSTRING,
-    blip_image_caption,
     claude35_text_extraction,
-    clip,
     closest_box_distance,
     closest_mask_distance,
-    countgd_example_based_counting,
     countgd_object_detection,
     countgd_sam2_object_detection,
+    countgd_example_based_counting,
     depth_anything_v2,
     detr_segmentation,
-    dpt_hybrid_midas,
     extract_frames_and_timestamps,
-    florence2_image_caption,
     florence2_ocr,
     florence2_phrase_grounding,
     florence2_phrase_grounding_video,
-    florence2_roberta_vqa,
     florence2_sam2_image,
     florence2_sam2_video_tracking,
     flux_image_inpainting,
     generate_pose_image,
-    generate_soft_edge_image,
     get_tool_documentation,
     get_tool_recommender,
-    git_vqa_v2,
     gpt4o_image_vqa,
     gpt4o_video_vqa,
-    grounding_dino,
-    grounding_sam,
-    ixc25_image_vqa,
-    ixc25_video_vqa,
     load_image,
-    loca_visual_prompt_counting,
-    loca_zero_shot_counting,
     minimum_distance,
     ocr,
     overlay_bounding_boxes,

vision_agent/tools/tool_utils.py

@@ -27,6 +27,7 @@ _LND_API_URL_v2 = f"{_LND_BASE_URL}/v1/tools"
 
 class ToolCallTrace(BaseModel):
     endpoint_url: str
+    type: str
     request: MutableMapping[str, Any]
     response: MutableMapping[str, Any]
     error: Optional[Error]
@@ -221,7 +222,6 @@ def _call_post(
         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
@@ -229,6 +229,7 @@ def _call_post(
         )
         tool_call_trace = ToolCallTrace(
             endpoint_url=url,
+            type="tool_call",
             request=tool_call_trace_payload,
             response={},
             error=None,
@@ -252,7 +253,6 @@ def _call_post(
     finally:
         if tool_call_trace is not None:
             trace = tool_call_trace.model_dump()
-            trace["type"] = "tool_call"
             display({MimeType.APPLICATION_JSON: trace}, raw=True)