basicmachines-co
diff --git a/‎src/basic_memory/mcp/tools/build_context.py‎
Lines changed: 1 addition & 3 deletions b/‎src/basic_memory/mcp/tools/build_context.py‎
Lines changed: 1 addition & 3 deletions
diff --git a/‎src/basic_memory/mcp/tools/write_note.py‎
Lines changed: 1 addition & 1 deletion b/‎src/basic_memory/mcp/tools/write_note.py‎
Lines changed: 1 addition & 1 deletion
@@ -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,
 
@@ -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: