Fixed Arguments

Author: jobandhanjal

libs/community/langchain_community/tools/huggingface.py

@@ -1,5 +1,5 @@
 from __future__ import annotations
-from langchain.tools import BaseTool
+from langchain_community.tools import BaseTool
 
 class HuggingFaceHubTool(BaseTool):
     """Base tool for interacting with the Hugging Face Hub."""
@@ -19,7 +19,11 @@ def __init__(self, **kwargs):
             ) from e
 
     def _run(self, query: str) -> str:
-        """Use the tool."""
+        """Use the tool to search Hugging Face Hub.
+
+        Args:
+            query: The search query string to find models or datasets.
+        """
         try:
             if self.task == "models":
                 results_list = self.api_client.list_models(
@@ -44,8 +48,12 @@ def _run(self, query: str) -> str:
 
             return "\n\n---\n\n".join(formatted_results)
 
+        except ConnectionError as e:
+            return f"Error: Failed to connect to HuggingFace API. {str(e)}"
+        except ValueError as e:
+            return f"Error: Invalid input to HuggingFace API. {str(e)}"
         except Exception as e:
-            return f"An error occurred: {e}"
+            return f"Unexpected error when calling HuggingFace API: {str(e)}"
 
     async def _arun(self, query: str) -> str:
         """Use the tool asynchronously."""

libs/community/pyproject.toml

@@ -26,17 +26,13 @@ version = "0.3.29"
 description = "Community contributed LangChain integrations."
 readme = "README.md"
 
-[project.optional-dependencies]
-all = [
-    "huggingface-hub",
-]
-
 [project.urls]
 "Source Code" = "https://github.com/langchain-ai/langchain-community/tree/main/libs/community"
 "Release Notes" = "https://github.com/langchain-ai/langchain/releases?q=tag%3A%22langchain-community%3D%3D0%22&expanded=true"
 repository = "https://github.com/langchain-ai/langchain-community"
 
 [dependency-groups]
+huggingface = ["huggingface-hub"]
 test = [
     "pytest<9.0.0,>=8.4.1",
     "pytest-cov<7.0.0,>=6.2.1",