package cn.iocoder.yudao.module.ai.framework.ai.config;
|
|
import lombok.Data;
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
|
import java.time.Duration;
|
|
/**
|
* 超级管理员 AI 配置类
|
*/
|
@ConfigurationProperties(prefix = "yudao.ai")
|
@Data
|
public class YudaoAiProperties {
|
|
/**
|
* 单次大模型调用超时。
|
* <p>
|
* 必须有默认值：模型排队时若不设超时，请求会一直挂到 TCP 层超时，
|
* 同期没有超时的前端 axios 也不会主动断开，用户界面会永远转圈。
|
*/
|
private Duration timeout = Duration.ofSeconds(60);
|
|
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;
|
}
|
}
|
|
}
|
