feat: fetch parser list for SegmentSetModal (#81)

This commit is contained in:
balibabu
2024-02-28 18:57:10 +08:00
committed by GitHub
parent 6e8781e0e6
commit 332d4874d0
4 changed files with 24 additions and 26 deletions

View File

@ -65,14 +65,14 @@ const model: DvaModel<KFModelState> = {
const { data } = yield call(kbService.createKb, payload); const { data } = yield call(kbService.createKb, payload);
const { retcode } = data; const { retcode } = data;
if (retcode === 0) { if (retcode === 0) {
message.success('创建成功!'); message.success('Created!');
} }
}, },
*updateKf({ payload = {} }, { call }) { *updateKf({ payload = {} }, { call }) {
const { data } = yield call(kbService.updateKb, payload); const { data } = yield call(kbService.updateKb, payload);
const { retcode } = data; const { retcode } = data;
if (retcode === 0) { if (retcode === 0) {
message.success('修改成功!'); message.success('Modified!');
} }
}, },
*getKfDetail({ payload = {} }, { call }) { *getKfDetail({ payload = {} }, { call }) {
@ -120,7 +120,7 @@ const model: DvaModel<KFModelState> = {
); );
const { retcode } = data; const { retcode } = data;
if (retcode === 0) { if (retcode === 0) {
message.success('修改成功!'); message.success('Modified!');
put({ put({
type: 'getKfList', type: 'getKfList',
payload: { kb_id: payload.kb_id }, payload: { kb_id: payload.kb_id },
@ -133,7 +133,7 @@ const model: DvaModel<KFModelState> = {
}); });
const { retcode } = data; const { retcode } = data;
if (retcode === 0) { if (retcode === 0) {
message.success('删除成功!'); message.success('Deleted!');
yield put({ yield put({
type: 'getKfList', type: 'getKfList',
payload: { kb_id: payload.kb_id }, payload: { kb_id: payload.kb_id },
@ -171,7 +171,7 @@ const model: DvaModel<KFModelState> = {
isShowCEFwModal: false, isShowCEFwModal: false,
}, },
}); });
message.success('创建成功!'); message.success('Created!');
} }
return retcode; return retcode;
}, },
@ -202,7 +202,7 @@ const model: DvaModel<KFModelState> = {
isShowSegmentSetModal: false, isShowSegmentSetModal: false,
}, },
}); });
message.success('修改成功!'); message.success('Modified!');
} }
return retcode; return retcode;
}, },

View File

@ -56,15 +56,13 @@ const ParsingActionCell = ({
}); });
}; };
const onRename = () => {};
const chunkItems: MenuProps['items'] = [ const chunkItems: MenuProps['items'] = [
{ {
key: '1', key: '1',
label: ( label: (
<div> <div>
<Button type="link" onClick={showSegmentSetModal}> <Button type="link" onClick={showSegmentSetModal}>
Parser type
</Button> </Button>
</div> </div>
), ),

View File

@ -1,6 +1,6 @@
import { useFetchParserList, useSelectParserList } from '@/hooks/knowledgeHook';
import { Modal, Space, Tag } from 'antd'; import { Modal, Space, Tag } from 'antd';
import React, { useEffect, useState } from 'react'; import React, { useEffect, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { useDispatch, useSelector } from 'umi'; import { useDispatch, useSelector } from 'umi';
import styles from './index.less'; import styles from './index.less';
const { CheckableTag } = Tag; const { CheckableTag } = Tag;
@ -16,17 +16,13 @@ const SegmentSetModal: React.FC<kFProps> = ({
}) => { }) => {
const dispatch = useDispatch(); const dispatch = useDispatch();
const kFModel = useSelector((state: any) => state.kFModel); const kFModel = useSelector((state: any) => state.kFModel);
const settingModel = useSelector((state: any) => state.settingModel);
const [selectedTag, setSelectedTag] = useState(''); const [selectedTag, setSelectedTag] = useState('');
const parser_ids = settingModel?.tenantIfo?.parser_ids ?? '';
const { isShowSegmentSetModal } = kFModel; const { isShowSegmentSetModal } = kFModel;
const { t } = useTranslation(); const parserList = useSelectParserList();
useFetchParserList();
useEffect(() => { useEffect(() => {
dispatch({
type: 'settingModel/getTenantInfo',
payload: {},
});
setSelectedTag(parser_id); setSelectedTag(parser_id);
}, [parser_id]); }, [parser_id]);
@ -48,32 +44,34 @@ const SegmentSetModal: React.FC<kFProps> = ({
}, },
}); });
retcode === 0 && getKfList && getKfList(); if (retcode === 0 && getKfList) {
getKfList();
handleCancel();
}
}; };
const handleChange = (tag: string, checked: boolean) => { const handleChange = (tag: string, checked: boolean) => {
const nextSelectedTag = checked ? tag : selectedTag; const nextSelectedTag = checked ? tag : selectedTag;
console.log('You are interested in: ', nextSelectedTag);
setSelectedTag(nextSelectedTag); setSelectedTag(nextSelectedTag);
}; };
return ( return (
<Modal <Modal
title="Basic Modal" title="Parser Type"
open={isShowSegmentSetModal} open={isShowSegmentSetModal}
onOk={handleOk} onOk={handleOk}
onCancel={handleCancel} onCancel={handleCancel}
> >
<Space size={[0, 8]} wrap> <Space size={[0, 8]} wrap>
<div className={styles.tags}> <div className={styles.tags}>
{parser_ids.split(',').map((tag: string) => { {parserList.map((x) => {
return ( return (
<CheckableTag <CheckableTag
key={tag} key={x.value}
checked={selectedTag === tag} checked={selectedTag === x.value}
onChange={(checked) => handleChange(tag, checked)} onChange={(checked) => handleChange(x.value, checked)}
> >
{tag} {x.label}
</CheckableTag> </CheckableTag>
); );
})} })}

View File

@ -110,10 +110,12 @@ const model: DvaModel<ChatModelState> = {
return data.retcode; return data.retcode;
}, },
*getConversation({ payload }, { call, put }) { *getConversation({ payload }, { call, put }) {
const needToBeSaved =
payload.needToBeSaved === undefined ? true : payload.needToBeSaved;
const { data } = yield call(chatService.getConversation, { const { data } = yield call(chatService.getConversation, {
conversation_id: payload.conversation_id, conversation_id: payload.conversation_id,
}); });
if (data.retcode === 0 && payload.needToBeSaved) { if (data.retcode === 0 && needToBeSaved) {
yield put({ yield put({
type: 'kFModel/fetch_document_thumbnails', type: 'kFModel/fetch_document_thumbnails',
payload: { payload: {