dify/api/controllers/console/feature.py

29 lines
729 B
Python
Raw Normal View History

from flask_login import current_user
from flask_restful import Resource
from libs.login import login_required
from services.feature_service import FeatureService
from . import api
from .setup import setup_required
from .wraps import account_initialization_required, cloud_utm_record
class FeatureApi(Resource):
@setup_required
@login_required
@account_initialization_required
@cloud_utm_record
def get(self):
return FeatureService.get_features(current_user.current_tenant_id).dict()
2024-05-15 16:14:49 +08:00
class SystemFeatureApi(Resource):
2024-04-18 17:33:32 +08:00
def get(self):
2024-05-15 16:14:49 +08:00
return FeatureService.get_system_features().dict()
2024-04-18 17:33:32 +08:00
api.add_resource(FeatureApi, '/features')
2024-05-15 16:14:49 +08:00
api.add_resource(SystemFeatureApi, '/system-features')