balibabu commited on
Commit
06526fb
·
1 Parent(s): 440a37a

feat: fetch parser list for SegmentSetModal (#81)

Browse files
web/src/pages/add-knowledge/components/knowledge-file/model.ts CHANGED
@@ -65,14 +65,14 @@ const model: DvaModel<KFModelState> = {
65
  const { data } = yield call(kbService.createKb, payload);
66
  const { retcode } = data;
67
  if (retcode === 0) {
68
- message.success('创建成功!');
69
  }
70
  },
71
  *updateKf({ payload = {} }, { call }) {
72
  const { data } = yield call(kbService.updateKb, payload);
73
  const { retcode } = data;
74
  if (retcode === 0) {
75
- message.success('修改成功!');
76
  }
77
  },
78
  *getKfDetail({ payload = {} }, { call }) {
@@ -120,7 +120,7 @@ const model: DvaModel<KFModelState> = {
120
  );
121
  const { retcode } = data;
122
  if (retcode === 0) {
123
- message.success('修改成功!');
124
  put({
125
  type: 'getKfList',
126
  payload: { kb_id: payload.kb_id },
@@ -133,7 +133,7 @@ const model: DvaModel<KFModelState> = {
133
  });
134
  const { retcode } = data;
135
  if (retcode === 0) {
136
- message.success('删除成功!');
137
  yield put({
138
  type: 'getKfList',
139
  payload: { kb_id: payload.kb_id },
@@ -171,7 +171,7 @@ const model: DvaModel<KFModelState> = {
171
  isShowCEFwModal: false,
172
  },
173
  });
174
- message.success('创建成功!');
175
  }
176
  return retcode;
177
  },
@@ -202,7 +202,7 @@ const model: DvaModel<KFModelState> = {
202
  isShowSegmentSetModal: false,
203
  },
204
  });
205
- message.success('修改成功!');
206
  }
207
  return retcode;
208
  },
 
65
  const { data } = yield call(kbService.createKb, payload);
66
  const { retcode } = data;
67
  if (retcode === 0) {
68
+ message.success('Created!');
69
  }
70
  },
71
  *updateKf({ payload = {} }, { call }) {
72
  const { data } = yield call(kbService.updateKb, payload);
73
  const { retcode } = data;
74
  if (retcode === 0) {
75
+ message.success('Modified!');
76
  }
77
  },
78
  *getKfDetail({ payload = {} }, { call }) {
 
120
  );
121
  const { retcode } = data;
122
  if (retcode === 0) {
123
+ message.success('Modified!');
124
  put({
125
  type: 'getKfList',
126
  payload: { kb_id: payload.kb_id },
 
133
  });
134
  const { retcode } = data;
135
  if (retcode === 0) {
136
+ message.success('Deleted!');
137
  yield put({
138
  type: 'getKfList',
139
  payload: { kb_id: payload.kb_id },
 
171
  isShowCEFwModal: false,
172
  },
173
  });
174
+ message.success('Created!');
175
  }
176
  return retcode;
177
  },
 
202
  isShowSegmentSetModal: false,
203
  },
204
  });
205
+ message.success('Modified!');
206
  }
207
  return retcode;
208
  },
web/src/pages/add-knowledge/components/knowledge-file/parsing-action-cell/index.tsx CHANGED
@@ -56,15 +56,13 @@ const ParsingActionCell = ({
56
  });
57
  };
58
 
59
- const onRename = () => {};
60
-
61
  const chunkItems: MenuProps['items'] = [
62
  {
63
  key: '1',
64
  label: (
65
  <div>
66
  <Button type="link" onClick={showSegmentSetModal}>
67
- 分段设置
68
  </Button>
69
  </div>
70
  ),
 
56
  });
57
  };
58
 
 
 
