more typings and add docstrings

Author: patrickkang

examples/example-fastmcp-mcp/src/auth0/__init__.py

@@ -24,6 +24,20 @@
 
 
 class Auth0Mcp:
+    """
+    Auth0 integration for FastMCP servers.
+
+    Provides authentication middleware, authorization decorators,
+    and OAuth2 Protected Resource metadata endpoints for MCP servers.
+
+    Args:
+        name: Human-readable name for the MCP server
+        audience: Auth0 API identifier (OAuth2 audience claim)
+        domain: Auth0 tenant domain (e.g., 'tenant.us.auth0.com')
+
+    Raises:
+        RuntimeError: If audience or domain are not provided
+    """
     def __init__(self, name: str, audience: str, domain: str, mcp_server_url: str | None = None):
         self.name = name
         self.audience = audience
@@ -77,7 +91,7 @@ def exception_handlers(self) -> dict[int | type[Exception], Callable[[Request, E
             Exception: self._generic_exception_handler,
         }
 
-    def _auth_error_handler(self, request: Request, exc: Exception):
+    def _auth_error_handler(self, request: Request, exc: Exception)  -> JSONResponse:
         """
         Handle auth errors: malformed authorization requests, missing auth, invalid tokens, and insufficient scopes.
         """

examples/example-fastmcp-mcp/src/tools.py

@@ -2,16 +2,17 @@
 
 from mcp.server.fastmcp import Context
 
+from .auth0 import Auth0Mcp
 from .auth0.authz import require_scopes
 
 
-def register_tools(auth0Mcp) -> None:
+def register_tools(auth0_mcp: Auth0Mcp) -> None:
     """
     Register all tools with the MCP server.
     """
-    mcp = auth0Mcp.mcp
+    mcp = auth0_mcp.mcp
     # Register scopes used by tools for Protected Resource Metadata
-    auth0Mcp.register_scopes(["tool:greet", "tool:whoami"])
+    auth0_mcp.register_scopes(["tool:greet", "tool:whoami"])
 
     # Tool without required scopes
     @mcp.tool()