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

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

@@ -1,5 +1,23 @@
 # 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
+
+- Fix the conversion of cohere chunks to Haystack streaming chunks (#2968)
+
+
 ## [integrations/cohere-v8.0.0] - 2026-03-11
 
 ### 🚀 Features

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cohere-haystack
-Version: 8.0.1
+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.0.1 → 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 \
@@ -84,9 +86,17 @@ line-length = 120
 [tool.ruff.lint]
 select = [
   "A",
+  "ANN",
   "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",
@@ -124,6 +134,8 @@ ignore = [
   # Misc
   "B008",
   "S101",
+  # Allow `Any` - used legitimately for dynamic types and SDK boundaries
+  "ANN401",
 ]
 
 [tool.ruff.lint.isort]
@@ -133,12 +145,13 @@ known-first-party = ["haystack_integrations"]
 ban-relative-imports = "parents"
 
 [tool.ruff.lint.per-file-ignores]
-# Tests can use magic values, assertions, and relative imports
-"tests/**/*" = ["PLR2004", "S101", "TID252"]
+# Tests can use magic values, assertions, relative imports, and don't need type annotations
+"tests/**/*" = ["PLR2004", "S101", "TID252", "D", "ANN"]
 
 [tool.coverage.run]
 source = ["haystack_integrations"]
 branch = true
+relative_files = true
 parallel = false
 
 

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

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 from dataclasses import replace
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import Document, component, default_from_dict, default_to_dict
 from haystack.utils import Secret, deserialize_secrets_inplace
@@ -36,10 +36,20 @@ class CohereDocumentEmbedder:
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "embed-v4.0",
+        "embed-english-v3.0",
+        "embed-english-light-v3.0",
+        "embed-multilingual-v3.0",
+        "embed-multilingual-light-v3.0",
+    ]
+    """A non-exhaustive list of embed models supported by this component.
+    See https://docs.cohere.com/docs/models#embed for the full list."""
+
     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",
@@ -49,18 +59,15 @@ class CohereDocumentEmbedder:
         meta_fields_to_embed: list[str] | None = None,
         embedding_separator: str = "\n",
         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
@@ -169,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.0.1 → cohere_haystack-9.0.0}/src/haystack_integrations/components/embedders/cohere/document_image_embedder.py

@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from dataclasses import replace
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import Document, component, default_from_dict, default_to_dict, logging
 from haystack.components.converters.image.image_utils import (
@@ -59,6 +59,16 @@ class CohereDocumentImageEmbedder:
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "embed-v4.0",
+        "embed-english-v3.0",
+        "embed-english-light-v3.0",
+        "embed-multilingual-v3.0",
+        "embed-multilingual-light-v3.0",
+    ]
+    """A non-exhaustive list of embed models supported by this component.
+    See https://docs.cohere.com/docs/models#embed for the full list."""
+
     def __init__(
         self,
         *,

{cohere_haystack-8.0.1 → cohere_haystack-9.0.0}/src/haystack_integrations/components/embedders/cohere/embedding_types.py

@@ -21,7 +21,7 @@ class EmbeddingTypes(Enum):
     BINARY = "binary"
     UBINARY = "ubinary"
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.value
 
     @staticmethod

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

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
 #
 # SPDX-License-Identifier: Apache-2.0
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, default_from_dict, default_to_dict
 from haystack.utils import Secret, deserialize_secrets_inplace
@@ -32,27 +32,34 @@ class CohereTextEmbedder:
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "embed-v4.0",
+        "embed-english-v3.0",
+        "embed-english-light-v3.0",
+        "embed-multilingual-v3.0",
+        "embed-multilingual-light-v3.0",
+    ]
+    """A non-exhaustive list of embed models supported by this component.
+    See https://docs.cohere.com/docs/models#embed for the full list."""
+
     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",
         timeout: float = 120.0,
         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.0.1 → 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.0.1 → cohere_haystack-9.0.0}/src/haystack_integrations/components/generators/cohere/chat/chat_generator.py

@@ -1,6 +1,6 @@
 import json
 from collections.abc import AsyncIterator, Iterator
-from typing import Any, Literal, get_args
+from typing import Any, ClassVar, Literal, get_args
 
 from haystack import component, default_from_dict, default_to_dict, logging
 from haystack.components.generators.utils import _convert_streaming_chunks_to_chat_message
