From ae841ca7b688d8fac427ff16293cb711d36de5e6 Mon Sep 17 00:00:00 2001
From: Mingrui Zhang <mingruizhang50@gmail.com>
Date: Tue, 27 Aug 2024 07:45:08 +0800
Subject: [PATCH] type

---
 vision_agent/agent/vision_agent_coder.py | 2 +-
 vision_agent/tools/__init__.py           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index d253d8ba..97a522a1 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -750,7 +750,7 @@ def chat_with_workflow(
             plans = write_plans(
                 int_chat,
                 T.get_tool_descriptions_by_names(
-                    customized_tool_names, T.FUNCTION_TOOLS, T.UTIL_TOOLS
+                    customized_tool_names, T.FUNCTION_TOOLS, T.UTIL_TOOLS  # type: ignore
                 ),
                 format_memory(working_memory),
                 self.planner,
diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
index 1ba9c11a..a90b7181 100644
--- a/vision_agent/tools/__init__.py
+++ b/vision_agent/tools/__init__.py
@@ -2,6 +2,7 @@
 
 from .meta_tools import META_TOOL_DOCSTRING
 from .prompts import CHOOSE_PARAMS, SYSTEM_PROMPT
+from .tool_utils import get_tool_descriptions_by_names
 from .tools import (
     FUNCTION_TOOLS,
     TOOL_DESCRIPTIONS,
@@ -27,7 +28,6 @@
     florence2_sam2_video,
     generate_pose_image,
     generate_soft_edge_image,
-    get_tool_descriptions_by_names,
     get_tool_documentation,
     git_vqa_v2,
     grounding_dino,