§ öŽ’j:ãó¼—dZddlmZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZmZdd lmZeGd „d¦«¦«ZdS)zÑ TaskSupport - Configuration for experimental task support. This module provides the TaskSupport class which encapsulates all the infrastructure needed for task-augmented requests: store, queue, and handler. é)Ú AsyncIterator)Úasynccontextmanager)Ú dataclassÚfieldN)Ú TaskGroup)ÚTaskResultHandler)Únew_session_scope)Ú ServerSession)ÚInMemoryTaskStore)ÚInMemoryTaskMessageQueueÚTaskMessageQueue)Ú TaskStorecóò—eZdZUdZeed<eed<ed¬¦«Ze ed<edd¬¦«Z e dzed <dd „Z e d e fd „¦«Zed edfd „¦«Zddœdeded dfd„Zedd„¦«ZdS)Ú TaskSupporta  Configuration for experimental task support. Encapsulates the task store, message queue, result handler, and task group for spawning background work. When enabled on a server, this automatically: - Configures response routing for each session - Provides default handlers for task operations - Manages a task group for background task execution Example: # Simple in-memory setup server.experimental.enable_tasks() # Custom store/queue for distributed systems server.experimental.enable_tasks( store=RedisTaskStore(redis_url), queue=RedisTaskMessageQueue(redis_url), ) ÚstoreÚqueueF)ÚinitÚhandlerN)rÚdefaultÚ _task_groupÚreturncóD—t|j|j¦«|_dS)z/Create the result handler from store and queue.N)rrrr©Úselfs úRC:\Projects\DSCons\.venv\Lib\site-packages\mcp\server\experimental\task_support.pyÚ __post_init__zTaskSupport.__post_init__4s€å(¨¬°T´ZÑ@Ô@ˆŒ ˆ ˆ ócó<—|j€td¦«‚|jS)z‚Get the task group for spawning background work. Raises: RuntimeError: If not within a run() context Nz7TaskSupport not running. Ensure Server.run() is active.)rÚ RuntimeErrorrs rÚ task_groupzTaskSupport.task_group8s%€ð Ô Ð #ÝÐXÑYÔYÐ YØÔÐrcóÊK—tj¦«4ƒd{V—†}||_ dWV—d|_n #d|_wxYw ddd¦«ƒd{V—†dS#1ƒd{V—†swxYwYdS)a" Run the task support lifecycle. This creates a task group for spawning background task work. Called automatically by Server.run(). Usage: async with task_support.run(): # Task group is now available ... N)ÚanyioÚcreate_task_groupr)rÚtgs rÚrunzTaskSupport.runCs èè€õÔ*Ñ,Ô,ð (ð (ð (ð (ð (ð (ð (°Ø!ˆDÔ ð (Øà#'Ô Ð ø 4Ô Ð'Ð'Ð'Ð'Ð'ð  (ð (ð (ñ (ô (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (øøøð (ð (ð (ð (ð (ð (s#œA¥2ªA² ;»AÁ AÁA)Ú statelessÚsessionr&cóˆ—| |j¦«|s#|j}|j€t ¦«|_dSdSdS)a} Configure a session for task support. This registers the result handler as a response router so that responses to queued requests (elicitation, sampling) are routed back to the waiting resolvers. It also assigns the session a task session scope. Task IDs generated by `run_task()` embed this scope, and the default task handlers only operate on tasks created by the requesting session. Stateless sessions are not assigned a scope: each request runs on a fresh session, so a task created by one request could never be retrieved by a later one if tasks were bound to the session that created them. Called automatically by Server.run() for each new session. Args: session: The session to configure stateless: Whether the session belongs to a stateless server run N)Úadd_response_routerrÚ _experimentalÚtask_session_scoper )rr'r&Úfeaturess rÚconfigure_sessionzTaskSupport.configure_sessionWsZ€ð* ×#Ò# D¤LÑ1Ô1Ð1Øð BØÔ,ˆHØÔ*Ð2Ý.?Ñ.AÔ.AÔ+Ð+Ð+ð Bð Bà2Ð2rcóL—|t¦«t¦«¬¦«S)a Create in-memory task support. Suitable for development, testing, and single-process servers. For distributed systems, provide custom store and queue implementations. Returns: TaskSupport configured with in-memory store and queue )rr)r r )Úclss rÚ in_memoryzTaskSupport.in_memoryrs/€ðˆsÝ#Ñ%Ô%Ý*Ñ,Ô,ð ñ ô ð r)rN)rr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ__annotations__r rrrrrrÚpropertyr rrr%r Úboolr-Ú classmethodr0©rrrrsL€€€€€€ððð, ÐÐÑØ ÐÐÑØ!& ¨EÐ!2Ñ!2Ô!2€GÐ Ð2Ð2Ñ2Ø$) E¨u¸dÐ$CÑ$CÔ$C€K˜TÑ!ÐCÐCÑCðAðAðAðAðð ˜Ið ð ð ñ„Xð ðð(˜=¨Ô.ð(ð(ð(ñÔð(ð&NSðBðBðB¨ðBÀdðBÐW[ðBðBðBðBð6ð  ð  ð  ñ„[ð  ð  ð  rr)r4Úcollections.abcrÚ contextlibrÚ dataclassesrrr"Ú anyio.abcrÚ+mcp.server.experimental.task_result_handlerrÚ"mcp.server.experimental.task_scoper Úmcp.server.sessionr Ú2mcp.shared.experimental.tasks.in_memory_task_storer Ú+mcp.shared.experimental.tasks.message_queuer r Ú#mcp.shared.experimental.tasks.storerrr9rrúrDs"ðððð*Ð)Ð)Ð)Ð)Ð)Ø*Ð*Ð*Ð*Ð*Ð*Ø(Ð(Ð(Ð(Ð(Ð(Ð(Ð(à € € € ØÐÐÐÐÐàIÐIÐIÐIÐIÐIØ@Ð@Ð@Ð@Ð@Ð@Ø,Ð,Ð,Ð,Ð,Ð,ØPÐPÐPÐPÐPÐPØbÐbÐbÐbÐbÐbÐbÐbØ9Ð9Ð9Ð9Ð9Ð9ð ðh ðh ðh ðh ðh ñh ôh ñ „ðh ðh ðh r