diff --git a/python/src/mcp_server/features/documents/version_tools.py b/python/src/mcp_server/features/documents/version_tools.py
index 376593b..b5033c6 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 @@ def register_version_tools(mcp: FastMCP):
         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 67b5b49..8686a75 100644
--- a/python/src/mcp_server/modules/rag_module.py
+++ b/python/src/mcp_server/modules/rag_module.py
@@ -78,7 +78,7 @@ def register_rag_tools(mcp: FastMCP):
 
     @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 @@ def register_rag_tools(mcp: FastMCP):
 
     @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.