diff --git a/langchain/chains/__init__.py b/langchain/chains/__init__.py index ccc8a2a9..46011f37 100644 --- a/langchain/chains/__init__.py +++ b/langchain/chains/__init__.py @@ -2,6 +2,10 @@ from langchain.chains.api.base import APIChain from langchain.chains.api.openapi.chain import OpenAPIEndpointChain from langchain.chains.combine_documents.base import AnalyzeDocumentChain +from langchain.chains.combine_documents.map_reduce import MapReduceDocumentsChain +from langchain.chains.combine_documents.map_rerank import MapRerankDocumentsChain +from langchain.chains.combine_documents.refine import RefineDocumentsChain +from langchain.chains.combine_documents.stuff import StuffDocumentsChain from langchain.chains.constitutional_ai.base import ConstitutionalChain from langchain.chains.conversation.base import ConversationChain from langchain.chains.conversational_retrieval.base import ( @@ -95,4 +99,8 @@ __all__ = [ "create_tagging_chain_pydantic", "load_chain", "create_citation_fuzzy_match_chain", + "StuffDocumentsChain", + "MapRerankDocumentsChain", + "MapReduceDocumentsChain", + "RefineDocumentsChain", ]