瀏覽代碼

切片新增,修改,删除,查询。

S0025136190 6 月之前
父節點
當前提交
8fefdb5f29
共有 3 個文件被更改,包括 59 次插入29 次删除
  1. 8 2
      src/apis/index.ts
  2. 35 18
      src/pages/takai/knowledgeLib/slice/detail/index.tsx
  3. 16 9
      src/pages/takai/knowledgeLib/slice/index.tsx

+ 8 - 2
src/apis/index.ts

@@ -96,7 +96,7 @@ export type FetchSliceListApiParams = {
     pageSize: number,
 };
 
-export type ModifySliceApiParams = {
+export type addAndModifySliceApiParams = {
     knowledgeId: string,
     sliceId: string,
     sliceText: string,
@@ -149,10 +149,11 @@ export type ModifyTakaiDocumentSettingLibApi = (id: string, data: ModifyDocument
 export type FetchTakaiSliceListApi = (data: FetchSliceListApiParams) => Promise<any>;
 export type deleteTakaiSliceLibApi = (sliceId: string, knowledgeId: string, documentId: string) => Promise<any>;
 export type FetchTakaiSliceDetailLibApi = (sliceId: string, knowledgeId: string) => Promise<any>;
-export type modifyTakaiSliceInfoLibApi = (data: ModifySliceApiParams) => Promise<any>;
+export type modifyTakaiSliceInfoLibApi = (data: addAndModifySliceApiParams) => Promise<any>;
 export type FetchTakaiChatHistoryListApi = (data: FetchChatHistoryListApiParams) => Promise<any>;
 export type ExportTakaiChatHistoryApi = (id: string) => Promise<any>;
 export type FetchTakaiApplicationListApi = () => Promise<any>;
+export type addTakaiSliceLibApi = (data: addAndModifySliceApiParams) => Promise<any>;
 
 // 登录
 const loginApi: LoginApi = async (data) => {
@@ -400,6 +401,10 @@ const fetchTakaiApplicationListApi: FetchTakaiApplicationListApi = async () => {
     return api.get('/deepseek/api/application/list');
 };
 
+// 新增takai切片
+const addTakaiSliceApi: addTakaiSliceLibApi = async (data) => {
+    return api.post(`/deepseek/api/add/slice/`, data);
+};
 
 export const apis = {
     login: loginApi,
@@ -451,4 +456,5 @@ export const apis = {
     fetchTakaiChatHistoryList: fetchTakaiChatHistoryListApi,
     exportTakaiChatHistory: exportTakaiChatHistoryApi,
     fetchTakaiApplicationList: fetchTakaiApplicationListApi,
+    addTakaiSlice: addTakaiSliceApi,
 };

+ 35 - 18
src/pages/takai/knowledgeLib/slice/detail/index.tsx

@@ -47,7 +47,9 @@ const SliceDetail: React.FC = () => {
     };
 
     const init = async () => {
-        await appApi.fetchList();
+        if (params.sliceId) {
+            await appApi.fetchList();
+        }
     }
 
     React.useEffect(() => {
@@ -83,7 +85,7 @@ const SliceDetail: React.FC = () => {
                             router.navigate(-1);
                         }}>返回</Button>
                         <Button
-                            style={{ marginLeft: '10px' }} 
+                            style={{ marginLeft: '10px' }}
                             type='primary'
                             onClick={() => {
                                 form.validateFields().then(async (values) => {
@@ -95,33 +97,48 @@ const SliceDetail: React.FC = () => {
                                     const data = values;
                                     const info = {
                                         knowledgeId: params.knowledgeId,
-                                        sliceId: params.sliceId,
+                                        sliceId: params.sliceId === null ? '' : params.sliceId,
                                         sliceText: values.slice_text,
                                         documentId: params.documentId,
                                     };
-                                    // 编辑应用
-                                    const res = await apis.modifyTakaiSliceInfo(info);
+                                    console.log(params.sliceId, 'params.sliceId');
+                                    let res = null;
+                                    if (params.sliceId && params.sliceId !== 'null') {
+                                        console.log('11111111');
+                                        // 编辑应用
+                                        res = await apis.modifyTakaiSliceInfo(info);
+                                    } else {
+                                        console.log('2222222');
+                                        res = await apis.addTakaiSlice(info);
+                                    }
                                     if (res.code === 200 && res.data === 1) {
-                                        const pageInfo = {
-                                            knowledge_id: params.knowledgeId,
-                                            document_id: params.documentId,
-                                            text: '',
-                                            pageNum: 1,
-                                            pageSize: 10,
+                                        if(params.sliceId && params.sliceId !== 'null'){
+                                            message.success('修改成功');
+                                        }else{
+                                            message.success('新增成功');
                                         }
-                                        // 刷新列表
-                                        await router.navigate({ pathname: '/deepseek/knowledgeLib/slice/' + params.documentId + '/' + params.knowledgeId + '/' + params.embeddingId });
-                                        // await apis.fetchTakaiSliceList(pageInfo);
-                                        message.success('修改成功');
+                                    }else{
+                                        if(params.sliceId && params.sliceId !== 'null'){
+                                            message.error('修改失败');
+                                        }else{
+                                            message.error('新增失败');
+                                        }
+                                    }
+                                    const pageInfo = {
+                                        knowledge_id: params.knowledgeId,
+                                        document_id: params.documentId,
+                                        text: '',
+                                        pageNum: 1,
+                                        pageSize: 10,
                                     }
+                                    // 刷新列表
+                                    await router.navigate({ pathname: '/deepseek/knowledgeLib/slice/' + params.documentId + '/' + params.knowledgeId + '/' + params.embeddingId });
+
                                 }).catch((error) => {
                                     console.error(error);
                                 });
                             }}
                         >保存</Button>
-
-
-
                     </Form>
                 </div>
             </div>

+ 16 - 9
src/pages/takai/knowledgeLib/slice/index.tsx

@@ -2,7 +2,7 @@ import * as React from 'react';
 import { observer } from 'mobx-react';
 import { useParams } from 'react-router-dom';
 import { List, Button, Divider, Empty, Modal, Form, Input, message } from 'antd';
-import { SettingOutlined, DeleteOutlined, ArrowLeftOutlined } from '@ant-design/icons';
+import { SettingOutlined, DeleteOutlined, ArrowLeftOutlined, PlusOutlined } from '@ant-design/icons';
 import { apis } from '@/apis';
 import { PaginationConfig } from 'antd/es/pagination';
 import router from '@/router';
@@ -47,19 +47,20 @@ const SliceList: React.FC = () => {
                     pageSize: page.pageSize,
                     pageNum: page.pageNum
                 })
-                const list = res.data.rows.map((item: any) => {
+                console.log('fetchTakaiSliceList', res);
+                const list = res.rows.map((item: any) => {
                     return {
-                        slice_id: item.slice_id,
-                        document_id: item.document_id,
-                        slice_text: item.slice_text,
-                        slice_char_len: item.slice_char_len,
+                        slice_id: item.sliceId,
+                        document_id: item.documentId,
+                        slice_text: item.sliceText,
+                        slice_char_len: item.sliceText.length,
                     }
                 });
                 setList(list);
                 setPage({
                     pageNum: page.pageNum,
                     pageSize: page.pageSize,
-                    total: res.data.total,
+                    total: res.total,
                 });
             } catch (error) {
                 console.error(error);
@@ -148,13 +149,19 @@ const SliceList: React.FC = () => {
                         </FormItem>
                     </Form>
 
-                    <div>
+                    <div >
+                        <Button style={{ marginRight: 16 }} type='primary' icon={<PlusOutlined />} onClick={() => {
+                            router.navigate({
+                                pathname: '/deepseek/knowledgeLib/slice/detail/' + null + '/' + params.knowledgeId + '/' + params.documentId + '/'
+                                    + params.embeddingId
+                            });
+                        }}>新增切片</Button>
+
                         <Button type='primary' icon={<ArrowLeftOutlined />} onClick={() => {
                             router.navigate(-1);
                         }}>返回</Button>
                     </div>
                 </div>
-
             </div>
             {
                 list.length