mirror of https://github.com/langgenius/dify.git
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from collections.abc import Callable
|
|
from functools import wraps
|
|
from typing import Optional, Union
|
|
|
|
from controllers.console.app.error import AppNotFoundError
|
|
from extensions.ext_database import db
|
|
from libs.login import current_user
|
|
from models.model import App, AppMode, ChatbotAppEngine
|
|
|
|
|
|
def get_app_model(view: Optional[Callable] = None, *,
|
|
mode: Union[AppMode, list[AppMode]] = None,
|
|
app_engine: ChatbotAppEngine = None):
|
|
def decorator(view_func):
|
|
@wraps(view_func)
|
|
def decorated_view(*args, **kwargs):
|
|
if not kwargs.get('app_id'):
|
|
raise ValueError('missing app_id in path parameters')
|
|
|
|
app_id = kwargs.get('app_id')
|
|
app_id = str(app_id)
|
|
|
|
del kwargs['app_id']
|
|
|
|
app_model = db.session.query(App).filter(
|
|
App.id == app_id,
|
|
App.tenant_id == current_user.current_tenant_id,
|
|
App.status == 'normal'
|
|
).first()
|
|
|
|
if not app_model:
|
|
raise AppNotFoundError()
|
|
|
|
app_mode = AppMode.value_of(app_model.mode)
|
|
if mode is not None:
|
|
if isinstance(mode, list):
|
|
modes = mode
|
|
else:
|
|
modes = [mode]
|
|
|
|
if app_mode not in modes:
|
|
mode_values = {m.value for m in modes}
|
|
raise AppNotFoundError(f"App mode is not in the supported list: {mode_values}")
|
|
|
|
if app_engine is not None:
|
|
if app_mode not in [AppMode.CHAT, AppMode.WORKFLOW]:
|
|
raise AppNotFoundError(f"App mode is not supported for {app_engine.value} app engine.")
|
|
|
|
if app_mode == AppMode.CHAT:
|
|
# fetch current app model config
|
|
app_model_config = app_model.app_model_config
|
|
if not app_model_config or app_model_config.chatbot_app_engine != app_engine.value:
|
|
raise AppNotFoundError(f"{app_engine.value} app engine is not supported.")
|
|
|
|
kwargs['app_model'] = app_model
|
|
|
|
return view_func(*args, **kwargs)
|
|
return decorated_view
|
|
|
|
if view is None:
|
|
return decorator
|
|
else:
|
|
return decorator(view)
|