feat: confirm before deleting knowledge base and add ChunkToolBar (#56)

* feat: confirm before deleting knowledge base

* feat: add ChunkToolBar
This commit is contained in:
balibabu
2024-02-05 19:26:03 +08:00
committed by GitHub
parent 407b2523b6
commit eb381963b3
16 changed files with 311 additions and 80 deletions

View File

@@ -5,8 +5,13 @@ import {
} from '@/hooks/knowledgeHook';
import { Pagination } from '@/interfaces/common';
import { IKnowledgeFile } from '@/interfaces/database/knowledge';
import { getOneNamespaceEffectsLoading } from '@/utils/stroreUtil';
import { PlusOutlined, SearchOutlined } from '@ant-design/icons';
import { getOneNamespaceEffectsLoading } from '@/utils/storeUtil';
import {
FileOutlined,
FileTextOutlined,
PlusOutlined,
SearchOutlined,
} from '@ant-design/icons';
import type { MenuProps } from 'antd';
import {
Button,
@@ -21,14 +26,13 @@ import {
import type { ColumnsType } from 'antd/es/table';
import { PaginationProps } from 'antd/lib';
import React, { useEffect, useMemo, useState } from 'react';
import { useDispatch, useNavigate, useSelector } from 'umi';
import { Link, useDispatch, useNavigate, useSelector } from 'umi';
import CreateEPModal from './createEFileModal';
import styles from './index.less';
import ParsingActionCell from './parsing-action-cell';
import ParsingStatusCell from './parsing-status-cell';
import RenameModal from './rename-modal';
import SegmentSetModal from './segmentSetModal';
import UploadFile from './upload';
const KnowledgeFile = () => {
const dispatch = useDispatch();
@@ -155,24 +159,32 @@ const KnowledgeFile = () => {
key: '1',
label: (
<div>
<UploadFile kb_id={knowledgeBaseId} getKfList={getKfList} />
<Button type="link">
<Link to={`/knowledge/dataset/upload?id=${knowledgeBaseId}`}>
<Space>
<FileTextOutlined />
Local files
</Space>
</Link>
</Button>
</div>
),
},
{ type: 'divider' },
{
key: '2',
label: (
<div>
<Button type="link" onClick={showCEFModal}>
{' '}
<FileOutlined />
Create empty file
</Button>
</div>
),
// disabled: true,
},
];
}, [knowledgeBaseId]);
}, []);
const chunkItems: MenuProps['items'] = [
{
key: '1',
@@ -191,7 +203,7 @@ const KnowledgeFile = () => {
<div>
<Button type="link" onClick={onRmDocument}>
{' '}
Delete
</Button>
</div>
),

View File

@@ -168,8 +168,8 @@ const model: DvaModel<KFModelState> = {
return retcode;
},
*document_create({ payload = {} }, { call, put }) {
const { data, response } = yield call(kbService.document_create, payload);
const { retcode, data: res, retmsg } = data;
const { data } = yield call(kbService.document_create, payload);
const { retcode, data: res } = data;
if (retcode === 0) {
put({
type: 'kFModel/updateState',
@@ -181,6 +181,14 @@ const model: DvaModel<KFModelState> = {
}
return retcode;
},
*document_run({ payload = {} }, { call, put }) {
const { data } = yield call(kbService.document_run, payload);
const { retcode } = data;
if (retcode === 0) {
message.success('Run successfully ');
}
return retcode;
},
*document_change_parser({ payload = {} }, { call, put }) {
const { data, response } = yield call(
kbService.document_change_parser,

View File

@@ -55,7 +55,8 @@ export const ParsingStatusCell = ({ record }: IProps) => {
{isRunning ? (
<Space>
<Badge color={runningStatus.color} />
`${runningStatus.label}${record.progress * 100}%`
{runningStatus.label}
<span>{record.progress * 100}%</span>
</Space>
) : (
runningStatus.label