From cc62cc77ea2147228bd2243fb060ae6b0616ad94 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 25 Jun 2025 13:58:26 +0800
Subject: [PATCH] r2

---
 api/fields/dataset_fields.py | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/api/fields/dataset_fields.py b/api/fields/dataset_fields.py
index 32a88cc5db..a1e558968e 100644
--- 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,