Added cohere under dependencies in pyproject.toml

deepset-ai · Mar 31, 2024 · 7fbe9bf · 7fbe9bf
1 parent f94f0bf
commit 7fbe9bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/haystack/components/rankers/cohere.py b/haystack/components/rankers/cohere.py
@@ -1,12 +1,14 @@
 from typing import Any, Dict, List, Optional
 
-import cohere
-
 from haystack import Document, component, default_from_dict, default_to_dict, logging
+from haystack.lazy_imports import LazyImport
 from haystack.utils import Secret, deserialize_secrets_inplace
 
 logger = logging.getLogger(__name__)
 
+with LazyImport(message="Run 'pip install \"cohere==5.1.7\"'") as cohere_import:
+    import cohere
+
 
 @component
 class CohereRanker:

diff --git a/pyproject.toml b/pyproject.toml
@@ -62,6 +62,7 @@ dependencies = [
   "requests",
   "numpy",
   "python-dateutil",
+  "cohere==5.1.7", # CohereRanker
 ]
 
 [tool.hatch.envs.default]
@@ -106,7 +107,6 @@ extra-dependencies = [
   "spacy>=3.7,<3.8",  # NamedEntityExtractor
   "spacy-curated-transformers>=0.2,<=0.3",  # NamedEntityExtractor
   "en-core-web-trf @ https://github.com/explosion/spacy-models/releases/download/en_core_web_trf-3.7.3/en_core_web_trf-3.7.3-py3-none-any.whl",  # NamedEntityExtractor
-  "cohere==5.1.7", # CohereRanker
 
   # Converters
   "pypdf",  # PyPDFConverter