46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
"""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)
|