@@ -71,7 +71,9 @@ async def test_stdio_elicitation():
7171 create_ask_user_tool (mcp )
7272
7373 # Create a custom handler for elicitation requests
74- async def elicitation_callback (context : RequestContext [ClientSession , None ], params : ElicitRequestParams ): # pragma: no cover
74+ async def elicitation_callback (
75+ context : RequestContext [ClientSession , None ], params : ElicitRequestParams
76+ ): # pragma: no cover
7577 if params .message == "Tool wants to ask: What is your name?" :
7678 return ElicitResult (action = "accept" , content = {"answer" : "Test User" })
7779 else :
@@ -126,7 +128,9 @@ class InvalidNestedSchema(BaseModel):
126128 create_validation_tool ("nested_model" , InvalidNestedSchema )
127129
128130 # Dummy callback (won't be called due to validation failure)
129- async def elicitation_callback (context : RequestContext [ClientSession , None ], params : ElicitRequestParams ): # pragma: no cover
131+ async def elicitation_callback (
132+ context : RequestContext [ClientSession , None ], params : ElicitRequestParams
133+ ): # pragma: no cover
130134 return ElicitResult (action = "accept" , content = {})
131135
132136 async with create_connected_server_and_client_session (
@@ -203,7 +207,9 @@ async def invalid_optional_tool(ctx: Context[ServerSession, None]) -> str: # pr
203207 except TypeError as e :
204208 return f"Validation failed: { str (e )} "
205209
206- async def elicitation_callback (context : RequestContext [ClientSession , None ], params : ElicitRequestParams ): # pragma: no cover
210+ async def elicitation_callback (
211+ context : RequestContext [ClientSession , None ], params : ElicitRequestParams
212+ ): # pragma: no cover
207213 return ElicitResult (action = "accept" , content = {})
208214
209215 await call_tool_and_assert (
0 commit comments