Kevin Hu
commited on
Commit
·
449650c
1
Parent(s):
f7ee6db
Fix voyage embedding. (#3818)
Browse files### 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
CHANGED
@@ -680,11 +680,10 @@ class VoyageEmbed(Base):
|
|
680 |
return np.array(res.embeddings), res.total_tokens
|
681 |
|
682 |
def encode_queries(self, text):
|
683 |
-
res = self.client.embed
|
684 |
res = self.client.embed(
|
685 |
texts=text, model=self.model_name, input_type="query"
|
686 |
)
|
687 |
-
return np.array(res.embeddings), res.total_tokens
|
688 |
|
689 |
|
690 |
class HuggingFaceEmbed(Base):
|
|
|
680 |
return np.array(res.embeddings), res.total_tokens
|
681 |
|
682 |
def encode_queries(self, text):
|
|
|
683 |
res = self.client.embed(
|
684 |
texts=text, model=self.model_name, input_type="query"
|
685 |
)
|
686 |
+
return np.array(res.embeddings)[0], res.total_tokens
|
687 |
|
688 |
|
689 |
class HuggingFaceEmbed(Base):
|