Spaces:
Paused
Paused
| from flask import current_app | |
| from flask_restful import fields, marshal_with | |
| from controllers.console import api | |
| from controllers.console.app.error import AppUnavailableError | |
| from controllers.console.explore.wraps import InstalledAppResource | |
| from models.model import AppMode, InstalledApp | |
| from services.app_service import AppService | |
| class AppParameterApi(InstalledAppResource): | |
| """Resource for app variables.""" | |
| variable_fields = { | |
| 'key': fields.String, | |
| 'name': fields.String, | |
| 'description': fields.String, | |
| 'type': fields.String, | |
| 'default': fields.String, | |
| 'max_length': fields.Integer, | |
| 'options': fields.List(fields.String) | |
| } | |
| system_parameters_fields = { | |
| 'image_file_size_limit': fields.String | |
| } | |
| parameters_fields = { | |
| 'opening_statement': fields.String, | |
| 'suggested_questions': fields.Raw, | |
| 'suggested_questions_after_answer': fields.Raw, | |
| 'speech_to_text': fields.Raw, | |
| 'text_to_speech': fields.Raw, | |
| 'retriever_resource': fields.Raw, | |
| 'annotation_reply': fields.Raw, | |
| 'more_like_this': fields.Raw, | |
| 'user_input_form': fields.Raw, | |
| 'sensitive_word_avoidance': fields.Raw, | |
| 'file_upload': fields.Raw, | |
| 'system_parameters': fields.Nested(system_parameters_fields) | |
| } | |
| def get(self, installed_app: InstalledApp): | |
| """Retrieve app parameters.""" | |
| app_model = installed_app.app | |
| if app_model.mode in [AppMode.ADVANCED_CHAT.value, AppMode.WORKFLOW.value]: | |
| workflow = app_model.workflow | |
| if workflow is None: | |
| raise AppUnavailableError() | |
| features_dict = workflow.features_dict | |
| user_input_form = workflow.user_input_form(to_old_structure=True) | |
| else: | |
| app_model_config = app_model.app_model_config | |
| features_dict = app_model_config.to_dict() | |
| user_input_form = features_dict.get('user_input_form', []) | |
| return { | |
| 'opening_statement': features_dict.get('opening_statement'), | |
| 'suggested_questions': features_dict.get('suggested_questions', []), | |
| 'suggested_questions_after_answer': features_dict.get('suggested_questions_after_answer', | |
| {"enabled": False}), | |
| 'speech_to_text': features_dict.get('speech_to_text', {"enabled": False}), | |
| 'text_to_speech': features_dict.get('text_to_speech', {"enabled": False}), | |
| 'retriever_resource': features_dict.get('retriever_resource', {"enabled": False}), | |
| 'annotation_reply': features_dict.get('annotation_reply', {"enabled": False}), | |
| 'more_like_this': features_dict.get('more_like_this', {"enabled": False}), | |
| 'user_input_form': user_input_form, | |
| 'sensitive_word_avoidance': features_dict.get('sensitive_word_avoidance', | |
| {"enabled": False, "type": "", "configs": []}), | |
| 'file_upload': features_dict.get('file_upload', {"image": { | |
| "enabled": False, | |
| "number_limits": 3, | |
| "detail": "high", | |
| "transfer_methods": ["remote_url", "local_file"] | |
| }}), | |
| 'system_parameters': { | |
| 'image_file_size_limit': current_app.config.get('UPLOAD_IMAGE_FILE_SIZE_LIMIT') | |
| } | |
| } | |
| class ExploreAppMetaApi(InstalledAppResource): | |
| def get(self, installed_app: InstalledApp): | |
| """Get app meta""" | |
| app_model = installed_app.app | |
| return AppService().get_app_meta(app_model) | |
| api.add_resource(AppParameterApi, '/installed-apps/<uuid:installed_app_id>/parameters', | |
| endpoint='installed_app_parameters') | |
| api.add_resource(ExploreAppMetaApi, '/installed-apps/<uuid:installed_app_id>/meta', endpoint='installed_app_meta') | |