99 lines
2.1 KiB
TypeScript
99 lines
2.1 KiB
TypeScript
import request from '/@/utils/request';
|
|
|
|
// 数据集查询参数
|
|
export interface knowledgeQueryParams {
|
|
keyword?: string;
|
|
pageNum: number;
|
|
pageSize: number;
|
|
datasetType?: number; // 1=文本，2=视频
|
|
}
|
|
|
|
// 创建知识库参数
|
|
export interface CreateknowledgeParams {
|
|
name: string; // 必传
|
|
datasetType: number; // 1=文本，2=视频
|
|
description?: string; // 可选
|
|
embeddingModel?: string; // 向量模型
|
|
status?: string; // 状态
|
|
}
|
|
|
|
// 更新知识库参数
|
|
export interface UpdateknowledgeParams {
|
|
id: string; // 必传
|
|
name?: string; // 可选
|
|
datasetType?: number; // 1=文本，2=视频
|
|
description?: string; // 可选
|
|
embeddingModel?: string; // 向量模型
|
|
status?: string; // 状态
|
|
}
|
|
|
|
// 数据集信息
|
|
export interface knowledgeInfo {
|
|
id?: string;
|
|
name: string;
|
|
datasetType?: number; // 1=文本，2=视频
|
|
description?: string;
|
|
fileCount?: number;
|
|
totalSize?: number;
|
|
createdAt?: string;
|
|
updatedAt?: string;
|
|
embeddingModel?: string; // 向量模型
|
|
documentCount?: number; // 文档数量
|
|
charCount?: number; // 字符数量
|
|
status?: string; // 状态
|
|
}
|
|
|
|
// 获取知识库列表
|
|
export function listknowledges(params: knowledgeQueryParams) {
|
|
return request({
|
|
url: '/rag/dataset/list',
|
|
method: 'get',
|
|
params,
|
|
});
|
|
}
|
|
|
|
// 创建知识库
|
|
export function createknowledge(data: CreateknowledgeParams) {
|
|
return request({
|
|
url: '/rag/dataset/create',
|
|
method: 'post',
|
|
data,
|
|
});
|
|
}
|
|
|
|
// 更新知识库
|
|
export function updateknowledge(data: UpdateknowledgeParams) {
|
|
return request({
|
|
url: '/rag/dataset/update',
|
|
method: 'put',
|
|
data,
|
|
});
|
|
}
|
|
|
|
// 删除知识库
|
|
export function deleteknowledge(id: string) {
|
|
return request({
|
|
url: '/rag/dataset/delete',
|
|
method: 'delete',
|
|
params: { id },
|
|
});
|
|
}
|
|
|
|
// 获取知识库详情
|
|
export function getknowledge(id: string) {
|
|
return request({
|
|
url: '/rag/dataset/detail',
|
|
method: 'get',
|
|
params: { id },
|
|
});
|
|
}
|
|
|
|
// 更新知识库状态
|
|
export function updateknowledgeStatus(data: { id: string; status: string }) {
|
|
return request({
|
|
url: '/rag/dataset/updateStatus',
|
|
method: 'put',
|
|
data,
|
|
});
|
|
}
|