File tree Expand file tree Collapse file tree 2 files changed +6
-3
lines changed
redisvl/utils/vectorize/text Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Original file line number Diff line number Diff 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 """
Original file line number Diff line number Diff line change @@ -43,7 +43,10 @@ def vectorizer(request):
4343@pytest .mark .skipif (skip_vectorizer_test , reason = "Skipping vectorizer tests" )
4444def 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 ()
You can’t perform that action at this time.
0 commit comments