diff --git a/docs/experimental/tasks-client.md b/docs/experimental/tasks-client.md index 0374ed86b5..890a1fa3bb 100644 --- a/docs/experimental/tasks-client.md +++ b/docs/experimental/tasks-client.md @@ -211,7 +211,7 @@ from mcp.types import ( CreateTaskResult, GetTaskResult, GetTaskPayloadResult, TaskMetadata, ElicitRequestParams, ) -from mcp.shared.experimental.tasks import InMemoryTaskStore +from mcp.shared.experimental.tasks.in_memory_task_store import InMemoryTaskStore # Client-side task store client_store = InMemoryTaskStore() diff --git a/docs/experimental/tasks.md b/docs/experimental/tasks.md index 2d4d06a025..7d69498dd2 100644 --- a/docs/experimental/tasks.md +++ b/docs/experimental/tasks.md @@ -114,7 +114,7 @@ The `ttl` (time-to-live) specifies how long the task and result are retained aft Servers persist task state in a `TaskStore`. The SDK provides `InMemoryTaskStore` for development: ```python -from mcp.shared.experimental.tasks import InMemoryTaskStore +from mcp.shared.experimental.tasks.in_memory_task_store import InMemoryTaskStore store = InMemoryTaskStore() ```