+api rework

backend/src/core/task_manager/manager.py

@@ -353,8 +353,8 @@ class TaskManager:
                     if task.status == status:
                         should_remove = True
                 else:
-                    # Clear all non-active tasks
-                    if task.status not in [TaskStatus.RUNNING]:
+                    # Clear all non-active tasks (keep RUNNING, AWAITING_INPUT, AWAITING_MAPPING)
+                    if task.status not in [TaskStatus.RUNNING, TaskStatus.AWAITING_INPUT, TaskStatus.AWAITING_MAPPING]:
                         should_remove = True
                 
                 if should_remove:

backend/src/core/task_manager/persistence.py

@@ -42,6 +42,7 @@ class TaskPersistenceService:
             cursor.execute("""
                 CREATE TABLE IF NOT EXISTS persistent_tasks (
                     id TEXT PRIMARY KEY,
+                    plugin_id TEXT NOT NULL,
                     status TEXT NOT NULL,
                     created_at TEXT NOT NULL,
                     updated_at TEXT NOT NULL,
@@ -68,10 +69,11 @@ class TaskPersistenceService:
                 if task.status == TaskStatus.AWAITING_INPUT:
                     cursor.execute("""
                         INSERT OR REPLACE INTO persistent_tasks
-                        (id, status, created_at, updated_at, input_request, context)
-                        VALUES (?, ?, ?, ?, ?, ?)
+                        (id, plugin_id, status, created_at, updated_at, input_request, context)
+                        VALUES (?, ?, ?, ?, ?, ?, ?)
                     """, (
                         task.id,
+                        task.plugin_id,
                         task.status.value,
                         task.started_at.isoformat() if task.started_at else datetime.utcnow().isoformat(),
                         datetime.utcnow().isoformat(),
@@ -98,16 +100,30 @@ class TaskPersistenceService:
             conn = sqlite3.connect(str(self.db_path))
             cursor = conn.cursor()
             
-            cursor.execute("SELECT id, status, created_at, input_request, context FROM persistent_tasks")
+            # Check if plugin_id column exists (migration for existing db)
+            cursor.execute("PRAGMA table_info(persistent_tasks)")
+            columns = [info[1] for info in cursor.fetchall()]
+            has_plugin_id = "plugin_id" in columns
+
+            if has_plugin_id:
+                cursor.execute("SELECT id, plugin_id, status, created_at, input_request, context FROM persistent_tasks")
+            else:
+                cursor.execute("SELECT id, status, created_at, input_request, context FROM persistent_tasks")
+            
             rows = cursor.fetchall()
             
             loaded_tasks = []
             for row in rows:
-                task_id, status, created_at, input_request_json, context_json = row
+                if has_plugin_id:
+                    task_id, plugin_id, status, created_at, input_request_json, context_json = row
+                else:
+                    task_id, status, created_at, input_request_json, context_json = row
+                    plugin_id = "superset-migration" # Default fallback
+
                 try:
                     task = Task(
                         id=task_id,
-                        plugin_id="migration",  # Default, assumes migration context for now
+                        plugin_id=plugin_id,
                         status=TaskStatus(status),
                         started_at=datetime.fromisoformat(created_at),
                         input_required=True,