From 17c127531aa0937531654fc2817c80574d7b1218 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Tue, 4 Jun 2024 09:48:25 -0700
Subject: [PATCH] community[patch]: deprecate all HF classes (#22444)

---
 .../langchain_community/chat_models/huggingface.py          | 4 +---
 .../community/langchain_community/embeddings/huggingface.py | 6 ++++++
 .../langchain_community/embeddings/huggingface_hub.py       | 6 ++++++
 .../langchain_community/llms/huggingface_endpoint.py        | 2 +-
 libs/community/langchain_community/llms/huggingface_hub.py  | 6 +++++-
 .../langchain_community/llms/huggingface_pipeline.py        | 2 +-
 .../llms/huggingface_text_gen_inference.py                  | 6 +++++-
 7 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/libs/community/langchain_community/chat_models/huggingface.py b/libs/community/langchain_community/chat_models/huggingface.py
index 0603d893824..f03c0c75228 100644
--- a/libs/community/langchain_community/chat_models/huggingface.py
+++ b/libs/community/langchain_community/chat_models/huggingface.py
@@ -39,9 +39,7 @@ DEFAULT_SYSTEM_PROMPT = """You are a helpful, respectful, and honest assistant."
 @deprecated(
     since="0.0.37",
     removal="0.3",
-    alternative_import=(
-        "from langchain_huggingface.chat_models.huggingface import ChatHuggingFace"
-    ),
+    alternative_import=("from langchain_huggingface import ChatHuggingFace"),
 )
 class ChatHuggingFace(BaseChatModel):
     """
diff --git a/libs/community/langchain_community/embeddings/huggingface.py b/libs/community/langchain_community/embeddings/huggingface.py
index 7eb175dfafc..fe36c8b825c 100644
--- a/libs/community/langchain_community/embeddings/huggingface.py
+++ b/libs/community/langchain_community/embeddings/huggingface.py
@@ -1,6 +1,7 @@
 from typing import Any, Dict, List, Optional
 
 import requests
+from langchain_core._api import deprecated
 from langchain_core.embeddings import Embeddings
 from langchain_core.pydantic_v1 import BaseModel, Extra, Field, SecretStr
 
@@ -17,6 +18,11 @@ DEFAULT_QUERY_BGE_INSTRUCTION_EN = (
 DEFAULT_QUERY_BGE_INSTRUCTION_ZH = "为这个句子生成表示以用于检索相关文章："
 
 
+@deprecated(
+    since="0.2.2",
+    removal="0.3.0",
+    alternative_import="from langchain_huggingface import HuggingFaceEmbeddings",
+)
 class HuggingFaceEmbeddings(BaseModel, Embeddings):
     """HuggingFace sentence_transformers embedding models.
 
diff --git a/libs/community/langchain_community/embeddings/huggingface_hub.py b/libs/community/langchain_community/embeddings/huggingface_hub.py
index 70424ac0b21..80578ca38a0 100644
--- a/libs/community/langchain_community/embeddings/huggingface_hub.py
+++ b/libs/community/langchain_community/embeddings/huggingface_hub.py
@@ -2,6 +2,7 @@ import json
 import os
 from typing import Any, Dict, List, Optional
 
+from langchain_core._api import deprecated
 from langchain_core.embeddings import Embeddings
 from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator
 
@@ -9,6 +10,11 @@ DEFAULT_MODEL = "sentence-transformers/all-mpnet-base-v2"
 VALID_TASKS = ("feature-extraction",)
 
 
+@deprecated(
+    since="0.2.2",
+    removal="0.3.0",
+    alternative_import="from langchain_huggingface import HuggingFaceEndpointEmbeddings",  # noqa: E501
+)
 class HuggingFaceHubEmbeddings(BaseModel, Embeddings):
     """HuggingFaceHub embedding models.
 
diff --git a/libs/community/langchain_community/llms/huggingface_endpoint.py b/libs/community/langchain_community/llms/huggingface_endpoint.py
index a0c96ee210d..6afa1ca67c7 100644
--- a/libs/community/langchain_community/llms/huggingface_endpoint.py
+++ b/libs/community/langchain_community/llms/huggingface_endpoint.py
@@ -25,7 +25,7 @@ VALID_TASKS = (
 @deprecated(
     since="0.0.37",
     removal="0.3",
-    alternative_import="from langchain_huggingface.llms import HuggingFaceEndpoint",
+    alternative_import="from langchain_huggingface import HuggingFaceEndpoint",
 )
 class HuggingFaceEndpoint(LLM):
     """
diff --git a/libs/community/langchain_community/llms/huggingface_hub.py b/libs/community/langchain_community/llms/huggingface_hub.py
index 58eeeb1c717..efc74d2b520 100644
--- a/libs/community/langchain_community/llms/huggingface_hub.py
+++ b/libs/community/langchain_community/llms/huggingface_hub.py
@@ -20,7 +20,11 @@ VALID_TASKS_DICT = {
 }
 
 
-@deprecated("0.0.21", removal="0.3.0", alternative="HuggingFaceEndpoint")
+@deprecated(
+    "0.0.21",
+    removal="0.3.0",
+    alternative_import="from langchain_huggingface import HuggingFaceEndpoint",
+)
 class HuggingFaceHub(LLM):
     """HuggingFaceHub  models.
     ! This class is deprecated, you should use HuggingFaceEndpoint instead.
diff --git a/libs/community/langchain_community/llms/huggingface_pipeline.py b/libs/community/langchain_community/llms/huggingface_pipeline.py
index e05cc1e89fb..bd32c98a275 100644
--- a/libs/community/langchain_community/llms/huggingface_pipeline.py
+++ b/libs/community/langchain_community/llms/huggingface_pipeline.py
@@ -26,7 +26,7 @@ logger = logging.getLogger(__name__)
 @deprecated(
     since="0.0.37",
     removal="0.3",
-    alternative_import="from rom langchain_huggingface.llms import HuggingFacePipeline",
+    alternative_import="from langchain_huggingface import HuggingFacePipeline",
 )
 class HuggingFacePipeline(BaseLLM):
     """HuggingFace Pipeline API.
diff --git a/libs/community/langchain_community/llms/huggingface_text_gen_inference.py b/libs/community/langchain_community/llms/huggingface_text_gen_inference.py
index 46af5a83033..e2f6c666c6f 100644
--- a/libs/community/langchain_community/llms/huggingface_text_gen_inference.py
+++ b/libs/community/langchain_community/llms/huggingface_text_gen_inference.py
@@ -14,7 +14,11 @@ from langchain_core.utils import get_pydantic_field_names
 logger = logging.getLogger(__name__)
 
 
-@deprecated("0.0.21", removal="0.3.0", alternative="HuggingFaceEndpoint")
+@deprecated(
+    "0.0.21",
+    removal="0.3.0",
+    alternative_import="from langchain_huggingface import HuggingFaceEndpoint",
+)
 class HuggingFaceTextGenInference(LLM):
     """
     HuggingFace text generation API.