| """ | |
| 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] | |