From 9e3a749d5017f73f0debdb7de4029c617f0c3010 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Tue, 24 Jun 2025 14:08:29 +0800
Subject: [PATCH] add dataset info in response

---
 api/controllers/console/datasets/datasets_document.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/controllers/console/datasets/datasets_document.py b/api/controllers/console/datasets/datasets_document.py
index c9cb035a4c..da96170a88 100644
--- a/api/controllers/console/datasets/datasets_document.py
+++ b/api/controllers/console/datasets/datasets_document.py
@@ -288,6 +288,8 @@ class DatasetDocumentListApi(Resource):
 
         try:
             documents, batch = DocumentService.save_document_with_dataset_id(dataset, knowledge_config, current_user)
+            dataset = DatasetService.get_dataset(dataset_id)
+
         except ProviderTokenNotInitError as ex:
             raise ProviderNotInitializeError(ex.description)
         except QuotaExceededError: