(improvement)(common) When EmbeddingModel retrieval fails, use the de…

…fault embedded model. (#1176)
tencentmusic · Jun 20, 2024 · 5555dc8 · 5555dc8
1 parent 3831f9f
commit 5555dc8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 16 deletions.
diff --git a/common/src/main/java/dev/langchain4j/model/embedding/EmbeddingModelConfig.java b/common/src/main/java/dev/langchain4j/model/embedding/EmbeddingModelConfig.java
diff --git a/common/src/main/java/dev/langchain4j/store/embedding/InMemoryS2EmbeddingStore.java b/common/src/main/java/dev/langchain4j/store/embedding/InMemoryS2EmbeddingStore.java
@@ -3,10 +3,12 @@
 import com.tencent.supersonic.common.config.EmbeddingConfig;
 import com.tencent.supersonic.common.util.ContextUtils;
 import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.model.embedding.BgeSmallZhEmbeddingModel;
 import dev.langchain4j.model.embedding.EmbeddingModel;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections.MapUtils;
 import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.NoSuchBeanDefinitionException;
 
 import java.io.IOException;
 import java.nio.file.Files;
@@ -95,7 +97,13 @@ public void addQuery(String collectionName, List<EmbeddingQuery> queries) {
     }
 
     private static EmbeddingModel getEmbeddingModel() {
-        return ContextUtils.getBean(EmbeddingModel.class);
+        EmbeddingModel embeddingModel;
+        try {
+            embeddingModel = ContextUtils.getBean(EmbeddingModel.class);
+        } catch (NoSuchBeanDefinitionException e) {
+            embeddingModel = new BgeSmallZhEmbeddingModel();
+        }
+        return embeddingModel;
     }
 
     private InMemoryEmbeddingStore<EmbeddingQuery> getEmbeddingStore(String collectionName) {