add institute-of-science-tokyo/llama-3.1-swallow-70b-instruct-v0.1 an…

…d institute-of-science-tokyo/llama-3.1-swallow-8b-instruct-v0.1 to set of known chat models
langchain-ai · Oct 8, 2024 · f0db608 · f0db608
1 parent c19b5fe
commit f0db608
Showing 1 changed file with 12 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
@@ -442,6 +442,18 @@ def validate_client(self) -> "Model":
         client="ChatNVIDIA",
         supports_structured_output=True,
     ),
+    "institute-of-science-tokyo/llama-3.1-swallow-8b-instruct-v0.1": Model(
+        id="institute-of-science-tokyo/llama-3.1-swallow-8b-instruct-v0.1",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_structured_output=True,
+    ),
+    "institute-of-science-tokyo/llama-3.1-swallow-70b-instruct-v0.1": Model(
+        id="institute-of-science-tokyo/llama-3.1-swallow-70b-instruct-v0.1",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_structured_output=True,
+    ),
 }
 
 QA_MODEL_TABLE = {