add nvidia/embed-qa-4 to set of supported embedding models

langchain-ai · Sep 16, 2024 · 8d83a81 · 8d83a81
1 parent cb117ce
commit 8d83a81
Showing 1 changed file with 5 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
@@ -521,6 +521,11 @@ def validate_client(cls, client: str, values: dict) -> str:
         model_type="embedding",
         client="NVIDIAEmbeddings",
     ),
+    "nvidia/embed-qa-4": Model(
+        id="nvidia/embed-qa-4",
+        model_type="embedding",
+        client="NVIDIAEmbeddings",
+    ),
 }
 
 RANKING_MODEL_TABLE = {