idZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z e dZ e d Zeej!d zZd ZeGd dZeGddZy)zTaskConfig for MCP SEP-1686 background task execution modes. This module defines the configuration for how tools, resources, and prompts handle task-augmented execution as specified in SEP-1686. ) annotationsN)Callable) dataclass) timedelta)AnyLiteral) forbiddenoptionalrequired)secondsii`c.eZdZUdZdZded<dZded<y)TaskMetaaeMetadata for task-augmented execution requests. When passed to call_tool/read_resource/get_prompt, signals that the operation should be submitted as a background task. Attributes: ttl: Client-requested TTL in milliseconds. If None, uses server default. fn_key: Docket routing key. Auto-derived from component name if None. Nz int | Nonettlz str | Nonefn_key)__name__ __module__ __qualname____doc__r__annotations__rk/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/config.pyrrsCFJrrcPeZdZUdZdZded<eZded<ed dZ d dZ d d Z y ) TaskConfigaConfiguration for MCP background task execution (SEP-1686). Controls how a component handles task-augmented requests: - "forbidden": Component does not support task execution. Clients must not request task augmentation; server returns -32601 if they do. - "optional": Component supports both synchronous and task execution. Client may request task augmentation or call normally. - "required": Component requires task execution. Clients must request task augmentation; server returns -32601 if they don't. Important: Task-enabled components must be available at server startup to be registered with all Docket workers. Components added dynamically after startup will not be registered for background execution. Example: ```python from fastmcp import FastMCP from fastmcp.server.tasks import TaskConfig mcp = FastMCP("MyServer") # Background execution required @mcp.tool(task=TaskConfig(mode="required")) async def long_running_task(): ... # Supports both modes (default when task=True) @mcp.tool(task=TaskConfig(mode="optional")) async def flexible_task(): ... ``` r TaskModemoder poll_intervalc&||rdSdS)zConvert boolean task flag to TaskConfig. Args: value: True for "optional" mode, False for "forbidden" mode. Returns: TaskConfig with appropriate mode. r r rr)clsvalues r from_boolzTaskConfig.from_boolNse ====rc |jdk7S)zCheck if this component supports task execution. Returns: True if mode is "optional" or "required", False if "forbidden". r r )selfs rsupports_taskszTaskConfig.supports_tasksZs yyK''rc&|jsyddlm}|d|d|}tj|st |r |j }t|tr |j}tj|std|dy)aValidate that function is compatible with this task config. Task execution requires: 1. fastmcp[tasks] to be installed (pydocket) 2. Async functions Raises ImportError if mode is "optional" or "required" but pydocket is not installed. Raises ValueError if function is synchronous. Args: fn: The function to validate (handles callable classes and staticmethods). name: Name for error messages. Raises: ImportError: If task execution is enabled but pydocket not installed. ValueError: If task execution is enabled but function is sync. Nr)require_docketz`task=True` on function ''z`' uses a sync function but has task execution enabled. Background tasks require async functions.) r&fastmcp.server.dependenciesr(inspect isroutinecallable__call__ isinstance staticmethod__func__iscoroutinefunction ValueError)r%fnnamer( fn_to_checks rvalidate_functionzTaskConfig.validate_functionbs$""$  ?24&:;   $"++K k< 0%..K**;7D6<< 8rN)r"boolreturnr)r9r8)r4zCallable[..., Any]r5strr9None) rrrrrrDEFAULT_POLL_INTERVALr classmethodr#r&r7rrrrr(s:B D(4M94 > >(&rr)r __future__rr+collections.abcr dataclassesrdatetimertypingrrrr<int total_secondsDEFAULT_POLL_INTERVAL_MSDEFAULT_TTL_MSrrrrrrGs #$! 6 7"!,4BBDtKL      __ _r