address server side model renaming for ai-llama3-8b, ai-llama3-70b, a…

…nd ai-phi-3-mini
langchain-ai · Apr 25, 2024 · 460eb55 · 460eb55
1 parent cf179e0
commit 460eb55
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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"