Knowledge optimization (#3755)

Co-authored-by: crazywoola <427733928@qq.com>
Co-authored-by: JzoNg <jzongcode@gmail.com>
This commit is contained in:
Jyong
2024-04-24 15:02:29 +08:00
committed by GitHub
parent 3cd8e6f5c6
commit f257f2c396
75 changed files with 2756 additions and 266 deletions

View File

@@ -27,6 +27,11 @@ dataset_retrieval_model_fields = {
'score_threshold': fields.Float
}
tag_fields = {
'id': fields.String,
'name': fields.String,
'type': fields.String
}
dataset_detail_fields = {
'id': fields.String,
@@ -46,7 +51,8 @@ dataset_detail_fields = {
'embedding_model': fields.String,
'embedding_model_provider': fields.String,
'embedding_available': fields.Boolean,
'retrieval_model_dict': fields.Nested(dataset_retrieval_model_fields)
'retrieval_model_dict': fields.Nested(dataset_retrieval_model_fields),
'tags': fields.List(fields.Nested(tag_fields))
}
dataset_query_detail_fields = {