From f0db60839223efccf20b654757a5e59a52066535 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Tue, 8 Oct 2024 09:44:39 -0400 Subject: [PATCH] add institute-of-science-tokyo/llama-3.1-swallow-70b-instruct-v0.1 and institute-of-science-tokyo/llama-3.1-swallow-8b-instruct-v0.1 to set of known chat models --- .../langchain_nvidia_ai_endpoints/_statics.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index 3c1f0bfc..1f5d2604 100644 --- 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 = {