Makes NVIDIA package dependency optional, updates docs

jupyterlab · Feb 1, 2024 · afea0af · afea0af
1 parent 8082fa4
commit afea0af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/users/index.md b/docs/source/users/index.md
@@ -150,7 +150,7 @@ Jupyter AI supports the following model providers:
 | ERNIE-Bot           | `qianfan`            | `QIANFAN_AK`, `QIANFAN_SK` | `qianfan`                       |
 | GPT4All             | `gpt4all`            | N/A                        | `gpt4all`                       |
 | Hugging Face Hub    | `huggingface_hub`    | `HUGGINGFACEHUB_API_TOKEN` | `huggingface_hub`, `ipywidgets`, `pillow` |
-| NVIDIA              | `nvidia-chat`        | `NVIDIA_API_KEY`           | N/A                             |
+| NVIDIA              | `nvidia-chat`        | `NVIDIA_API_KEY`           | `langchain_nvidia_ai_endpoints` |
 | OpenAI              | `openai`             | `OPENAI_API_KEY`           | `openai`                        |
 | OpenAI (chat)       | `openai-chat`        | `OPENAI_API_KEY`           | `openai`                        |
 | SageMaker           | `sagemaker-endpoint` | N/A                        | `boto3`                         |

diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml
@@ -27,7 +27,6 @@ dependencies = [
     "typing_extensions>=4.5.0",
     "click~=8.0",
     "jsonpath-ng>=1.5.3,<2",
-    "langchain_nvidia_ai_endpoints",
 ]
 
 [project.optional-dependencies]
@@ -42,6 +41,7 @@ all = [
     "gpt4all",
     "huggingface_hub",
     "ipywidgets",
+    "langchain_nvidia_ai_endpoints",
     "pillow",
     "openai~=1.6.1",
     "boto3",