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,