add mistral-large and mixtral-8x22b to the set of known models

langchain-ai · Apr 17, 2024 · 268f791 · 268f791
1 parent 92af5f5
commit 268f791
Showing 1 changed file with 8 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
@@ -118,6 +118,14 @@ class Model(BaseModel):
             "model_name": "google/recurrentgemma-2b",
         },
         "ai-gemma-2b": {"model_type": "chat", "model_name": "google/gemma-2b"},
+        "ai-mistral-large": {
+            "model_type": "chat",
+            "model_name": "mistralai/mistral-large",
+        },
+        "ai-mixtral-8x22b": {
+            "model_type": "completion",
+            "model_name": "mistralai/mixtral-8x22b-v0.1",
+        },
     }
 )