Skip to content

Commit a5b4218

Browse files
committed
Corrections
1 parent 55f6380 commit a5b4218

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

redisvl/utils/vectorize/text/voyageai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,11 @@ class VoyageAITextVectorizer(BaseVectorizer):
3636
)
3737
query_embedding = vectorizer.embed(
3838
text="your input query text here",
39-
input_type="search_query"
39+
input_type="query"
4040
)
4141
doc_embeddings = vectorizer.embed_many(
4242
texts=["your document text", "more document text"],
43-
input_type="search_document"
43+
input_type="document"
4444
)
4545
4646
"""

tests/integration/test_rerankers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def hfCrossEncoderRerankerWithCustomModel():
4040

4141

4242
# Test for basic ranking functionality
43-
def test_rank_documents_cohere(reranker):
43+
def test_rank_documents(reranker):
4444
docs = ["document one", "document two", "document three"]
4545
query = "search query"
4646

0 commit comments

Comments
 (0)