langchain 0.3.26__py3-none-any.whl → 0.3.27__py3-none-any.whl

langchain/chains/question_answering/chain.py

@@ -29,13 +29,16 @@ class LoadingCallable(Protocol):
     """Interface for loading the combine documents chain."""
 
     def __call__(
-        self, llm: BaseLanguageModel, **kwargs: Any
+        self,
+        llm: BaseLanguageModel,
+        **kwargs: Any,
     ) -> BaseCombineDocumentsChain:
         """Callable to load the combine documents chain."""
 
 
 def _load_map_rerank_chain(
     llm: BaseLanguageModel,
+    *,
     prompt: BasePromptTemplate = MAP_RERANK_PROMPT,
     verbose: bool = False,
     document_variable_name: str = "context",
@@ -65,6 +68,7 @@ def _load_map_rerank_chain(
 
 def _load_stuff_chain(
     llm: BaseLanguageModel,
+    *,
     prompt: Optional[BasePromptTemplate] = None,
     document_variable_name: str = "context",
     verbose: Optional[bool] = None,
@@ -93,6 +97,7 @@ def _load_stuff_chain(
 
 def _load_map_reduce_chain(
     llm: BaseLanguageModel,
+    *,
     question_prompt: Optional[BasePromptTemplate] = None,
     combine_prompt: Optional[BasePromptTemplate] = None,
     combine_document_variable_name: str = "summaries",
@@ -138,10 +143,11 @@ def _load_map_reduce_chain(
     if collapse_prompt is None:
         collapse_chain = None
         if collapse_llm is not None:
-            raise ValueError(
+            msg = (
                 "collapse_llm provided, but collapse_prompt was not: please "
                 "provide one or stop providing collapse_llm."
             )
+            raise ValueError(msg)
     else:
         _collapse_llm = collapse_llm or llm
         collapse_chain = StuffDocumentsChain(
@@ -175,6 +181,7 @@ def _load_map_reduce_chain(
 
 def _load_refine_chain(
     llm: BaseLanguageModel,
+    *,
     question_prompt: Optional[BasePromptTemplate] = None,
     refine_prompt: Optional[BasePromptTemplate] = None,
     document_variable_name: str = "context_str",
@@ -189,7 +196,7 @@ def _load_refine_chain(
         question_prompt or refine_prompts.QUESTION_PROMPT_SELECTOR.get_prompt(llm)
     )
     _refine_prompt = refine_prompt or refine_prompts.REFINE_PROMPT_SELECTOR.get_prompt(
-        llm
+        llm,
     )
     initial_chain = LLMChain(
         llm=llm,
@@ -224,10 +231,10 @@ def _load_refine_chain(
     message=(
         "This class is deprecated. See the following migration guides for replacements "
         "based on `chain_type`:\n"
-        "stuff: https://python.langchain.com/docs/versions/migrating_chains/stuff_docs_chain\n"  # noqa: E501
-        "map_reduce: https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain\n"  # noqa: E501
-        "refine: https://python.langchain.com/docs/versions/migrating_chains/refine_chain\n"  # noqa: E501
-        "map_rerank: https://python.langchain.com/docs/versions/migrating_chains/map_rerank_docs_chain\n"  # noqa: E501
+        "stuff: https://python.langchain.com/docs/versions/migrating_chains/stuff_docs_chain\n"
+        "map_reduce: https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain\n"
+        "refine: https://python.langchain.com/docs/versions/migrating_chains/refine_chain\n"
+        "map_rerank: https://python.langchain.com/docs/versions/migrating_chains/map_rerank_docs_chain\n"
         "\nSee also guides on retrieval and question-answering here: "
         "https://python.langchain.com/docs/how_to/#qa-with-rag"
     ),
@@ -235,7 +242,7 @@ def _load_refine_chain(
 def load_qa_chain(
     llm: BaseLanguageModel,
     chain_type: str = "stuff",
-    verbose: Optional[bool] = None,
+    verbose: Optional[bool] = None,  # noqa: FBT001
     callback_manager: Optional[BaseCallbackManager] = None,
     **kwargs: Any,
 ) -> BaseCombineDocumentsChain:
@@ -259,10 +266,14 @@ def load_qa_chain(
         "map_rerank": _load_map_rerank_chain,
     }
     if chain_type not in loader_mapping:
-        raise ValueError(
+        msg = (
             f"Got unsupported chain type: {chain_type}. "
             f"Should be one of {loader_mapping.keys()}"
         )
+        raise ValueError(msg)
     return loader_mapping[chain_type](
-        llm, verbose=verbose, callback_manager=callback_manager, **kwargs
+        llm,
+        verbose=verbose,
+        callback_manager=callback_manager,
+        **kwargs,
     )

langchain/chains/question_answering/map_reduce_prompt.py

@@ -1,5 +1,3 @@
-# flake8: noqa
-from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
 from langchain_core.prompts.chat import (
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
@@ -7,18 +5,20 @@ from langchain_core.prompts.chat import (
 )
 from langchain_core.prompts.prompt import PromptTemplate
 
-question_prompt_template = """Use the following portion of a long document to see if any of the text is relevant to answer the question. 
+from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
+
+question_prompt_template = """Use the following portion of a long document to see if any of the text is relevant to answer the question.
 Return any relevant text verbatim.
 {context}
 Question: {question}
-Relevant text, if any:"""
+Relevant text, if any:"""  # noqa: E501
 QUESTION_PROMPT = PromptTemplate(
     template=question_prompt_template, input_variables=["context", "question"]
 )
-system_template = """Use the following portion of a long document to see if any of the text is relevant to answer the question. 
+system_template = """Use the following portion of a long document to see if any of the text is relevant to answer the question.
 Return any relevant text verbatim.
 ______________________
-{context}"""
+{context}"""  # noqa: E501
 messages = [
     SystemMessagePromptTemplate.from_template(system_template),
     HumanMessagePromptTemplate.from_template("{question}"),
@@ -30,7 +30,7 @@ QUESTION_PROMPT_SELECTOR = ConditionalPromptSelector(
     default_prompt=QUESTION_PROMPT, conditionals=[(is_chat_model, CHAT_QUESTION_PROMPT)]
 )
 
-combine_prompt_template = """Given the following extracted parts of a long document and a question, create a final answer. 
+combine_prompt_template = """Given the following extracted parts of a long document and a question, create a final answer.
 If you don't know the answer, just say that you don't know. Don't try to make up an answer.
 
 QUESTION: Which state/country's law governs the interpretation of the contract?
@@ -45,13 +45,13 @@ FINAL ANSWER: This Agreement is governed by English law.
 
 QUESTION: What did the president say about Michael Jackson?
 =========
-Content: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  \n\nLast year COVID-19 kept us apart. This year we are finally together again. \n\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \n\nWith a duty to one another to the American people to the Constitution. \n\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \n\nSix days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \n\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \n\nHe met the Ukrainian people. \n\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \n\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.
+Content: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  \n\nLast year COVID-19 kept us apart. This year we are finally together again. \n\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \n\nWith a duty to one another to the American people to the Constitution. \n\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \n\nSix days ago, Russia's Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \n\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \n\nHe met the Ukrainian people. \n\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \n\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.
 
-Content: And we won’t stop. \n\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \n\nLet’s use this moment to reset. Let’s stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease.  \n\nLet’s stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans.  \n\nWe can’t change how divided we’ve been. But we can change how we move forward—on COVID-19 and other issues we must face together. \n\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \n\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \n\nOfficer Mora was 27 years old. \n\nOfficer Rivera was 22. \n\nBoth Dominican Americans who’d grown up on the same streets they later chose to patrol as police officers. \n\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.
+Content: And we won't stop. \n\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \n\nLet's use this moment to reset. Let's stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease.  \n\nLet's stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans.  \n\nWe can't change how divided we've been. But we can change how we move forward—on COVID-19 and other issues we must face together. \n\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \n\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \n\nOfficer Mora was 27 years old. \n\nOfficer Rivera was 22. \n\nBoth Dominican Americans who'd grown up on the same streets they later chose to patrol as police officers. \n\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.
 
-Content: And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  \n\nTo all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. \n\nAnd I’m taking robust action to make sure the pain of our sanctions  is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. \n\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  \n\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  \n\nThese steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. \n\nBut I want you to know that we are going to be okay.
+Content: And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  \n\nTo all Americans, I will be honest with you, as I've always promised. A Russian dictator, invading a foreign country, has costs around the world. \n\nAnd I'm taking robust action to make sure the pain of our sanctions  is targeted at Russia's economy. And I will use every tool at our disposal to protect American businesses and consumers. \n\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  \n\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  \n\nThese steps will help blunt gas prices here at home. And I know the news about what's happening can seem alarming. \n\nBut I want you to know that we are going to be okay.
 
-Content: More support for patients and families. \n\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \n\nIt’s based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more.  \n\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer’s, diabetes, and more. \n\nA unity agenda for the nation. \n\nWe can do this. \n\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \n\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \n\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \n\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \n\nNow is the hour. \n\nOur moment of responsibility. \n\nOur test of resolve and conscience, of history itself. \n\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \n\nWell I know this nation.
+Content: More support for patients and families. \n\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \n\nIt's based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more.  \n\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer's, diabetes, and more. \n\nA unity agenda for the nation. \n\nWe can do this. \n\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \n\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \n\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \n\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \n\nNow is the hour. \n\nOur moment of responsibility. \n\nOur test of resolve and conscience, of history itself. \n\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \n\nWell I know this nation.
 =========
 FINAL ANSWER: The president did not mention Michael Jackson.
 
@@ -59,15 +59,15 @@ QUESTION: {question}
 =========
 {summaries}
 =========
-FINAL ANSWER:"""
+FINAL ANSWER:"""  # noqa: E501
 COMBINE_PROMPT = PromptTemplate(
     template=combine_prompt_template, input_variables=["summaries", "question"]
 )
 
-system_template = """Given the following extracted parts of a long document and a question, create a final answer. 
+system_template = """Given the following extracted parts of a long document and a question, create a final answer.
 If you don't know the answer, just say that you don't know. Don't try to make up an answer.
 ______________________
-{summaries}"""
+{summaries}"""  # noqa: E501
 messages = [
     SystemMessagePromptTemplate.from_template(system_template),
     HumanMessagePromptTemplate.from_template("{question}"),

langchain/chains/question_answering/map_rerank_prompt.py

@@ -1,7 +1,7 @@
-# flake8: noqa
-from langchain.output_parsers.regex import RegexParser
 from langchain_core.prompts import PromptTemplate
 
+from langchain.output_parsers.regex import RegexParser
+
 output_parser = RegexParser(
     regex=r"(.*?)\nScore: (\d*)",
     output_keys=["answer", "score"],
@@ -58,7 +58,7 @@ Context:
 {context}
 ---------
 Question: {question}
-Helpful Answer:"""
+Helpful Answer:"""  # noqa: E501
 PROMPT = PromptTemplate(
     template=prompt_template,
     input_variables=["context", "question"],

langchain/chains/question_answering/refine_prompts.py

@@ -1,13 +1,10 @@
-# flake8: noqa
-from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
 from langchain_core.prompts.chat import (
-    AIMessagePromptTemplate,
     ChatPromptTemplate,
-    HumanMessagePromptTemplate,
-    SystemMessagePromptTemplate,
 )
 from langchain_core.prompts.prompt import PromptTemplate
 
+from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
+
 DEFAULT_REFINE_PROMPT_TMPL = (
     "The original question is as follows: {question}\n"
     "We have provided an existing answer: {existing_answer}\n"

langchain/chains/question_answering/stuff_prompt.py

@@ -1,5 +1,3 @@
-# flake8: noqa
-from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
 from langchain_core.prompts import PromptTemplate
 from langchain_core.prompts.chat import (
     ChatPromptTemplate,
@@ -7,20 +5,22 @@ from langchain_core.prompts.chat import (
     SystemMessagePromptTemplate,
 )
 
+from langchain.chains.prompt_selector import ConditionalPromptSelector, is_chat_model
+
 prompt_template = """Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.
 
 {context}
 
 Question: {question}
-Helpful Answer:"""
+Helpful Answer:"""  # noqa: E501
 PROMPT = PromptTemplate(
     template=prompt_template, input_variables=["context", "question"]
 )
 
-system_template = """Use the following pieces of context to answer the user's question. 
+system_template = """Use the following pieces of context to answer the user's question.
 If you don't know the answer, just say that you don't know, don't try to make up an answer.
 ----------------
-{context}"""
+{context}"""  # noqa: E501
 messages = [
     SystemMessagePromptTemplate.from_template(system_template),
     HumanMessagePromptTemplate.from_template("{question}"),

langchain/chains/retrieval.py

@@ -60,10 +60,8 @@ def create_retrieval_chain(
     else:
         retrieval_docs = (lambda x: x["input"]) | retriever
 
-    retrieval_chain = (
+    return (
         RunnablePassthrough.assign(
             context=retrieval_docs.with_config(run_name="retrieve_documents"),
         ).assign(answer=combine_docs_chain)
     ).with_config(run_name="retrieval_chain")
-
-    return retrieval_chain

langchain/chains/retrieval_qa/base.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import inspect
-import warnings
 from abc import abstractmethod
 from typing import Any, Optional
 
@@ -69,7 +68,7 @@ class BaseRetrievalQA(Chain):
         """
         _output_keys = [self.output_key]
         if self.return_source_documents:
-            _output_keys = _output_keys + ["source_documents"]
+            _output_keys = [*_output_keys, "source_documents"]
         return _output_keys
 
     @classmethod
@@ -84,10 +83,14 @@ class BaseRetrievalQA(Chain):
         """Initialize from LLM."""
         _prompt = prompt or PROMPT_SELECTOR.get_prompt(llm)
         llm_chain = LLMChain(
-            llm=llm, prompt=_prompt, callbacks=callbacks, **(llm_chain_kwargs or {})
+            llm=llm,
+            prompt=_prompt,
+            callbacks=callbacks,
+            **(llm_chain_kwargs or {}),
         )
         document_prompt = PromptTemplate(
-            input_variables=["page_content"], template="Context:\n{page_content}"
+            input_variables=["page_content"],
+            template="Context:\n{page_content}",
         )
         combine_documents_chain = StuffDocumentsChain(
             llm_chain=llm_chain,
@@ -113,7 +116,9 @@ class BaseRetrievalQA(Chain):
         """Load chain from chain type."""
         _chain_type_kwargs = chain_type_kwargs or {}
         combine_documents_chain = load_qa_chain(
-            llm, chain_type=chain_type, **_chain_type_kwargs
+            llm,
+            chain_type=chain_type,
+            **_chain_type_kwargs,
         )
         return cls(combine_documents_chain=combine_documents_chain, **kwargs)
 
@@ -152,13 +157,14 @@ class BaseRetrievalQA(Chain):
         else:
             docs = self._get_docs(question)  # type: ignore[call-arg]
         answer = self.combine_documents_chain.run(
-            input_documents=docs, question=question, callbacks=_run_manager.get_child()
+            input_documents=docs,
+            question=question,
+            callbacks=_run_manager.get_child(),
         )
 
         if self.return_source_documents:
             return {self.output_key: answer, "source_documents": docs}
-        else:
-            return {self.output_key: answer}
+        return {self.output_key: answer}
 
     @abstractmethod
     async def _aget_docs(
@@ -195,13 +201,14 @@ class BaseRetrievalQA(Chain):
         else:
             docs = await self._aget_docs(question)  # type: ignore[call-arg]
         answer = await self.combine_documents_chain.arun(
-            input_documents=docs, question=question, callbacks=_run_manager.get_child()
+            input_documents=docs,
+            question=question,
+            callbacks=_run_manager.get_child(),
         )
 
         if self.return_source_documents:
             return {self.output_key: answer, "source_documents": docs}
-        else:
-            return {self.output_key: answer}
+        return {self.output_key: answer}
 
 
 @deprecated(
@@ -269,7 +276,8 @@ class RetrievalQA(BaseRetrievalQA):
     ) -> list[Document]:
         """Get docs."""
         return self.retriever.invoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
 
     async def _aget_docs(
@@ -280,7 +288,8 @@ class RetrievalQA(BaseRetrievalQA):
     ) -> list[Document]:
         """Get docs."""
         return await self.retriever.ainvoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
 
     @property
@@ -310,15 +319,6 @@ class VectorDBQA(BaseRetrievalQA):
     search_kwargs: dict[str, Any] = Field(default_factory=dict)
     """Extra search args."""
 
-    @model_validator(mode="before")
-    @classmethod
-    def raise_deprecation(cls, values: dict) -> Any:
-        warnings.warn(
-            "`VectorDBQA` is deprecated - "
-            "please use `from langchain.chains import RetrievalQA`"
-        )
-        return values
-
     @model_validator(mode="before")
     @classmethod
     def validate_search_type(cls, values: dict) -> Any:
@@ -326,7 +326,8 @@ class VectorDBQA(BaseRetrievalQA):
         if "search_type" in values:
             search_type = values["search_type"]
             if search_type not in ("similarity", "mmr"):
-                raise ValueError(f"search_type of {search_type} not allowed.")
+                msg = f"search_type of {search_type} not allowed."
+                raise ValueError(msg)
         return values
 
     def _get_docs(
@@ -338,14 +339,19 @@ class VectorDBQA(BaseRetrievalQA):
         """Get docs."""
         if self.search_type == "similarity":
             docs = self.vectorstore.similarity_search(
-                question, k=self.k, **self.search_kwargs
+                question,
+                k=self.k,
+                **self.search_kwargs,
             )
         elif self.search_type == "mmr":
             docs = self.vectorstore.max_marginal_relevance_search(
-                question, k=self.k, **self.search_kwargs
+                question,
+                k=self.k,
+                **self.search_kwargs,
             )
         else:
-            raise ValueError(f"search_type of {self.search_type} not allowed.")
+            msg = f"search_type of {self.search_type} not allowed."
+            raise ValueError(msg)
         return docs
 
     async def _aget_docs(
@@ -355,7 +361,8 @@ class VectorDBQA(BaseRetrievalQA):
         run_manager: AsyncCallbackManagerForChainRun,
     ) -> list[Document]:
         """Get docs."""
-        raise NotImplementedError("VectorDBQA does not support async")
+        msg = "VectorDBQA does not support async"
+        raise NotImplementedError(msg)
 
     @property
     def _chain_type(self) -> str:

langchain/chains/retrieval_qa/prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts import PromptTemplate
 
 prompt_template = """Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.
@@ -6,7 +5,7 @@ prompt_template = """Use the following pieces of context to answer the question
 {context}
 
 Question: {question}
-Helpful Answer:"""
+Helpful Answer:"""  # noqa: E501
 PROMPT = PromptTemplate(
     template=prompt_template, input_variables=["context", "question"]
 )

langchain/chains/router/__init__.py

@@ -4,9 +4,9 @@ from langchain.chains.router.multi_prompt import MultiPromptChain
 from langchain.chains.router.multi_retrieval_qa import MultiRetrievalQAChain
 
 __all__ = [
-    "RouterChain",
-    "MultiRouteChain",
+    "LLMRouterChain",
     "MultiPromptChain",
     "MultiRetrievalQAChain",
-    "LLMRouterChain",
+    "MultiRouteChain",
+    "RouterChain",
 ]

langchain/chains/router/base.py

@@ -43,7 +43,9 @@ class RouterChain(Chain, ABC):
         return Route(result["destination"], result["next_inputs"])
 
     async def aroute(
-        self, inputs: dict[str, Any], callbacks: Callbacks = None
+        self,
+        inputs: dict[str, Any],
+        callbacks: Callbacks = None,
     ) -> Route:
         result = await self.acall(inputs, callbacks=callbacks)
         return Route(result["destination"], result["next_inputs"])
@@ -59,7 +61,7 @@ class MultiRouteChain(Chain):
     default_chain: Chain
     """Default chain to use when none of the destination chains are suitable."""
     silent_errors: bool = False
-    """If True, use default_chain when an invalid destination name is provided. 
+    """If True, use default_chain when an invalid destination name is provided.
     Defaults to False."""
 
     model_config = ConfigDict(
@@ -93,20 +95,20 @@ class MultiRouteChain(Chain):
         route = self.router_chain.route(inputs, callbacks=callbacks)
 
         _run_manager.on_text(
-            str(route.destination) + ": " + str(route.next_inputs), verbose=self.verbose
+            str(route.destination) + ": " + str(route.next_inputs),
+            verbose=self.verbose,
         )
         if not route.destination:
             return self.default_chain(route.next_inputs, callbacks=callbacks)
-        elif route.destination in self.destination_chains:
+        if route.destination in self.destination_chains:
             return self.destination_chains[route.destination](
-                route.next_inputs, callbacks=callbacks
+                route.next_inputs,
+                callbacks=callbacks,
             )
-        elif self.silent_errors:
+        if self.silent_errors:
             return self.default_chain(route.next_inputs, callbacks=callbacks)
-        else:
-            raise ValueError(
-                f"Received invalid destination chain name '{route.destination}'"
-            )
+        msg = f"Received invalid destination chain name '{route.destination}'"
+        raise ValueError(msg)
 
     async def _acall(
         self,
@@ -118,21 +120,23 @@ class MultiRouteChain(Chain):
         route = await self.router_chain.aroute(inputs, callbacks=callbacks)
 
         await _run_manager.on_text(
-            str(route.destination) + ": " + str(route.next_inputs), verbose=self.verbose
+            str(route.destination) + ": " + str(route.next_inputs),
+            verbose=self.verbose,
         )
         if not route.destination:
             return await self.default_chain.acall(
-                route.next_inputs, callbacks=callbacks
+                route.next_inputs,
+                callbacks=callbacks,
             )
-        elif route.destination in self.destination_chains:
+        if route.destination in self.destination_chains:
             return await self.destination_chains[route.destination].acall(
-                route.next_inputs, callbacks=callbacks
+                route.next_inputs,
+                callbacks=callbacks,
             )
-        elif self.silent_errors:
+        if self.silent_errors:
             return await self.default_chain.acall(
-                route.next_inputs, callbacks=callbacks
-            )
-        else:
-            raise ValueError(
-                f"Received invalid destination chain name '{route.destination}'"
+                route.next_inputs,
+                callbacks=callbacks,
             )
+        msg = f"Received invalid destination chain name '{route.destination}'"
+        raise ValueError(msg)

langchain/chains/router/embedding_router.py

@@ -63,10 +63,12 @@ class EmbeddingRouterChain(RouterChain):
         """Convenience constructor."""
         documents = []
         for name, descriptions in names_and_descriptions:
-            for description in descriptions:
-                documents.append(
+            documents.extend(
+                [
                     Document(page_content=description, metadata={"name": name})
-                )
+                    for description in descriptions
+                ]
+            )
         vectorstore = vectorstore_cls.from_documents(documents, embeddings)
         return cls(vectorstore=vectorstore, **kwargs)
 
@@ -80,10 +82,12 @@ class EmbeddingRouterChain(RouterChain):
     ) -> EmbeddingRouterChain:
         """Convenience constructor."""
         documents = []
-        for name, descriptions in names_and_descriptions:
-            for description in descriptions:
-                documents.append(
-                    Document(page_content=description, metadata={"name": name})
-                )
+        documents.extend(
+            [
+                Document(page_content=description, metadata={"name": name})
+                for name, descriptions in names_and_descriptions
+                for description in descriptions
+            ]
+        )
         vectorstore = await vectorstore_cls.afrom_documents(documents, embeddings)
         return cls(vectorstore=vectorstore, **kwargs)