diff --git a/backend/main.py b/backend/main.py index 3f58aa8b..c9d7e7f7 100644 --- a/backend/main.py +++ b/backend/main.py @@ -1616,8 +1616,7 @@ async def clear_all_tasks(): task_manager = get_task_manager() progress_manager = get_progress_manager() - task_manager._active_downloads.clear() - task_manager._active_generations.clear() + task_manager.clear_all() with progress_manager._lock: progress_manager._progress.clear() diff --git a/backend/utils/tasks.py b/backend/utils/tasks.py index da13011a..8baf71c3 100644 --- a/backend/utils/tasks.py +++ b/backend/utils/tasks.py @@ -76,6 +76,11 @@ class TaskManager: """Cancel/dismiss a download task (removes it from active list).""" return self._active_downloads.pop(model_name, None) is not None + def clear_all(self) -> None: + """Clear all download and generation tasks.""" + self._active_downloads.clear() + self._active_generations.clear() + def is_download_active(self, model_name: str) -> bool: """Check if a download is active.""" return model_name in self._active_downloads