| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871 | 
							- import json
 - import logging
 - from collections.abc import Sequence
 - from typing import cast
 - 
 - from flask import abort, request
 - from flask_restx import Resource, inputs, marshal_with, reqparse
 - from sqlalchemy.orm import Session
 - from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
 - 
 - import services
 - from configs import dify_config
 - from controllers.console import api
 - from controllers.console.app.error import ConversationCompletedError, DraftWorkflowNotExist, DraftWorkflowNotSync
 - from controllers.console.app.wraps import get_app_model
 - from controllers.console.wraps import account_initialization_required, setup_required
 - from controllers.web.error import InvokeRateLimitError as InvokeRateLimitHttpError
 - from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 - from core.app.apps.base_app_queue_manager import AppQueueManager
 - from core.app.entities.app_invoke_entities import InvokeFrom
 - from core.file.models import File
 - from core.helper.trace_id_helper import get_external_trace_id
 - from extensions.ext_database import db
 - from factories import file_factory, variable_factory
 - from fields.workflow_fields import workflow_fields, workflow_pagination_fields
 - from fields.workflow_run_fields import workflow_run_node_execution_fields
 - from libs import helper
 - from libs.helper import TimestampField, uuid_value
 - from libs.login import current_user, login_required
 - from models import App
 - from models.account import Account
 - from models.model import AppMode
 - from models.workflow import Workflow
 - from services.app_generate_service import AppGenerateService
 - from services.errors.app import WorkflowHashNotEqualError
 - from services.errors.llm import InvokeRateLimitError
 - from services.workflow_service import DraftWorkflowDeletionError, WorkflowInUseError, WorkflowService
 - 
 - logger = logging.getLogger(__name__)
 - 
 - 
 - # TODO(QuantumGhost): Refactor existing node run API to handle file parameter parsing
 - # at the controller level rather than in the workflow logic. This would improve separation
 - # of concerns and make the code more maintainable.
 - def _parse_file(workflow: Workflow, files: list[dict] | None = None) -> Sequence[File]:
 -     files = files or []
 - 
 -     file_extra_config = FileUploadConfigManager.convert(workflow.features_dict, is_vision=False)
 -     file_objs: Sequence[File] = []
 -     if file_extra_config is None:
 -         return file_objs
 -     file_objs = file_factory.build_from_mappings(
 -         mappings=files,
 -         tenant_id=workflow.tenant_id,
 -         config=file_extra_config,
 -     )
 -     return file_objs
 - 
 - 
 - class DraftWorkflowApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_fields)
 -     def get(self, app_model: App):
 -         """
 -         Get draft workflow
 -         """
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         assert isinstance(current_user, Account)
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         # fetch draft workflow by app_model
 -         workflow_service = WorkflowService()
 -         workflow = workflow_service.get_draft_workflow(app_model=app_model)
 - 
 -         if not workflow:
 -             raise DraftWorkflowNotExist()
 - 
 -         # return workflow, if not found, return None (initiate graph by frontend)
 -         return workflow
 - 
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def post(self, app_model: App):
 -         """
 -         Sync draft workflow
 -         """
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         assert isinstance(current_user, Account)
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         content_type = request.headers.get("Content-Type", "")
 - 
 -         if "application/json" in content_type:
 -             parser = reqparse.RequestParser()
 -             parser.add_argument("graph", type=dict, required=True, nullable=False, location="json")
 -             parser.add_argument("features", type=dict, required=True, nullable=False, location="json")
 -             parser.add_argument("hash", type=str, required=False, location="json")
 -             parser.add_argument("environment_variables", type=list, required=True, location="json")
 -             parser.add_argument("conversation_variables", type=list, required=False, location="json")
 -             args = parser.parse_args()
 -         elif "text/plain" in content_type:
 -             try:
 -                 data = json.loads(request.data.decode("utf-8"))
 -                 if "graph" not in data or "features" not in data:
 -                     raise ValueError("graph or features not found in data")
 - 
 -                 if not isinstance(data.get("graph"), dict) or not isinstance(data.get("features"), dict):
 -                     raise ValueError("graph or features is not a dict")
 - 
 -                 args = {
 -                     "graph": data.get("graph"),
 -                     "features": data.get("features"),
 -                     "hash": data.get("hash"),
 -                     "environment_variables": data.get("environment_variables"),
 -                     "conversation_variables": data.get("conversation_variables"),
 -                 }
 -             except json.JSONDecodeError:
 -                 return {"message": "Invalid JSON data"}, 400
 -         else:
 -             abort(415)
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 - 
 -         workflow_service = WorkflowService()
 - 
 -         try:
 -             environment_variables_list = args.get("environment_variables") or []
 -             environment_variables = [
 -                 variable_factory.build_environment_variable_from_mapping(obj) for obj in environment_variables_list
 -             ]
 -             conversation_variables_list = args.get("conversation_variables") or []
 -             conversation_variables = [
 -                 variable_factory.build_conversation_variable_from_mapping(obj) for obj in conversation_variables_list
 -             ]
 -             workflow = workflow_service.sync_draft_workflow(
 -                 app_model=app_model,
 -                 graph=args["graph"],
 -                 features=args["features"],
 -                 unique_hash=args.get("hash"),
 -                 account=current_user,
 -                 environment_variables=environment_variables,
 -                 conversation_variables=conversation_variables,
 -             )
 -         except WorkflowHashNotEqualError:
 -             raise DraftWorkflowNotSync()
 - 
 -         return {
 -             "result": "success",
 -             "hash": workflow.unique_hash,
 -             "updated_at": TimestampField().format(workflow.updated_at or workflow.created_at),
 -         }
 - 
 - 
 - class AdvancedChatDraftWorkflowRunApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT])
 -     def post(self, app_model: App):
 -         """
 -         Run draft workflow
 -         """
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         assert isinstance(current_user, Account)
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, location="json")
 -         parser.add_argument("query", type=str, required=True, location="json", default="")
 -         parser.add_argument("files", type=list, location="json")
 -         parser.add_argument("conversation_id", type=uuid_value, location="json")
 -         parser.add_argument("parent_message_id", type=uuid_value, required=False, location="json")
 - 
 -         args = parser.parse_args()
 - 
 -         external_trace_id = get_external_trace_id(request)
 -         if external_trace_id:
 -             args["external_trace_id"] = external_trace_id
 - 
 -         try:
 -             response = AppGenerateService.generate(
 -                 app_model=app_model, user=current_user, args=args, invoke_from=InvokeFrom.DEBUGGER, streaming=True
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except services.errors.conversation.ConversationNotExistsError:
 -             raise NotFound("Conversation Not Exists.")
 -         except services.errors.conversation.ConversationCompletedError:
 -             raise ConversationCompletedError()
 -         except InvokeRateLimitError as ex:
 -             raise InvokeRateLimitHttpError(ex.description)
 -         except ValueError as e:
 -             raise e
 -         except Exception:
 -             logger.exception("internal server error.")
 -             raise InternalServerError()
 - 
 - 
 - class AdvancedChatDraftRunIterationNodeApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT])
 -     def post(self, app_model: App, node_id: str):
 -         """
 -         Run draft workflow iteration node
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, location="json")
 -         args = parser.parse_args()
 - 
 -         try:
 -             response = AppGenerateService.generate_single_iteration(
 -                 app_model=app_model, user=current_user, node_id=node_id, args=args, streaming=True
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except services.errors.conversation.ConversationNotExistsError:
 -             raise NotFound("Conversation Not Exists.")
 -         except services.errors.conversation.ConversationCompletedError:
 -             raise ConversationCompletedError()
 -         except ValueError as e:
 -             raise e
 -         except Exception:
 -             logger.exception("internal server error.")
 -             raise InternalServerError()
 - 
 - 
 - class WorkflowDraftRunIterationNodeApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.WORKFLOW])
 -     def post(self, app_model: App, node_id: str):
 -         """
 -         Run draft workflow iteration node
 -         """
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, location="json")
 -         args = parser.parse_args()
 - 
 -         try:
 -             response = AppGenerateService.generate_single_iteration(
 -                 app_model=app_model, user=current_user, node_id=node_id, args=args, streaming=True
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except services.errors.conversation.ConversationNotExistsError:
 -             raise NotFound("Conversation Not Exists.")
 -         except services.errors.conversation.ConversationCompletedError:
 -             raise ConversationCompletedError()
 -         except ValueError as e:
 -             raise e
 -         except Exception:
 -             logger.exception("internal server error.")
 -             raise InternalServerError()
 - 
 - 
 - class AdvancedChatDraftRunLoopNodeApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT])
 -     def post(self, app_model: App, node_id: str):
 -         """
 -         Run draft workflow loop node
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, location="json")
 -         args = parser.parse_args()
 - 
 -         try:
 -             response = AppGenerateService.generate_single_loop(
 -                 app_model=app_model, user=current_user, node_id=node_id, args=args, streaming=True
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except services.errors.conversation.ConversationNotExistsError:
 -             raise NotFound("Conversation Not Exists.")
 -         except services.errors.conversation.ConversationCompletedError:
 -             raise ConversationCompletedError()
 -         except ValueError as e:
 -             raise e
 -         except Exception:
 -             logger.exception("internal server error.")
 -             raise InternalServerError()
 - 
 - 
 - class WorkflowDraftRunLoopNodeApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.WORKFLOW])
 -     def post(self, app_model: App, node_id: str):
 -         """
 -         Run draft workflow loop node
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, location="json")
 -         args = parser.parse_args()
 - 
 -         try:
 -             response = AppGenerateService.generate_single_loop(
 -                 app_model=app_model, user=current_user, node_id=node_id, args=args, streaming=True
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except services.errors.conversation.ConversationNotExistsError:
 -             raise NotFound("Conversation Not Exists.")
 -         except services.errors.conversation.ConversationCompletedError:
 -             raise ConversationCompletedError()
 -         except ValueError as e:
 -             raise e
 -         except Exception:
 -             logger.exception("internal server error.")
 -             raise InternalServerError()
 - 
 - 
 - class DraftWorkflowRunApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.WORKFLOW])
 -     def post(self, app_model: App):
 -         """
 -         Run draft workflow
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, required=True, nullable=False, location="json")
 -         parser.add_argument("files", type=list, required=False, location="json")
 -         args = parser.parse_args()
 - 
 -         external_trace_id = get_external_trace_id(request)
 -         if external_trace_id:
 -             args["external_trace_id"] = external_trace_id
 - 
 -         try:
 -             response = AppGenerateService.generate(
 -                 app_model=app_model,
 -                 user=current_user,
 -                 args=args,
 -                 invoke_from=InvokeFrom.DEBUGGER,
 -                 streaming=True,
 -             )
 - 
 -             return helper.compact_generate_response(response)
 -         except InvokeRateLimitError as ex:
 -             raise InvokeRateLimitHttpError(ex.description)
 - 
 - 
 - class WorkflowTaskStopApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def post(self, app_model: App, task_id: str):
 -         """
 -         Stop workflow task
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         AppQueueManager.set_stop_flag(task_id, InvokeFrom.DEBUGGER, current_user.id)
 - 
 -         return {"result": "success"}
 - 
 - 
 - class DraftWorkflowNodeRunApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_run_node_execution_fields)
 -     def post(self, app_model: App, node_id: str):
 -         """
 -         Run draft workflow node
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("inputs", type=dict, required=True, nullable=False, location="json")
 -         parser.add_argument("query", type=str, required=False, location="json", default="")
 -         parser.add_argument("files", type=list, location="json", default=[])
 -         args = parser.parse_args()
 - 
 -         user_inputs = args.get("inputs")
 -         if user_inputs is None:
 -             raise ValueError("missing inputs")
 - 
 -         workflow_srv = WorkflowService()
 -         # fetch draft workflow by app_model
 -         draft_workflow = workflow_srv.get_draft_workflow(app_model=app_model)
 -         if not draft_workflow:
 -             raise ValueError("Workflow not initialized")
 -         files = _parse_file(draft_workflow, args.get("files"))
 -         workflow_service = WorkflowService()
 - 
 -         workflow_node_execution = workflow_service.run_draft_workflow_node(
 -             app_model=app_model,
 -             draft_workflow=draft_workflow,
 -             node_id=node_id,
 -             user_inputs=user_inputs,
 -             account=current_user,
 -             query=args.get("query", ""),
 -             files=files,
 -         )
 - 
 -         return workflow_node_execution
 - 
 - 
 - class PublishedWorkflowApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_fields)
 -     def get(self, app_model: App):
 -         """
 -         Get published workflow
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         # fetch published workflow by app_model
 -         workflow_service = WorkflowService()
 -         workflow = workflow_service.get_published_workflow(app_model=app_model)
 - 
 -         # return workflow, if not found, return None
 -         return workflow
 - 
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def post(self, app_model: App):
 -         """
 -         Publish workflow
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("marked_name", type=str, required=False, default="", location="json")
 -         parser.add_argument("marked_comment", type=str, required=False, default="", location="json")
 -         args = parser.parse_args()
 - 
 -         # Validate name and comment length
 -         if args.marked_name and len(args.marked_name) > 20:
 -             raise ValueError("Marked name cannot exceed 20 characters")
 -         if args.marked_comment and len(args.marked_comment) > 100:
 -             raise ValueError("Marked comment cannot exceed 100 characters")
 - 
 -         workflow_service = WorkflowService()
 -         with Session(db.engine) as session:
 -             workflow = workflow_service.publish_workflow(
 -                 session=session,
 -                 app_model=app_model,
 -                 account=current_user,
 -                 marked_name=args.marked_name or "",
 -                 marked_comment=args.marked_comment or "",
 -             )
 - 
 -             app_model.workflow_id = workflow.id
 -             db.session.commit()  # NOTE: this is necessary for update app_model.workflow_id
 - 
 -             workflow_created_at = TimestampField().format(workflow.created_at)
 - 
 -             session.commit()
 - 
 -         return {
 -             "result": "success",
 -             "created_at": workflow_created_at,
 -         }
 - 
 - 
 - class DefaultBlockConfigsApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def get(self, app_model: App):
 -         """
 -         Get default block config
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         # Get default block configs
 -         workflow_service = WorkflowService()
 -         return workflow_service.get_default_block_configs()
 - 
 - 
 - class DefaultBlockConfigApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def get(self, app_model: App, block_type: str):
 -         """
 -         Get default block config
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("q", type=str, location="args")
 -         args = parser.parse_args()
 - 
 -         q = args.get("q")
 - 
 -         filters = None
 -         if q:
 -             try:
 -                 filters = json.loads(args.get("q", ""))
 -             except json.JSONDecodeError:
 -                 raise ValueError("Invalid filters")
 - 
 -         # Get default block configs
 -         workflow_service = WorkflowService()
 -         return workflow_service.get_default_block_config(node_type=block_type, filters=filters)
 - 
 - 
 - class ConvertToWorkflowApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.CHAT, AppMode.COMPLETION])
 -     def post(self, app_model: App):
 -         """
 -         Convert basic mode of chatbot app to workflow mode
 -         Convert expert mode of chatbot app to workflow mode
 -         Convert Completion App to Workflow App
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # The role of the current user in the ta table must be admin, owner, or editor
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         if request.data:
 -             parser = reqparse.RequestParser()
 -             parser.add_argument("name", type=str, required=False, nullable=True, location="json")
 -             parser.add_argument("icon_type", type=str, required=False, nullable=True, location="json")
 -             parser.add_argument("icon", type=str, required=False, nullable=True, location="json")
 -             parser.add_argument("icon_background", type=str, required=False, nullable=True, location="json")
 -             args = parser.parse_args()
 -         else:
 -             args = {}
 - 
 -         # convert to workflow mode
 -         workflow_service = WorkflowService()
 -         new_app_model = workflow_service.convert_to_workflow(app_model=app_model, account=current_user, args=args)
 - 
 -         # return app id
 -         return {
 -             "new_app_id": new_app_model.id,
 -         }
 - 
 - 
 - class WorkflowConfigApi(Resource):
 -     """Resource for workflow configuration."""
 - 
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def get(self, app_model: App):
 -         return {
 -             "parallel_depth_limit": dify_config.WORKFLOW_PARALLEL_DEPTH_LIMIT,
 -         }
 - 
 - 
 - class PublishedAllWorkflowApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_pagination_fields)
 -     def get(self, app_model: App):
 -         """
 -         Get published workflows
 -         """
 - 
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("page", type=inputs.int_range(1, 99999), required=False, default=1, location="args")
 -         parser.add_argument("limit", type=inputs.int_range(1, 100), required=False, default=20, location="args")
 -         parser.add_argument("user_id", type=str, required=False, location="args")
 -         parser.add_argument("named_only", type=inputs.boolean, required=False, default=False, location="args")
 -         args = parser.parse_args()
 -         page = int(args.get("page", 1))
 -         limit = int(args.get("limit", 10))
 -         user_id = args.get("user_id")
 -         named_only = args.get("named_only", False)
 - 
 -         if user_id:
 -             if user_id != current_user.id:
 -                 raise Forbidden()
 -             user_id = cast(str, user_id)
 - 
 -         workflow_service = WorkflowService()
 -         with Session(db.engine) as session:
 -             workflows, has_more = workflow_service.get_all_published_workflow(
 -                 session=session,
 -                 app_model=app_model,
 -                 page=page,
 -                 limit=limit,
 -                 user_id=user_id,
 -                 named_only=named_only,
 -             )
 - 
 -             return {
 -                 "items": workflows,
 -                 "page": page,
 -                 "limit": limit,
 -                 "has_more": has_more,
 -             }
 - 
 - 
 - class WorkflowByIdApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_fields)
 -     def patch(self, app_model: App, workflow_id: str):
 -         """
 -         Update workflow attributes
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # Check permission
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         parser = reqparse.RequestParser()
 -         parser.add_argument("marked_name", type=str, required=False, location="json")
 -         parser.add_argument("marked_comment", type=str, required=False, location="json")
 -         args = parser.parse_args()
 - 
 -         # Validate name and comment length
 -         if args.marked_name and len(args.marked_name) > 20:
 -             raise ValueError("Marked name cannot exceed 20 characters")
 -         if args.marked_comment and len(args.marked_comment) > 100:
 -             raise ValueError("Marked comment cannot exceed 100 characters")
 - 
 -         # Prepare update data
 -         update_data = {}
 -         if args.get("marked_name") is not None:
 -             update_data["marked_name"] = args["marked_name"]
 -         if args.get("marked_comment") is not None:
 -             update_data["marked_comment"] = args["marked_comment"]
 - 
 -         if not update_data:
 -             return {"message": "No valid fields to update"}, 400
 - 
 -         workflow_service = WorkflowService()
 - 
 -         # Create a session and manage the transaction
 -         with Session(db.engine, expire_on_commit=False) as session:
 -             workflow = workflow_service.update_workflow(
 -                 session=session,
 -                 workflow_id=workflow_id,
 -                 tenant_id=app_model.tenant_id,
 -                 account_id=current_user.id,
 -                 data=update_data,
 -             )
 - 
 -             if not workflow:
 -                 raise NotFound("Workflow not found")
 - 
 -             # Commit the transaction in the controller
 -             session.commit()
 - 
 -         return workflow
 - 
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     def delete(self, app_model: App, workflow_id: str):
 -         """
 -         Delete workflow
 -         """
 -         if not isinstance(current_user, Account):
 -             raise Forbidden()
 -         # Check permission
 -         if not current_user.has_edit_permission:
 -             raise Forbidden()
 - 
 -         workflow_service = WorkflowService()
 - 
 -         # Create a session and manage the transaction
 -         with Session(db.engine) as session:
 -             try:
 -                 workflow_service.delete_workflow(
 -                     session=session, workflow_id=workflow_id, tenant_id=app_model.tenant_id
 -                 )
 -                 # Commit the transaction in the controller
 -                 session.commit()
 -             except WorkflowInUseError as e:
 -                 abort(400, description=str(e))
 -             except DraftWorkflowDeletionError as e:
 -                 abort(400, description=str(e))
 -             except ValueError as e:
 -                 raise NotFound(str(e))
 - 
 -         return None, 204
 - 
 - 
 - class DraftWorkflowNodeLastRunApi(Resource):
 -     @setup_required
 -     @login_required
 -     @account_initialization_required
 -     @get_app_model(mode=[AppMode.ADVANCED_CHAT, AppMode.WORKFLOW])
 -     @marshal_with(workflow_run_node_execution_fields)
 -     def get(self, app_model: App, node_id: str):
 -         srv = WorkflowService()
 -         workflow = srv.get_draft_workflow(app_model)
 -         if not workflow:
 -             raise NotFound("Workflow not found")
 -         node_exec = srv.get_node_last_run(
 -             app_model=app_model,
 -             workflow=workflow,
 -             node_id=node_id,
 -         )
 -         if node_exec is None:
 -             raise NotFound("last run not found")
 -         return node_exec
 - 
 - 
 - api.add_resource(
 -     DraftWorkflowApi,
 -     "/apps/<uuid:app_id>/workflows/draft",
 - )
 - api.add_resource(
 -     WorkflowConfigApi,
 -     "/apps/<uuid:app_id>/workflows/draft/config",
 - )
 - api.add_resource(
 -     AdvancedChatDraftWorkflowRunApi,
 -     "/apps/<uuid:app_id>/advanced-chat/workflows/draft/run",
 - )
 - api.add_resource(
 -     DraftWorkflowRunApi,
 -     "/apps/<uuid:app_id>/workflows/draft/run",
 - )
 - api.add_resource(
 -     WorkflowTaskStopApi,
 -     "/apps/<uuid:app_id>/workflow-runs/tasks/<string:task_id>/stop",
 - )
 - api.add_resource(
 -     DraftWorkflowNodeRunApi,
 -     "/apps/<uuid:app_id>/workflows/draft/nodes/<string:node_id>/run",
 - )
 - api.add_resource(
 -     AdvancedChatDraftRunIterationNodeApi,
 -     "/apps/<uuid:app_id>/advanced-chat/workflows/draft/iteration/nodes/<string:node_id>/run",
 - )
 - api.add_resource(
 -     WorkflowDraftRunIterationNodeApi,
 -     "/apps/<uuid:app_id>/workflows/draft/iteration/nodes/<string:node_id>/run",
 - )
 - api.add_resource(
 -     AdvancedChatDraftRunLoopNodeApi,
 -     "/apps/<uuid:app_id>/advanced-chat/workflows/draft/loop/nodes/<string:node_id>/run",
 - )
 - api.add_resource(
 -     WorkflowDraftRunLoopNodeApi,
 -     "/apps/<uuid:app_id>/workflows/draft/loop/nodes/<string:node_id>/run",
 - )
 - api.add_resource(
 -     PublishedWorkflowApi,
 -     "/apps/<uuid:app_id>/workflows/publish",
 - )
 - api.add_resource(
 -     PublishedAllWorkflowApi,
 -     "/apps/<uuid:app_id>/workflows",
 - )
 - api.add_resource(
 -     DefaultBlockConfigsApi,
 -     "/apps/<uuid:app_id>/workflows/default-workflow-block-configs",
 - )
 - api.add_resource(
 -     DefaultBlockConfigApi,
 -     "/apps/<uuid:app_id>/workflows/default-workflow-block-configs/<string:block_type>",
 - )
 - api.add_resource(
 -     ConvertToWorkflowApi,
 -     "/apps/<uuid:app_id>/convert-to-workflow",
 - )
 - api.add_resource(
 -     WorkflowByIdApi,
 -     "/apps/<uuid:app_id>/workflows/<string:workflow_id>",
 - )
 - api.add_resource(
 -     DraftWorkflowNodeLastRunApi,
 -     "/apps/<uuid:app_id>/workflows/draft/nodes/<string:node_id>/last-run",
 - )
 
 
  |