Skip to content

Commit

Permalink
feat: only expose FUNCTION_TOOLS to tool info for user to select from…
Browse files Browse the repository at this point in the history
… UI (#224)

done
  • Loading branch information
MingruiZhang authored Sep 5, 2024
1 parent e24f77a commit d3cb1ac
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions vision_agent/tools/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -1820,7 +1820,6 @@ def overlay_counting_results(

FUNCTION_TOOLS = [
owl_v2,
extract_frames,
ocr,
clip,
vit_image_classification,
Expand All @@ -1841,6 +1840,7 @@ def overlay_counting_results(
]

UTIL_TOOLS = [
extract_frames,
save_json,
load_image,
save_image,
Expand All @@ -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,
Expand Down

0 comments on commit d3cb1ac

Please sign in to comment.