§ )`ƒikãó°—dZddlmZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZdd 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)Ú ServerSession)ÚInMemoryTaskStore)ÚInMemoryTaskMessageQueueÚTaskMessageQueue)Ú TaskStorecóè—eZdZUdZeed<eed<ed¬¦«Ze ed<edd¬¦«Z e dzed <dd „Z e d e fd „¦«Zed edfd „¦«Zded 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 úx/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/experimental/task_support.pyÚ __post_init__zTaskSupport.__post_init__3s€å(¨¬°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_group7s%€ð Ô Ð #ÝÐ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.runBs èè€õÔ*Ñ,Ô,ð (ð (ð (ð (ð (ð (ð (°Ø!ˆDÔ ð (Øà#'Ô Ð ø 4Ô Ð'Ð'Ð'Ð'Ð'ð  (ð (ð (ñ (ô (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (ð (øøøð (ð (ð (ð (ð (ð (s#œA¥2ªA² ;»AÁ AÁAÚsessioncó:—| |j¦«dS)ao 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. Called automatically by Server.run() for each new session. Args: session: The session to configure N)Úadd_response_routerr)rr%s rÚconfigure_sessionzTaskSupport.configure_sessionVs €ð ×#Ò# D¤LÑ1Ô1Ð1Ð1Ð1rcó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_memoryes/€ðˆsÝ#Ñ%Ô%Ý*Ñ,Ô,ð ñ ô ð r)rN)rr)Ú__name__Ú __module__Ú __qualname__Ú__doc__r Ú__annotations__r rrrrrrÚpropertyrrrr$r r(Ú classmethodr+©rrrrs*€€€€€€ððð, ÐÐÑØ ÐÐÑØ!& ¨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ð ðð(˜=¨Ô.ð(ð(ð(ñÔð(ð& 2¨ð 2¸4ð 2ð 2ð 2ð 2ðð  ð  ð  ñ„[ð  ð  ð  rr)r/Úcollections.abcrÚ contextlibrÚ dataclassesrrr!Ú anyio.abcrÚ+mcp.server.experimental.task_result_handlerrÚmcp.server.sessionr Ú2mcp.shared.experimental.tasks.in_memory_task_storer Ú+mcp.shared.experimental.tasks.message_queuer r Ú#mcp.shared.experimental.tasks.storer rr3rrúr=sðððð*Ð)Ð)Ð)Ð)Ð)Ø*Ð*Ð*Ð*Ð*Ð*Ø(Ð(Ð(Ð(Ð(Ð(Ð(Ð(à € € € ØÐÐÐÐÐà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ð ð\ ð\ ð\ ð\ ð\ ñ\ ô\ ñ „ð\ ð\ ð\ r