diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 6cb697a3..6d20f4f9 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -1337,6 +1337,7 @@ def florencev2_fine_tuning(bboxes: List[Dict[str, Any]], task: str) -> UUID:
     )
 
 
+# TODO: add this function to the imports so that is picked in the agent
 def florencev2_fine_tuned_object_detection(
     image: np.ndarray, prompt: str, model_id: UUID, task: str
 ) -> List[Dict[str, Any]]: