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: prevent the user from entering the knowledge base if he is not logged in (#45)

Browse Source
This commit is contained in:
balibabu
2024-01-29 19:28:39 +08:00
committed by GitHub
parent e1bc1d46e6
commit 04aba1bb65
20 changed files with 700 additions and 591 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,5 @@
import { Effect, Reducer, Subscription } from 'umi'
import { message } from 'antd';
import kbService from '@/services/kbService';
import { Effect, Reducer } from 'umi';
export interface chunkModelState {
loading: boolean;
@ -9,7 +8,7 @@ export interface chunkModelState {
isShowCreateModal: boolean;
chunk_id: string;
doc_id: string;
chunkInfo: any
chunkInfo: any;
}
export interface chunkgModelType {
namespace: 'chunkModel';
@ -24,7 +23,7 @@ export interface chunkgModelType {
reducers: {
updateState: Reducer<chunkModelState>;
};
subscriptions: { setup: Subscription };
// subscriptions: { setup: Subscription };
}
const Model: chunkgModelType = {
namespace: 'chunkModel',
@ -35,91 +34,86 @@ const Model: chunkgModelType = {
isShowCreateModal: false,
chunk_id: '',
doc_id: '',
chunkInfo: {}
},
subscriptions: {
setup({ dispatch, history }) {
history.listen(location => {
console.log(location)
});
}
chunkInfo: {},
},
// subscriptions: {
// setup({ dispatch, history }) {
// history.listen(location => {
// console.log(location)
// });
// }
// },
effects: {
* chunk_list({ payload = {}, callback }, { call, put }) {
*chunk_list({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.chunk_list, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
console.log(res)
console.log(res);
yield put({
type: 'updateState',
payload: {
data: res.chunks,
total: res.total,
loading: false
}
loading: false,
},
});
callback && callback()
callback && callback();
}
},
*switch_chunk({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.switch_chunk, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
callback && callback()
callback && callback();
}
},
*rm_chunk({ payload = {}, callback }, { call, put }) {
console.log('shanchu')
console.log('shanchu');
const { data, response } = yield call(kbService.rm_chunk, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
callback && callback()
callback && callback();
}
},
* get_chunk({ payload = {}, callback }, { call, put }) {
*get_chunk({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.get_chunk, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
yield put({
type: 'updateState',
payload: {
chunkInfo: res
}
chunkInfo: res,
},
});
callback && callback(res)
callback && callback(res);
}
},
*create_hunk({ payload = {} }, { call, put }) {
yield put({
type: 'updateState',
payload: {
loading: true
}
loading: true,
},
});
let service = kbService.create_chunk
let service = kbService.create_chunk;
if (payload.chunk_id) {
service = kbService.set_chunk
service = kbService.set_chunk;
}
const { data, response } = yield call(service, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
yield put({
type: 'updateState',
payload: {
loading: false
}
loading: false,
},
});
if (retcode === 0) {
yield put({
type: 'updateState',
payload: {
isShowCreateModal: false
}
isShowCreateModal: false,
},
});
}
},
@ -128,9 +122,9 @@ const Model: chunkgModelType = {
updateState(state, { payload }) {
return {
...state,
...payload
...payload,
};
}
}
},
},
};
export default Model;

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

@ -1,6 +1,6 @@
import { message } from 'antd';
import { Effect, Reducer, Subscription } from 'umi'
import kbService from '@/services/kbService';
import { message } from 'antd';
import { Effect, Reducer, Subscription } from 'umi';
export interface kFModelState {
isShowCEFwModal: boolean;
@ -8,7 +8,7 @@ export interface kFModelState {
isShowSegmentSetModal: boolean;
loading: boolean;
tenantIfo: any;
data: any[]
data: any[];
}
export interface kFModelType {
namespace: 'kFModel';
@ -36,60 +36,60 @@ const Model: kFModelType = {
isShowSegmentSetModal: false,
loading: false,
tenantIfo: {},
data: []
data: [],
},
subscriptions: {
setup({ dispatch, history }) {
history.listen(location => {
});
}
history.listen((location) => {});
},
},
effects: {
* createKf({ payload = {}, callback }, { call, put }) {
*createKf({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.createKb, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('创建成功！');
}
},
* updateKf({ payload = {}, callback }, { call, put }) {
*updateKf({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.updateKb, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('修改成功！');
}
},
*getKfDetail({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.get_kb_detail, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
// localStorage.setItem('userInfo',res.)
callback && callback(res)
callback && callback(res);
}
},
*getKfList({ payload = {} }, { call, put }) {
yield put({
type: 'updateState',
payload: {
loading: true
}
loading: true,
},
});
const { data, response } = yield call(kbService.get_document_list, payload);
const { retcode, data: res, retmsg } = data
const { data, response } = yield call(
kbService.get_document_list,
payload,
);
const { retcode, data: res, retmsg } = data;
yield put({
type: 'updateState',
payload: {
loading: false
}
loading: false,
},
});
if (retcode === 0) {
yield put({
type: 'updateState',
payload: {
data: res
}
data: res,
},
});
}
},
@ -97,58 +97,60 @@ const Model: kFModelType = {
yield put({
type: 'updateState',
payload: {
loading: true
}
loading: true,
},
});
const { data, response } = yield call(kbService.document_change_status, payload);
const { retcode, data: res, retmsg } = data
const { data, response } = yield call(
kbService.document_change_status,
payload,
);
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('修改成功！');
yield put({
type: 'updateState',
payload: {
loading: false
}
loading: false,
},
});
callback && callback()
callback && callback();
}
},
*document_rm({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.document_rm, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('删除成功！');
callback && callback()
callback && callback();
}
},
*document_create({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.document_create, payload);
const { retcode, data: res, retmsg } = data
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('创建成功！');
callback && callback()
callback && callback();
}
},
*document_change_parser({ payload = {}, callback }, { call, put }) {
const { data, response } = yield call(kbService.document_change_parser, payload);
const { retcode, data: res, retmsg } = data
const { data, response } = yield call(
kbService.document_change_parser,
payload,
);
const { retcode, data: res, retmsg } = data;
if (retcode === 0) {
message.success('修改成功！');
callback && callback()
callback && callback();
}
},
},
reducers: {
updateState(state, { payload }) {
return {
...state,
...payload
...payload,
};
}
}
},
},
};
export default Model;