diff --git a/python/src/mcp_server/features/documents/version_tools.py b/python/src/mcp_server/features/documents/version_tools.py
index 376593bfc9..b5033c6d3e 100644
--- a/python/src/mcp_server/features/documents/version_tools.py
+++ b/python/src/mcp_server/features/documents/version_tools.py
@@ -28,7 +28,7 @@ async def create_version(
         ctx: Context,
         project_id: str,
         field_name: str,
-        content: Any,
+        content: dict[str, Any] | list[dict[str, Any]],
         change_summary: str | None = None,
         document_id: str | None = None,
         created_by: str = "system",
diff --git a/python/src/mcp_server/modules/rag_module.py b/python/src/mcp_server/modules/rag_module.py
index 67b5b498ea..8686a75c11 100644
--- a/python/src/mcp_server/modules/rag_module.py
+++ b/python/src/mcp_server/modules/rag_module.py
@@ -78,7 +78,7 @@ async def get_available_sources(ctx: Context) -> str:
 
     @mcp.tool()
     async def perform_rag_query(
-        ctx: Context, query: str, source_domain: str = None, match_count: int = 5
+        ctx: Context, query: str, source_domain: str | None = None, match_count: int = 5
     ) -> str:
         """
         Search knowledge base for relevant content using RAG.
@@ -135,7 +135,7 @@ async def perform_rag_query(
 
     @mcp.tool()
     async def search_code_examples(
-        ctx: Context, query: str, source_domain: str = None, match_count: int = 5
+        ctx: Context, query: str, source_domain: str | None = None, match_count: int = 5
     ) -> str:
         """
         Search for relevant code examples in the knowledge base.