Underground-Digital/Workflow-Engine

1from flask_restful import fields2 3from fields.workflow_fields import workflow_partial_fields4from libs.helper import AppIconUrlField, TimestampField5 6app_detail_kernel_fields = {7    "id": fields.String,8    "name": fields.String,9    "description": fields.String,10    "mode": fields.String(attribute="mode_compatible_with_agent"),11    "icon_type": fields.String,12    "icon": fields.String,13    "icon_background": fields.String,14    "icon_url": AppIconUrlField,15}16 17related_app_list = {18    "data": fields.List(fields.Nested(app_detail_kernel_fields)),19    "total": fields.Integer,20}21 22model_config_fields = {23    "opening_statement": fields.String,24    "suggested_questions": fields.Raw(attribute="suggested_questions_list"),25    "suggested_questions_after_answer": fields.Raw(attribute="suggested_questions_after_answer_dict"),26    "speech_to_text": fields.Raw(attribute="speech_to_text_dict"),27    "text_to_speech": fields.Raw(attribute="text_to_speech_dict"),28    "retriever_resource": fields.Raw(attribute="retriever_resource_dict"),29    "annotation_reply": fields.Raw(attribute="annotation_reply_dict"),30    "more_like_this": fields.Raw(attribute="more_like_this_dict"),31    "sensitive_word_avoidance": fields.Raw(attribute="sensitive_word_avoidance_dict"),32    "external_data_tools": fields.Raw(attribute="external_data_tools_list"),33    "model": fields.Raw(attribute="model_dict"),34    "user_input_form": fields.Raw(attribute="user_input_form_list"),35    "dataset_query_variable": fields.String,36    "pre_prompt": fields.String,37    "agent_mode": fields.Raw(attribute="agent_mode_dict"),38    "prompt_type": fields.String,39    "chat_prompt_config": fields.Raw(attribute="chat_prompt_config_dict"),40    "completion_prompt_config": fields.Raw(attribute="completion_prompt_config_dict"),41    "dataset_configs": fields.Raw(attribute="dataset_configs_dict"),42    "file_upload": fields.Raw(attribute="file_upload_dict"),43    "created_by": fields.String,44    "created_at": TimestampField,45    "updated_by": fields.String,46    "updated_at": TimestampField,47}48 49app_detail_fields = {50    "id": fields.String,51    "name": fields.String,52    "description": fields.String,53    "mode": fields.String(attribute="mode_compatible_with_agent"),54    "icon": fields.String,55    "icon_background": fields.String,56    "enable_site": fields.Boolean,57    "enable_api": fields.Boolean,58    "model_config": fields.Nested(model_config_fields, attribute="app_model_config", allow_null=True),59    "workflow": fields.Nested(workflow_partial_fields, allow_null=True),60    "tracing": fields.Raw,61    "use_icon_as_answer_icon": fields.Boolean,62    "created_by": fields.String,63    "created_at": TimestampField,64    "updated_by": fields.String,65    "updated_at": TimestampField,66}67 68prompt_config_fields = {69    "prompt_template": fields.String,70}71 72model_config_partial_fields = {73    "model": fields.Raw(attribute="model_dict"),74    "pre_prompt": fields.String,75    "created_by": fields.String,76    "created_at": TimestampField,77    "updated_by": fields.String,78    "updated_at": TimestampField,79}80 81tag_fields = {"id": fields.String, "name": fields.String, "type": fields.String}82 83app_partial_fields = {84    "id": fields.String,85    "name": fields.String,86    "max_active_requests": fields.Raw(),87    "description": fields.String(attribute="desc_or_prompt"),88    "mode": fields.String(attribute="mode_compatible_with_agent"),89    "icon_type": fields.String,90    "icon": fields.String,91    "icon_background": fields.String,92    "icon_url": AppIconUrlField,93    "model_config": fields.Nested(model_config_partial_fields, attribute="app_model_config", allow_null=True),94    "workflow": fields.Nested(workflow_partial_fields, allow_null=True),95    "use_icon_as_answer_icon": fields.Boolean,96    "created_by": fields.String,97    "created_at": TimestampField,98    "updated_by": fields.String,99    "updated_at": TimestampField,100    "tags": fields.List(fields.Nested(tag_fields)),101}102 103 104app_pagination_fields = {105    "page": fields.Integer,106    "limit": fields.Integer(attribute="per_page"),107    "total": fields.Integer,108    "has_more": fields.Boolean(attribute="has_next"),109    "data": fields.List(fields.Nested(app_partial_fields), attribute="items"),110}111 112template_fields = {113    "name": fields.String,114    "icon": fields.String,115    "icon_background": fields.String,116    "description": fields.String,117    "mode": fields.String,118    "model_config": fields.Nested(model_config_fields),119}120 121template_list_fields = {122    "data": fields.List(fields.Nested(template_fields)),123}124 125site_fields = {126    "access_token": fields.String(attribute="code"),127    "code": fields.String,128    "title": fields.String,129    "icon_type": fields.String,130    "icon": fields.String,131    "icon_background": fields.String,132    "icon_url": AppIconUrlField,133    "description": fields.String,134    "default_language": fields.String,135    "chat_color_theme": fields.String,136    "chat_color_theme_inverted": fields.Boolean,137    "customize_domain": fields.String,138    "copyright": fields.String,139    "privacy_policy": fields.String,140    "custom_disclaimer": fields.String,141    "customize_token_strategy": fields.String,142    "prompt_public": fields.Boolean,143    "app_base_url": fields.String,144    "show_workflow_steps": fields.Boolean,145    "use_icon_as_answer_icon": fields.Boolean,146    "created_by": fields.String,147    "created_at": TimestampField,148    "updated_by": fields.String,149    "updated_at": TimestampField,150}151 152app_detail_fields_with_site = {153    "id": fields.String,154    "name": fields.String,155    "description": fields.String,156    "mode": fields.String(attribute="mode_compatible_with_agent"),157    "icon_type": fields.String,158    "icon": fields.String,159    "icon_background": fields.String,160    "icon_url": AppIconUrlField,161    "enable_site": fields.Boolean,162    "enable_api": fields.Boolean,163    "model_config": fields.Nested(model_config_fields, attribute="app_model_config", allow_null=True),164    "workflow": fields.Nested(workflow_partial_fields, allow_null=True),165    "site": fields.Nested(site_fields),166    "api_base_url": fields.String,167    "use_icon_as_answer_icon": fields.Boolean,168    "created_by": fields.String,169    "created_at": TimestampField,170    "updated_by": fields.String,171    "updated_at": TimestampField,172    "deleted_tools": fields.List(fields.String),173}174 175app_site_fields = {176    "app_id": fields.String,177    "access_token": fields.String(attribute="code"),178    "code": fields.String,179    "title": fields.String,180    "icon": fields.String,181    "icon_background": fields.String,182    "description": fields.String,183    "default_language": fields.String,184    "customize_domain": fields.String,185    "copyright": fields.String,186    "privacy_policy": fields.String,187    "custom_disclaimer": fields.String,188    "customize_token_strategy": fields.String,189    "prompt_public": fields.Boolean,190    "show_workflow_steps": fields.Boolean,191    "use_icon_as_answer_icon": fields.Boolean,192}193