admin-ui/src/api/knowledge/document/index.ts

import request from '/@/utils/request';

// 文档查询参数
export interface DocumentQueryParams {
	keyword?: string;
	title?: string;
	datasetId?: string;
	knowledgeId?: string;
	fileType?: string;
	status?: string;
	pageNum: number;
	pageSize: number;
}

// 创建文档参数
export interface CreateDocumentParams {
	datasetId: string;
	filePath: string;
	fileSize: number;
	format: string;
	title: string;
}

// 更新文档参数
export interface UpdateDocumentParams {
	id: string;
	datasetId?: string;
	filePath?: string;
	fileSize?: number;
	format?: string;
	title?: string;
	status?: number;
}

// 文档分段查询参数
export interface DocumentChunkQueryParams {
	documentId: string;
	pageNum: number;
	pageSize: number;
}

// 文档信息
export interface DocumentInfo {
	id?: string;
	name: string;
	datasetId: string;
	fileType: string;
	fileSize?: number;
	filePath?: string;
	chunkCount?: number;
	parseStatus?: string;
	enabled?: boolean;
	createdAt?: string;
	updatedAt?: string;
}

// 获取文档列表
export function listDocuments(params: DocumentQueryParams) {
	return request({
		url: '/rag/document/list',
		method: 'get',
		params,
	});
}

// 获取文档详情
export function getDocument(id: string) {
	return request({
		url: '/rag/document/get',
		method: 'get',
		params: { id },
	});
}

// 创建文档
export function createDocument(data: CreateDocumentParams) {
	return request({
		url: '/rag/document/create',
		method: 'post',
		data,
	});
}

// 更新文档
export function updateDocument(data: UpdateDocumentParams) {
	return request({
		url: '/rag/document/update',
		method: 'put',
		data,
	});
}

// 公共文件上传（OSS），返回文件路径
export function uploadFile(file: File) {
	const formData = new FormData();
	formData.append('file', file);
	return request({
		url: '/oss/file/uploadFile',
		method: 'post',
		data: formData,
		headers: { 'Content-Type': 'multipart/form-data' },
	});
}

// 上传文档
export function uploadDocument(data: FormData) {
	return request({
		url: '/rag/document/create',
		method: 'post',
		data,
		headers: {
			'Content-Type': 'multipart/form-data',
		},
	});
}

// 预览文档
export function previewDocument(id: string) {
	return request({
		url: '/rag/document/preview',
		method: 'get',
		params: { id },
	});
}

// 删除文档
export function deleteDocument(id: string) {
	return request({
		url: '/rag/document/delete',
		method: 'delete',
		params: { id },
	});
}

// 批量删除文档
export function batchDeleteDocuments(ids: string[]) {
	return request({
		url: '/rag/document/batchDeleteDocument',
		method: 'delete',
		data: { ids },
	});
}

// 重新处理文档
export function reprocessDocument(id: string) {
	return request({
		url: '/rag/document/reprocessDocument',
		method: 'post',
		data: { id },
	});
}

// 获取文档分段列表
export function listDocumentChunks(params: DocumentChunkQueryParams) {
	return request({
		url: '/rag/document/listDocumentChunk',
		method: 'get',
		params,
	});
}

// 更新文档分段
export function updateDocumentChunk(data: { id: string; content: string }) {
	return request({
		url: '/rag/document/updateDocumentChunk',
		method: 'put',
		data,
	});
}

// 删除文档分段
export function deleteDocumentChunk(id: string) {
	return request({
		url: '/rag/document/deleteDocumentChunk',
		method: 'delete',
		params: { id },
	});
}

// 获取文件向量化处理进度
export function getDocumentProcess(id: string) {
	return request({
		url: '/rag/document/getProcess',
		method: 'get',
		params: { id },
	});
}

// 生成向量
export function generateVector(id: string, datasetId: string) {
	return request({
		url: '/rag/document/vectorization',
		method: 'post',
		data: { id, datasetId },
	});
}