add mistralai/mixtral-8x22b-instruct-v0.1, meta/llama3-8b, meta/llama…

…3-70b to set of known models
langchain-ai · Apr 18, 2024 · 7aac656 · 7aac656
1 parent 78eca72
commit 7aac656
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -126,6 +126,15 @@ class Model(BaseModel):
             "model_type": "completion",
             "model_name": "mistralai/mixtral-8x22b-v0.1",
         },
+        "ai-mixtral-8x22b-instruct": {
+            "model_type": "chat",
+            "model_name": "mistralai/mixtral-8x22b-instruct-v0.1",
+        },
+        "ai-llama3-8b": {"model_type": "chat", "model_name": "meta/llama3-8b"},
+        "ai-llama3-70b": {
+            "model_type": "chat",
+            "model_name": "meta/llama3-70b",
+        },
     }
 )