diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index e3f50c61..62a174cb 100644 --- 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 index d7b060d6..22004e1f 100644 --- 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"