add nvidia/nv-embed-v1 to set of NVIDIAEmbeddings models

langchain-ai · Jun 7, 2024 · c67baa3 · c67baa3
1 parent b6d68b4
commit c67baa3
Showing 1 changed file with 6 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
@@ -261,6 +261,12 @@ def __hash__(self) -> int:
             "nvolveqa_40k",
         ],
     ),
+    "nvidia/nv-embed-v1": Model(
+        id="nvidia/nv-embed-v1",
+        model_type="embedding",
+        client="NVIDIAEmbeddings",
+        aliases=["ai-nv-embed-v1"],
+    ),
 }
 
 RANKING_MODEL_TABLE = {