update input_type to always be query

aurelio-labs · Nov 11, 2023 · 8c9c694 · 8c9c694
1 parent ad200b0
commit 8c9c694
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py
@@ -20,9 +20,9 @@ def __init__(
     def __call__(self, texts: list[str]) -> list[list[float]]:
         if self.client is None:
             raise ValueError("Cohere client is not initialized.")
-        if len(texts) == 1:
-            input_type = "search_query"
-        else:
-            input_type = "search_document"
-        embeds = self.client.embed(texts, input_type=input_type, model=self.name)
+        embeds = self.client.embed(
+            texts,
+            input_type="search_query",
+            model=self.name
+        )
         return embeds.embeddings