From a2fdabdad27c846999cb5865b3346f3d33b5b75a Mon Sep 17 00:00:00 2001 From: Rashmi Pawar <168514198+raspawar@users.noreply.github.com> Date: Tue, 7 May 2024 01:14:58 +0530 Subject: [PATCH] mark NemoEmbeddings as deprecated (#21239) The NemoEmbeddings is deprecated, instead use langchain-nvidia-ai-endpoints NVIDIAEmbeddings interface. cc: @mattf --------- Co-authored-by: Daniel Glogowski <167348611+dglogo@users.noreply.github.com> Co-authored-by: andyjessen <62343929+andyjessen@users.noreply.github.com> Co-authored-by: Chris Germann <88305668+TAAGECH9@users.noreply.github.com> Co-authored-by: gere Co-authored-by: Chester Curme --- libs/community/langchain_community/embeddings/nemo.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/libs/community/langchain_community/embeddings/nemo.py b/libs/community/langchain_community/embeddings/nemo.py index f0c33c67969..b21c0d7e000 100644 --- a/libs/community/langchain_community/embeddings/nemo.py +++ b/libs/community/langchain_community/embeddings/nemo.py @@ -6,6 +6,7 @@ from typing import Any, Dict, List, Optional import aiohttp import requests +from langchain_core._api.deprecation import deprecated from langchain_core.embeddings import Embeddings from langchain_core.pydantic_v1 import BaseModel, root_validator @@ -41,6 +42,15 @@ def is_endpoint_live(url: str, headers: Optional[dict], payload: Any) -> bool: raise Exception(f"Error querying the endpoint: {e}") +@deprecated( + since="0.0.37", + removal="0.2.0", + message=( + "Directly instantiating a NeMoEmbeddings from langchain-community is " + "deprecated. Please use langchain-nvidia-ai-endpoints NVIDIAEmbeddings " + "interface." + ), +) class NeMoEmbeddings(BaseModel, Embeddings): """NeMo embedding models."""