63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
from flask_restx import Api, Namespace, fields
|
|
|
|
from libs.helper import AppIconUrlField
|
|
|
|
parameters__system_parameters = {
|
|
"image_file_size_limit": fields.Integer,
|
|
"video_file_size_limit": fields.Integer,
|
|
"audio_file_size_limit": fields.Integer,
|
|
"file_size_limit": fields.Integer,
|
|
"workflow_file_upload_limit": fields.Integer,
|
|
}
|
|
|
|
|
|
def build_system_parameters_model(api_or_ns: Api | Namespace):
|
|
"""Build the system parameters model for the API or Namespace."""
|
|
return api_or_ns.model("SystemParameters", parameters__system_parameters)
|
|
|
|
|
|
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(parameters__system_parameters),
|
|
}
|
|
|
|
|
|
def build_parameters_model(api_or_ns: Api | Namespace):
|
|
"""Build the parameters model for the API or Namespace."""
|
|
copied_fields = parameters_fields.copy()
|
|
copied_fields["system_parameters"] = fields.Nested(build_system_parameters_model(api_or_ns))
|
|
return api_or_ns.model("Parameters", copied_fields)
|
|
|
|
|
|
site_fields = {
|
|
"title": fields.String,
|
|
"chat_color_theme": fields.String,
|
|
"chat_color_theme_inverted": fields.Boolean,
|
|
"icon_type": fields.String,
|
|
"icon": fields.String,
|
|
"icon_background": fields.String,
|
|
"icon_url": AppIconUrlField,
|
|
"description": fields.String,
|
|
"copyright": fields.String,
|
|
"privacy_policy": fields.String,
|
|
"custom_disclaimer": fields.String,
|
|
"default_language": fields.String,
|
|
"show_workflow_steps": fields.Boolean,
|
|
"use_icon_as_answer_icon": fields.Boolean,
|
|
}
|
|
|
|
|
|
def build_site_model(api_or_ns: Api | Namespace):
|
|
"""Build the site model for the API or Namespace."""
|
|
return api_or_ns.model("Site", site_fields)
|