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

import { newService } from '/@/utils/request';

// 文档查询参数
export interface DocumentQueryParams {
	keyword?: string;
	datasetId?: 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;
}

// 文档信息
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 newService({
		url: '/rag-knowledge/document/listDocument',
		method: 'get',
		params,
	});
}

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

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

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

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

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

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

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