forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.4 KiB
Python
106 lines
2.4 KiB
Python
"""Main entrypoint into package."""
|
|
|
|
from typing import Optional
|
|
|
|
from langchain.agents import MRKLChain, ReActChain, SelfAskWithSearchChain
|
|
from langchain.cache import BaseCache
|
|
from langchain.callbacks import (
|
|
set_default_callback_manager,
|
|
set_handler,
|
|
set_tracing_callback_manager,
|
|
)
|
|
from langchain.chains import (
|
|
ConversationChain,
|
|
LLMBashChain,
|
|
LLMChain,
|
|
LLMCheckerChain,
|
|
LLMMathChain,
|
|
PALChain,
|
|
QAWithSourcesChain,
|
|
SQLDatabaseChain,
|
|
VectorDBQA,
|
|
VectorDBQAWithSourcesChain,
|
|
)
|
|
from langchain.docstore import InMemoryDocstore, Wikipedia
|
|
from langchain.llms import (
|
|
Anthropic,
|
|
Banana,
|
|
CerebriumAI,
|
|
Cohere,
|
|
ForefrontAI,
|
|
GooseAI,
|
|
HuggingFaceHub,
|
|
Modal,
|
|
OpenAI,
|
|
Petals,
|
|
StochasticAI,
|
|
Writer,
|
|
)
|
|
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
|
from langchain.prompts import (
|
|
BasePromptTemplate,
|
|
FewShotPromptTemplate,
|
|
Prompt,
|
|
PromptTemplate,
|
|
)
|
|
from langchain.sql_database import SQLDatabase
|
|
from langchain.utilities.google_search import GoogleSearchAPIWrapper
|
|
from langchain.utilities.google_serper import GoogleSerperAPIWrapper
|
|
from langchain.utilities.searx_search import SearxSearchWrapper
|
|
from langchain.utilities.serpapi import SerpAPIWrapper
|
|
from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
|
|
from langchain.vectorstores import FAISS, ElasticVectorSearch
|
|
|
|
verbose: bool = False
|
|
llm_cache: Optional[BaseCache] = None
|
|
set_default_callback_manager()
|
|
|
|
# For backwards compatibility
|
|
SerpAPIChain = SerpAPIWrapper
|
|
|
|
__all__ = [
|
|
"LLMChain",
|
|
"LLMBashChain",
|
|
"LLMCheckerChain",
|
|
"LLMMathChain",
|
|
"SelfAskWithSearchChain",
|
|
"SerpAPIWrapper",
|
|
"SerpAPIChain",
|
|
"SearxSearchWrapper",
|
|
"GoogleSearchAPIWrapper",
|
|
"GoogleSerperAPIWrapper",
|
|
"WolframAlphaAPIWrapper",
|
|
"Anthropic",
|
|
"Banana",
|
|
"CerebriumAI",
|
|
"Cohere",
|
|
"ForefrontAI",
|
|
"GooseAI",
|
|
"Modal",
|
|
"OpenAI",
|
|
"Petals",
|
|
"StochasticAI",
|
|
"Writer",
|
|
"BasePromptTemplate",
|
|
"Prompt",
|
|
"FewShotPromptTemplate",
|
|
"PromptTemplate",
|
|
"ReActChain",
|
|
"Wikipedia",
|
|
"HuggingFaceHub",
|
|
"HuggingFacePipeline",
|
|
"SQLDatabase",
|
|
"SQLDatabaseChain",
|
|
"FAISS",
|
|
"MRKLChain",
|
|
"VectorDBQA",
|
|
"ElasticVectorSearch",
|
|
"InMemoryDocstore",
|
|
"ConversationChain",
|
|
"VectorDBQAWithSourcesChain",
|
|
"QAWithSourcesChain",
|
|
"PALChain",
|
|
"set_handler",
|
|
"set_tracing_callback_manager",
|
|
]
|