kKj dZddlmZddlmZmZmZddlZddl m Z ddlm Z m Z ddl mZddlmZerdd lmZdd lmZdd lmZdd lmZed Z d ddZy)zjTask routing helper for MCP components. Provides unified task mode enforcement and docket routing logic. ) annotations) TYPE_CHECKINGAnyLiteralN)McpError)METHOD_NOT_FOUND ErrorData)TaskMeta)submit_to_docket)Prompt)Resource)ResourceTemplate)Tool)toolresourcetemplatepromptcK|j}t|jd|jxs |jd}|j dk(r|st tt|d|js|rt tt|d|sy|jxs |j}t|||||d{S7w)a-Check task mode and submit to background if requested. Args: component: The MCP component task_type: Type of task ("tool", "resource", "template", "prompt") arguments: Arguments for tool/prompt/template execution task_meta: Task execution metadata. If provided, execute as background task. Returns: CreateTaskResult if submitted to docket, None for sync execution Raises: McpError: If mode="required" but no task metadata, or mode="forbidden" but task metadata is present z ''requiredz" requires task-augmented execution)codemessagez* does not support task-augmented executionN) task_configtype__name__titlekeymoderr rsupports_tasksfn_keyr ) component task_type arguments task_metar entity_labelr s m/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/routing.pycheck_background_taskr's*''K9o../r)//2RY]]1SSTUL:%i %'.(JK     % % 'I %'.(RS       .F!)VY 9U UU UsC CCC)NN) r!z+Tool | Resource | ResourceTemplate | Promptr"TaskTyper#zdict[str, Any] | Noner$zTaskMeta | Nonereturnz!mcp.types.CreateTaskResult | None)__doc__ __future__rtypingrrr mcp.typesmcpmcp.shared.exceptionsrrr fastmcp.server.tasks.configr fastmcp.server.tasks.handlersr fastmcp.prompts.baser fastmcp.resources.baser fastmcp.resources.templaterfastmcp.tools.baserr(r'r&r8s #..*10:+/;' ; < (,!% 2V:2V2V%2V 2V ' 2Vr7