From 6014e880efbdd36d4e7b270c64c08e6c34b75f49 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Wed, 4 Jun 2025 10:24:03 -0400
Subject: [PATCH] Update test_tool_manager.py

---
 tests/tools/test_tool_manager.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/tools/test_tool_manager.py b/tests/tools/test_tool_manager.py
index 7902effa9..8f2bb32fa 100644
--- a/tests/tools/test_tool_manager.py
+++ b/tests/tools/test_tool_manager.py
@@ -693,6 +693,7 @@ class TestCustomToolNames:
         # The tool is stored under the custom name and its .name is also set to custom_name
         assert manager.get_tool("custom_name") is not None
         assert tool.name == "custom_name"
+        assert isinstance(tool, FunctionTool)
         assert tool.fn.__name__ == "original_fn"
         # The tool should not be accessible via its original function name
         with pytest.raises(NotFoundError, match="Unknown tool: original_fn"):
@@ -764,6 +765,7 @@ class TestCustomToolNames:
         assert stored_tool.name == "test_tool"
 
         # But the function is different
+        assert isinstance(stored_tool, FunctionTool)
         assert stored_tool.fn.__name__ == "replacement_fn"