fix: check empty Cohere key in rerank

Cinnamon · Oct 1, 2024 · 76ab3fd · 76ab3fd
1 parent 8d3a752
commit 76ab3fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/kotaemon/kotaemon/rerankings/cohere.py b/libs/kotaemon/kotaemon/rerankings/cohere.py
@@ -34,7 +34,7 @@ def run(self, documents: list[Document], query: str) -> list[Document]:
                 "Please install Cohere " "`pip install cohere` to use Cohere Reranking"
             )
 
-        if not self.cohere_api_key:
+        if not self.cohere_api_key or "COHERE_API_KEY" in self.cohere_api_key:
             print("Cohere API key not found. Skipping rerankings.")
             return documents