Feat/assistant app (#2086)

Co-authored-by: chenhe <guchenhe@gmail.com>
Co-authored-by: Pascal M <11357019+perzeuss@users.noreply.github.com>
This commit is contained in:
Yeuoly
2024-01-23 19:58:23 +08:00
committed by GitHub
parent 7bbe12b2bd
commit 86286e1ac8
175 changed files with 11619 additions and 1235 deletions

View File

@@ -40,6 +40,7 @@ app_detail_fields = {
'id': fields.String,
'name': fields.String,
'mode': fields.String,
'is_agent': fields.Boolean,
'icon': fields.String,
'icon_background': fields.String,
'enable_site': fields.Boolean,
@@ -64,6 +65,7 @@ app_partial_fields = {
'id': fields.String,
'name': fields.String,
'mode': fields.String,
'is_agent': fields.Boolean,
'icon': fields.String,
'icon_background': fields.String,
'enable_site': fields.Boolean,
@@ -120,11 +122,13 @@ app_detail_fields_with_site = {
'enable_api': fields.Boolean,
'api_rpm': fields.Integer,
'api_rph': fields.Integer,
'is_agent': fields.Boolean,
'is_demo': fields.Boolean,
'model_config': fields.Nested(model_config_fields, attribute='app_model_config'),
'site': fields.Nested(site_fields),
'api_base_url': fields.String,
'created_at': TimestampField
'created_at': TimestampField,
'deleted_tools': fields.List(fields.String),
}
app_site_fields = {

View File

@@ -39,6 +39,20 @@ message_file_fields = {
'id': fields.String,
'type': fields.String,
'url': fields.String,
'belongs_to': fields.String(default='user'),
}
agent_thought_fields = {
'id': fields.String,
'chain_id': fields.String,
'message_id': fields.String,
'position': fields.Integer,
'thought': fields.String,
'tool': fields.String,
'tool_input': fields.String,
'created_at': TimestampField,
'observation': fields.String,
'files': fields.List(fields.String)
}
message_detail_fields = {
@@ -58,6 +72,7 @@ message_detail_fields = {
'annotation': fields.Nested(annotation_fields, allow_null=True),
'annotation_hit_history': fields.Nested(annotation_hit_history_fields, allow_null=True),
'created_at': TimestampField,
'agent_thoughts': fields.List(fields.Nested(agent_thought_fields)),
'message_files': fields.List(fields.Nested(message_file_fields), attribute='files'),
}

View File

@@ -17,6 +17,7 @@ installed_app_fields = {
'last_used_at': TimestampField,
'editable': fields.Boolean,
'uninstallable': fields.Boolean,
'is_agent': fields.Boolean,
}
installed_app_list_fields = {

View File

@@ -25,20 +25,57 @@ retriever_resource_fields = {
'created_at': TimestampField
}
feedback_fields = {
'rating': fields.String
}
agent_thought_fields = {
'id': fields.String,
'chain_id': fields.String,
'message_id': fields.String,
'position': fields.Integer,
'thought': fields.String,
'tool': fields.String,
'tool_input': fields.String,
'created_at': TimestampField,
'observation': fields.String,
'files': fields.List(fields.String)
}
retriever_resource_fields = {
'id': fields.String,
'message_id': fields.String,
'position': fields.Integer,
'dataset_id': fields.String,
'dataset_name': fields.String,
'document_id': fields.String,
'document_name': fields.String,
'data_source_type': fields.String,
'segment_id': fields.String,
'score': fields.Float,
'hit_count': fields.Integer,
'word_count': fields.Integer,
'segment_position': fields.Integer,
'index_node_hash': fields.String,
'content': fields.String,
'created_at': TimestampField
}
message_fields = {
'id': fields.String,
'conversation_id': fields.String,
'inputs': fields.Raw,
'query': fields.String,
'answer': fields.String,
'message_files': fields.List(fields.Nested(message_file_fields), attribute='files'),
'feedback': fields.Nested(feedback_fields, attribute='user_feedback', allow_null=True),
'retriever_resources': fields.List(fields.Nested(retriever_resource_fields)),
'created_at': TimestampField
'created_at': TimestampField,
'agent_thoughts': fields.List(fields.Nested(agent_thought_fields)),
'message_files': fields.List(fields.Nested(message_file_fields), attribute='files')
}
message_infinite_scroll_pagination_fields = {
'limit': fields.Integer,
'has_more': fields.Boolean,
'data': fields.List(fields.Nested(message_fields))
}
}