mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
from flask_restful import fields, marshal_with
|
|
|
|
from configs import dify_config
|
|
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.Integer,
|
|
"video_file_size_limit": fields.Integer,
|
|
"audio_file_size_limit": fields.Integer,
|
|
"file_size_limit": fields.Integer,
|
|
}
|
|
|
|
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),
|
|
}
|
|
|
|
@marshal_with(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": dify_config.UPLOAD_IMAGE_FILE_SIZE_LIMIT,
|
|
"video_file_size_limit": dify_config.UPLOAD_VIDEO_FILE_SIZE_LIMIT,
|
|
"audio_file_size_limit": dify_config.UPLOAD_AUDIO_FILE_SIZE_LIMIT,
|
|
"file_size_limit": dify_config.UPLOAD_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")
|