add mistralai/mathstral-7b-v0.1, rakuten/rakutenai-7b-instruct, rakut…

…en/rakutenai-7b-chat to set of supported chat models
langchain-ai · Aug 13, 2024 · 8f4deaa · 8f4deaa
1 parent c554bd5
commit 8f4deaa
Showing 1 changed file with 15 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
@@ -339,6 +339,21 @@ def validate_client(cls, client: str, values: dict) -> str:
         supports_tools=True,
         supports_structured_output=True,
     ),
+    "mistralai/mathstral-7b-v0.1": Model(
+        id="mistralai/mathstral-7b-v0.1",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "rakuten/rakutenai-7b-instruct": Model(
+        id="rakuten/rakutenai-7b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "rakuten/rakutenai-7b-chat": Model(
+        id="rakuten/rakutenai-7b-chat",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {