Browse Source

feat: fetch parser list for SegmentSetModal (#81)

tags/v0.1.0
balibabu 1 year ago
parent
commit
332d4874d0
No account linked to committer's email address

+ 6
- 6
web/src/pages/add-knowledge/components/knowledge-file/model.ts View File

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 }) {
); );
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 },
}); });
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 },
isShowCEFwModal: false, isShowCEFwModal: false,
}, },
}); });
message.success('创建成功!');
message.success('Created!');
} }
return retcode; return retcode;
}, },
isShowSegmentSetModal: false, isShowSegmentSetModal: false,
}, },
}); });
message.success('修改成功!');
message.success('Modified!');
} }
return retcode; return retcode;
}, },

+ 1
- 3
web/src/pages/add-knowledge/components/knowledge-file/parsing-action-cell/index.tsx View File

}); });
}; };


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>
), ),

+ 14
- 16
web/src/pages/add-knowledge/components/knowledge-file/segmentSetModal.tsx View File

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;
}) => { }) => {
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]);
}, },
}); });
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}
checked={selectedTag === tag}
onChange={(checked) => handleChange(tag, checked)}
key={x.value}
checked={selectedTag === x.value}
onChange={(checked) => handleChange(x.value, checked)}
> >
{tag}
{x.label}
</CheckableTag> </CheckableTag>
); );
})} })}

+ 3
- 1
web/src/pages/chat/model.ts View File

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: {

Loading…
Cancel
Save