From afea0af0662385f04aa00a5e820f9823044b63b4 Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Thu, 1 Feb 2024 09:22:03 -0800 Subject: [PATCH] Makes NVIDIA package dependency optional, updates docs --- docs/source/users/index.md | 2 +- packages/jupyter-ai-magics/pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/users/index.md b/docs/source/users/index.md index e784b1262..9016444c6 100644 --- 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 index 9fe5c0721..138b79ee3 100644 --- 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",