mirror of
https://github.com/langgenius/dify.git
synced 2025-07-04 23:57:12 +00:00
10b738a296
This pull request introduces a feature aimed at improving the debugging experience during workflow editing. With the addition of variable persistence, the system will automatically retain the output variables from previously executed nodes. These persisted variables can then be reused when debugging subsequent nodes, eliminating the need for repetitive manual input. By streamlining this aspect of the workflow, the feature minimizes user errors and significantly reduces debugging effort, offering a smoother and more efficient experience. Key highlights of this change: - Automatic persistence of output variables for executed nodes. - Reuse of persisted variables to simplify input steps for nodes requiring them (e.g., `code`, `template`, `variable_assigner`). - Enhanced debugging experience with reduced friction. Closes #19735.
61 lines
1.7 KiB
Python
61 lines
1.7 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 import App, AppMode
|
|
|
|
|
|
def _load_app_model(app_id: str) -> Optional[App]:
|
|
app_model = (
|
|
db.session.query(App)
|
|
.filter(App.id == app_id, App.tenant_id == current_user.current_tenant_id, App.status == "normal")
|
|
.first()
|
|
)
|
|
return app_model
|
|
|
|
|
|
def get_app_model(view: Optional[Callable] = None, *, mode: Union[AppMode, list[AppMode], None] = 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 = _load_app_model(app_id)
|
|
|
|
if not app_model:
|
|
raise AppNotFoundError()
|
|
|
|
app_mode = AppMode.value_of(app_model.mode)
|
|
if app_mode == AppMode.CHANNEL:
|
|
raise AppNotFoundError()
|
|
|
|
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}")
|
|
|
|
kwargs["app_model"] = app_model
|
|
|
|
return view_func(*args, **kwargs)
|
|
|
|
return decorated_view
|
|
|
|
if view is None:
|
|
return decorator
|
|
else:
|
|
return decorator(view)
|