diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index 04400c4b..3f92a9a6 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py @@ -133,12 +133,15 @@ class Model(BaseModel): "model_type": "chat", "model_name": "mistralai/mixtral-8x22b-instruct-v0.1", }, - "ai-llama3-8b": {"model_type": "chat", "model_name": "meta/llama3-8b"}, + "ai-llama3-8b": {"model_type": "chat", "model_name": "meta/llama3-8b-instruct"}, "ai-llama3-70b": { "model_type": "chat", - "model_name": "meta/llama3-70b", + "model_name": "meta/llama3-70b-instruct", + }, + "ai-phi-3-mini": { + "model_type": "chat", + "model_name": "microsoft/phi-3-mini-128k-instruct", }, - "ai-phi-3-mini": {"model_type": "chat", "model_name": "microsoft/phi-3-mini"}, "ai-arctic": {"model_type": "chat", "model_name": "snowflake/arctic"}, } ) diff --git a/libs/ai-endpoints/pyproject.toml b/libs/ai-endpoints/pyproject.toml index 0613e76e..45c028b9 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.10" +version = "0.0.11" description = "An integration package connecting NVIDIA AI Endpoints and LangChain" authors = [] readme = "README.md"