added overlays to utilities

landing-ai · May 29, 2024 · b7fb5c2 · b7fb5c2
1 parent ed60afe
commit b7fb5c2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -734,5 +734,12 @@ def get_tools_df(funcs: List[Callable[..., Any]]) -> pd.DataFrame:
 TOOL_DESCRIPTIONS = get_tool_descriptions(TOOLS) # type: ignore
 TOOL_DOCSTRING = get_tool_documentation(TOOLS) # type: ignore
 UTILITIES_DOCSTRING = get_tool_documentation(
- [save_json, load_image, save_image, overlay_bounding_boxes]
+ [
+ save_json,
+ load_image,
+ save_image,
+ overlay_bounding_boxes,
+ overlay_segmentation_masks,
+ overlay_heat_map,
+ ]
 )