From 0a9c01cf33ecb417b20d50dea3cccead3aa497e1 Mon Sep 17 00:00:00 2001
From: yujiangping <yujiangping@taofen8.com>
Date: Wed, 17 Dec 2025 14:16:25 +0800
Subject: [PATCH] feat:create cusotm text dataset

---
 .../[knowledgeBaseId]/CreateDataset.tsx       |  4 +--
 .../[knowledgeBaseId]/Private.tsx             |  1 +
 .../components/CreateContentModal.tsx         | 27 ++++++++++++++-----
 3 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/web/src/views/KnowledgeBase/[knowledgeBaseId]/CreateDataset.tsx b/web/src/views/KnowledgeBase/[knowledgeBaseId]/CreateDataset.tsx
index 7139230d..d4a9084f 100644
--- a/web/src/views/KnowledgeBase/[knowledgeBaseId]/CreateDataset.tsx
+++ b/web/src/views/KnowledgeBase/[knowledgeBaseId]/CreateDataset.tsx
@@ -552,8 +552,8 @@ const CreateDataset = () => {
             {rechunkFileIds.length > 0 ? (
               <Table
                 ref={tableRef}
-                apiUrl={`/documents/${knowledgeBaseId}/${parentId}/documents`}
-                apiParams={{
+                apiUrl={`/documents/${knowledgeBaseId}/documents`}
+                apiParams={{                    
                     document_ids: rechunkFileIds.join(','),
                 }}
                 columns={columns}
diff --git a/web/src/views/KnowledgeBase/[knowledgeBaseId]/Private.tsx b/web/src/views/KnowledgeBase/[knowledgeBaseId]/Private.tsx
index 96b1af9d..ed909515 100644
--- a/web/src/views/KnowledgeBase/[knowledgeBaseId]/Private.tsx
+++ b/web/src/views/KnowledgeBase/[knowledgeBaseId]/Private.tsx
@@ -605,6 +605,7 @@ const Private: FC = () => {
   }
 
   const handleRefreshTable = () => {
+    debugger
     // 刷新表格数据
     tableRef.current?.loadData();
   }
diff --git a/web/src/views/KnowledgeBase/components/CreateContentModal.tsx b/web/src/views/KnowledgeBase/components/CreateContentModal.tsx
index 5ad0b4f8..63292dec 100644
--- a/web/src/views/KnowledgeBase/components/CreateContentModal.tsx
+++ b/web/src/views/KnowledgeBase/components/CreateContentModal.tsx
@@ -1,4 +1,6 @@
 import { forwardRef, useImperativeHandle, useState } from 'react';
+import { useNavigate } from 'react-router-dom';
+
 import { Form, Input } from 'antd';
 import { useTranslation } from 'react-i18next';
 import RbModal from '@/components/RbModal';
@@ -12,6 +14,7 @@ interface ContentFormData {
 const CreateContentModal = forwardRef<CreateSetModalRef, CreateSetMoealRefProps>(
   ({ refreshTable }, ref) => {
     const { t } = useTranslation();
+    const navigate = useNavigate();
     const [visible, setVisible] = useState(false);
     const [form] = Form.useForm<ContentFormData>();
     const [loading, setLoading] = useState(false);
@@ -45,13 +48,11 @@ const CreateContentModal = forwardRef<CreateSetModalRef, CreateSetMoealRefProps>
           parent_id: parentId,
         };
 
-        // 模拟API调用
-        await new Promise(resolve => setTimeout(resolve, 1000));
-        await createDocumentAndUpload(values, params)
-        if (refreshTable) {
-          await refreshTable();
-        }
 
+        const response = await createDocumentAndUpload(values, params)
+        if(response){
+          handleChunking(response.kb_id,parentId,response.id)
+        }
         handleClose();
       } catch (err) {
         console.error('创建内容失败:', err);
@@ -59,7 +60,19 @@ const CreateContentModal = forwardRef<CreateSetModalRef, CreateSetMoealRefProps>
         setLoading(false);
       }
     };
-
+    const handleChunking = (kb_id: string, parent_id: string, file_id: string) => {
+      if (!kb_id) return;
+      const targetFileId = file_id
+      navigate(`/knowledge-base/${kb_id}/create-dataset`, {
+        state: {
+          source: 'local',
+          knowledgeBaseId: kb_id,
+          parentId: parent_id ?? kb_id,
+          startStep: 'parameterSettings',
+          fileId: targetFileId,
+        },
+      });
+    }
     useImperativeHandle(ref, () => ({
       handleOpen,
     }));