danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix mcp test

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-03-24 13:36:14 +02:00
parent 8f5b6c299d
commit 5474465610
No known key found for this signature in database
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_mcp.py
View file

@ -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