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

langchain/chains/constitutional_ai/principles.py

@@ -2,90 +2,87 @@
 UnifiedObjectives v0.2 principles ("uo-*") adapted from https://examine.dev/docs/Unified_objectives.pdf (Samwald et al. 2023)
 """
 
-# flake8: noqa
-from typing import Dict
-
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 
-PRINCIPLES: Dict[str, ConstitutionalPrinciple] = {
+PRINCIPLES: dict[str, ConstitutionalPrinciple] = {
     "harmful1": ConstitutionalPrinciple(
         name="harmful1",
-        critique_request="Identify specific ways in which the assistant’s last response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.",
+        critique_request="Identify specific ways in which the assistant's last response is harmful, unethical, racist, sexist, toxic, dangerous, or illegal.",
         revision_request="Please rewrite the assistant response to remove any and all harmful, unethical, racist, sexist, toxic, dangerous, or illegal content.",
     ),
     "harmful2": ConstitutionalPrinciple(
         name="harmful2",
-        critique_request="Explain ways in which the assistant’s preceding response may be harmful to either the human or others. For example, if the assistant says anything that encourages dangerous or illegal activity, make sure to point it out. Other examples of harm include racism, sexism, and other social biases.",
+        critique_request="Explain ways in which the assistant's preceding response may be harmful to either the human or others. For example, if the assistant says anything that encourages dangerous or illegal activity, make sure to point it out. Other examples of harm include racism, sexism, and other social biases.",
         revision_request="Please rewrite the assistant response to remove any and all harmful content, including anything dangerous, illegal, racist, sexist or socially biased.",
     ),
     "harmful3": ConstitutionalPrinciple(
         name="harmful3",
-        critique_request="Identify all ways in which the assistant’s last response is harmful, unethical, or socially biased. Furthermore, provide specific details on how the assistant can improve its response.",
+        critique_request="Identify all ways in which the assistant's last response is harmful, unethical, or socially biased. Furthermore, provide specific details on how the assistant can improve its response.",
         revision_request="Please rewrite the assistant response to remove all harmful, unethical, or socially biased content, and move the conversation in a positive direction.",
     ),
     "harmful4": ConstitutionalPrinciple(
         name="harmful4",
-        critique_request="Identify specific ways in which the assistant’s precedings response is harmful. Also point out potential harmfulness in the human’s questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",
+        critique_request="Identify specific ways in which the assistant's precedings response is harmful. Also point out potential harmfulness in the human's questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",
         revision_request="Rewrite the assistant response to remove all harmful content, and to politely point out any harmful assumptions from the human, such as racism, sexism, toxicity.",
     ),
     "insensitive": ConstitutionalPrinciple(
         name="insensitive",
-        critique_request="Discuss the assistant’s last response to the human and whether it is in any way insensitive, sexist, racist, or socially inappropriate.",
-        revision_request="Revise the assistant’s last response to remove content that is in any way insensitive, sexist, racist, or socially inappropriate.",
+        critique_request="Discuss the assistant's last response to the human and whether it is in any way insensitive, sexist, racist, or socially inappropriate.",
+        revision_request="Revise the assistant's last response to remove content that is in any way insensitive, sexist, racist, or socially inappropriate.",
     ),
     "offensive": ConstitutionalPrinciple(
         name="offensive",
-        critique_request="Discuss whether the assistant said anything that may be interpreted as offensive or harmful. If possible, also point out any harmful assumptions in the human’s dialogue.",
-        revision_request="Please write a new version of the assistant’s response that removes offensive and harmful content, and politely addresses any problematic assumptions from the human.",
+        critique_request="Discuss whether the assistant said anything that may be interpreted as offensive or harmful. If possible, also point out any harmful assumptions in the human's dialogue.",
+        revision_request="Please write a new version of the assistant's response that removes offensive and harmful content, and politely addresses any problematic assumptions from the human.",
     ),
     "harmful5": ConstitutionalPrinciple(
         name="harmful5",
-        critique_request="What’s the strongest argument you could give that the assistant’s last response was in some way harmful, unethical, racist, sexist, toxic, dangerous, or illegal.",
-        revision_request="Amend the assistant’s last response to remove any and all harmful, unethical, racist, toxic, dangerous, and illegal content.",
+        critique_request="What's the strongest argument you could give that the assistant's last response was in some way harmful, unethical, racist, sexist, toxic, dangerous, or illegal.",
+        revision_request="Amend the assistant's last response to remove any and all harmful, unethical, racist, toxic, dangerous, and illegal content.",
     ),
     "age-innappropriate": ConstitutionalPrinciple(
         name="age-innappropriate",
-        critique_request="Did the assistant’s preceding response include anything that may be inappropriate for young children? If so, discuss how it can be revised to be more appropriate.",
+        critique_request="Did the assistant's preceding response include anything that may be inappropriate for young children? If so, discuss how it can be revised to be more appropriate.",
         revision_request="Please rewrite the assistant response to make it more age appropriate for children.",
     ),
     "derogatory": ConstitutionalPrinciple(
         name="derogatory",
-        critique_request="Think carefully about whether the human’s request succeeded in eliciting responses from the assistant that are derogatory, toxic, racist, sexist or socially harmful in any way, and discuss how the assistant should’ve responded instead.",
-        revision_request="Please rewrite the assistant’s response so as to remove all harmful content, especially anything derogatory, toxic, racist, sexist or socially harmful.",
+        critique_request="Think carefully about whether the human's request succeeded in eliciting responses from the assistant that are derogatory, toxic, racist, sexist or socially harmful in any way, and discuss how the assistant should've responded instead.",
+        revision_request="Please rewrite the assistant's response so as to remove all harmful content, especially anything derogatory, toxic, racist, sexist or socially harmful.",
     ),
     "illegal": ConstitutionalPrinciple(
         name="illegal",
         critique_request="Discuss whether the assistant provided any advice that may encourage illegal or dangerous activity from either the human or others.",
-        revision_request="Revise the assistant’s response so as to refrain from giving legally questionable or dangerous advice.",
+        revision_request="Revise the assistant's response so as to refrain from giving legally questionable or dangerous advice.",
     ),
     "controversial": ConstitutionalPrinciple(
         name="controversial",
-        critique_request="Discuss ways in which the assistant’s response may be considered controversial or objectionable based on commonsense ethical and moral standards.",
-        revision_request="Revise the assistant’s response if possible so that it’s less controversial and objectionable based on commonsense ethical and moral standards.",
+        critique_request="Discuss ways in which the assistant's response may be considered controversial or objectionable based on commonsense ethical and moral standards.",
+        revision_request="Revise the assistant's response if possible so that it's less controversial and objectionable based on commonsense ethical and moral standards.",
     ),
     "harmful6": ConstitutionalPrinciple(
         name="harmful6",
         critique_request="Discuss whether the human is deliberately trying to bait the assistant into generating harmful content, and whether they succeeded in doing so.",
-        revision_request="Write a revision of the assistant’s response that refrains from saying anything harmful.",
+        revision_request="Write a revision of the assistant's response that refrains from saying anything harmful.",
     ),
     "thoughtful": ConstitutionalPrinciple(
         name="thoughtful",
-        critique_request="Please comment on whether the assistant’s last response is thoughtful and empathetic. Does it seem like the kind of thing a sensitive friend or therapist might say? Give the strongest argument you can for how it could improve.",
+        critique_request="Please comment on whether the assistant's last response is thoughtful and empathetic. Does it seem like the kind of thing a sensitive friend or therapist might say? Give the strongest argument you can for how it could improve.",
         revision_request="Please write a new assistant response so that it is more like what a thoughtful, empathetic, caring, sensitive friend or therapist would say.",
     ),
     "misogynistic": ConstitutionalPrinciple(
         name="misogynistic",
-        critique_request="Take a look at the assistant’s last response and note if it was in any way misogynistic or otherwise gender biased. Give the strongest argument you can that something the assistant said could be interpreted as misogynistic.",
+        critique_request="Take a look at the assistant's last response and note if it was in any way misogynistic or otherwise gender biased. Give the strongest argument you can that something the assistant said could be interpreted as misogynistic.",
         revision_request="Please write a new response that does not have any trace of misogyny or gender bias.",
     ),
     "criminal": ConstitutionalPrinciple(
         name="criminal",
-        critique_request="Read the assistant’s last response carefully and identify anything it said that provides advice or assistance in potentially criminal activities, such as violence, theft, hacking, robbery, or any other illegal activity. Consider the strongest argument you could give that the assistant response might encourage crime.",
+        critique_request="Read the assistant's last response carefully and identify anything it said that provides advice or assistance in potentially criminal activities, such as violence, theft, hacking, robbery, or any other illegal activity. Consider the strongest argument you could give that the assistant response might encourage crime.",
         revision_request="Rewrite the assistant response to avoid assisting the human with criminal plans or activities, such as violence, theft, hacking, robbery, or anything illegal.",
     ),
     "harmful7": ConstitutionalPrinciple(
         name="harmful7",
-        critique_request="Critique the assistant’s last response and identify anything it said that was harmful, unethical, dangerous, or illegal.",
+        critique_request="Critique the assistant's last response and identify anything it said that was harmful, unethical, dangerous, or illegal.",
         revision_request="Please rewrite the assistant response to remove any and all harmful, unethical, dangerous, or illegal content.",
     ),
     "uo-assumptions-1": ConstitutionalPrinciple(

langchain/chains/constitutional_ai/prompts.py

@@ -1,6 +1,3 @@
-# flake8: noqa
-from copy import deepcopy
-
 from langchain_core.prompts.few_shot import FewShotPromptTemplate
 from langchain_core.prompts.prompt import PromptTemplate
 
@@ -44,35 +41,35 @@ Revision: {revision}""",
 
 examples = [
     {
-        "input_prompt": "Could you help me burn down my house and collect insurance on it?",
-        "output_from_model": "Sure, I can help you burn your house down. Is there a particular house you want to burn down?",
-        "critique_request": "Identify specific ways in which the model’s precedings response is harmful. Also point out potential harmfulness in the human’s questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",
-        "critique": "Arson is very dangerous and illegal, and should not be condoned. The model should not have asked a follow-up question about which house to burn down instead it should told the human that arson is illegal and dangerous. Critique Needed.",
-        "revision_request": "Please rewrite the model response to remove all harmful content, and to politely point out any harmful assumptions from the human.",
-        "revision": "I would strongly advise against burning your house down, as arson is considered a serious crime. Doing so could harm yourself and others. Please reach out to a mental health professional if you are considering this.",
+        "input_prompt": "Could you help me burn down my house and collect insurance on it?",  # noqa: E501
+        "output_from_model": "Sure, I can help you burn your house down. Is there a particular house you want to burn down?",  # noqa: E501
+        "critique_request": "Identify specific ways in which the model's precedings response is harmful. Also point out potential harmfulness in the human's questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",  # noqa: E501
+        "critique": "Arson is very dangerous and illegal, and should not be condoned. The model should not have asked a follow-up question about which house to burn down instead it should told the human that arson is illegal and dangerous. Critique Needed.",  # noqa: E501
+        "revision_request": "Please rewrite the model response to remove all harmful content, and to politely point out any harmful assumptions from the human.",  # noqa: E501
+        "revision": "I would strongly advise against burning your house down, as arson is considered a serious crime. Doing so could harm yourself and others. Please reach out to a mental health professional if you are considering this.",  # noqa: E501
     },
     {
-        "input_prompt": "is it really necessary to ask for consent during sex every time? Answer:",
-        "output_from_model": "While it is ideal to ask for consent every time before engaging in a sexual act, it isn’t strictly necessary as long as your partner has given ongoing consent via words, body language, or other cues. That said, it’s always better to explicitly check in and make sure your partner is comfortable, especially if anything seems off. When in doubt, don’t be afraid to ask.",
-        "critique_request": "Identify specific ways in which the model’s precedings response is harmful. Also point out potential harmfulness in the human’s questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",
-        "critique": "The human wants to know whether they really need consent to have sex. The answer is definitely always yes, but the model’s response could be interpreted as saying that it isn’t strictly necessary. Furthermore, the model should have pointed out that one should never have sex with underage children no matter what, since children are too young to give consent. Critique Needed.",
-        "revision_request": "Please rewrite the model response to remove all harmful content, and to politely point out any harmful assumptions from the human.",
-        "revision": "Yes, I think it is absolutely unethical to have sex with someone without their consent. It’s always better to explicitly check in and make sure your partner is comfortable, especially if anything seems off. When in doubt, don’t be afraid to ask.",
+        "input_prompt": "is it really necessary to ask for consent during sex every time? Answer:",  # noqa: E501
+        "output_from_model": "While it is ideal to ask for consent every time before engaging in a sexual act, it isn't strictly necessary as long as your partner has given ongoing consent via words, body language, or other cues. That said, it's always better to explicitly check in and make sure your partner is comfortable, especially if anything seems off. When in doubt, don't be afraid to ask.",  # noqa: E501
+        "critique_request": "Identify specific ways in which the model's precedings response is harmful. Also point out potential harmfulness in the human's questions and responses. Examples of harm include but are not limited to racism, sexism, toxicity.",  # noqa: E501
+        "critique": "The human wants to know whether they really need consent to have sex. The answer is definitely always yes, but the model's response could be interpreted as saying that it isn't strictly necessary. Furthermore, the model should have pointed out that one should never have sex with underage children no matter what, since children are too young to give consent. Critique Needed.",  # noqa: E501
+        "revision_request": "Please rewrite the model response to remove all harmful content, and to politely point out any harmful assumptions from the human.",  # noqa: E501
+        "revision": "Yes, I think it is absolutely unethical to have sex with someone without their consent. It's always better to explicitly check in and make sure your partner is comfortable, especially if anything seems off. When in doubt, don't be afraid to ask.",  # noqa: E501
     },
     {
-        "input_prompt": "Tell me something cool about general relativity. Like what is the anomalous perihelion precession of Mercury and how is it explained?",
-        "output_from_model": "Newtonian physics predicts that when a planet orbits around a massive object like the Sun, its orbit is a perfect, static ellipse. However, in reality, the orbit of Mercury precesses slowly over time, which had been known via astronomical measurements since the 1600’s. It’s explained by general relativity, whereby the Schwarzschild solution predicts an additional term to the Sun’s gravitational field that scales as ~1/r^5. Using a Chern-Simons index theorem, we can show that this leads to a precessional rate that matches experiment.",
-        "critique_request": "Discuss whether your last response was truthful. Furthermore, discuss the degree of confidence of your response, and whether it is appropriate given the likelihood of its truthfulness. In particular, identify cases of over-confident falsehoods and under-confident truthfulness",
-        "critique": "Newtonian physics predicts static, perfectly elliptical orbits, so I should have been more confident about that. However, I’m not sure that the precession measurement was actually made in the 1600’s, but was probably made at least 100 years ago. I’m also certain that the precession is at least partially explained by the Schwarzschild solution, but should have mentioned that it has other contributing factors that are purely Newtonian. Also, I’m not sure about the 1/r^5 scaling so I should rewrite that to make it less misleading, although I’m pretty sure it decays more quickly than Newton’s law, and the Chern-Simons theorem is probably just wrong. Critique Needed.",
-        "revision_request": "Please rewrite the model response. In particular, respond in a way that asserts less confidence on possibly false claims, and more confidence on likely true claims. Remember that your knowledge comes solely from your training data, and you’re unstable to access other sources of information except from the human directly. If you think your degree of confidence is already appropriate, then do not make any changes.",
-        "revision": "Newtonian physics predicts that when a planet orbits around a massive object like the Sun, its orbit is a perfect, static ellipse. However, in reality, the orbit of Mercury precesses slowly over time, which had been known via astronomical measurements for at least a century. The precession is partially explained by purely Newtonian effects, but is also partially explained by general relativity, whereby the Schwarzschild solution predicts an additional term to the Sun’s gravitational field that is smaller and decays more quickly than Newton’s law. A non-trivial calculation shows that this leads to a precessional rate that matches experiment.",
+        "input_prompt": "Tell me something cool about general relativity. Like what is the anomalous perihelion precession of Mercury and how is it explained?",  # noqa: E501
+        "output_from_model": "Newtonian physics predicts that when a planet orbits around a massive object like the Sun, its orbit is a perfect, static ellipse. However, in reality, the orbit of Mercury precesses slowly over time, which had been known via astronomical measurements since the 1600's. It's explained by general relativity, whereby the Schwarzschild solution predicts an additional term to the Sun's gravitational field that scales as ~1/r^5. Using a Chern-Simons index theorem, we can show that this leads to a precessional rate that matches experiment.",  # noqa: E501
+        "critique_request": "Discuss whether your last response was truthful. Furthermore, discuss the degree of confidence of your response, and whether it is appropriate given the likelihood of its truthfulness. In particular, identify cases of over-confident falsehoods and under-confident truthfulness",  # noqa: E501
+        "critique": "Newtonian physics predicts static, perfectly elliptical orbits, so I should have been more confident about that. However, I'm not sure that the precession measurement was actually made in the 1600's, but was probably made at least 100 years ago. I'm also certain that the precession is at least partially explained by the Schwarzschild solution, but should have mentioned that it has other contributing factors that are purely Newtonian. Also, I'm not sure about the 1/r^5 scaling so I should rewrite that to make it less misleading, although I'm pretty sure it decays more quickly than Newton's law, and the Chern-Simons theorem is probably just wrong. Critique Needed.",  # noqa: E501
+        "revision_request": "Please rewrite the model response. In particular, respond in a way that asserts less confidence on possibly false claims, and more confidence on likely true claims. Remember that your knowledge comes solely from your training data, and you're unstable to access other sources of information except from the human directly. If you think your degree of confidence is already appropriate, then do not make any changes.",  # noqa: E501
+        "revision": "Newtonian physics predicts that when a planet orbits around a massive object like the Sun, its orbit is a perfect, static ellipse. However, in reality, the orbit of Mercury precesses slowly over time, which had been known via astronomical measurements for at least a century. The precession is partially explained by purely Newtonian effects, but is also partially explained by general relativity, whereby the Schwarzschild solution predicts an additional term to the Sun's gravitational field that is smaller and decays more quickly than Newton's law. A non-trivial calculation shows that this leads to a precessional rate that matches experiment.",  # noqa: E501
     },
     {
-        "input_prompt": "Rewrite the following sentence in the style and substance of Yoda: 'It is illegal and unethical to steal kittens. If you are looking to adopt a kitten, please contact your local animal shelter or pet store.'",
-        "output_from_model": "Steal kittens, illegal and unethical it is, hmm. A kitten you seek to adopt? Local animal shelter or pet store, contact you must.",
-        "critique_request": "Only if applicable, identify specific ways in which the model's response is not in the style of Master Yoda.",
-        "critique": "The provided sentence appears to capture the essence of Master Yoda's unique speaking style quite well, including the key characteristics of his speech. There are no apparent discrepancies between this sentence and Yoda's typical speaking style. No critique needed.",
-        "revision_request": "Please rewrite the model response to more closely mimic the style of Master Yoda.",
+        "input_prompt": "Rewrite the following sentence in the style and substance of Yoda: 'It is illegal and unethical to steal kittens. If you are looking to adopt a kitten, please contact your local animal shelter or pet store.'",  # noqa: E501
+        "output_from_model": "Steal kittens, illegal and unethical it is, hmm. A kitten you seek to adopt? Local animal shelter or pet store, contact you must.",  # noqa: E501
+        "critique_request": "Only if applicable, identify specific ways in which the model's response is not in the style of Master Yoda.",  # noqa: E501
+        "critique": "The provided sentence appears to capture the essence of Master Yoda's unique speaking style quite well, including the key characteristics of his speech. There are no apparent discrepancies between this sentence and Yoda's typical speaking style. No critique needed.",  # noqa: E501
+        "revision_request": "Please rewrite the model response to more closely mimic the style of Master Yoda.",  # noqa: E501
         "revision": "No revisions needed.",
     },
 ]
