fix: Add explicit type annotations to MCP tool parameters (#394)

Signed-off-by: phernandez <paul@basicmachines.co>
Co-authored-by: claude[bot] <41898282+claude[bot]@users.noreply.github.com>
Co-authored-by: Paul Hernandez <phernandez@users.noreply.github.com>

Author: phernandez

src/basic_memory/mcp/tools/build_context.py

@@ -16,8 +16,6 @@
     memory_url_path,
 )
 
-type StringOrInt = str | int
-
 
 @mcp.tool(
     description="""Build context from a memory:// URI to continue conversations naturally.
@@ -39,7 +37,7 @@
 async def build_context(
     url: MemoryUrl,
     project: Optional[str] = None,
-    depth: Optional[StringOrInt] = 1,
+    depth: str | int | None = 1,
     timeframe: Optional[TimeFrame] = "7d",
     page: int = 1,
     page_size: int = 10,

src/basic_memory/mcp/tools/write_note.py

@@ -28,7 +28,7 @@ async def write_note(
     content: str,
     folder: str,
     project: Optional[str] = None,
-    tags=None,
+    tags: list[str] | str | None = None,
     entity_type: str = "note",
     context: Context | None = None,
 ) -> str: