pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: remove loading from model and use DvaModel instead of redundant types such as kAModelType (#47)

Browse Source 
* feat: use DvaModel instead of redundant types such as kAModelType

* feat: set the type for registerServer

* feat: remove loading from model
This commit is contained in:
balibabu
2024-01-30 19:26:29 +08:00
committed by GitHub
parent 96a1a44cb6
commit 362ec6c364
29 changed files with 1911 additions and 1938 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
web/src/pages/add-knowledge/components/knowledge-chunk/model.ts
View File

@ -1,8 +1,7 @@
import kbService from '@/services/kbService';
import { Effect, Reducer } from 'umi';
import { DvaModel } from 'umi';
export interface chunkModelState {
loading: boolean;
export interface ChunkModelState {
data: any[];
total: number;
isShowCreateModal: boolean;
@ -10,25 +9,10 @@ export interface chunkModelState {
doc_id: string;
chunkInfo: any;
}
export interface chunkgModelType {
namespace: 'chunkModel';
state: chunkModelState;
effects: {
chunk_list: Effect;
get_chunk: Effect;
create_hunk: Effect;
switch_chunk: Effect;
rm_chunk: Effect;
};
reducers: {
updateState: Reducer<chunkModelState>;
};
// subscriptions: { setup: Subscription };
}
const Model: chunkgModelType = {
const model: DvaModel<ChunkModelState> = {
namespace: 'chunkModel',
state: {
loading: false,
data: [],
total: 0,
isShowCreateModal: false,
@ -36,6 +20,14 @@ const Model: chunkgModelType = {
doc_id: '',
chunkInfo: {},
},
reducers: {
updateState(state, { payload }) {
return {
...state,
...payload,
};
},
},
// subscriptions: {
// setup({ dispatch, history }) {
// history.listen(location => {
@ -44,7 +36,7 @@ const Model: chunkgModelType = {
// }
// },
effects: {
*chunk_list({ payload = {}, callback }, { call, put }) {
*chunk_list({ payload = {} }, { call, put }) {
const { data, response } = yield call(kbService.chunk_list, payload);
const { retcode, data: res, retmsg } = data;
@ -55,28 +47,23 @@ const Model: chunkgModelType = {
payload: {
data: res.chunks,
total: res.total,
loading: false,
},
});
callback && callback();
}
},
*switch_chunk({ payload = {}, callback }, { call, put }) {
*switch_chunk({ payload = {} }, { call, put }) {
const { data, response } = yield call(kbService.switch_chunk, payload);
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
callback && callback();
}
return retcode;
},
*rm_chunk({ payload = {}, callback }, { call, put }) {
*rm_chunk({ payload = {} }, { call, put }) {
console.log('shanchu');
const { data, response } = yield call(kbService.rm_chunk, payload);
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
callback && callback();
}
return retcode;
},
*get_chunk({ payload = {}, callback }, { call, put }) {
*get_chunk({ payload = {} }, { call, put }) {
const { data, response } = yield call(kbService.get_chunk, payload);
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
@ -86,28 +73,16 @@ const Model: chunkgModelType = {
chunkInfo: res,
},
});
callback && callback(res);
}
return data;
},
*create_hunk({ payload = {} }, { call, put }) {
yield put({
type: 'updateState',
payload: {
loading: true,
},
});
let service = kbService.create_chunk;
if (payload.chunk_id) {
service = kbService.set_chunk;
}
const { data, response } = yield call(service, payload);
const { retcode, data: res, retmsg } = data;
yield put({
type: 'updateState',
payload: {
loading: false,
},
});
if (retcode === 0) {
yield put({
type: 'updateState',
@ -118,13 +93,5 @@ const Model: chunkgModelType = {
}
},
},
reducers: {
updateState(state, { payload }) {
return {
...state,
...payload,
};
},
},
};
export default Model;
export default model;