update Cohere provider dependencies in docs

jupyterlab · Jun 21, 2024 · ebf3dc0 · ebf3dc0
1 parent 1b50b62
commit ebf3dc0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/source/users/index.md b/docs/source/users/index.md
@@ -160,7 +160,7 @@ Jupyter AI supports the following model providers:
 | Anthropic (chat)    | `anthropic-chat`     | `ANTHROPIC_API_KEY`        | `langchain-anthropic`           |
 | Bedrock             | `bedrock`            | N/A                        | `boto3`                         |
 | Bedrock (chat)      | `bedrock-chat`       | N/A                        | `boto3`                         |
-| Cohere              | `cohere`             | `COHERE_API_KEY`           | `cohere`                        |
+| Cohere              | `cohere`             | `COHERE_API_KEY`           | `langchain_cohere`              |
 | ERNIE-Bot           | `qianfan`            | `QIANFAN_AK`, `QIANFAN_SK` | `qianfan`                       |
 | Gemini              | `gemini`             | `GOOGLE_API_KEY`           | `langchain-google-genai`        |
 | GPT4All             | `gpt4all`            | N/A                        | `gpt4all`                       |

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/cohere.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/cohere.py
@@ -19,7 +19,7 @@ class CohereProvider(BaseProvider, ChatCohere):
         "command-r",
     ]
     model_id_key = "model"
-    pypi_package_deps = ["cohere"]
+    pypi_package_deps = ["langchain_cohere"]
     auth_strategy = EnvAuthStrategy(name="COHERE_API_KEY")
 
 
@@ -36,5 +36,5 @@ class CohereEmbeddingsProvider(BaseEmbeddingsProvider, CohereEmbeddings):
         "embed-multilingual-light-v3.0",
     ]
     model_id_key = "model"
-    pypi_package_deps = ["cohere"]
+    pypi_package_deps = ["langchain_cohere"]
     auth_strategy = EnvAuthStrategy(name="COHERE_API_KEY")