package org.xyzh.ai.client.dto;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;

/**
 * @description 创建知识库请求
 * @filename DatasetCreateRequest.java
 * @author AI Assistant
 * @copyright xyzh
 * @since 2025-11-04
 */
@Data
public class DatasetCreateRequest {

    /**
     * 知识库名称
     */
    private String name;

    /**
     * 知识库描述
     */
    private String description;

    /**
     * 索引方式：high_quality（高质量）、economy（经济）
     */
    @JsonProperty("indexing_technique")
    private String indexingTechnique = "high_quality";

    /**
     * Embedding模型
     */
    @JsonProperty("embedding_model")
    private String embeddingModel;

    /**
     * Embedding模型提供商
     */
    @JsonProperty("embedding_model_provider")
    private String embeddingModelProvider;

    /**
     * 检索模型配置（包含 Rerank、Top K、Score 阈值等）
     */
    @JsonProperty("retrieval_model")
    private RetrievalModel retrievalModel;

    /**
     * 权限：only_me（仅自己）、all_team_members（团队所有成员）
     */
    private String permission = "only_me";
}

