diff --git a/modules/cms-ai/pom.xml b/modules/cms-ai/pom.xml index c3a375f5..5e32ac78 100644 --- a/modules/cms-ai/pom.xml +++ b/modules/cms-ai/pom.xml @@ -43,11 +43,11 @@ spring-ai-starter-model-openai - org.springframework.ai spring-ai-starter-model-ollama - --> + @@ -65,19 +65,19 @@ httpclient5 - org.springframework.ai spring-ai-starter-vector-store-pgvector - --> + - org.springframework.ai spring-ai-starter-vector-store-elasticsearch - --> + - org.springframework.ai spring-ai-starter-vector-store-milvus @@ -92,7 +92,7 @@ io.netty netty-resolver-dns-native-macos osx-aarch_64 - --> + diff --git a/modules/cms-ai/src/main/java/com/jeesite/modules/cms/ai/properties/CmsAiProperties.java b/modules/cms-ai/src/main/java/com/jeesite/modules/cms/ai/properties/CmsAiProperties.java index 8adaf470..fd924f25 100644 --- a/modules/cms-ai/src/main/java/com/jeesite/modules/cms/ai/properties/CmsAiProperties.java +++ b/modules/cms-ai/src/main/java/com/jeesite/modules/cms/ai/properties/CmsAiProperties.java @@ -1,6 +1,7 @@ package com.jeesite.modules.cms.ai.properties; import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.boot.context.properties.NestedConfigurationProperty; @ConfigurationProperties("spring.ai") public class CmsAiProperties { @@ -20,6 +21,12 @@ public class CmsAiProperties { */ private String defaultPromptTemplate = ""; + /** + * 向量数据库设置 + */ + @NestedConfigurationProperty + private final Vectorstore vectorstore = new Vectorstore(); + public Boolean getToolCalls() { return toolCalls; } @@ -43,4 +50,24 @@ public class CmsAiProperties { public void setDefaultPromptTemplate(String defaultPromptTemplate) { this.defaultPromptTemplate = defaultPromptTemplate; } + + public Vectorstore getVectorstore() { + return vectorstore; + } + + public static class Vectorstore { + + /** + * 向量库类型选择:chroma、pgvector、elasticsearch、milvus + */ + private String type; + + public String getType() { + return type; + } + + public void setType(String type) { + this.type = type; + } + } } diff --git a/modules/cms-ai/src/main/resources/config/jeesite-cms-ai.yml b/modules/cms-ai/src/main/resources/config/jeesite-cms-ai.yml index 94ceb69f..39413a39 100644 --- a/modules/cms-ai/src/main/resources/config/jeesite-cms-ai.yml +++ b/modules/cms-ai/src/main/resources/config/jeesite-cms-ai.yml @@ -3,6 +3,13 @@ spring: ai: + # 模型选择:openai、ollama + model: + chat: ollama + embedding: ${spring.ai.model.chat} + image: ${spring.ai.model.chat} + audio: ${spring.ai.model.chat} + # 在线大模型【请在 pom.xml 中打开 openai 的注释,并注释上其它模型】 openai: @@ -83,6 +90,9 @@ spring: # 向量数据库配置 vectorstore: + # 向量库类型:chroma、pgvector、elasticsearch、milvus + type: chroma + # Chroma 向量数据库【请在 pom.xml 中打开 chroma 的注释,并注释上其它向量库】 chroma: client: diff --git a/web-ai/pom.xml b/web-ai/pom.xml index 05462ade..8c9e951a 100644 --- a/web-ai/pom.xml +++ b/web-ai/pom.xml @@ -13,7 +13,7 @@ jeesite-web-ai war - Web AI 服务,也可为分离端提供接口服务 + Web AI 服务 JeeSite Web AI http://jeesite.com