diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py index b16f63be4a4ce1becda1f24e536a60182bb3b7e2..bea75ba20a3fd3f137b998e348d51c651ab01038 100644 --- a/semantic_router/encoders/cohere.py +++ b/semantic_router/encoders/cohere.py @@ -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