@@ -480,6 +480,24 @@ class CohereChatGenerator:
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "command-a-03-2025",
+        "command-r7b-12-2024",
+        "command-a-translate-08-2025",
+        "command-a-reasoning-08-2025",
+        "command-a-vision-07-2025",
+        "command-r-08-2024",
+        "command-r-plus-08-2024",
+        "command-r-03-2024",
+        "command-r-plus-04-2024",
+        "command-r-plus",
+        "command-r",
+        "command-light",
+        "command",
+    ]
+    """A non-exhaustive list of chat models supported by this component.
+    See https://docs.cohere.com/docs/models#command for the full list."""
+
     def __init__(
         self,
         api_key: Secret = Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"]),
@@ -491,7 +509,7 @@ class CohereChatGenerator:
         *,
         timeout: float | None = None,
         max_retries: int | None = None,
-    ):
+    ) -> None:
         """
         Initialize the CohereChatGenerator instance.
 

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

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 from collections.abc import Callable
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, logging
 from haystack.dataclasses import ChatMessage
@@ -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.
@@ -30,6 +31,24 @@ class CohereGenerator(CohereChatGenerator):
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "command-a-03-2025",
+        "command-r7b-12-2024",
+        "command-a-translate-08-2025",
+        "command-a-reasoning-08-2025",
+        "command-a-vision-07-2025",
+        "command-r-08-2024",
+        "command-r-plus-08-2024",
+        "command-r-03-2024",
+        "command-r-plus-04-2024",
+        "command-r-plus",
+        "command-r",
+        "command-light",
+        "command",
+    ]
+    """A non-exhaustive list of chat models supported by this component.
+    See https://docs.cohere.com/docs/models#command for the full list."""
+
     def __init__(
         self,
         api_key: Secret = Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"]),
@@ -37,7 +56,7 @@ class CohereGenerator(CohereChatGenerator):
         streaming_callback: Callable | None = None,
         api_base_url: str | None = None,
         **kwargs: Any,
-    ):
+    ) -> None:
         """
         Instantiates a `CohereGenerator` component.
 
@@ -97,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.0.1 → cohere_haystack-9.0.0}/src/haystack_integrations/components/rankers/cohere/ranker.py

@@ -41,7 +41,7 @@ class CohereRanker:
         meta_fields_to_embed: list[str] | None = None,
         meta_data_separator: str = "\n",
         max_tokens_per_doc: int = 4096,
-    ):
+    ) -> None:
         """
         Creates an instance of the 'CohereRanker'.
 
@@ -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.0.1 → cohere_haystack-9.0.0}/tests/test_chat_generator.py

@@ -148,6 +148,13 @@ class TestFormatMessage:
 
 
 class TestCohereChatGenerator:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = CohereChatGenerator.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("COHERE_API_KEY", "test-api-key")
 

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

@@ -15,11 +15,18 @@ COHERE_API_URL = "https://api.cohere.com"
 
 
 class TestCohereDocumentEmbedder:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = CohereDocumentEmbedder.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         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"
@@ -63,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",
@@ -115,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",
@@ -130,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"
@@ -246,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"}),
@@ -269,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.0.1 → cohere_haystack-9.0.0}/tests/test_document_image_embedder.py

@@ -20,6 +20,13 @@ IMPORT_PATH = "haystack_integrations.components.embedders.cohere.document_image_
 
 
 class TestCohereDocumentImageEmbedder:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = CohereDocumentImageEmbedder.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("COHERE_API_KEY", "test-api-key")
         embedder = CohereDocumentImageEmbedder()

{cohere_haystack-8.0.1 → cohere_haystack-9.0.0}/tests/test_generator.py

@@ -15,6 +15,13 @@ COHERE_API_URL = "https://api.cohere.com"
 
 
 class TestCohereGenerator:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = CohereGenerator.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("COHERE_API_KEY", "foo")
         component = CohereGenerator()

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

@@ -13,6 +13,13 @@ COHERE_API_URL = "https://api.cohere.com"
 
 
 class TestCohereTextEmbedder:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = CohereTextEmbedder.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         """
         Test default initialization parameters for CohereTextEmbedder.
@@ -21,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"
@@ -58,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",
@@ -101,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",
@@ -113,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"
@@ -141,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
@@ -155,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"])