mirror of
https://github.com/langgenius/dify.git
synced 2025-11-24 00:42:41 +00:00
2f50f3fd4b
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from flask_restx import Resource, fields
|
|
|
|
from libs.login import current_user, login_required
|
|
from models.account import Account
|
|
from services.feature_service import FeatureService
|
|
|
|
from . import api, console_ns
|
|
from .wraps import account_initialization_required, cloud_utm_record, setup_required
|
|
|
|
|
|
@console_ns.route("/features")
|
|
class FeatureApi(Resource):
|
|
@api.doc("get_tenant_features")
|
|
@api.doc(description="Get feature configuration for current tenant")
|
|
@api.response(
|
|
200,
|
|
"Success",
|
|
api.model("FeatureResponse", {"features": fields.Raw(description="Feature configuration object")}),
|
|
)
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@cloud_utm_record
|
|
def get(self):
|
|
"""Get feature configuration for current tenant"""
|
|
assert isinstance(current_user, Account)
|
|
assert current_user.current_tenant_id is not None
|
|
return FeatureService.get_features(current_user.current_tenant_id).model_dump()
|
|
|
|
|
|
@console_ns.route("/system-features")
|
|
class SystemFeatureApi(Resource):
|
|
@api.doc("get_system_features")
|
|
@api.doc(description="Get system-wide feature configuration")
|
|
@api.response(
|
|
200,
|
|
"Success",
|
|
api.model("SystemFeatureResponse", {"features": fields.Raw(description="System feature configuration object")}),
|
|
)
|
|
def get(self):
|
|
"""Get system-wide feature configuration"""
|
|
return FeatureService.get_system_features().model_dump()
|