59
  const chunkItems: MenuProps['items'] = [
60
  {
61
  key: '1',
62
  label: (
63
  <div>
64
  <Button type="link" onClick={showSegmentSetModal}>
65
+ Parser type
66
  </Button>
67
  </div>
68
  ),
web/src/pages/add-knowledge/components/knowledge-file/segmentSetModal.tsx CHANGED
@@ -1,6 +1,6 @@
 
1
  import { Modal, Space, Tag } from 'antd';
2
  import React, { useEffect, useState } from 'react';
3
- import { useTranslation } from 'react-i18next';
4
  import { useDispatch, useSelector } from 'umi';
5
  import styles from './index.less';
6
  const { CheckableTag } = Tag;
@@ -16,17 +16,13 @@ const SegmentSetModal: React.FC<kFProps> = ({
16
  }) => {
17
  const dispatch = useDispatch();
18
  const kFModel = useSelector((state: any) => state.kFModel);
19
- const settingModel = useSelector((state: any) => state.settingModel);
20
  const [selectedTag, setSelectedTag] = useState('');
21
- const parser_ids = settingModel?.tenantIfo?.parser_ids ?? '';
22
  const { isShowSegmentSetModal } = kFModel;
23
- const { t } = useTranslation();
 
 
24
 
25
  useEffect(() => {
26
- dispatch({
27
- type: 'settingModel/getTenantInfo',
28
- payload: {},
29
- });
30
  setSelectedTag(parser_id);
31
  }, [parser_id]);
32
 
@@ -48,32 +44,34 @@ const SegmentSetModal: React.FC<kFProps> = ({
48
  },
49
  });
50
 
51
- retcode === 0 && getKfList && getKfList();
 
 
 
52
  };
53
 
54
  const handleChange = (tag: string, checked: boolean) => {
55
  const nextSelectedTag = checked ? tag : selectedTag;
56
- console.log('You are interested in: ', nextSelectedTag);
57
  setSelectedTag(nextSelectedTag);
58
  };
59
 
60
  return (
61
  <Modal
62
- title="Basic Modal"
63
  open={isShowSegmentSetModal}
64
  onOk={handleOk}
65
  onCancel={handleCancel}
66
  >
67
  <Space size={[0, 8]} wrap>
68
  <div className={styles.tags}>
69
- {parser_ids.split(',').map((tag: string) => {
70
  return (
71
  <CheckableTag
72
- key={tag}
73
- checked={selectedTag === tag}
74
- onChange={(checked) => handleChange(tag, checked)}
75
  >
76
- {tag}
77
  </CheckableTag>
78
  );
79
  })}
 
1
+ import { useFetchParserList, useSelectParserList } from '@/hooks/knowledgeHook';
2
  import { Modal, Space, Tag } from 'antd';
3
  import React, { useEffect, useState } from 'react';
 
4
  import { useDispatch, useSelector } from 'umi';
5
  import styles from './index.less';
6
  const { CheckableTag } = Tag;
 
16
  }) => {
17
  const dispatch = useDispatch();
18
  const kFModel = useSelector((state: any) => state.kFModel);
 
19
  const [selectedTag, setSelectedTag] = useState('');
 
20
  const { isShowSegmentSetModal } = kFModel;
21
+ const parserList = useSelectParserList();
22
+
23
+ useFetchParserList();
24
 
25
  useEffect(() => {
 
 
 
 
26
  setSelectedTag(parser_id);
27
  }, [parser_id]);
28
 
 
44
  },
45
  });
46
 
47
+ if (retcode === 0 && getKfList) {
48
+ getKfList();
49
+ handleCancel();
50
+ }
51
  };
52
 
53
  const handleChange = (tag: string, checked: boolean) => {
54
  const nextSelectedTag = checked ? tag : selectedTag;
 
55
  setSelectedTag(nextSelectedTag);
56
  };
57
 
58
  return (
59
  <Modal
60
+ title="Parser Type"
61
  open={isShowSegmentSetModal}
62
  onOk={handleOk}
63
  onCancel={handleCancel}
64
  >
65
  <Space size={[0, 8]} wrap>
66
  <div className={styles.tags}>
67
+ {parserList.map((x) => {
68
  return (
69
  <CheckableTag
70
+ key={x.value}
71
+ checked={selectedTag === x.value}
72
+ onChange={(checked) => handleChange(x.value, checked)}
73
  >
74
+ {x.label}
75
  </CheckableTag>
76
  );
77
  })}
web/src/pages/chat/model.ts CHANGED
@@ -110,10 +110,12 @@ const model: DvaModel<ChatModelState> = {
110
  return data.retcode;
111
  },
112
  *getConversation({ payload }, { call, put }) {
 
 
113
  const { data } = yield call(chatService.getConversation, {
114
  conversation_id: payload.conversation_id,
115
  });
116
- if (data.retcode === 0 && payload.needToBeSaved) {
117
  yield put({
118
  type: 'kFModel/fetch_document_thumbnails',
119
  payload: {
 
110
  return data.retcode;
111
  },
112
  *getConversation({ payload }, { call, put }) {
113
+ const needToBeSaved =
114
+ payload.needToBeSaved === undefined ? true : payload.needToBeSaved;
115
  const { data } = yield call(chatService.getConversation, {
116
  conversation_id: payload.conversation_id,
117
  });
118
+ if (data.retcode === 0 && needToBeSaved) {
119
  yield put({
120
  type: 'kFModel/fetch_document_thumbnails',
121
  payload: {