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

* 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

View File

@ -19,101 +19,83 @@ const {
get_chunk,
switch_chunk,
rm_chunk,
retrieval_test } = api;
interface kbService {
createKb: () => void;
updateKb: () => void;
rmKb: () => void;
get_kb_detail: () => void;
getList: () => void;
get_document_list: () => void;
document_change_status: () => void;
document_rm: () => void;
document_create: () => void;
document_change_parser: () => void;
chunk_list: () => void;
create_chunk: () => void;
set_chunk: () => void;
get_chunk: () => void;
switch_chunk: () => void;
rm_chunk: () => void;
retrieval_test: () => void;
}
const kbService: kbService = registerServer(
{
// 知识库管理
createKb: {
url: create_kb,
method: 'post'
},
updateKb: {
url: update_kb,
method: 'post'
},
rmKb: {
url: rm_kb,
method: 'post'
},
get_kb_detail: {
url: get_kb_detail,
method: 'get'
},
getList: {
url: kb_list,
method: 'get'
},
// 文件管理
get_document_list: {
url: get_document_list,
method: 'get'
},
document_change_status: {
url: document_change_status,
method: 'post'
},
document_rm: {
url: document_rm,
method: 'post'
},
document_create: {
url: document_create,
method: 'post'
},
document_change_parser: {
url: document_change_parser,
method: 'post'
},
// chunk管理
chunk_list: {
url: chunk_list,
method: 'post'
},
create_chunk: {
url: create_chunk,
method: 'post'
},
set_chunk: {
url: set_chunk,
method: 'post'
},
get_chunk: {
url: get_chunk,
method: 'get'
},
switch_chunk: {
url: switch_chunk,
method: 'post'
},
rm_chunk: {
url: rm_chunk,
method: 'post'
},
retrieval_test: {
url: retrieval_test,
method: 'post'
},
retrieval_test,
} = api;
const methods = {
// 知识库管理
createKb: {
url: create_kb,
method: 'post',
},
request
);
updateKb: {
url: update_kb,
method: 'post',
},
rmKb: {
url: rm_kb,
method: 'post',
},
get_kb_detail: {
url: get_kb_detail,
method: 'get',
},
getList: {
url: kb_list,
method: 'get',
},
// 文件管理
get_document_list: {
url: get_document_list,
method: 'get',
},
document_change_status: {
url: document_change_status,
method: 'post',
},
document_rm: {
url: document_rm,
method: 'post',
},
document_create: {
url: document_create,
method: 'post',
},
document_change_parser: {
url: document_change_parser,
method: 'post',
},
// chunk管理
chunk_list: {
url: chunk_list,
method: 'post',
},
create_chunk: {
url: create_chunk,
method: 'post',
},
set_chunk: {
url: set_chunk,
method: 'post',
},
get_chunk: {
url: get_chunk,
method: 'get',
},
switch_chunk: {
url: switch_chunk,
method: 'post',
},
rm_chunk: {
url: rm_chunk,
method: 'post',
},
retrieval_test: {
url: retrieval_test,
method: 'post',
},
};
const kbService = registerServer<keyof typeof methods>(methods, request);
export default kbService;

View File

@ -3,55 +3,61 @@ import registerServer from '@/utils/registerServer';
import request from '@/utils/request';
const {
login, register, setting, user_info, tenant_info, factories_list, llm_list, my_llm, set_api_key, set_tenant_info } = api;
interface userServiceType {
login: (params: any) => void
}
const userService = registerServer(
{
login: {
url: login,
method: 'post',
login,
register,
setting,
user_info,
tenant_info,
factories_list,
llm_list,
my_llm,
set_api_key,
set_tenant_info,
} = api;
},
register: {
url: register,
method: 'post'
},
setting: {
url: setting,
method: 'post'
},
user_info: {
url: user_info,
method: 'get'
},
get_tenant_info: {
url: tenant_info,
method: 'get'
},
set_tenant_info: {
url: set_tenant_info,
method: 'post'
},
factories_list: {
url: factories_list,
method: 'get'
},
llm_list: {
url: llm_list,
method: 'get'
},
my_llm: {
url: my_llm,
method: 'get'
},
set_api_key: {
url: set_api_key,
method: 'post'
},
const methods = {
login: {
url: login,
method: 'post',
},
request
);
register: {
url: register,
method: 'post',
},
setting: {
url: setting,
method: 'post',
},
user_info: {
url: user_info,
method: 'get',
},
get_tenant_info: {
url: tenant_info,
method: 'get',
},
set_tenant_info: {
url: set_tenant_info,
method: 'post',
},
factories_list: {
url: factories_list,
method: 'get',
},
llm_list: {
url: llm_list,
method: 'get',
},
my_llm: {
url: my_llm,
method: 'get',
},
set_api_key: {
url: set_api_key,
method: 'post',
},
} as const;
const userService = registerServer<keyof typeof methods>(methods, request);
export default userService;