From bc50f68127a786f0de60cc2583df6932d54b0442 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=9F=E4=B8=8D=E6=B1=9F?= <74400272+Seaver-Zhu@users.noreply.github.com> Date: Thu, 25 Jul 2024 11:43:43 +0800 Subject: [PATCH] fix embedding_model (#1698) ### What problem does this PR solve? fix embedding_model #1692 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Signed-off-by: seaver --- rag/llm/embedding_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index c342d06f4..d9eb484c8 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -128,8 +128,8 @@ class LocalAIEmbed(Base): ) # local embedding for LmStudio donot count tokens def encode_queries(self, text): - res = self.client.embeddings.create(text, model=self.model_name) - return np.array(res.data[0].embedding), 1024 + embds, cnt = self.encode([text]) + return np.array(embds[0]), cnt class AzureEmbed(OpenAIEmbed):