diff --git a/api/controllers/console/snippets/snippet_workflow.py b/api/controllers/console/snippets/snippet_workflow.py index ff46f71b0f..17309fc4d4 100644 --- a/api/controllers/console/snippets/snippet_workflow.py +++ b/api/controllers/console/snippets/snippet_workflow.py @@ -36,6 +36,7 @@ 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 extensions.ext_database import db +from extensions.ext_redis import redis_client from factories import variable_factory from libs import helper from libs.helper import TimestampField @@ -535,6 +536,6 @@ class SnippetWorkflowTaskStopApi(Resource): AppQueueManager.set_stop_flag_no_user_check(task_id) # New graph engine command channel mechanism - GraphEngineManager.send_stop_command(task_id) + GraphEngineManager(redis_client).send_stop_command(task_id) return {"result": "success"}