nit

Tapan Chugh · Tapan Chugh · commit 136b2ee01b1f · 2025-08-28T22:15:27.000-07:00
diff --git a/tests/server/fastmcp/test_elicitation.py b/tests/server/fastmcp/test_elicitation.py
@@ -11,7 +11,7 @@
 from mcp.server.fastmcp import Context, FastMCP
 from mcp.server.session import ServerSession
 from mcp.shared.memory import create_connected_server_and_client_session
-from mcp.types import ElicitRequestParams, ElicitResult, ErrorData, TextContent
+from mcp.types import ElicitRequestParams, ElicitResult, TextContent
 
 
 # Shared schema for basic tests
@@ -225,9 +225,7 @@ async def valid_multiselect_tool(ctx: Context[ServerSession, None]) -> str:
             return f"Name: {result.data.name}, Tags: {', '.join(result.data.tags)}"
         return f"User {result.action}"
 
-    async def multiselect_callback(
-        context: RequestContext[ClientSession, Any], params: ElicitRequestParams
-    ) -> ElicitResult | ErrorData:
+    async def multiselect_callback(context: RequestContext[ClientSession, Any], params: ElicitRequestParams):
         if "Please provide tags" in params.message:
             return ElicitResult(action="accept", content={"name": "Test", "tags": ["tag1", "tag2"]})
         return ElicitResult(action="decline")
@@ -301,9 +299,7 @@ async def select_color_deprecated(ctx: Context[ServerSession, None]) -> str:
             return f"User: {result.data.user_name}, Color: {result.data.color}"
         return f"User {result.action}"
 
-    async def enum_callback(
-        context: RequestContext[ClientSession, Any], params: ElicitRequestParams
-    ) -> ElicitResult | ErrorData:
+    async def enum_callback(context: RequestContext[ClientSession, Any], params: ElicitRequestParams):
         if "colors" in params.message and "deprecated" not in params.message:
             return ElicitResult(action="accept", content={"user_name": "Bob", "favorite_colors": ["red", "green"]})
         elif "color" in params.message:
