From a82c4b24af04b1081718b98ebc812429969c1610 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Wed, 29 May 2024 13:51:56 -0700
Subject: [PATCH] added overlays to utilities

---
 vision_agent/tools/tools.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index a6a40348..d47c2e2e 100644
--- 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,
+    ]
 )