From 3557ea50fa444022c6fec206dec0864fd22e1952 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Tue, 8 Apr 2025 17:24:33 +0800
Subject: [PATCH] fix: Missing parameters such as character length when
 creating a knowledge base (#2827)

---
 apps/dataset/serializers/dataset_serializers.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/apps/dataset/serializers/dataset_serializers.py b/apps/dataset/serializers/dataset_serializers.py
index 5b94c6c0c..895443d99 100644
--- a/apps/dataset/serializers/dataset_serializers.py
+++ b/apps/dataset/serializers/dataset_serializers.py
@@ -455,11 +455,13 @@ class DataSetSerializers(serializers.ModelSerializer):
             # 批量插入关联问题
             QuerySet(ProblemParagraphMapping).bulk_create(problem_paragraph_mapping_list) if len(
                 problem_paragraph_mapping_list) > 0 else None
-
             # 响应数据
             return {**DataSetSerializers(dataset).data,
-                'document_list': DocumentSerializers.Query(data={'dataset_id': dataset_id}).list(
-                    with_valid=True)}, dataset_id
+                    'user_id': user_id,
+                    'document_list': document_model_list,
+                    "document_count": len(document_model_list),
+                    "char_length": reduce(lambda x, y: x + y, [d.char_length for d in document_model_list],
+                                           0)}, dataset_id
 
         @staticmethod
         def get_last_url_path(url):