modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: split RequestContext between server and client (#1987)

Browse source
This commit is contained in:
Marcelo Trylesinski 2026-02-03 14:35:07 +01:00 • committed by GitHub
commit b1f7eec3cd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 210 additions and 212 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/experimental/tasks/test_elicitation_scenarios.py
View file

@ -53,7 +53,7 @@ def create_client_task_handlers(
task_complete_events: dict[str, Event] = {}
async def handle_augmented_elicitation(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: ElicitRequestParams,
task_metadata: TaskMetadata,
) -> CreateTaskResult:
@ -72,7 +72,7 @@ def create_client_task_handlers(
return CreateTaskResult(task=task)
async def handle_get_task(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: Any,
) -> GetTaskResult:
"""Handle tasks/get from server."""
@ -89,7 +89,7 @@ def create_client_task_handlers(
)
async def handle_get_task_result(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: Any,
) -> GetTaskPayloadResult | ErrorData:
"""Handle tasks/result from server."""
@ -121,7 +121,7 @@ def create_sampling_task_handlers(
task_complete_events: dict[str, Event] = {}
async def handle_augmented_sampling(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: CreateMessageRequestParams,
task_metadata: TaskMetadata,
) -> CreateTaskResult:
@ -140,7 +140,7 @@ def create_sampling_task_handlers(
return CreateTaskResult(task=task)
async def handle_get_task(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: Any,
) -> GetTaskResult:
"""Handle tasks/get from server."""
@ -157,7 +157,7 @@ def create_sampling_task_handlers(
)
async def handle_get_task_result(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: Any,
) -> GetTaskPayloadResult | ErrorData:
"""Handle tasks/result from server."""
@ -211,7 +211,7 @@ async def test_scenario1_normal_tool_normal_elicitation() -> None:
# Elicitation callback for client
async def elicitation_callback(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: ElicitRequestParams,
) -> ElicitResult:
elicit_received.set()
@ -379,7 +379,7 @@ async def test_scenario3_task_augmented_tool_normal_elicitation() -> None:
# Elicitation callback for client
async def elicitation_callback(
context: RequestContext[ClientSession, Any],
context: RequestContext[ClientSession],
params: ElicitRequestParams,
) -> ElicitResult:
elicit_received.set()