diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index 863cfd9a..aab45112 100644 --- 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 index 8148b794..c8f0b2f5 100644 --- 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"