r2

2025-12-06 19:42:42 +08:00 · 2025-06-25 13:58:26 +08:00
1 changed files with 13 additions and 10 deletions
--- a/api/fields/dataset_fields.py
+++ b/api/fields/dataset_fields.py
@@ -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,