From 5474465610b1c36a1b1cced939e2121b781e1f8b Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Tue, 24 Mar 2026 13:36:14 +0200 Subject: [PATCH] Fix mcp test --- tests/test_mcp.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_mcp.py b/tests/test_mcp.py index a3743168..3ba1dcdf 100644 --- a/tests/test_mcp.py +++ b/tests/test_mcp.py @@ -141,7 +141,8 @@ class TestMCPWriteTools: async with HaikuRAG(temp_db_path, create=True): pass mcp = create_mcp_server(temp_db_path, read_only=False) - tool_names = list(mcp._tool_manager._tools.keys()) + tools = await mcp.list_tools() + tool_names = [t.name for t in tools] assert "add_document_from_text" in tool_names assert "add_document_from_file" in tool_names assert "add_document_from_url" in tool_names @@ -152,7 +153,8 @@ class TestMCPWriteTools: async with HaikuRAG(temp_db_path, create=True): pass mcp = create_mcp_server(temp_db_path, read_only=True) - tool_names = list(mcp._tool_manager._tools.keys()) + tools = await mcp.list_tools() + tool_names = [t.name for t in tools] assert "add_document_from_text" not in tool_names assert "delete_document" not in tool_names