From ec21d9a98f011452ae055df600309a32eb88bf46 Mon Sep 17 00:00:00 2001 From: Stephen Hu Date: Wed, 23 Jul 2025 10:51:48 +0800 Subject: [PATCH] Refactor:remove use less convert for FastEmbed (#8984) ### What problem does this PR solve? remove use less convert for FastEmbed ### Type of change - [x] Refactoring --- rag/llm/embedding_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 98f9bcbc1..9774e7dd9 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -352,8 +352,7 @@ class FastEmbed(DefaultEmbedding): # Using the internal tokenizer to encode the texts and get the total # number of tokens encoding = self._model.model.tokenizer.encode(text) - embedding = next(self._model.query_embed(text)).tolist() - + embedding = next(self._model.query_embed(text)) return np.array(embedding), len(encoding.ids)