Skip to content
Snippets Groups Projects
Unverified Commit 5ffc38ac authored by Siraj R Aizlewood's avatar Siraj R Aizlewood
Browse files

Update cohere.py

Need to specify texts is the first arg for cohere v^5.00.
parent 6344a7bc
No related branches found
No related tags found
No related merge requests found
...@@ -42,7 +42,7 @@ class CohereEncoder(BaseEncoder): ...@@ -42,7 +42,7 @@ class CohereEncoder(BaseEncoder):
raise ValueError("Cohere client is not initialized.") raise ValueError("Cohere client is not initialized.")
try: try:
embeds = self.client.embed( embeds = self.client.embed(
docs, input_type=self.input_type, model=self.name texts=docs, input_type=self.input_type, model=self.name
) )
return embeds.embeddings return embeds.embeddings
except Exception as e: except Exception as e:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment