diff --git a/docs/source/users/index.md b/docs/source/users/index.md index 7fc2af6cd..4b339f13c 100644 --- 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 index 3286e8e35..d9b4a9e7b 100644 --- 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")