mirror of
https://github.com/hwchase17/langchain.git
synced 2026-02-21 14:43:07 +00:00
9becdeaadf
Add LLM wrappers and examples for Banana, Writer, Modal, Stochastic AI Added rigid json format for Banana and Modal
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",
|
|
]
|