Skip to content

Commit 96998ec

Browse files
Fix vectorizer tests and docstrings (#121)
1 parent c06a787 commit 96998ec

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

redisvl/utils/vectorize/text/cohere.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ class CohereTextVectorizer(BaseVectorizer):
3838
)
3939
doc_embeddings = cohere.embed_many(
4040
texts=["your document text", "more document text"],
41-
input_type="search_documents"
41+
input_type="search_document"
4242
)
4343
4444
"""

tests/integration/test_vectorizers.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,10 @@ def vectorizer(request):
4343
@pytest.mark.skipif(skip_vectorizer_test, reason="Skipping vectorizer tests")
4444
def test_vectorizer_embed(vectorizer):
4545
text = "This is a test sentence."
46-
embedding = vectorizer.embed(text)
46+
if isinstance(vectorizer, CohereTextVectorizer):
47+
embedding = vectorizer.embed(text, input_type="search_document")
48+
else:
49+
embedding = vectorizer.embed(text)
4750

4851
assert isinstance(embedding, list)
4952
assert len(embedding) == vectorizer.dims
@@ -77,7 +80,7 @@ def test_vectorizer_bad_input(vectorizer):
7780

7881

7982
@pytest.fixture(params=[OpenAITextVectorizer])
80-
def avectorizer(request, openai_key):
83+
def avectorizer(request):
8184
# Here we use actual models for integration test
8285
if request.param == OpenAITextVectorizer:
8386
return request.param()

0 commit comments

Comments
 (0)