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

@ -28,7 +28,7 @@ import {
UploadProps,
} from 'antd';
import classNames from 'classnames';
import { ReactElement, useEffect, useState } from 'react';
import { ReactElement, useEffect, useRef, useState } from 'react';
import { Nullable } from 'typings';
import { Link, useDispatch, useNavigate, useSelector } from 'umi';
@ -72,11 +72,15 @@ const UploaderItem = ({
const content = (
<Radio.Group onChange={onChange} value={value}>
<Space direction="vertical">
{parserArray.map((x) => (
<Radio value={x} key={x}>
{x}
</Radio>
))}
{parserArray.map(
(
x, // value is lowercase, key is uppercase
) => (
<Radio value={x.toLowerCase()} key={x}>
{x}
</Radio>
),
)}
</Space>
</Radio.Group>
);
@ -147,6 +151,7 @@ const KnowledgeUploadFile = () => {
(state: any) => state.settingModel.tenantIfo,
);
const navigate = useNavigate();
const fileListRef = useRef<UploadFile[]>([]);
const parserArray = tenantIfo?.parser_ids.split(',') ?? [];
@ -168,6 +173,7 @@ const KnowledgeUploadFile = () => {
name: 'file',
multiple: true,
itemRender(originNode, file, fileList, actions) {
fileListRef.current = fileList;
return (
<UploaderItem
isUpload={isUpload}
@ -185,8 +191,17 @@ const KnowledgeUploadFile = () => {
},
};
const runSelectedDocument = () => {
const ids = fileListRef.current.map((x) => x.response.id);
dispatch({
type: 'kFModel/document_run',
payload: { doc_ids: ids, run: 1 },
});
};
const handleNextClick = () => {
if (!isUpload) {
runSelectedDocument();
navigate(`/knowledge/${KnowledgeRouteKey.Dataset}?id=${knowledgeBaseId}`);
} else {
setIsUpload(false);