From ce0019b646e4a70ac01d8daa37575bfd364cb3a6 Mon Sep 17 00:00:00 2001
From: Harrison Chase <hw.chase.17@gmail.com>
Date: Tue, 10 Oct 2023 22:11:32 -0700
Subject: [PATCH] make utils conditional (#11646)

---
 libs/langchain/langchain/agents/load_tools.py |   6 +-
 .../langchain/tools/openweathermap/tool.py    |   2 +-
 libs/langchain/langchain/tools/python/tool.py |   2 +-
 .../langchain/langchain/utilities/__init__.py | 303 ++++++++++++++++--
 4 files changed, 275 insertions(+), 38 deletions(-)

diff --git a/libs/langchain/langchain/agents/load_tools.py b/libs/langchain/langchain/agents/load_tools.py
index 115edbd12f..0ff690605c 100644
--- a/libs/langchain/langchain/agents/load_tools.py
+++ b/libs/langchain/langchain/agents/load_tools.py
@@ -43,9 +43,9 @@ from langchain.tools.wolfram_alpha.tool import WolframAlphaQueryRun
 from langchain.tools.openweathermap.tool import OpenWeatherMapQueryRun
 from langchain.tools.dataforseo_api_search import DataForSeoAPISearchRun
 from langchain.tools.dataforseo_api_search import DataForSeoAPISearchResults
-from langchain.utilities import ArxivAPIWrapper
-from langchain.utilities import GoldenQueryAPIWrapper
-from langchain.utilities import PubMedAPIWrapper
+from langchain.utilities.arxiv import ArxivAPIWrapper
+from langchain.utilities.golden_query import GoldenQueryAPIWrapper
+from langchain.utilities.pubmed import PubMedAPIWrapper
 from langchain.utilities.bing_search import BingSearchAPIWrapper
 from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
 from langchain.utilities.google_search import GoogleSearchAPIWrapper
diff --git a/libs/langchain/langchain/tools/openweathermap/tool.py b/libs/langchain/langchain/tools/openweathermap/tool.py
index 4e122b13d2..cf365e8ca8 100644
--- a/libs/langchain/langchain/tools/openweathermap/tool.py
+++ b/libs/langchain/langchain/tools/openweathermap/tool.py
@@ -5,7 +5,7 @@ from typing import Optional
 from langchain.callbacks.manager import CallbackManagerForToolRun
 from langchain.pydantic_v1 import Field
 from langchain.tools.base import BaseTool
-from langchain.utilities import OpenWeatherMapAPIWrapper
+from langchain.utilities.openweathermap import OpenWeatherMapAPIWrapper
 
 
 class OpenWeatherMapQueryRun(BaseTool):
diff --git a/libs/langchain/langchain/tools/python/tool.py b/libs/langchain/langchain/tools/python/tool.py
index 81f9174557..b86f65a42a 100644
--- a/libs/langchain/langchain/tools/python/tool.py
+++ b/libs/langchain/langchain/tools/python/tool.py
@@ -14,7 +14,7 @@ from langchain.callbacks.manager import (
 )
 from langchain.pydantic_v1 import BaseModel, Field, root_validator
 from langchain.tools.base import BaseTool
-from langchain.utilities import PythonREPL
+from langchain.utilities.python import PythonREPL
 
 
 def _get_default_python_repl() -> PythonREPL:
diff --git a/libs/langchain/langchain/utilities/__init__.py b/libs/langchain/langchain/utilities/__init__.py
index 21f8568780..cc4a460acc 100644
--- a/libs/langchain/langchain/utilities/__init__.py
+++ b/libs/langchain/langchain/utilities/__init__.py
@@ -3,40 +3,277 @@
 Other LangChain classes use **Utilities** to interact with third-part systems
 and packages.
 """
-from langchain.utilities.alpha_vantage import AlphaVantageAPIWrapper
-from langchain.utilities.apify import ApifyWrapper
-from langchain.utilities.arcee import ArceeWrapper
-from langchain.utilities.arxiv import ArxivAPIWrapper
-from langchain.utilities.awslambda import LambdaWrapper
-from langchain.utilities.bibtex import BibtexparserWrapper
-from langchain.utilities.bing_search import BingSearchAPIWrapper
-from langchain.utilities.brave_search import BraveSearchWrapper
-from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
-from langchain.utilities.golden_query import GoldenQueryAPIWrapper
-from langchain.utilities.google_places_api import GooglePlacesAPIWrapper
-from langchain.utilities.google_search import GoogleSearchAPIWrapper
-from langchain.utilities.google_serper import GoogleSerperAPIWrapper
-from langchain.utilities.graphql import GraphQLAPIWrapper
-from langchain.utilities.jira import JiraAPIWrapper
-from langchain.utilities.max_compute import MaxComputeAPIWrapper
-from langchain.utilities.metaphor_search import MetaphorSearchAPIWrapper
-from langchain.utilities.openweathermap import OpenWeatherMapAPIWrapper
-from langchain.utilities.portkey import Portkey
-from langchain.utilities.powerbi import PowerBIDataset
-from langchain.utilities.pubmed import PubMedAPIWrapper
-from langchain.utilities.python import PythonREPL
+from typing import Any
+
 from langchain.utilities.requests import Requests, RequestsWrapper, TextRequestsWrapper
-from langchain.utilities.scenexplain import SceneXplainAPIWrapper
-from langchain.utilities.searchapi import SearchApiAPIWrapper
-from langchain.utilities.searx_search import SearxSearchWrapper
-from langchain.utilities.serpapi import SerpAPIWrapper
-from langchain.utilities.spark_sql import SparkSQL
-from langchain.utilities.sql_database import SQLDatabase
-from langchain.utilities.tensorflow_datasets import TensorflowDatasets
-from langchain.utilities.twilio import TwilioAPIWrapper
-from langchain.utilities.wikipedia import WikipediaAPIWrapper
-from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
-from langchain.utilities.zapier import ZapierNLAWrapper
+
+
+def _import_alpha_vantage() -> Any:
+    from langchain.utilities.alpha_vantage import AlphaVantageAPIWrapper
+
+    return AlphaVantageAPIWrapper
+
+
+def _import_apify() -> Any:
+    from langchain.utilities.apify import ApifyWrapper
+
+    return ApifyWrapper
+
+
+def _import_arcee() -> Any:
+    from langchain.utilities.arcee import ArceeWrapper
+
+    return ArceeWrapper
+
+
+def _import_arxiv() -> Any:
+    from langchain.utilities.arxiv import ArxivAPIWrapper
+
+    return ArxivAPIWrapper
+
+
+def _import_awslambda() -> Any:
+    from langchain.utilities.awslambda import LambdaWrapper
+
+    return LambdaWrapper
+
+
+def _import_bibtex() -> Any:
+    from langchain.utilities.bibtex import BibtexparserWrapper
+
+    return BibtexparserWrapper
+
+
+def _import_bing_search() -> Any:
+    from langchain.utilities.bing_search import BingSearchAPIWrapper
+
+    return BingSearchAPIWrapper
+
+
+def _import_brave_search() -> Any:
+    from langchain.utilities.brave_search import BraveSearchWrapper
+
+    return BraveSearchWrapper
+
+
+def _import_duckduckgo_search() -> Any:
+    from langchain.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
+
+    return DuckDuckGoSearchAPIWrapper
+
+
+def _import_golden_query() -> Any:
+    from langchain.utilities.golden_query import GoldenQueryAPIWrapper
+
+    return GoldenQueryAPIWrapper
+
+
+def _import_google_places_api() -> Any:
+    from langchain.utilities.google_places_api import GooglePlacesAPIWrapper
+
+    return GooglePlacesAPIWrapper
+
+
+def _import_google_search() -> Any:
+    from langchain.utilities.google_search import GoogleSearchAPIWrapper
+
+    return GoogleSearchAPIWrapper
+
+
+def _import_google_serper() -> Any:
+    from langchain.utilities.google_serper import GoogleSerperAPIWrapper
+
+    return GoogleSerperAPIWrapper
+
+
+def _import_graphql() -> Any:
+    from langchain.utilities.graphql import GraphQLAPIWrapper
+
+    return GraphQLAPIWrapper
+
+
+def _import_jira() -> Any:
+    from langchain.utilities.jira import JiraAPIWrapper
+
+    return JiraAPIWrapper
+
+
+def _import_max_compute() -> Any:
+    from langchain.utilities.max_compute import MaxComputeAPIWrapper
+
+    return MaxComputeAPIWrapper
+
+
+def _import_metaphor_search() -> Any:
+    from langchain.utilities.metaphor_search import MetaphorSearchAPIWrapper
+
+    return MetaphorSearchAPIWrapper
+
+
+def _import_openweathermap() -> Any:
+    from langchain.utilities.openweathermap import OpenWeatherMapAPIWrapper
+
+    return OpenWeatherMapAPIWrapper
+
+
+def _import_portkey() -> Any:
+    from langchain.utilities.portkey import Portkey
+
+    return Portkey
+
+
+def _import_powerbi() -> Any:
+    from langchain.utilities.powerbi import PowerBIDataset
+
+    return PowerBIDataset
+
+
+def _import_pubmed() -> Any:
+    from langchain.utilities.pubmed import PubMedAPIWrapper
+
+    return PubMedAPIWrapper
+
+
+def _import_python() -> Any:
+    from langchain.utilities.python import PythonREPL
+
+    return PythonREPL
+
+
+def _import_scenexplain() -> Any:
+    from langchain.utilities.scenexplain import SceneXplainAPIWrapper
+
+    return SceneXplainAPIWrapper
+
+
+def _import_searchapi() -> Any:
+    from langchain.utilities.searchapi import SearchApiAPIWrapper
+
+    return SearchApiAPIWrapper
+
+
+def _import_searx_search() -> Any:
+    from langchain.utilities.searx_search import SearxSearchWrapper
+
+    return SearxSearchWrapper
+
+
+def _import_serpapi() -> Any:
+    from langchain.utilities.serpapi import SerpAPIWrapper
+
+    return SerpAPIWrapper
+
+
+def _import_spark_sql() -> Any:
+    from langchain.utilities.spark_sql import SparkSQL
+
+    return SparkSQL
+
+
+def _import_sql_database() -> Any:
+    from langchain.utilities.sql_database import SQLDatabase
+
+    return SQLDatabase
+
+
+def _import_tensorflow_datasets() -> Any:
+    from langchain.utilities.tensorflow_datasets import TensorflowDatasets
+
+    return TensorflowDatasets
+
+
+def _import_twilio() -> Any:
+    from langchain.utilities.twilio import TwilioAPIWrapper
+
+    return TwilioAPIWrapper
+
+
+def _import_wikipedia() -> Any:
+    from langchain.utilities.wikipedia import WikipediaAPIWrapper
+
+    return WikipediaAPIWrapper
+
+
+def _import_wolfram_alpha() -> Any:
+    from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
+
+    return WolframAlphaAPIWrapper
+
+
+def _import_zapier() -> Any:
+    from langchain.utilities.zapier import ZapierNLAWrapper
+
+    return ZapierNLAWrapper
+
+
+def __getattr__(name: str) -> Any:
+    if name == "AlphaVantageAPIWrapper":
+        return _import_alpha_vantage()
+    elif name == "ApifyWrapper":
+        return _import_apify()
+    elif name == "ArxivAPIWrapper":
+        return _import_arxiv()
+    elif name == "LambdaWrapper":
+        return _import_awslambda()
+    elif name == "BibtexparserWrapper":
+        return _import_bibtex()
+    elif name == "BingSearchAPIWrapper":
+        return _import_bing_search()
+    elif name == "BraveSearchWrapper":
+        return _import_brave_search()
+    elif name == "DuckDuckGoSearchAPIWrapper":
+        return _import_duckduckgo_search()
+    elif name == "GoldenQueryAPIWrapper":
+        return _import_golden_query()
+    elif name == "GooglePlacesAPIWrapper":
+        return _import_google_places_api()
+    elif name == "GoogleSearchAPIWrapper":
+        return _import_google_search()
+    elif name == "GoogleSerperAPIWrapper":
+        return _import_google_serper()
+    elif name == "GraphQLAPIWrapper":
+        return _import_graphql()
+    elif name == "JiraAPIWrapper":
+        return _import_jira()
+    elif name == "MaxComputeAPIWrapper":
+        return _import_max_compute()
+    elif name == "MetaphorSearchAPIWrapper":
+        return _import_metaphor_search()
+    elif name == "OpenWeatherMapAPIWrapper":
+        return _import_openweathermap()
+    elif name == "Portkey":
+        return _import_portkey()
+    elif name == "PowerBIDataset":
+        return _import_powerbi()
+    elif name == "PubMedAPIWrapper":
+        return _import_pubmed()
+    elif name == "PythonREPL":
+        return _import_python()
+    elif name == "SceneXplainAPIWrapper":
+        return _import_scenexplain()
+    elif name == "SearchApiAPIWrapper":
+        return _import_searchapi()
+    elif name == "SearxSearchWrapper":
+        return _import_searx_search()
+    elif name == "SerpAPIWrapper":
+        return _import_serpapi()
+    elif name == "SparkSQL":
+        return _import_spark_sql()
+    elif name == "SQLDatabase":
+        return _import_sql_database()
+    elif name == "TensorflowDatasets":
+        return _import_tensorflow_datasets()
+    elif name == "TwilioAPIWrapper":
+        return _import_twilio()
+    elif name == "WikipediaAPIWrapper":
+        return _import_wikipedia()
+    elif name == "WolframAlphaAPIWrapper":
+        return _import_wolfram_alpha()
+    elif name == "ZapierNLAWrapper":
+        return _import_zapier()
+    else:
+        raise AttributeError(f"Could not find: {name}")
+
 
 __all__ = [
     "AlphaVantageAPIWrapper",