From c33d6c33ed28f5155403a7af727466903bbb1171 Mon Sep 17 00:00:00 2001
From: awinml <97467100+awinml@users.noreply.github.com>
Date: Fri, 8 Dec 2023 17:58:54 +0530
Subject: [PATCH] fix model names

---
 .../cohere/src/cohere_haystack/embedders/document_embedder.py   | 2 +-
 .../cohere/src/cohere_haystack/embedders/text_embedder.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/integrations/cohere/src/cohere_haystack/embedders/document_embedder.py b/integrations/cohere/src/cohere_haystack/embedders/document_embedder.py
index 5ad027a29..0f72a8dc2 100644
--- a/integrations/cohere/src/cohere_haystack/embedders/document_embedder.py
+++ b/integrations/cohere/src/cohere_haystack/embedders/document_embedder.py
@@ -54,7 +54,7 @@ def __init__(
         :param api_key: The Cohere API key. It can be explicitly provided or automatically read from the environment
             variable COHERE_API_KEY (recommended).
         :param model_name: The name of the model to use, defaults to `"embed-english-v2.0"`. Supported Models are:
-            `"embed-english-v3.0"`, `"embed-english-light-3.0"`, `"embed-multilingual-v3.0"`,
+            `"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 on the
             [model documentation](https://docs.cohere.com/docs/models#representation).
diff --git a/integrations/cohere/src/cohere_haystack/embedders/text_embedder.py b/integrations/cohere/src/cohere_haystack/embedders/text_embedder.py
index 28a1d2f7d..19052aa63 100644
--- a/integrations/cohere/src/cohere_haystack/embedders/text_embedder.py
+++ b/integrations/cohere/src/cohere_haystack/embedders/text_embedder.py
@@ -48,7 +48,7 @@ def __init__(
         :param api_key: The Cohere API key. It can be explicitly provided or automatically read from the environment
             variable COHERE_API_KEY (recommended).
         :param model_name: The name of the model to use, defaults to `"embed-english-v2.0"`. Supported Models are:
-            `"embed-english-v3.0"`, `"embed-english-light-3.0"`, `"embed-multilingual-v3.0"`,
+            `"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 on the
             [model documentation](https://docs.cohere.com/docs/models#representation).