mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
"""Tool contract tests for MCP tool signatures."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import inspect
|
|
|
|
from basic_memory.mcp import tools
|
|
|
|
|
|
EXPECTED_TOOL_SIGNATURES: dict[str, list[str]] = {
|
|
"build_context": ["url", "project", "depth", "timeframe", "page", "page_size", "max_related"],
|
|
"canvas": ["nodes", "edges", "title", "directory", "project"],
|
|
"cloud_info": [],
|
|
"create_memory_project": ["project_name", "project_path", "set_default"],
|
|
"delete_note": ["identifier", "is_directory", "project"],
|
|
"delete_project": ["project_name"],
|
|
"edit_note": [
|
|
"identifier",
|
|
"operation",
|
|
"content",
|
|
"project",
|
|
"section",
|
|
"find_text",
|
|
"expected_replacements",
|
|
],
|
|
"fetch": ["id"],
|
|
"list_directory": ["dir_name", "depth", "file_name_glob", "project"],
|
|
"list_memory_projects": [],
|
|
"move_note": ["identifier", "destination_path", "is_directory", "project"],
|
|
"read_content": ["path", "project"],
|
|
"read_note": ["identifier", "project", "page", "page_size", "output_format"],
|
|
"release_notes": [],
|
|
"recent_activity": ["type", "depth", "timeframe", "project"],
|
|
"search": ["query"],
|
|
"search_by_metadata": ["filters", "project", "limit", "offset"],
|
|
"search_notes": [
|
|
"query",
|
|
"project",
|
|
"page",
|
|
"page_size",
|
|
"search_type",
|
|
"output_format",
|
|
"types",
|
|
"entity_types",
|
|
"after_date",
|
|
"metadata_filters",
|
|
"tags",
|
|
"status",
|
|
],
|
|
"view_note": ["identifier", "project", "page", "page_size"],
|
|
"write_note": ["title", "content", "directory", "project", "tags", "note_type"],
|
|
}
|
|
|
|
|
|
TOOL_FUNCTIONS: dict[str, object] = {
|
|
"build_context": tools.build_context,
|
|
"canvas": tools.canvas,
|
|
"cloud_info": tools.cloud_info,
|
|
"create_memory_project": tools.create_memory_project,
|
|
"delete_note": tools.delete_note,
|
|
"delete_project": tools.delete_project,
|
|
"edit_note": tools.edit_note,
|
|
"fetch": tools.fetch,
|
|
"list_directory": tools.list_directory,
|
|
"list_memory_projects": tools.list_memory_projects,
|
|
"move_note": tools.move_note,
|
|
"read_content": tools.read_content,
|
|
"read_note": tools.read_note,
|
|
"release_notes": tools.release_notes,
|
|
"recent_activity": tools.recent_activity,
|
|
"search": tools.search,
|
|
"search_by_metadata": tools.search_by_metadata,
|
|
"search_notes": tools.search_notes,
|
|
"view_note": tools.view_note,
|
|
"write_note": tools.write_note,
|
|
}
|
|
|
|
|
|
def _signature_params(tool_obj: object) -> list[str]:
|
|
fn = tool_obj.fn
|
|
params = []
|
|
for param in inspect.signature(fn).parameters.values():
|
|
if param.name == "context":
|
|
continue
|
|
params.append(param.name)
|
|
return params
|
|
|
|
|
|
def test_mcp_tool_signatures_are_stable():
|
|
assert set(TOOL_FUNCTIONS.keys()) == set(EXPECTED_TOOL_SIGNATURES.keys())
|
|
|
|
for tool_name, tool_obj in TOOL_FUNCTIONS.items():
|
|
assert _signature_params(tool_obj) == EXPECTED_TOOL_SIGNATURES[tool_name]
|