Skip to content

Commit 3557ea5

Browse files
authored
fix: Missing parameters such as character length when creating a knowledge base (#2827)
1 parent 43702e4 commit 3557ea5

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

apps/dataset/serializers/dataset_serializers.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -455,11 +455,13 @@ def save(self, instance: Dict, with_valid=True):
455455
# 批量插入关联问题
456456
QuerySet(ProblemParagraphMapping).bulk_create(problem_paragraph_mapping_list) if len(
457457
problem_paragraph_mapping_list) > 0 else None
458-
459458
# 响应数据
460459
return {**DataSetSerializers(dataset).data,
461-
'document_list': DocumentSerializers.Query(data={'dataset_id': dataset_id}).list(
462-
with_valid=True)}, dataset_id
460+
'user_id': user_id,
461+
'document_list': document_model_list,
462+
"document_count": len(document_model_list),
463+
"char_length": reduce(lambda x, y: x + y, [d.char_length for d in document_model_list],
464+
0)}, dataset_id
463465

464466
@staticmethod
465467
def get_last_url_path(url):

0 commit comments

Comments
 (0)