package cn.iocoder.yudao.module.ai.framework.ai.config;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;

/**
 * 超级管理员 AI 配置类
 */
@ConfigurationProperties(prefix = "yudao.ai")
@Data
public class YudaoAiProperties {

    private WebSearch webSearch;

    /** 向量库配置 */
    private VectorStore vectorStore = new VectorStore();

    @Data
    public static class WebSearch {
        private boolean enable;
        private String apiKey;
    }

    @Data
    public static class VectorStore {

        /** 向量库类型：milvus */
        private String type = "milvus";

        /** Milvus 配置 */
        private Milvus milvus = new Milvus();

        @Data
        public static class Milvus {
            /** 是否初始化 Schema */
            private boolean initializeSchema = true;
            /** 数据库名称 */
            private String databaseName = "default";
            /** 集合名称 */
            private String collectionName = "knowledge_segment";
            /** 主机地址 */
            private String host = "127.0.0.1";
            /** 端口 */
            private int port = 19530;
        }
    }

}