address all comments

wandb · Dec 23, 2024 · 7011748 · 7011748
1 parent 55b12f1
commit 7011748
Show file tree

Hide file tree

Showing 25 changed files with 217 additions and 842 deletions.
diff --git a/docs/docs/guides/integrations/index.md b/docs/docs/guides/integrations/index.md
@@ -20,7 +20,7 @@ LLM providers are the vendors that offer access to large language models for gen
 - **[Groq](/guides/integrations/groq)**
 - **[Open Router](/guides/integrations/openrouter)**
 - **[LiteLLM](/guides/integrations/litellm)**
-- - **[NVIDIA NIM](/guides/integrations/nvidia_nim)**
+- **[NVIDIA NIM](/guides/integrations/nvidia_nim)**
 
 
 

diff --git a/noxfile.py b/noxfile.py
@@ -8,6 +8,7 @@
     "cohere",
     "dspy",
     "langchain",
+    "langchain_nvidia_ai_endpoints",
     "litellm",
     "notdiamond",
     "google_ai_studio",
@@ -40,6 +41,7 @@ def lint(session):
         "google_ai_studio",
         "groq",
         "instructor",
+        "langchain_nvidia_ai_endpoints",
         "langchain",
         "litellm",
         "llamaindex",
@@ -73,6 +75,10 @@ def tests(session, shard):
     if shard == "google_ai_studio":
         env["GOOGLE_API_KEY"] = session.env.get("GOOGLE_API_KEY")
 
+    # Add the NVIDIA_API_KEY environment variable for the "langchain_nvidia_ai_endpoints" shard
+    if shard == "langchain_nvidia_ai_endpoints":
+        env["NVIDIA_API_KEY"] = session.env.get("NVIDIA_API_KEY")
+
     # we are doing some integration test in test_llm_integrations.py that requires
     # setting some environment variables for the LLM providers
     if shard == "scorers_tests":

diff --git a/pyproject.toml b/pyproject.toml
@@ -65,6 +65,9 @@ langchain = [
   "chromadb>=0.5.0",
   "pysqlite3",
   "opentelemetry-exporter-otlp",
+]
+langchain_nvidia_ai_endpoints = [
+  "langchain-core>=0.2.1",
   "langchain-nvidia-ai-endpoints",
 ]
 litellm = ["litellm>=1.36.1"]

diff --git a/tests/integrations/__init__.py b/tests/integrations/__init__.py
diff --git a/...ain/cassettes/langchain_nv_ai_endpoints_test/test_chatnvidia_async_stream_quickstart.yaml b/...ain/cassettes/langchain_nv_ai_endpoints_test/test_chatnvidia_async_stream_quickstart.yaml