From f76bf27bf93e523ae4572d9e10e9ca541c324185 Mon Sep 17 00:00:00 2001
From: zhangwencui <1064582902@qq.com>
Date: 星期二, 03 十一月 2026 17:18:55 +0800
Subject: [PATCH] 部分改动
---
src/main/java/com/ruoyi/ai/config/EmbeddingStoreConfig.java | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/src/main/java/com/ruoyi/ai/config/EmbeddingStoreConfig.java b/src/main/java/com/ruoyi/ai/config/EmbeddingStoreConfig.java
index eaa487c..18c0b53 100644
--- a/src/main/java/com/ruoyi/ai/config/EmbeddingStoreConfig.java
+++ b/src/main/java/com/ruoyi/ai/config/EmbeddingStoreConfig.java
@@ -3,6 +3,7 @@
import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.store.embedding.EmbeddingStore;
+import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
import dev.langchain4j.store.embedding.pinecone.PineconeEmbeddingStore;
import dev.langchain4j.store.embedding.pinecone.PineconeServerlessIndexConfig;
import io.pinecone.clients.Index;
@@ -26,6 +27,9 @@
@Value("${pinecone.namespace:knowledge-base}")
private String namespace;
+ @Value("${pinecone.enabled:false}")
+ private boolean pineconeEnabled;
+
@Bean
public Pinecone pinecone() {
return new Pinecone.Builder(pineconeApiKey).build();
@@ -38,6 +42,11 @@
@Bean
public EmbeddingStore<TextSegment> embeddingStore(EmbeddingModel embeddingModel) {
+ // 椤圭洰绠$悊绛夋牳蹇冧笟鍔′笉搴斿洜澶栭儴鍚戦噺搴撲笉鍙敤鑰屾棤娉曞惎鍔ㄣ��
+ // 鐢熶骇鐜鏄庣‘閰嶇疆 pinecone.enabled=true 鏃朵粛浣跨敤鍘� Pinecone 瀛樺偍銆�
+ if (!pineconeEnabled) {
+ return new InMemoryEmbeddingStore<>();
+ }
return PineconeEmbeddingStore.builder()
.apiKey(pineconeApiKey)
.index(indexName)
@@ -49,4 +58,4 @@
.build())
.build();
}
-}
\ No newline at end of file
+}
--
Gitblit v1.9.3