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