From 6a0583f5ade3182342f5422dccaa538e825909f9 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 3 Dec 2024 09:33:54 +0800
Subject: [PATCH] Fix voyage embedding. (#3818)

### What problem does this PR solve?

#3816

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 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 1e82bd621..6d70ada4b 100644
--- a/rag/llm/embedding_model.py
+++ b/rag/llm/embedding_model.py
@@ -680,11 +680,10 @@ class VoyageEmbed(Base):
         return np.array(res.embeddings), res.total_tokens
 
     def encode_queries(self, text):
-        res = self.client.embed
         res = self.client.embed(
             texts=text, model=self.model_name, input_type="query"
             )
-        return np.array(res.embeddings), res.total_tokens
+        return np.array(res.embeddings)[0], res.total_tokens
 
 
 class HuggingFaceEmbed(Base):