youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

add AdvancedChatAppGenerateTaskPipeline

Browse Source
This commit is contained in:
takatost
2024-03-04 02:04:40 +08:00
parent 602bc67495
commit be709d4b84
14 changed files with 570 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
api/services/workflow_service.py
View File

@ -4,6 +4,7 @@ from typing import Optional
from core.app.apps.advanced_chat.app_config_manager import AdvancedChatAppConfigManager
from core.app.apps.workflow.app_config_manager import WorkflowAppConfigManager
from core.workflow.workflow_engine_manager import WorkflowEngineManager
from extensions.ext_database import db
from models.account import Account
from models.model import App, AppMode
@ -21,15 +22,10 @@ class WorkflowService:
"""
Get draft workflow
"""
# fetch draft workflow by app_model
workflow = db.session.query(Workflow).filter(
Workflow.tenant_id == app_model.tenant_id,
Workflow.app_id == app_model.id,
Workflow.version == 'draft'
).first()
workflow_engine_manager = WorkflowEngineManager()
# return draft workflow
return workflow
return workflow_engine_manager.get_draft_workflow(app_model=app_model)
def get_published_workflow(self, app_model: App) -> Optional[Workflow]:
"""
@ -38,15 +34,10 @@ class WorkflowService:
if not app_model.workflow_id:
return None
# fetch published workflow by workflow_id
workflow = db.session.query(Workflow).filter(
Workflow.tenant_id == app_model.tenant_id,
Workflow.app_id == app_model.id,
Workflow.id == app_model.workflow_id
).first()
workflow_engine_manager = WorkflowEngineManager()
# return published workflow
return workflow
return workflow_engine_manager.get_published_workflow(app_model=app_model)
def sync_draft_workflow(self, app_model: App,
graph: dict,