dify

dify/api/services/app_task_service.py

@@ -0,0 +1,45 @@
+"""Service for managing application task operations.
+
+This service provides centralized logic for task control operations
+like stopping tasks, handling both legacy Redis flag mechanism and
+new GraphEngine command channel mechanism.
+"""
+
+from core.app.apps.base_app_queue_manager import AppQueueManager
+from core.app.entities.app_invoke_entities import InvokeFrom
+from core.workflow.graph_engine.manager import GraphEngineManager
+from models.model import AppMode
+
+
+class AppTaskService:
+    """Service for managing application task operations."""
+
+    @staticmethod
+    def stop_task(
+        task_id: str,
+        invoke_from: InvokeFrom,
+        user_id: str,
+        app_mode: AppMode,
+    ) -> None:
+        """Stop a running task.
+
+        This method handles stopping tasks using both mechanisms:
+        1. Legacy Redis flag mechanism (for backward compatibility)
+        2. New GraphEngine command channel (for workflow-based apps)
+
+        Args:
+            task_id: The task ID to stop
+            invoke_from: The source of the invoke (e.g., DEBUGGER, WEB_APP, SERVICE_API)
+            user_id: The user ID requesting the stop
+            app_mode: The application mode (CHAT, AGENT_CHAT, ADVANCED_CHAT, WORKFLOW, etc.)
+
+        Returns:
+            None
+        """
+        # Legacy mechanism: Set stop flag in Redis
+        AppQueueManager.set_stop_flag(task_id, invoke_from, user_id)
+
+        # New mechanism: Send stop command via GraphEngine for workflow-based apps
+        # This ensures proper workflow status recording in the persistence layer
+        if app_mode in (AppMode.ADVANCED_CHAT, AppMode.WORKFLOW):
+            GraphEngineManager.send_stop_command(task_id)