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

fix bugs and add unit tests

Browse Source
This commit is contained in:
takatost
2024-02-22 15:15:42 +08:00
parent 8642354a2a
commit 3b234febf5
9 changed files with 292 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/models/workflow.py
View File

@ -5,7 +5,6 @@ from sqlalchemy.dialects.postgresql import UUID
from extensions.ext_database import db
from models.account import Account
from models.model import AppMode
class WorkflowType(Enum):
@ -29,13 +28,14 @@ class WorkflowType(Enum):
raise ValueError(f'invalid workflow type value {value}')
@classmethod
def from_app_mode(cls, app_mode: Union[str, AppMode]) -> 'WorkflowType':
def from_app_mode(cls, app_mode: Union[str, 'AppMode']) -> 'WorkflowType':
"""
Get workflow type from app mode.
:param app_mode: app mode
:return: workflow type
"""
from models.model import AppMode
app_mode = app_mode if isinstance(app_mode, AppMode) else AppMode.value_of(app_mode)
return cls.WORKFLOW if app_mode == AppMode.WORKFLOW else cls.CHAT