Merge pull request #128 from mattf/add-chat-models

add chat models: qwen/qwen2.5-coder-32b-instruct, qwen/qwen2.5-coder-7b-instruct, nvidia/llama-3.1-nemotron-70b-reward
langchain-ai · Jan 13, 2025 · 62b1552 · 62b1552
2 parents a7d8dc3 + e533395
commit 62b1552
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
@@ -493,6 +493,21 @@ def validate_client(self) -> "Model":
         supports_tools=True,
         supports_structured_output=True,
     ),
+    "qwen/qwen2.5-coder-32b-instruct": Model(
+        id="qwen/qwen2.5-coder-32b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "qwen/qwen2.5-coder-7b-instruct": Model(
+        id="qwen/qwen2.5-coder-7b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "nvidia/llama-3.1-nemotron-70b-reward": Model(
+        id="nvidia/llama-3.1-nemotron-70b-reward",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {