make warnings more modular (#11047)

libs/langchain/langchain/__init__.py

@@ -20,194 +20,289 @@ debug: bool = False
 llm_cache: Optional["BaseCache"] = None
 
 
-def __getattr__(name: str) -> Any:
+def _warn_on_import(name: str) -> None:
     warnings.warn(
         f"Importing {name} from langchain root module is no longer supported."
     )
+
+
+def __getattr__(name: str) -> Any:
     if name == "MRKLChain":
         from langchain.agents import MRKLChain
 
+        _warn_on_import(name)
+
         return MRKLChain
     elif name == "ReActChain":
         from langchain.agents import ReActChain
 
+        _warn_on_import(name)
+
         return ReActChain
     elif name == "SelfAskWithSearchChain":
         from langchain.agents import SelfAskWithSearchChain
 
+        _warn_on_import(name)
+
         return SelfAskWithSearchChain
     elif name == "ConversationChain":
         from langchain.chains import ConversationChain
 
+        _warn_on_import(name)
+
         return ConversationChain
     elif name == "LLMBashChain":
         from langchain.chains import LLMBashChain
 
+        _warn_on_import(name)
+
         return LLMBashChain
     elif name == "LLMChain":
         from langchain.chains import LLMChain
 
+        _warn_on_import(name)
+
         return LLMChain
     elif name == "LLMCheckerChain":
         from langchain.chains import LLMCheckerChain
 
+        _warn_on_import(name)
+
         return LLMCheckerChain
     elif name == "LLMMathChain":
         from langchain.chains import LLMMathChain
 
+        _warn_on_import(name)
+
         return LLMMathChain
     elif name == "QAWithSourcesChain":
         from langchain.chains import QAWithSourcesChain
 
+        _warn_on_import(name)
+
         return QAWithSourcesChain
     elif name == "VectorDBQA":
         from langchain.chains import VectorDBQA
 
+        _warn_on_import(name)
+
         return VectorDBQA
     elif name == "VectorDBQAWithSourcesChain":
         from langchain.chains import VectorDBQAWithSourcesChain
 
+        _warn_on_import(name)
+
         return VectorDBQAWithSourcesChain
     elif name == "InMemoryDocstore":
         from langchain.docstore import InMemoryDocstore
 
+        _warn_on_import(name)
+
         return InMemoryDocstore
     elif name == "Wikipedia":
         from langchain.docstore import Wikipedia
 
+        _warn_on_import(name)
+
         return Wikipedia
     elif name == "Anthropic":
         from langchain.llms import Anthropic
 
+        _warn_on_import(name)
+
         return Anthropic
     elif name == "Banana":
         from langchain.llms import Banana
 
+        _warn_on_import(name)
+
         return Banana
     elif name == "CerebriumAI":
         from langchain.llms import CerebriumAI
 
+        _warn_on_import(name)
+
         return CerebriumAI
     elif name == "Cohere":
         from langchain.llms import Cohere
 
+        _warn_on_import(name)
+
         return Cohere
     elif name == "ForefrontAI":
         from langchain.llms import ForefrontAI
 
+        _warn_on_import(name)
+
         return ForefrontAI
     elif name == "GooseAI":
         from langchain.llms import GooseAI
 
+        _warn_on_import(name)
+
         return GooseAI
     elif name == "HuggingFaceHub":
         from langchain.llms import HuggingFaceHub
 
+        _warn_on_import(name)
+
         return HuggingFaceHub
     elif name == "HuggingFaceTextGenInference":
         from langchain.llms import HuggingFaceTextGenInference
 
+        _warn_on_import(name)
+
         return HuggingFaceTextGenInference
     elif name == "LlamaCpp":
         from langchain.llms import LlamaCpp
 
+        _warn_on_import(name)
+
         return LlamaCpp
     elif name == "Modal":
         from langchain.llms import Modal
 
+        _warn_on_import(name)
+
         return Modal
     elif name == "OpenAI":
         from langchain.llms import OpenAI
 
+        _warn_on_import(name)
+
         return OpenAI
     elif name == "Petals":
         from langchain.llms import Petals
 
+        _warn_on_import(name)
+
         return Petals
     elif name == "PipelineAI":
         from langchain.llms import PipelineAI
 
+        _warn_on_import(name)
+
         return PipelineAI
     elif name == "SagemakerEndpoint":
         from langchain.llms import SagemakerEndpoint
 
+        _warn_on_import(name)
+
         return SagemakerEndpoint
     elif name == "StochasticAI":
         from langchain.llms import StochasticAI
 
+        _warn_on_import(name)
+
         return StochasticAI
     elif name == "Writer":
         from langchain.llms import Writer
 
+        _warn_on_import(name)
+
         return Writer
     elif name == "HuggingFacePipeline":
         from langchain.llms.huggingface_pipeline import HuggingFacePipeline
 
+        _warn_on_import(name)
+
         return HuggingFacePipeline
     elif name == "FewShotPromptTemplate":
         from langchain.prompts import FewShotPromptTemplate
 
+        _warn_on_import(name)
+
         return FewShotPromptTemplate
     elif name == "Prompt":
         from langchain.prompts import Prompt
 
+        _warn_on_import(name)
+
         return Prompt
     elif name == "PromptTemplate":
         from langchain.prompts import PromptTemplate
 
+        _warn_on_import(name)
+
         return PromptTemplate
     elif name == "BasePromptTemplate":
         from langchain.schema.prompt_template import BasePromptTemplate
 
+        _warn_on_import(name)
+
         return BasePromptTemplate
     elif name == "ArxivAPIWrapper":
         from langchain.utilities import ArxivAPIWrapper
 
+        _warn_on_import(name)
+
         return ArxivAPIWrapper
     elif name == "GoldenQueryAPIWrapper":
         from langchain.utilities import GoldenQueryAPIWrapper
 
+        _warn_on_import(name)
+
         return GoldenQueryAPIWrapper
     elif name == "GoogleSearchAPIWrapper":
         from langchain.utilities import GoogleSearchAPIWrapper
 
+        _warn_on_import(name)
+
         return GoogleSearchAPIWrapper
     elif name == "GoogleSerperAPIWrapper":
         from langchain.utilities import GoogleSerperAPIWrapper
 
+        _warn_on_import(name)
+
         return GoogleSerperAPIWrapper
     elif name == "PowerBIDataset":
         from langchain.utilities import PowerBIDataset
 
+        _warn_on_import(name)
+
         return PowerBIDataset
     elif name == "SearxSearchWrapper":
         from langchain.utilities import SearxSearchWrapper
 
+        _warn_on_import(name)
+
         return SearxSearchWrapper
     elif name == "WikipediaAPIWrapper":
         from langchain.utilities import WikipediaAPIWrapper
 
+        _warn_on_import(name)
+
         return WikipediaAPIWrapper
     elif name == "WolframAlphaAPIWrapper":
         from langchain.utilities import WolframAlphaAPIWrapper
 
+        _warn_on_import(name)
+
         return WolframAlphaAPIWrapper
     elif name == "SQLDatabase":
         from langchain.utilities import SQLDatabase
 
+        _warn_on_import(name)
+
         return SQLDatabase
     elif name == "FAISS":
         from langchain.vectorstores import FAISS
 
+        _warn_on_import(name)
+
         return FAISS
     elif name == "ElasticVectorSearch":
         from langchain.vectorstores import ElasticVectorSearch
 
+        _warn_on_import(name)
+
         return ElasticVectorSearch
     # For backwards compatibility
     elif name == "SerpAPIChain":
         from langchain.utilities import SerpAPIWrapper
 
+        _warn_on_import(name)
+
         return SerpAPIWrapper
     else:
         raise AttributeError(f"Could not find: {name}")