From d3cb1ac1e8a5a2c1880eb3cf8196d3b97575204b Mon Sep 17 00:00:00 2001
From: Mingrui Zhang <mingruizhang50@gmail.com>
Date: Thu, 5 Sep 2024 17:34:04 +0800
Subject: [PATCH] feat: only expose FUNCTION_TOOLS to tool info for user to
 select from UI (#224)

done
---
 vision_agent/tools/tools.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 8dddf8bc..f0a6bc29 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -1820,7 +1820,6 @@ def overlay_counting_results(
 
 FUNCTION_TOOLS = [
     owl_v2,
-    extract_frames,
     ocr,
     clip,
     vit_image_classification,
@@ -1841,6 +1840,7 @@ def overlay_counting_results(
 ]
 
 UTIL_TOOLS = [
+    extract_frames,
     save_json,
     load_image,
     save_image,
@@ -1856,7 +1856,7 @@ def overlay_counting_results(
 TOOLS_DF = get_tools_df(TOOLS)  # type: ignore
 TOOL_DESCRIPTIONS = get_tool_descriptions(TOOLS)  # type: ignore
 TOOL_DOCSTRING = get_tool_documentation(TOOLS)  # type: ignore
-TOOLS_INFO = get_tools_info(TOOLS)  # type: ignore
+TOOLS_INFO = get_tools_info(FUNCTION_TOOLS)  # type: ignore
 UTILITIES_DOCSTRING = get_tool_documentation(
     [
         save_json,