Skip to content
Snippets Groups Projects
Unverified Commit f2d4019d authored by James Briggs's avatar James Briggs
Browse files

lint

parent 8c9c6944
Branches
Tags
Loading
......@@ -20,9 +20,5 @@ class CohereEncoder(BaseEncoder):
def __call__(self, texts: list[str]) -> list[list[float]]:
if self.client is None:
raise ValueError("Cohere client is not initialized.")
embeds = self.client.embed(
texts,
input_type="search_query",
model=self.name
)
embeds = self.client.embed(texts, input_type="search_query", model=self.name)
return embeds.embeddings
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment