cohere-haystack 8.1.0__tar.gz → 9.0.0__tar.gz

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## [integrations/cohere-v8.1.0] - 2026-03-19
+
+### 🚀 Features
+
+- Add list of models supported by Cohere components (#2993)
+
+### 🧹 Chores
+
+- Enable ANN ruff ruleset for cohere integration (#2983)
+
+
 ## [integrations/cohere-v8.0.1] - 2026-03-17
 
 ### 🐛 Bug Fixes

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cohere-haystack
-Version: 8.1.0
+Version: 9.0.0
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/cohere#readme
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues
 Project-URL: Source, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/cohere
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/pyproject.toml

@@ -19,6 +19,7 @@ classifiers = [
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
@@ -66,7 +67,8 @@ dependencies = [
 unit = 'pytest -m "not integration" {args:tests}'
 integration = 'pytest -m "integration" {args:tests}'
 all = 'pytest {args:tests}'
-cov-retry = 'pytest --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x {args:tests}'
+unit-cov-retry = 'pytest --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x -m "not integration" {args:tests}'
+integration-cov-append-retry = 'pytest --cov=haystack_integrations --cov-append --reruns 3 --reruns-delay 30 -x -m "integration" {args:tests}'
 
 types = """mypy -p haystack_integrations.components.embedders.cohere \
 -p haystack_integrations.components.generators.cohere \
@@ -88,6 +90,13 @@ select = [
   "ARG",
   "B",
   "C",
+  "D102",   # Missing docstring in public method
+  "D103",   # Missing docstring in public function
+  "D205",   # 1 blank line required between summary line and description
+  "D209",   # Closing triple quotes go to new line
+  "D213",   # summary lines must be positioned on the second physical line of the docstring
+  "D417",   # Missing argument descriptions in the docstring
+  "D419",   # Docstring is empty
   "DTZ",
   "E",
   "EM",
@@ -137,11 +146,12 @@ ban-relative-imports = "parents"
 
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, relative imports, and don't need type annotations
-"tests/**/*" = ["PLR2004", "S101", "TID252", "ANN"]
+"tests/**/*" = ["PLR2004", "S101", "TID252", "D", "ANN"]
 
 [tool.coverage.run]
 source = ["haystack_integrations"]
 branch = true
+relative_files = true
 parallel = false
 
 

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/src/haystack_integrations/components/embedders/cohere/document_embedder.py

@@ -49,7 +49,7 @@ class CohereDocumentEmbedder:
     def __init__(
         self,
         api_key: Secret = Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"]),
-        model: str = "embed-english-v2.0",
+        model: str = "embed-v4.0",
         input_type: str = "search_document",
         api_base_url: str = "https://api.cohere.com",
         truncate: str = "END",
@@ -61,16 +61,13 @@ class CohereDocumentEmbedder:
         embedding_type: EmbeddingTypes | None = None,
     ) -> None:
         """
+        Initialize the CohereDocumentEmbedder.
+
         :param api_key: the Cohere API key.
-        :param model: the name of the model to use. Supported Models are:
-            `"embed-english-v3.0"`, `"embed-english-light-v3.0"`, `"embed-multilingual-v3.0"`,
-            `"embed-multilingual-light-v3.0"`, `"embed-english-v2.0"`, `"embed-english-light-v2.0"`,
-            `"embed-multilingual-v2.0"`. This list of all supported models can be found in the
-            [model documentation](https://docs.cohere.com/docs/models#representation).
+        :param model: the name of the model to use.
+            Read [Cohere documentation](https://docs.cohere.com/docs/models#embed) for a list of all supported models.
         :param input_type: specifies the type of input you're giving to the model. Supported values are
-            "search_document", "search_query", "classification" and "clustering". Not
-            required for older versions of the embedding models (meaning anything lower than v3), but is required for
-            more recent versions (meaning anything bigger than v2).
+            "search_document", "search_query", "classification" and "clustering".
         :param api_base_url: the Cohere API Base url.
         :param truncate: truncate embeddings that are too long from start or end, ("NONE"|"START"|"END").
             Passing "START" will discard the start of the input. "END" will discard the end of the input. In both
@@ -179,7 +176,8 @@ class CohereDocumentEmbedder:
 
     @component.output_types(documents=list[Document], meta=dict[str, Any])
     def run(self, documents: list[Document]) -> dict[str, list[Document] | dict[str, Any]]:
-        """Embed a list of `Documents`.
+        """
+        Embed a list of `Documents`.
 
         :param documents: documents to embed.
         :returns:  A dictionary with the following keys:

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/src/haystack_integrations/components/embedders/cohere/text_embedder.py

@@ -45,7 +45,7 @@ class CohereTextEmbedder:
     def __init__(
         self,
         api_key: Secret = Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"]),
-        model: str = "embed-english-v2.0",
+        model: str = "embed-v4.0",
         input_type: str = "search_query",
         api_base_url: str = "https://api.cohere.com",
         truncate: str = "END",
@@ -53,16 +53,13 @@ class CohereTextEmbedder:
         embedding_type: EmbeddingTypes | None = None,
     ) -> None:
         """
+        Initialize the CohereTextEmbedder.
+
         :param api_key: the Cohere API key.
-        :param model: the name of the model to use. Supported Models are:
-            `"embed-english-v3.0"`, `"embed-english-light-v3.0"`, `"embed-multilingual-v3.0"`,
-            `"embed-multilingual-light-v3.0"`, `"embed-english-v2.0"`, `"embed-english-light-v2.0"`,
-            `"embed-multilingual-v2.0"`. This list of all supported models can be found in the
-            [model documentation](https://docs.cohere.com/docs/models#representation).
+        :param model: the name of the model to use.
+            Read [Cohere documentation](https://docs.cohere.com/docs/models#embed) for a list of all supported models.
         :param input_type: specifies the type of input you're giving to the model. Supported values are
-        "search_document", "search_query", "classification" and "clustering". Not
-            required for older versions of the embedding models (meaning anything lower than v3), but is required for
-            more recent versions (meaning anything bigger than v2).
+        "search_document", "search_query", "classification" and "clustering".
         :param api_base_url: the Cohere API Base url.
         :param truncate: truncate embeddings that are too long from start or end, ("NONE"|"START"|"END").
             Passing "START" will discard the start of the input. "END" will discard the end of the input. In both

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/src/haystack_integrations/components/embedders/cohere/utils.py

@@ -18,7 +18,8 @@ async def get_async_response(
     truncate: str,
     embedding_type: EmbeddingTypes | None = None,
 ) -> tuple[list[list[float]], dict[str, Any]]:
-    """Embeds a list of texts asynchronously using the Cohere API.
+    """
+    Embeds a list of texts asynchronously using the Cohere API.
 
     :param cohere_async_client: the Cohere `AsyncClient`
     :param texts: the texts to embed
@@ -66,7 +67,8 @@ def get_response(
     progress_bar: bool = False,
     embedding_type: EmbeddingTypes | None = None,
 ) -> tuple[list[list[float]], dict[str, Any]]:
-    """Embeds a list of texts using the Cohere API.
+    """
+    Embeds a list of texts using the Cohere API.
 
     :param cohere_client: the Cohere `Client`
     :param texts: the texts to embed

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/src/haystack_integrations/components/generators/cohere/generator.py

@@ -15,7 +15,8 @@ logger = logging.getLogger(__name__)
 
 @component
 class CohereGenerator(CohereChatGenerator):
-    """Generates text using Cohere's models through Cohere's `generate` endpoint.
+    """
+    Generates text using Cohere's models through Cohere's `generate` endpoint.
 
     NOTE: Cohere discontinued the `generate` API, so this generator is a mere wrapper
     around `CohereChatGenerator` provided for backward compatibility.
@@ -115,6 +116,7 @@ class CohereGenerator(CohereChatGenerator):
     ) -> dict[str, list[str] | list[dict[str, Any]]]:
         """
         Queries the LLM asynchronously with the prompts to produce replies.
+
         :param prompt: the prompt to be sent to the generative model.
         :returns: A dictionary with the following keys:
             - `replies`: A list of replies generated by the model.

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/src/haystack_integrations/components/rankers/cohere/ranker.py

@@ -106,6 +106,7 @@ class CohereRanker:
     def _prepare_cohere_input_docs(self, documents: list[Document]) -> list[str]:
         """
         Prepare the input by concatenating the document text with the metadata fields specified.
+
         :param documents: The list of Document objects.
 
         :return: A list of strings to be given as input to Cohere model.

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/tests/test_document_embedder.py

@@ -26,7 +26,7 @@ class TestCohereDocumentEmbedder:
         monkeypatch.setenv("COHERE_API_KEY", "test-api-key")
         embedder = CohereDocumentEmbedder()
         assert embedder.api_key == Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"])
-        assert embedder.model == "embed-english-v2.0"
+        assert embedder.model == "embed-v4.0"
         assert embedder.input_type == "search_document"
         assert embedder.api_base_url == COHERE_API_URL
         assert embedder.truncate == "END"
@@ -70,7 +70,7 @@ class TestCohereDocumentEmbedder:
             "type": "haystack_integrations.components.embedders.cohere.document_embedder.CohereDocumentEmbedder",
             "init_parameters": {
                 "api_key": {"env_vars": ["COHERE_API_KEY", "CO_API_KEY"], "strict": True, "type": "env_var"},
-                "model": "embed-english-v2.0",
+                "model": "embed-v4.0",
                 "input_type": "search_document",
                 "api_base_url": COHERE_API_URL,
                 "truncate": "END",
@@ -122,7 +122,7 @@ class TestCohereDocumentEmbedder:
             "type": "haystack_integrations.components.embedders.cohere.document_embedder.CohereDocumentEmbedder",
             "init_parameters": {
                 "api_key": {"env_vars": ["COHERE_API_KEY", "CO_API_KEY"], "strict": True, "type": "env_var"},
-                "model": "embed-english-v2.0",
+                "model": "embed-v4.0",
                 "input_type": "search_document",
                 "api_base_url": COHERE_API_URL,
                 "truncate": "END",
@@ -137,7 +137,7 @@ class TestCohereDocumentEmbedder:
         }
         embedder = CohereDocumentEmbedder.from_dict(component_dict)
         assert embedder.api_key == Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"])
-        assert embedder.model == "embed-english-v2.0"
+        assert embedder.model == "embed-v4.0"
         assert embedder.input_type == "search_document"
         assert embedder.api_base_url == COHERE_API_URL
         assert embedder.truncate == "END"
@@ -253,7 +253,7 @@ class TestCohereDocumentEmbedder:
     )
     @pytest.mark.integration
     def test_live_run(self):
-        embedder = CohereDocumentEmbedder(model="embed-english-v2.0", embedding_type=EmbeddingTypes.FLOAT)
+        embedder = CohereDocumentEmbedder(model="embed-v4.0", embedding_type=EmbeddingTypes.FLOAT)
 
         docs = [
             Document(content="I love cheese", meta={"topic": "Cuisine"}),
@@ -276,7 +276,7 @@ class TestCohereDocumentEmbedder:
     )
     @pytest.mark.integration
     async def test_live_run_async(self):
-        embedder = CohereDocumentEmbedder(model="embed-english-v2.0", embedding_type=EmbeddingTypes.FLOAT)
+        embedder = CohereDocumentEmbedder(model="embed-v4.0", embedding_type=EmbeddingTypes.FLOAT)
 
         docs = [
             Document(content="I love cheese", meta={"topic": "Cuisine"}),

{cohere_haystack-8.1.0 → cohere_haystack-9.0.0}/tests/test_text_embedder.py

@@ -28,7 +28,7 @@ class TestCohereTextEmbedder:
         embedder = CohereTextEmbedder()
 
         assert embedder.api_key == Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"])
-        assert embedder.model == "embed-english-v2.0"
+        assert embedder.model == "embed-v4.0"
         assert embedder.input_type == "search_query"
         assert embedder.api_base_url == COHERE_API_URL
         assert embedder.truncate == "END"
@@ -65,7 +65,7 @@ class TestCohereTextEmbedder:
             "type": "haystack_integrations.components.embedders.cohere.text_embedder.CohereTextEmbedder",
             "init_parameters": {
                 "api_key": {"env_vars": ["COHERE_API_KEY", "CO_API_KEY"], "strict": True, "type": "env_var"},
-                "model": "embed-english-v2.0",
+                "model": "embed-v4.0",
                 "input_type": "search_query",
                 "api_base_url": COHERE_API_URL,
                 "truncate": "END",
@@ -108,7 +108,7 @@ class TestCohereTextEmbedder:
             "type": "haystack_integrations.components.embedders.cohere.text_embedder.CohereTextEmbedder",
             "init_parameters": {
                 "api_key": {"env_vars": ["COHERE_API_KEY", "CO_API_KEY"], "strict": True, "type": "env_var"},
-                "model": "embed-english-v2.0",
+                "model": "embed-v4.0",
                 "input_type": "search_query",
                 "api_base_url": COHERE_API_URL,
                 "truncate": "END",
@@ -120,7 +120,7 @@ class TestCohereTextEmbedder:
 
         embedder = CohereTextEmbedder.from_dict(component_dict)
         assert embedder.api_key == Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"])
-        assert embedder.model == "embed-english-v2.0"
+        assert embedder.model == "embed-v4.0"
         assert embedder.input_type == "search_query"
         assert embedder.api_base_url == COHERE_API_URL
         assert embedder.truncate == "END"
@@ -148,7 +148,7 @@ class TestCohereTextEmbedder:
         text = "The food was delicious"
         result = embedder.run(text=text)
 
-        assert len(result["embedding"]) == 4096
+        assert len(result["embedding"]) == 1536
         assert all(isinstance(x, float) for x in result["embedding"])
 
     @pytest.mark.asyncio
@@ -162,5 +162,5 @@ class TestCohereTextEmbedder:
         text = "The food was delicious"
         result = await embedder.run_async(text=text)
 
-        assert len(result["embedding"]) == 4096
+        assert len(result["embedding"]) == 1536
         assert all(isinstance(x, float) for x in result["embedding"])