Merge pull request #113 from langchain-ai/add-retriever-24.10

add new retriever models: nvidia/llama-3.2-nv-embedqa-1b-v1 and nvidia/llama-3.2-nv-rerankqa-1b-v1
langchain-ai · Oct 25, 2024 · cd48c4a · cd48c4a
2 parents 8e6d380 + 77b7982
commit cd48c4a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -596,6 +596,11 @@ def validate_client(self) -> "Model":
         model_type="embedding",
         client="NVIDIAEmbeddings",
     ),
+    "nvidia/llama-3.2-nv-embedqa-1b-v1": Model(
+        id="nvidia/llama-3.2-nv-embedqa-1b-v1",
+        model_type="embedding",
+        client="NVIDIAEmbeddings",
+    ),
 }
 
 RANKING_MODEL_TABLE = {
@@ -612,6 +617,12 @@ def validate_client(self) -> "Model":
         client="NVIDIARerank",
         endpoint="https://ai.api.nvidia.com/v1/retrieval/nvidia/nv-rerankqa-mistral-4b-v3/reranking",
     ),
+    "nvidia/llama-3.2-nv-rerankqa-1b-v1": Model(
+        id="nvidia/llama-3.2-nv-rerankqa-1b-v1",
+        model_type="ranking",
+        client="NVIDIARerank",
+        endpoint="https://ai.api.nvidia.com/v1/retrieval/nvidia/llama-3_2-nv-rerankqa-1b-v1/reranking",
+    ),
 }
 
 COMPLETION_MODEL_TABLE = {

diff --git a/libs/ai-endpoints/pyproject.toml b/libs/ai-endpoints/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-nvidia-ai-endpoints"
-version = "0.3.2"
+version = "0.3.3"
 description = "An integration package connecting NVIDIA AI Endpoints and LangChain"
 authors = []
 readme = "README.md"