From eb6f5cf84b234f61935ff7cd600329c5da453149 Mon Sep 17 00:00:00 2001
From: liunux4odoo <41217877+liunux4odoo@users.noreply.github.com>
Date: Fri, 29 Sep 2023 18:57:15 +0800
Subject: [PATCH] fix: embedding_model_dict used for bge-* models (#1628)

change to use `get_model_path`
---
 server/knowledge_base/kb_cache/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/knowledge_base/kb_cache/base.py b/server/knowledge_base/kb_cache/base.py
index 59426fa..bb4942a 100644
--- a/server/knowledge_base/kb_cache/base.py
+++ b/server/knowledge_base/kb_cache/base.py
@@ -133,7 +133,7 @@ class EmbeddingsPool(CachePool):
                     else:
                         # maybe ReRanker or else, just use empty string instead
                         query_instruction = ""
-                    embeddings = HuggingFaceBgeEmbeddings(model_name=embedding_model_dict[model],
+                    embeddings = HuggingFaceBgeEmbeddings(model_name=get_model_path(model),
                                                           model_kwargs={'device': device},
                                                           query_instruction=query_instruction)             
                     if model == "bge-large-zh-noinstruct":  # bge large -noinstruct embedding