-
+
-
- {data.map((item: any) => {
- return (
-
- {
- handleEditchunk(item.chunk_id);
- }}
- >
- 
-
-
-
- {item.content_ltks}
-
-
- {
- e.stopPropagation();
- e.nativeEvent.stopImmediatePropagation();
- switchChunk(item.chunk_id, checked);
- }}
- />
-
-
-
-
-
- {item.doc_num}文档
-
-
-
- {item.chunk_num}个
-
-
-
- {item.token_num}千字符
-
-
- {
- e.stopPropagation();
- e.nativeEvent.stopImmediatePropagation();
- console.log(confirm);
- confirm(item.chunk_id);
- }}
- okText="Yes"
- cancelText="No"
- >
- {
- e.stopPropagation();
- e.nativeEvent.stopImmediatePropagation();
- }}
- />
-
-
-
-
+
+ {data.map((item) => (
+ x === item.chunk_id)}
+ handleCheckboxClick={handleSingleCheckboxClick}
+ >
+ ))}
+
@@ -259,10 +187,10 @@ const Chunk = () => {
showLessItems
showQuickJumper
showSizeChanger
- onChange={onShowSizeChange}
- defaultPageSize={30}
- pageSizeOptions={[30, 60, 90]}
- defaultCurrent={pagination.page}
+ onChange={onPaginationChange}
+ defaultPageSize={10}
+ pageSizeOptions={[10, 30, 60, 90]}
+ defaultCurrent={pagination.current}
total={total}
/>
diff --git a/web/src/pages/add-knowledge/components/knowledge-chunk/model.ts b/web/src/pages/add-knowledge/components/knowledge-chunk/model.ts
index 2ebba517..2df2af88 100644
--- a/web/src/pages/add-knowledge/components/knowledge-chunk/model.ts
+++ b/web/src/pages/add-knowledge/components/knowledge-chunk/model.ts
@@ -1,13 +1,19 @@
+import { BaseState } from '@/interfaces/common';
+import { IKnowledgeFile } from '@/interfaces/database/knowledge';
import kbService from '@/services/kbService';
+import { message } from 'antd';
+// import { delay } from '@/utils/storeUtil';
import { DvaModel } from 'umi';
-export interface ChunkModelState {
+export interface ChunkModelState extends BaseState {
data: any[];
total: number;
isShowCreateModal: boolean;
chunk_id: string;
doc_id: string;
chunkInfo: any;
+ documentInfo: Partial
;
+ available?: number;
}
const model: DvaModel = {
@@ -19,6 +25,13 @@ const model: DvaModel = {
chunk_id: '',
doc_id: '',
chunkInfo: {},
+ documentInfo: {},
+ pagination: {
+ current: 1,
+ pageSize: 10,
+ },
+ searchString: '',
+ available: undefined, // set to undefined to select all
},
reducers: {
updateState(state, { payload }) {
@@ -27,33 +40,56 @@ const model: DvaModel = {
...payload,
};
},
+ setAvailable(state, { payload }) {
+ return { ...state, available: payload };
+ },
+ setSearchString(state, { payload }) {
+ return { ...state, searchString: payload };
+ },
+ setPagination(state, { payload }) {
+ return { ...state, pagination: { ...state.pagination, ...payload } };
+ },
+ resetFilter(state, { payload }) {
+ return {
+ ...state,
+ pagination: {
+ current: 1,
+ pageSize: 10,
+ },
+ searchString: '',
+ available: undefined,
+ };
+ },
},
- // subscriptions: {
- // setup({ dispatch, history }) {
- // history.listen(location => {
- // console.log(location)
- // });
- // }
- // },
effects: {
- *chunk_list({ payload = {} }, { call, put }) {
- const { data, response } = yield call(kbService.chunk_list, payload);
-
- const { retcode, data: res, retmsg } = data;
+ *chunk_list({ payload = {} }, { call, put, select }) {
+ const { available, searchString, pagination }: ChunkModelState =
+ yield select((state: any) => state.chunkModel);
+ const { data } = yield call(kbService.chunk_list, {
+ ...payload,
+ available_int: available,
+ keywords: searchString,
+ page: pagination.current,
+ size: pagination.pageSize,
+ });
+ const { retcode, data: res } = data;
if (retcode === 0) {
- console.log(res);
yield put({
type: 'updateState',
payload: {
data: res.chunks,
total: res.total,
+ documentInfo: res.doc,
},
});
}
},
*switch_chunk({ payload = {} }, { call, put }) {
- const { data, response } = yield call(kbService.switch_chunk, payload);
- const { retcode, data: res, retmsg } = data;
+ const { data } = yield call(kbService.switch_chunk, payload);
+ const { retcode } = data;
+ if (retcode === 0) {
+ message.success('Modified successfully ！');
+ }
return retcode;
},
*rm_chunk({ payload = {} }, { call, put }) {
diff --git a/web/src/pages/add-knowledge/components/knowledge-file/model.ts b/web/src/pages/add-knowledge/components/knowledge-file/model.ts
index 875665b7..1b673355 100644
--- a/web/src/pages/add-knowledge/components/knowledge-file/model.ts
+++ b/web/src/pages/add-knowledge/components/knowledge-file/model.ts
@@ -16,7 +16,6 @@ export interface KFModelState extends BaseState {
data: IKnowledgeFile[];
total: number;
currentRecord: Nullable;
- searchString: string;
}
const model: DvaModel = {
diff --git a/web/src/pages/knowledge/knowledge-card/index.tsx b/web/src/pages/knowledge/knowledge-card/index.tsx
index 07d1616f..199b7638 100644
--- a/web/src/pages/knowledge/knowledge-card/index.tsx
+++ b/web/src/pages/knowledge/knowledge-card/index.tsx
@@ -9,7 +9,6 @@ import {
UserOutlined,
} from '@ant-design/icons';
import { Avatar, Card, Dropdown, MenuProps, Space } from 'antd';
-import { MouseEvent } from 'react';
import { useDispatch, useNavigate } from 'umi';
import showDeleteConfirm from '@/components/deleting-confirm';
@@ -23,6 +22,15 @@ const KnowledgeCard = ({ item }: IProps) => {
const navigate = useNavigate();
const dispatch = useDispatch();
+ const removeKnowledge = () => {
+ return dispatch({
+ type: 'knowledgeModel/rmKb',
+ payload: {
+ kb_id: item.id,
+ },
+ });
+ };
+
const handleDelete = () => {
showDeleteConfirm({ onOk: removeKnowledge });
};
@@ -47,16 +55,7 @@ const KnowledgeCard = ({ item }: IProps) => {
}
};
- const removeKnowledge = () => {
- return dispatch({
- type: 'knowledgeModel/rmKb',
- payload: {
- kb_id: item.id,
- },
- });
- };
-
- const handleCardClick = (e: MouseEvent) => {
+ const handleCardClick = () => {
navigate(`/knowledge/${KnowledgeRouteKey.Dataset}?id=${item.id}`);
};
diff --git a/web/src/utils/storeUtil.ts b/web/src/utils/storeUtil.ts
index 4437d25f..dfa25ab9 100644
--- a/web/src/utils/storeUtil.ts
+++ b/web/src/utils/storeUtil.ts
@@ -7,3 +7,8 @@ export const getOneNamespaceEffectsLoading = (
(effectName) => effects[`${namespace}/${effectName}`],
);
};
+
+export const delay = (ms: number) =>
+ new Promise((resolve) => {
+ setTimeout(resolve, ms);
+ });
+ );
+};
+
+const ChunkCard = ({ item, checked, handleCheckboxClick }: IProps) => {
+ const dispatch = useDispatch();
+
+ const available = Number(item.available_int);
+ const [enabled, setEnabled] = useState(available === 1);
+
+ const switchChunk = () => {
+ dispatch({
+ type: 'chunkModel/switch_chunk',
+ payload: {
+ chunk_ids: [item.chunk_id],
+ available_int: available === 0 ? 1 : 0,
+ doc_id: item.doc_id,
+ },
+ });
+ };
+
+ const onChange = (checked: boolean) => {
+ setEnabled(checked);
+ switchChunk();
+ };
+
+ const handleCheck: CheckboxProps['onChange'] = (e) => {
+ handleCheckboxClick(item.chunk_id, e.target.checked);
+ };
+
+ return (
+