From 8f4deaa380373cf23132f13971eb83094aeef7c8 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Tue, 13 Aug 2024 14:16:47 -0400 Subject: [PATCH] add mistralai/mathstral-7b-v0.1, rakuten/rakutenai-7b-instruct, rakuten/rakutenai-7b-chat to set of supported chat models --- .../langchain_nvidia_ai_endpoints/_statics.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index e9b280e4..52e29679 100644 --- 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 = {