Spaces:
Running on Zero
Running on Zero
File size: 658 Bytes
fd02f25 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | """
MCP Tool: get_task_status
Query the processing progress and final results of an async image generation task.
"""
from .common import _TASKS_DB
from .error_schema import make_validation_error, make_not_found_error
def handle_get_task_status(task_id: str) -> dict:
"""Query the processing progress and final results of an async image generation task."""
if not task_id:
return make_validation_error(
"Parameter 'task_id' is required.",
missing_fields=["task_id"],
)
if task_id not in _TASKS_DB:
return make_not_found_error("task", task_id)
return _TASKS_DB[task_id]
|