Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update deprecated langchain imports with suggested new paths. #12164

Merged
merged 1 commit into from
Oct 23, 2023
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
94 changes: 49 additions & 45 deletions libs/langchain/langchain/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,25 +49,25 @@ def __getattr__(name: str) -> Any:
if name == "MRKLChain":
from langchain.agents import MRKLChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.agents.MRKLChain")

return MRKLChain
elif name == "ReActChain":
from langchain.agents import ReActChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.agents.ReActChain")

return ReActChain
elif name == "SelfAskWithSearchChain":
from langchain.agents import SelfAskWithSearchChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.agents.SelfAskWithSearchChain")

return SelfAskWithSearchChain
elif name == "ConversationChain":
from langchain.chains import ConversationChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.ConversationChain")

return ConversationChain
elif name == "LLMBashChain":
Expand All @@ -83,248 +83,252 @@ def __getattr__(name: str) -> Any:
elif name == "LLMChain":
from langchain.chains import LLMChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.LLMChain")

return LLMChain
elif name == "LLMCheckerChain":
from langchain.chains import LLMCheckerChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.LLMCheckerChain")

return LLMCheckerChain
elif name == "LLMMathChain":
from langchain.chains import LLMMathChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.LLMMathChain")

return LLMMathChain
elif name == "QAWithSourcesChain":
from langchain.chains import QAWithSourcesChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.QAWithSourcesChain")

return QAWithSourcesChain
elif name == "VectorDBQA":
from langchain.chains import VectorDBQA

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.VectorDBQA")

return VectorDBQA
elif name == "VectorDBQAWithSourcesChain":
from langchain.chains import VectorDBQAWithSourcesChain

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.chains.VectorDBQAWithSourcesChain")

return VectorDBQAWithSourcesChain
elif name == "InMemoryDocstore":
from langchain.docstore import InMemoryDocstore

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.docstore.InMemoryDocstore")

return InMemoryDocstore
elif name == "Wikipedia":
from langchain.docstore import Wikipedia

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.docstore.Wikipedia")

return Wikipedia
elif name == "Anthropic":
from langchain.llms import Anthropic

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Anthropic")

return Anthropic
elif name == "Banana":
from langchain.llms import Banana

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Banana")

return Banana
elif name == "CerebriumAI":
from langchain.llms import CerebriumAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.CerebriumAI")

return CerebriumAI
elif name == "Cohere":
from langchain.llms import Cohere

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Cohere")

return Cohere
elif name == "ForefrontAI":
from langchain.llms import ForefrontAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.ForefrontAI")

return ForefrontAI
elif name == "GooseAI":
from langchain.llms import GooseAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.GooseAI")

return GooseAI
elif name == "HuggingFaceHub":
from langchain.llms import HuggingFaceHub

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.HuggingFaceHub")

return HuggingFaceHub
elif name == "HuggingFaceTextGenInference":
from langchain.llms import HuggingFaceTextGenInference

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.HuggingFaceTextGenInference")

return HuggingFaceTextGenInference
elif name == "LlamaCpp":
from langchain.llms import LlamaCpp

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.LlamaCpp")

return LlamaCpp
elif name == "Modal":
from langchain.llms import Modal

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Modal")

return Modal
elif name == "OpenAI":
from langchain.llms import OpenAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.OpenAI")

return OpenAI
elif name == "Petals":
from langchain.llms import Petals

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Petals")

return Petals
elif name == "PipelineAI":
from langchain.llms import PipelineAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.PipelineAI")

return PipelineAI
elif name == "SagemakerEndpoint":
from langchain.llms import SagemakerEndpoint

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.SagemakerEndpoint")

return SagemakerEndpoint
elif name == "StochasticAI":
from langchain.llms import StochasticAI

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.StochasticAI")

return StochasticAI
elif name == "Writer":
from langchain.llms import Writer

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.llms.Writer")

return Writer
elif name == "HuggingFacePipeline":
from langchain.llms.huggingface_pipeline import HuggingFacePipeline

_warn_on_import(name)
_warn_on_import(
name, replacement="langchain.llms.huggingface_pipeline.HuggingFacePipeline"
)

return HuggingFacePipeline
elif name == "FewShotPromptTemplate":
from langchain.prompts import FewShotPromptTemplate

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.prompts.FewShotPromptTemplate")

return FewShotPromptTemplate
elif name == "Prompt":
from langchain.prompts import Prompt

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.prompts.Prompt")

return Prompt
elif name == "PromptTemplate":
from langchain.prompts import PromptTemplate

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.prompts.PromptTemplate")

return PromptTemplate
elif name == "BasePromptTemplate":
from langchain.schema.prompt_template import BasePromptTemplate

_warn_on_import(name)
_warn_on_import(
name, replacement="langchain.schema.prompt_template.BasePromptTemplate"
)

return BasePromptTemplate
elif name == "ArxivAPIWrapper":
from langchain.utilities import ArxivAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.ArxivAPIWrapper")

return ArxivAPIWrapper
elif name == "GoldenQueryAPIWrapper":
from langchain.utilities import GoldenQueryAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.GoldenQueryAPIWrapper")

return GoldenQueryAPIWrapper
elif name == "GoogleSearchAPIWrapper":
from langchain.utilities import GoogleSearchAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.GoogleSearchAPIWrapper")

return GoogleSearchAPIWrapper
elif name == "GoogleSerperAPIWrapper":
from langchain.utilities import GoogleSerperAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.GoogleSerperAPIWrapper")

return GoogleSerperAPIWrapper
elif name == "PowerBIDataset":
from langchain.utilities import PowerBIDataset

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.PowerBIDataset")

return PowerBIDataset
elif name == "SearxSearchWrapper":
from langchain.utilities import SearxSearchWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.SearxSearchWrapper")

return SearxSearchWrapper
elif name == "WikipediaAPIWrapper":
from langchain.utilities import WikipediaAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.WikipediaAPIWrapper")

return WikipediaAPIWrapper
elif name == "WolframAlphaAPIWrapper":
from langchain.utilities import WolframAlphaAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.WolframAlphaAPIWrapper")

return WolframAlphaAPIWrapper
elif name == "SQLDatabase":
from langchain.utilities import SQLDatabase

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.SQLDatabase")

return SQLDatabase
elif name == "FAISS":
from langchain.vectorstores import FAISS

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.vectorstores.FAISS")

return FAISS
elif name == "ElasticVectorSearch":
from langchain.vectorstores import ElasticVectorSearch

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.vectorstores.ElasticVectorSearch")

return ElasticVectorSearch
# For backwards compatibility
elif name == "SerpAPIChain" or name == "SerpAPIWrapper":
from langchain.utilities import SerpAPIWrapper

_warn_on_import(name)
_warn_on_import(name, replacement="langchain.utilities.SerpAPIWrapper")

return SerpAPIWrapper
elif name == "verbose":
Expand Down