diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index afc6434c..863cfd9a 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py @@ -189,6 +189,14 @@ class Model(BaseModel): "model_type": "image_in", "model_name": "microsoft/phi-3-vision-128k-instruct", }, + "ai-granite-8b-code-instruct": { + "model_type": "chat", + "model_name": "ibm/granite-8b-code-instruct", + }, + "ai-granite-34b-code-instruct": { + "model_type": "chat", + "model_name": "ibm/granite-34b-code-instruct", + }, } )