This commit is contained in:
jyong 2025-06-25 13:58:26 +08:00
parent a098825fcc
commit cc62cc77ea

View File

@ -2,16 +2,7 @@ from flask_restful import fields
from libs.helper import TimestampField
dataset_fields = {
"id": fields.String,
"name": fields.String,
"description": fields.String,
"permission": fields.String,
"data_source_type": fields.String,
"indexing_technique": fields.String,
"created_by": fields.String,
"created_at": TimestampField,
}
reranking_model_fields = {"reranking_provider_name": fields.String, "reranking_model_name": fields.String}
@ -56,6 +47,18 @@ external_knowledge_info_fields = {
doc_metadata_fields = {"id": fields.String, "name": fields.String, "type": fields.String}
dataset_fields = {
"id": fields.String,
"name": fields.String,
"description": fields.String,
"permission": fields.String,
"data_source_type": fields.String,
"indexing_technique": fields.String,
"created_by": fields.String,
"created_at": TimestampField,
"retrieval_model_dict": fields.Nested(dataset_retrieval_model_fields),
}
dataset_detail_fields = {
"id": fields.String,
"name": fields.String,