Function get_task_result

dash/mcp/tasks/tasks.py:106–171

Handle tasks/result — retrieve and format the callback result. Uses the framework's background callback retrieval functions (``_update_background_callback`` and ``_prepare_response``) with ``cache_key`` and ``job_id`` supplied directly, bypassing the request adapter query-param look

(task_id: str)

def get_task_result(task_id: str) -> Any:
    """Handle tasks/result — retrieve and format the callback result.

    Uses the framework's background callback retrieval functions
    (``_update_background_callback`` and ``_prepare_response``) with
    ``cache_key`` and ``job_id`` supplied directly, bypassing the
    request adapter query-param lookup.
    """
    tool_name, job_id, cache_key, _created_at = parse_task_id(task_id)

    manager = _get_callback_manager(tool_name)
    if manager is None:
        raise MCPError("No background callback manager configured.")

    app = get_app()
    adapter = app.mcp_callback_map.find_by_tool_name(tool_name)
    if adapter is None:
        raise MCPError(f"Task not found: {tool_name}")
    # pylint: disable-next=protected-access
    cb_info = adapter._cb_info
    background = cb_info.get("background")
    has_output = not cb_info.get("no_output")
    multi = adapter.output_id.startswith("..")
    output = split_callback_id(adapter.output_id)

    # Build the body to get output_spec, same as as_callback_body
    body = adapter.as_callback_body({})
    output_spec = body.get("outputs", [])

    callback_ctx = AttributeDict({"updated_props": {}})
    response: CallbackExecutionResponse = {"multi": True}

    output_value, has_update, skip = _update_background_callback(
        error_handler=None,
        callback_ctx=callback_ctx,
        response=response,
        kwargs={"background_callback_manager": manager},
        background=background,
        multi=multi,
        cache_key=cache_key,
        job_id=job_id,
    )

    if skip:
        # Result not ready — still polling
        raise MCPError(
            "Task result not ready. Poll tasks/get until status is 'completed'."
        )

    _prepare_response(
        output_value,
        output_spec,
        multi,
        response,
        callback_ctx,
        app,
        set(ComponentRegistry.registry),
        background,