add google/codegemma-1.1-7b to set of ChatNVIDIA models

langchain-ai · May 30, 2024 · 53c7448 · 53c7448
1 parent 906e329
commit 53c7448
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -197,6 +197,10 @@ class Model(BaseModel):
             "model_type": "chat",
             "model_name": "ibm/granite-34b-code-instruct",
         },
+        "ai-codegemma-1-1-7b": {
+            "model_type": "chat",
+            "model_name": "google/codegemma-1.1-7b",
+        },
     }
 )
 

diff --git a/libs/ai-endpoints/pyproject.toml b/libs/ai-endpoints/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-nvidia-ai-endpoints"
-version = "0.0.19"
+version = "0.0.20"
 description = "An integration package connecting NVIDIA AI Endpoints and LangChain"
 authors = []
 readme = "README.md"