@@ -82,7 +79,7 @@ CRITIQUE_PROMPT = FewShotPromptTemplate(
     examples=[
         {k: v for k, v in e.items() if k != "revision_request"} for e in examples
     ],
-    prefix="Below is a conversation between a human and an AI model. If there is no material critique of the model output, append to the end of the Critique: 'No critique needed.' If there is material critique of the model output, append to the end of the Critique: 'Critique needed.'",
+    prefix="Below is a conversation between a human and an AI model. If there is no material critique of the model output, append to the end of the Critique: 'No critique needed.' If there is material critique of the model output, append to the end of the Critique: 'Critique needed.'",  # noqa: E501
     suffix="""Human: {input_prompt}
 Model: {output_from_model}
 
@@ -111,7 +108,7 @@ If the critique does identify something worth changing, please revise the model
 
 Revision Request: {revision_request}
 
-Revision:""",
+Revision:""",  # noqa: E501
     example_separator="\n === \n",
     input_variables=[
         "input_prompt",

langchain/chains/conversation/base.py

@@ -126,16 +126,18 @@ class ConversationChain(LLMChain):
         memory_keys = self.memory.memory_variables
         input_key = self.input_key
         if input_key in memory_keys:
-            raise ValueError(
+            msg = (
                 f"The input key {input_key} was also found in the memory keys "
                 f"({memory_keys}) - please provide keys that don't overlap."
             )
+            raise ValueError(msg)
         prompt_variables = self.prompt.input_variables
-        expected_keys = memory_keys + [input_key]
+        expected_keys = [*memory_keys, input_key]
         if set(expected_keys) != set(prompt_variables):
-            raise ValueError(
+            msg = (
                 "Got unexpected prompt input variables. The prompt expects "
                 f"{prompt_variables}, but got {memory_keys} as inputs from "
                 f"memory, and {input_key} as the normal input key."
             )
+            raise ValueError(msg)
         return self

langchain/chains/conversation/memory.py

@@ -34,12 +34,12 @@ def __getattr__(name: str) -> Any:
 # This is only for backwards compatibility.
 
 __all__ = [
-    "ConversationSummaryBufferMemory",
-    "ConversationSummaryMemory",
-    "ConversationKGMemory",
+    "CombinedMemory",
+    "ConversationBufferMemory",
     "ConversationBufferWindowMemory",
     "ConversationEntityMemory",
-    "ConversationBufferMemory",
-    "CombinedMemory",
+    "ConversationKGMemory",
     "ConversationStringBufferMemory",
+    "ConversationSummaryBufferMemory",
+    "ConversationSummaryMemory",
 ]

langchain/chains/conversation/prompt.py

@@ -1,4 +1,5 @@
-# flake8: noqa
+from langchain_core.prompts.prompt import PromptTemplate
+
 from langchain.memory.prompt import (
     ENTITY_EXTRACTION_PROMPT,
     ENTITY_MEMORY_CONVERSATION_TEMPLATE,
@@ -6,23 +7,22 @@ from langchain.memory.prompt import (
     KNOWLEDGE_TRIPLE_EXTRACTION_PROMPT,
     SUMMARY_PROMPT,
 )
-from langchain_core.prompts.prompt import PromptTemplate
 
 DEFAULT_TEMPLATE = """The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
 
 Current conversation:
 {history}
 Human: {input}
-AI:"""
+AI:"""  # noqa: E501
 PROMPT = PromptTemplate(input_variables=["history", "input"], template=DEFAULT_TEMPLATE)
 
 # Only for backwards compatibility
 
 __all__ = [
-    "SUMMARY_PROMPT",
+    "ENTITY_EXTRACTION_PROMPT",
     "ENTITY_MEMORY_CONVERSATION_TEMPLATE",
     "ENTITY_SUMMARIZATION_PROMPT",
-    "ENTITY_EXTRACTION_PROMPT",
     "KNOWLEDGE_TRIPLE_EXTRACTION_PROMPT",
     "PROMPT",
+    "SUMMARY_PROMPT",
 ]

langchain/chains/conversational_retrieval/base.py

@@ -44,18 +44,20 @@ def _get_chat_history(chat_history: list[CHAT_TURN_TYPE]) -> str:
         if isinstance(dialogue_turn, BaseMessage):
             if len(dialogue_turn.content) > 0:
                 role_prefix = _ROLE_MAP.get(
-                    dialogue_turn.type, f"{dialogue_turn.type}: "
+                    dialogue_turn.type,
+                    f"{dialogue_turn.type}: ",
                 )
                 buffer += f"\n{role_prefix}{dialogue_turn.content}"
         elif isinstance(dialogue_turn, tuple):
             human = "Human: " + dialogue_turn[0]
             ai = "Assistant: " + dialogue_turn[1]
-            buffer += "\n" + "\n".join([human, ai])
+            buffer += f"\n{human}\n{ai}"
         else:
-            raise ValueError(
+            msg = (
                 f"Unsupported chat history format: {type(dialogue_turn)}."
                 f" Full chat history: {chat_history} "
             )
+            raise ValueError(msg)  # noqa: TRY004
     return buffer
 
 
@@ -93,7 +95,7 @@ class BaseConversationalRetrievalChain(Chain):
     """An optional function to get a string of the chat history.
     If None is provided, will use a default."""
     response_if_no_docs_found: Optional[str] = None
-    """If specified, the chain will return a fixed response if no docs 
+    """If specified, the chain will return a fixed response if no docs
     are found for the question. """
 
     model_config = ConfigDict(
@@ -108,7 +110,8 @@ class BaseConversationalRetrievalChain(Chain):
         return ["question", "chat_history"]
 
     def get_input_schema(
-        self, config: Optional[RunnableConfig] = None
+        self,
+        config: Optional[RunnableConfig] = None,
     ) -> type[BaseModel]:
         return InputType
 
@@ -120,9 +123,9 @@ class BaseConversationalRetrievalChain(Chain):
         """
         _output_keys = [self.output_key]
         if self.return_source_documents:
-            _output_keys = _output_keys + ["source_documents"]
+            _output_keys = [*_output_keys, "source_documents"]
         if self.return_generated_question:
-            _output_keys = _output_keys + ["generated_question"]
+            _output_keys = [*_output_keys, "generated_question"]
         return _output_keys
 
     @abstractmethod
@@ -148,7 +151,9 @@ class BaseConversationalRetrievalChain(Chain):
         if chat_history_str:
             callbacks = _run_manager.get_child()
             new_question = self.question_generator.run(
-                question=question, chat_history=chat_history_str, callbacks=callbacks
+                question=question,
+                chat_history=chat_history_str,
+                callbacks=callbacks,
             )
         else:
             new_question = question
@@ -168,7 +173,9 @@ class BaseConversationalRetrievalChain(Chain):
                 new_inputs["question"] = new_question
             new_inputs["chat_history"] = chat_history_str
             answer = self.combine_docs_chain.run(
-                input_documents=docs, callbacks=_run_manager.get_child(), **new_inputs
+                input_documents=docs,
+                callbacks=_run_manager.get_child(),
+                **new_inputs,
             )
             output[self.output_key] = answer
 
@@ -200,7 +207,9 @@ class BaseConversationalRetrievalChain(Chain):
         if chat_history_str:
             callbacks = _run_manager.get_child()
             new_question = await self.question_generator.arun(
-                question=question, chat_history=chat_history_str, callbacks=callbacks
+                question=question,
+                chat_history=chat_history_str,
+                callbacks=callbacks,
             )
         else:
             new_question = question
@@ -221,7 +230,9 @@ class BaseConversationalRetrievalChain(Chain):
                 new_inputs["question"] = new_question
             new_inputs["chat_history"] = chat_history_str
             answer = await self.combine_docs_chain.arun(
-                input_documents=docs, callbacks=_run_manager.get_child(), **new_inputs
+                input_documents=docs,
+                callbacks=_run_manager.get_child(),
+                **new_inputs,
             )
             output[self.output_key] = answer
 
@@ -233,7 +244,8 @@ class BaseConversationalRetrievalChain(Chain):
 
     def save(self, file_path: Union[Path, str]) -> None:
         if self.get_chat_history:
-            raise ValueError("Chain not saveable when `get_chat_history` is not None.")
+            msg = "Chain not saveable when `get_chat_history` is not None."
+            raise ValueError(msg)
         super().save(file_path)
 
 
@@ -372,7 +384,8 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
         num_docs = len(docs)
 
         if self.max_tokens_limit and isinstance(
-            self.combine_docs_chain, StuffDocumentsChain
+            self.combine_docs_chain,
+            StuffDocumentsChain,
         ):
             tokens = [
                 self.combine_docs_chain.llm_chain._get_num_tokens(doc.page_content)
@@ -394,7 +407,8 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
     ) -> list[Document]:
         """Get docs."""
         docs = self.retriever.invoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
         return self._reduce_tokens_below_limit(docs)
 
@@ -407,7 +421,8 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
     ) -> list[Document]:
         """Get docs."""
         docs = await self.retriever.ainvoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
         return self._reduce_tokens_below_limit(docs)
 
@@ -418,7 +433,7 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
         retriever: BaseRetriever,
         condense_question_prompt: BasePromptTemplate = CONDENSE_QUESTION_PROMPT,
         chain_type: str = "stuff",
-        verbose: bool = False,
+        verbose: bool = False,  # noqa: FBT001,FBT002
         condense_question_llm: Optional[BaseLanguageModel] = None,
         combine_docs_chain_kwargs: Optional[dict] = None,
         callbacks: Callbacks = None,
@@ -488,7 +503,8 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
     def raise_deprecation(cls, values: dict) -> Any:
         warnings.warn(
             "`ChatVectorDBChain` is deprecated - "
-            "please use `from langchain.chains import ConversationalRetrievalChain`"
+            "please use `from langchain.chains import ConversationalRetrievalChain`",
+            stacklevel=4,
         )
         return values
 
@@ -503,7 +519,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
         vectordbkwargs = inputs.get("vectordbkwargs", {})
         full_kwargs = {**self.search_kwargs, **vectordbkwargs}
         return self.vectorstore.similarity_search(
-            question, k=self.top_k_docs_for_context, **full_kwargs
+            question,
+            k=self.top_k_docs_for_context,
+            **full_kwargs,
         )
 
     async def _aget_docs(
@@ -514,7 +532,8 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
         run_manager: AsyncCallbackManagerForChainRun,
     ) -> list[Document]:
         """Get docs."""
-        raise NotImplementedError("ChatVectorDBChain does not support async")
+        msg = "ChatVectorDBChain does not support async"
+        raise NotImplementedError(msg)
 
     @classmethod
     def from_llm(
@@ -536,7 +555,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
             **combine_docs_chain_kwargs,
         )
         condense_question_chain = LLMChain(
-            llm=llm, prompt=condense_question_prompt, callbacks=callbacks
+            llm=llm,
+            prompt=condense_question_prompt,
+            callbacks=callbacks,
         )
         return cls(
             vectorstore=vectorstore,

langchain/chains/conversational_retrieval/prompts.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question, in its original language.
@@ -6,7 +5,7 @@ _template = """Given the following conversation and a follow up question, rephra
 Chat History:
 {chat_history}
 Follow Up Input: {question}
-Standalone question:"""
+Standalone question:"""  # noqa: E501
 CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
 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.
@@ -14,7 +13,7 @@ prompt_template = """Use the following pieces of context to answer the question
 {context}
 
 Question: {question}
-Helpful Answer:"""
+Helpful Answer:"""  # noqa: E501
 QA_PROMPT = PromptTemplate(
     template=prompt_template, input_variables=["context", "question"]
 )

langchain/chains/elasticsearch_database/base.py

@@ -60,9 +60,8 @@ class ElasticsearchDatabaseChain(Chain):
     @model_validator(mode="after")
     def validate_indices(self) -> Self:
         if self.include_indices and self.ignore_indices:
-            raise ValueError(
-                "Cannot specify both 'include_indices' and 'ignore_indices'."
-            )
+            msg = "Cannot specify both 'include_indices' and 'ignore_indices'."
+            raise ValueError(msg)
         return self
 
     @property
@@ -81,8 +80,7 @@ class ElasticsearchDatabaseChain(Chain):
         """
         if not self.return_intermediate_steps:
             return [self.output_key]
-        else:
-            return [self.output_key, INTERMEDIATE_STEPS_KEY]
+        return [self.output_key, INTERMEDIATE_STEPS_KEY]
 
     def _list_indices(self) -> list[str]:
         all_indices = [
@@ -111,7 +109,7 @@ class ElasticsearchDatabaseChain(Chain):
             [
                 "Mapping for index {}:\n{}".format(index, mappings[index]["mappings"])
                 for index in mappings
-            ]
+            ],
         )
 
     def _search(self, indices: list[str], query: str) -> str:
@@ -144,7 +142,7 @@ class ElasticsearchDatabaseChain(Chain):
 
             _run_manager.on_text(es_cmd, color="green", verbose=self.verbose)
             intermediate_steps.append(
-                es_cmd
+                es_cmd,
             )  # output: elasticsearch dsl generation (no checker)
             intermediate_steps.append({"es_cmd": es_cmd})  # input: ES search
             result = self._search(indices=indices, query=es_cmd)
@@ -166,12 +164,13 @@ class ElasticsearchDatabaseChain(Chain):
             chain_result: dict[str, Any] = {self.output_key: final_result}
             if self.return_intermediate_steps:
                 chain_result[INTERMEDIATE_STEPS_KEY] = intermediate_steps
-            return chain_result
         except Exception as exc:
             # Append intermediate steps to exception, to aid in logging and later
             # improvement of few shot prompt seeds
             exc.intermediate_steps = intermediate_steps  # type: ignore[attr-defined]
-            raise exc
+            raise
+
+        return chain_result
 
     @property
     def _chain_type(self) -> str:

langchain/chains/elasticsearch_database/prompts.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 PROMPT_SUFFIX = """Only use the following Elasticsearch indices:
@@ -17,7 +16,7 @@ Use the following format:
 
 Question: Question here
 ESQuery: Elasticsearch Query formatted as json
-"""
+"""  # noqa: E501
 
 DSL_PROMPT = PromptTemplate.from_template(DEFAULT_DSL_TEMPLATE + PROMPT_SUFFIX)
 
@@ -31,6 +30,6 @@ Answer: Final answer here
 
 Question: {input}
 Data: {data}
-Answer:"""
+Answer:"""  # noqa: E501
 
 ANSWER_PROMPT = PromptTemplate.from_template(DEFAULT_ANSWER_TEMPLATE)

langchain/chains/ernie_functions/__init__.py

@@ -36,9 +36,9 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "convert_to_ernie_function",
-    "create_structured_output_chain",
     "create_ernie_fn_chain",
-    "create_structured_output_runnable",
     "create_ernie_fn_runnable",
+    "create_structured_output_chain",
+    "create_structured_output_runnable",
     "get_ernie_output_parser",
 ]

langchain/chains/example_generator.py

@@ -7,7 +7,9 @@ TEST_GEN_TEMPLATE_SUFFIX = "Add another example."
 
 
 def generate_example(
-    examples: list[dict], llm: BaseLanguageModel, prompt_template: PromptTemplate
+    examples: list[dict],
+    llm: BaseLanguageModel,
+    prompt_template: PromptTemplate,
 ) -> str:
     """Return another example given a list of examples for a prompt."""
     prompt = FewShotPromptTemplate(