From 8c9c69446816162e67b6dc05e68b3c852b241130 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 11 Nov 2023 14:06:24 +0100
Subject: [PATCH] update input_type to always be query
---
semantic_router/encoders/cohere.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py
index 0ed2ecc0..b16f63be 100644
--- a/semantic_router/encoders/cohere.py
+++ b/semantic_router/encoders/cohere.py
@@ -20,9 +20,9 @@ class CohereEncoder(BaseEncoder):
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
--
GitLab