type

landing-ai · Aug 26, 2024 · ae841ca · ae841ca
1 parent a4552a3
commit ae841ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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,