72 lines
1.7 KiB
TypeScript
72 lines
1.7 KiB
TypeScript
import { get, post } from './request';
|
|
|
|
export interface RagStore {
|
|
id?: string;
|
|
storeCode: string;
|
|
storeName: string;
|
|
description?: string | null;
|
|
status?: string | null;
|
|
remark?: string | null;
|
|
createTime?: string | null;
|
|
updateTime?: string | null;
|
|
}
|
|
|
|
export interface RagStoreOverview {
|
|
totalStores: number;
|
|
totalDocuments: number;
|
|
totalChunks?: number | null;
|
|
retrievableStores: number;
|
|
}
|
|
|
|
export interface RagStoreDocumentOverview {
|
|
storeId: string;
|
|
storeName?: string | null;
|
|
documentCount: number;
|
|
enabledDocumentCount: number;
|
|
parsedDocumentCount: number;
|
|
indexedDocumentCount: number;
|
|
lastUploadTime?: string | null;
|
|
}
|
|
|
|
export interface RagStoreQueryRequest {
|
|
storeCode?: string;
|
|
storeName?: string;
|
|
status?: string;
|
|
}
|
|
|
|
export type RagStoreSaveRequest = RagStore;
|
|
|
|
export function listRagStores() {
|
|
return post<RagStore[]>('/rag/store/list');
|
|
}
|
|
|
|
export function queryRagStores(query?: RagStoreQueryRequest) {
|
|
return post<RagStore[], RagStoreQueryRequest | undefined>('/rag/store/query', query);
|
|
}
|
|
|
|
export function getRagStoreById(id: string) {
|
|
return get<RagStore>('/rag/store/detail', {
|
|
params: { id },
|
|
});
|
|
}
|
|
|
|
export function getRagStoreOverview() {
|
|
return get<RagStoreOverview>('/rag/store/overview');
|
|
}
|
|
|
|
export function getRagStoreDocumentOverview(storeId: string) {
|
|
return get<RagStoreDocumentOverview>('/rag/store/documentOverview', {
|
|
params: { storeId },
|
|
});
|
|
}
|
|
|
|
export function saveRagStore(data: RagStoreSaveRequest) {
|
|
return post<boolean, RagStoreSaveRequest>('/rag/store/save', data);
|
|
}
|
|
|
|
export function deleteRagStore(id: string) {
|
|
return post<boolean>('/rag/store/delete', undefined, {
|
|
params: { id },
|
|
});
|
|
}
|