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

// 数据集查询参数
export interface DatasetQueryParams {
	keyword?: string;
	status?: string;
	pageNum: number;
	pageSize: number;
}

// 数据集信息
export interface DatasetInfo {
	id?: string;
	name: string;
	description?: string;
	type: string; // text, qa, table
	documentCount?: number;
	charCount?: number;
	status: string; // enable, disable
	embeddingModel?: string;
	createdAt?: string;
	updatedAt?: string;
}

// 获取数据集列表
export function listDatasets(params: DatasetQueryParams) {
	return newService({
		url: '/knowledge/dataset/list',
		method: 'get',
		params,
	});
}

// 获取数据集详情
export function getDataset(id: string) {
	return newService({
		url: '/knowledge/dataset/detail',
		method: 'get',
		params: { id },
	});
}

// 创建数据集
export function createDataset(data: DatasetInfo) {
	return newService({
		url: '/knowledge/dataset/create',
		method: 'post',
		data,
	});
}

// 更新数据集
export function updateDataset(data: DatasetInfo) {
	return newService({
		url: '/knowledge/dataset/update',
		method: 'put',
		data,
	});
}

// 删除数据集
export function deleteDataset(id: string) {
	return newService({
		url: '/knowledge/dataset/delete',
		method: 'delete',
		params: { id },
	});
}

// 更新数据集状态
export function updateDatasetStatus(data: { id: string; status: string }) {
	return newService({
		url: '/knowledge/dataset/updateStatus',
		method: 'put',
		data,
	});
}