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

langchain/chains/router/llm_router.py

@@ -27,7 +27,7 @@ from langchain.chains.router.base import RouterChain
     message=(
         "Use RunnableLambda to select from multiple prompt templates. See example "
         "in API reference: "
-        "https://api.python.langchain.com/en/latest/chains/langchain.chains.router.llm_router.LLMRouterChain.html"  # noqa: E501
+        "https://api.python.langchain.com/en/latest/chains/langchain.chains.router.llm_router.LLMRouterChain.html"
     ),
 )
 class LLMRouterChain(RouterChain):
@@ -105,12 +105,13 @@ class LLMRouterChain(RouterChain):
     def validate_prompt(self) -> Self:
         prompt = self.llm_chain.prompt
         if prompt.output_parser is None:
-            raise ValueError(
+            msg = (
                 "LLMRouterChain requires base llm_chain prompt to have an output"
                 " parser that converts LLM text output to a dictionary with keys"
                 " 'destination' and 'next_inputs'. Received a prompt with no output"
                 " parser."
             )
+            raise ValueError(msg)
         return self
 
     @property
@@ -124,7 +125,7 @@ class LLMRouterChain(RouterChain):
     def _validate_outputs(self, outputs: dict[str, Any]) -> None:
         super()._validate_outputs(outputs)
         if not isinstance(outputs["next_inputs"], dict):
-            raise ValueError
+            raise ValueError  # noqa: TRY004
 
     def _call(
         self,
@@ -135,11 +136,10 @@ class LLMRouterChain(RouterChain):
         callbacks = _run_manager.get_child()
 
         prediction = self.llm_chain.predict(callbacks=callbacks, **inputs)
-        output = cast(
+        return cast(
             dict[str, Any],
             self.llm_chain.prompt.output_parser.parse(prediction),
         )
-        return output
 
     async def _acall(
         self,
@@ -148,15 +148,17 @@ class LLMRouterChain(RouterChain):
     ) -> dict[str, Any]:
         _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager()
         callbacks = _run_manager.get_child()
-        output = cast(
+        return cast(
             dict[str, Any],
             await self.llm_chain.apredict_and_parse(callbacks=callbacks, **inputs),
         )
-        return output
 
     @classmethod
     def from_llm(
-        cls, llm: BaseLanguageModel, prompt: BasePromptTemplate, **kwargs: Any
+        cls,
+        llm: BaseLanguageModel,
+        prompt: BasePromptTemplate,
+        **kwargs: Any,
     ) -> LLMRouterChain:
         """Convenience constructor."""
         llm_chain = LLMChain(llm=llm, prompt=prompt)
@@ -175,11 +177,11 @@ class RouterOutputParser(BaseOutputParser[dict[str, str]]):
             expected_keys = ["destination", "next_inputs"]
             parsed = parse_and_check_json_markdown(text, expected_keys)
             if not isinstance(parsed["destination"], str):
-                raise ValueError("Expected 'destination' to be a string.")
+                msg = "Expected 'destination' to be a string."
+                raise TypeError(msg)
             if not isinstance(parsed["next_inputs"], self.next_inputs_type):
-                raise ValueError(
-                    f"Expected 'next_inputs' to be {self.next_inputs_type}."
-                )
+                msg = f"Expected 'next_inputs' to be {self.next_inputs_type}."
+                raise TypeError(msg)
             parsed["next_inputs"] = {self.next_inputs_inner_key: parsed["next_inputs"]}
             if (
                 parsed["destination"].strip().lower()
@@ -188,8 +190,7 @@ class RouterOutputParser(BaseOutputParser[dict[str, str]]):
                 parsed["destination"] = None
             else:
                 parsed["destination"] = parsed["destination"].strip()
-            return parsed
         except Exception as e:
-            raise OutputParserException(
-                f"Parsing text\n{text}\n raised following error:\n{e}"
-            )
+            msg = f"Parsing text\n{text}\n raised following error:\n{e}"
+            raise OutputParserException(msg) from e
+        return parsed

langchain/chains/router/multi_prompt.py

@@ -21,7 +21,7 @@ from langchain.chains.router.multi_prompt_prompt import MULTI_PROMPT_ROUTER_TEMP
     removal="1.0",
     message=(
         "Please see migration guide here for recommended implementation: "
-        "https://python.langchain.com/docs/versions/migrating_chains/multi_prompt_chain/"  # noqa: E501
+        "https://python.langchain.com/docs/versions/migrating_chains/multi_prompt_chain/"
     ),
 )
 class MultiPromptChain(MultiRouteChain):
@@ -157,7 +157,7 @@ class MultiPromptChain(MultiRouteChain):
         destinations = [f"{p['name']}: {p['description']}" for p in prompt_infos]
         destinations_str = "\n".join(destinations)
         router_template = MULTI_PROMPT_ROUTER_TEMPLATE.format(
-            destinations=destinations_str
+            destinations=destinations_str,
         )
         router_prompt = PromptTemplate(
             template=router_template,

langchain/chains/router/multi_retrieval_qa.py

@@ -48,14 +48,15 @@ class MultiRetrievalQAChain(MultiRouteChain):
         **kwargs: Any,
     ) -> MultiRetrievalQAChain:
         if default_prompt and not default_retriever:
-            raise ValueError(
+            msg = (
                 "`default_retriever` must be specified if `default_prompt` is "
                 "provided. Received only `default_prompt`."
             )
+            raise ValueError(msg)
         destinations = [f"{r['name']}: {r['description']}" for r in retriever_infos]
         destinations_str = "\n".join(destinations)
         router_template = MULTI_RETRIEVAL_ROUTER_TEMPLATE.format(
-            destinations=destinations_str
+            destinations=destinations_str,
         )
         router_prompt = PromptTemplate(
             template=router_template,
@@ -74,15 +75,18 @@ class MultiRetrievalQAChain(MultiRouteChain):
             _default_chain = default_chain
         elif default_retriever:
             _default_chain = RetrievalQA.from_llm(
-                llm, prompt=default_prompt, retriever=default_retriever
+                llm,
+                prompt=default_prompt,
+                retriever=default_retriever,
             )
         else:
             prompt_template = DEFAULT_TEMPLATE.replace("input", "query")
             prompt = PromptTemplate(
-                template=prompt_template, input_variables=["history", "query"]
+                template=prompt_template,
+                input_variables=["history", "query"],
             )
             if default_chain_llm is None:
-                raise NotImplementedError(
+                msg = (
                     "conversation_llm must be provided if default_chain is not "
                     "specified. This API has been changed to avoid instantiating "
                     "default LLMs on behalf of users."
@@ -90,6 +94,7 @@ class MultiRetrievalQAChain(MultiRouteChain):
                     "from langchain_openai import ChatOpenAI\n"
                     "llm = ChatOpenAI()"
                 )
+                raise NotImplementedError(msg)
             _default_chain = ConversationChain(
                 llm=default_chain_llm,
                 prompt=prompt,

langchain/chains/sequential.py

@@ -48,17 +48,18 @@ class SequentialChain(Chain):
         """Validate that the correct inputs exist for all chains."""
         chains = values["chains"]
         input_variables = values["input_variables"]
-        memory_keys = list()
+        memory_keys = []
         if "memory" in values and values["memory"] is not None:
             """Validate that prompt input variables are consistent."""
             memory_keys = values["memory"].memory_variables
             if set(input_variables).intersection(set(memory_keys)):
                 overlapping_keys = set(input_variables) & set(memory_keys)
-                raise ValueError(
+                msg = (
                     f"The input key(s) {''.join(overlapping_keys)} are found "
                     f"in the Memory keys ({memory_keys}) - please use input and "
                     f"memory keys that don't overlap."
                 )
+                raise ValueError(msg)
 
         known_variables = set(input_variables + memory_keys)
 
@@ -68,15 +69,15 @@ class SequentialChain(Chain):
                 missing_vars = missing_vars.difference(chain.memory.memory_variables)
 
             if missing_vars:
-                raise ValueError(
+                msg = (
                     f"Missing required input keys: {missing_vars}, "
                     f"only had {known_variables}"
                 )
+                raise ValueError(msg)
             overlapping_keys = known_variables.intersection(chain.output_keys)
             if overlapping_keys:
-                raise ValueError(
-                    f"Chain returned keys that already exist: {overlapping_keys}"
-                )
+                msg = f"Chain returned keys that already exist: {overlapping_keys}"
+                raise ValueError(msg)
 
             known_variables |= set(chain.output_keys)
 
@@ -89,9 +90,8 @@ class SequentialChain(Chain):
         else:
             missing_vars = set(values["output_variables"]).difference(known_variables)
             if missing_vars:
-                raise ValueError(
-                    f"Expected output variables that were not found: {missing_vars}."
-                )
+                msg = f"Expected output variables that were not found: {missing_vars}."
+                raise ValueError(msg)
 
         return values
 
@@ -102,7 +102,7 @@ class SequentialChain(Chain):
     ) -> dict[str, str]:
         known_values = inputs.copy()
         _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager()
-        for i, chain in enumerate(self.chains):
+        for _i, chain in enumerate(self.chains):
             callbacks = _run_manager.get_child()
             outputs = chain(known_values, return_only_outputs=True, callbacks=callbacks)
             known_values.update(outputs)
@@ -116,9 +116,11 @@ class SequentialChain(Chain):
         known_values = inputs.copy()
         _run_manager = run_manager or AsyncCallbackManagerForChainRun.get_noop_manager()
         callbacks = _run_manager.get_child()
-        for i, chain in enumerate(self.chains):
+        for _i, chain in enumerate(self.chains):
             outputs = await chain.acall(
-                known_values, return_only_outputs=True, callbacks=callbacks
+                known_values,
+                return_only_outputs=True,
+                callbacks=callbacks,
             )
             known_values.update(outputs)
         return {k: known_values[k] for k in self.output_variables}
@@ -158,15 +160,17 @@ class SimpleSequentialChain(Chain):
         """Validate that chains are all single input/output."""
         for chain in self.chains:
             if len(chain.input_keys) != 1:
-                raise ValueError(
+                msg = (
                     "Chains used in SimplePipeline should all have one input, got "
                     f"{chain} with {len(chain.input_keys)} inputs."
                 )
+                raise ValueError(msg)
             if len(chain.output_keys) != 1:
-                raise ValueError(
+                msg = (
                     "Chains used in SimplePipeline should all have one output, got "
                     f"{chain} with {len(chain.output_keys)} outputs."
                 )
+                raise ValueError(msg)
         return self
 
     def _call(
@@ -179,12 +183,16 @@ class SimpleSequentialChain(Chain):
         color_mapping = get_color_mapping([str(i) for i in range(len(self.chains))])
         for i, chain in enumerate(self.chains):
             _input = chain.run(
-                _input, callbacks=_run_manager.get_child(f"step_{i + 1}")
+                _input,
+                callbacks=_run_manager.get_child(f"step_{i + 1}"),
             )
             if self.strip_outputs:
                 _input = _input.strip()
             _run_manager.on_text(
-                _input, color=color_mapping[str(i)], end="\n", verbose=self.verbose
+                _input,
+                color=color_mapping[str(i)],
+                end="\n",
+                verbose=self.verbose,
             )
         return {self.output_key: _input}
 
@@ -198,11 +206,15 @@ class SimpleSequentialChain(Chain):
         color_mapping = get_color_mapping([str(i) for i in range(len(self.chains))])
         for i, chain in enumerate(self.chains):
             _input = await chain.arun(
-                _input, callbacks=_run_manager.get_child(f"step_{i + 1}")
+                _input,
+                callbacks=_run_manager.get_child(f"step_{i + 1}"),
             )
             if self.strip_outputs:
                 _input = _input.strip()
             await _run_manager.on_text(
-                _input, color=color_mapping[str(i)], end="\n", verbose=self.verbose
+                _input,
+                color=color_mapping[str(i)],
+                end="\n",
+                verbose=self.verbose,
             )
         return {self.output_key: _input}

langchain/chains/sql_database/prompt.py

@@ -1,8 +1,6 @@
-# flake8: noqa
 from langchain_core.output_parsers.list import CommaSeparatedListOutputParser
 from langchain_core.prompts.prompt import PromptTemplate
 
-
 PROMPT_SUFFIX = """Only use the following tables:
 {table_info}
 
@@ -21,7 +19,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "dialect", "top_k"],
@@ -35,7 +33,7 @@ Question: {query}
 
 Table Names: {table_names}
 
-Relevant Table Names:"""
+Relevant Table Names:"""  # noqa: E501
 DECIDER_PROMPT = PromptTemplate(
     input_variables=["query", "table_names"],
     template=_DECIDER_TEMPLATE,
@@ -46,7 +44,7 @@ _cratedb_prompt = """You are a CrateDB expert. Given an input question, first cr
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per CrateDB. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (") to denote them as delimited identifiers.
 Pay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.
-Pay attention to use CURRENT_DATE function to get the current date, if the question involves "today". 
+Pay attention to use CURRENT_DATE function to get the current date, if the question involves "today".
 
 Use the following format:
 
@@ -55,7 +53,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 CRATEDB_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -75,7 +73,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 DUCKDB_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -95,7 +93,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 GOOGLESQL_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -116,7 +114,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 MSSQL_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -137,7 +135,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 MYSQL_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -158,7 +156,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 MARIADB_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -179,7 +177,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 ORACLE_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -200,7 +198,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 POSTGRES_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -221,7 +219,7 @@ SQLQuery: SQL Query to run
 SQLResult: Result of the SQLQuery
 Answer: Final answer here
 
-"""
+"""  # noqa: E501
 
 SQLITE_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -241,7 +239,7 @@ SQLQuery: "SQL Query to run"
 SQLResult: "Result of the SQLQuery"
 Answer: "Final answer here"
 
-"""
+"""  # noqa: E501
 
 CLICKHOUSE_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],
@@ -261,7 +259,7 @@ SQLQuery: "SQL Query to run"
 SQLResult: "Result of the SQLQuery"
 Answer: "Final answer here"
 
-"""
+"""  # noqa: E501
 
 PRESTODB_PROMPT = PromptTemplate(
     input_variables=["input", "table_info", "top_k"],

langchain/chains/sql_database/query.py

@@ -121,26 +121,27 @@ def create_sql_query_chain(
     else:
         prompt_to_use = PROMPT
     if {"input", "top_k", "table_info"}.difference(
-        prompt_to_use.input_variables + list(prompt_to_use.partial_variables)
+        prompt_to_use.input_variables + list(prompt_to_use.partial_variables),
     ):
-        raise ValueError(
+        msg = (
             f"Prompt must have input variables: 'input', 'top_k', "
             f"'table_info'. Received prompt with input variables: "
             f"{prompt_to_use.input_variables}. Full prompt:\n\n{prompt_to_use}"
         )
+        raise ValueError(msg)
     if "dialect" in prompt_to_use.input_variables:
         prompt_to_use = prompt_to_use.partial(dialect=db.dialect)
 
     table_info_kwargs = {}
     if get_col_comments:
         if db.dialect not in ("postgresql", "mysql", "oracle"):
-            raise ValueError(
+            msg = (
                 f"get_col_comments=True is only supported for dialects "
                 f"'postgresql', 'mysql', and 'oracle'. Received dialect: "
                 f"{db.dialect}"
             )
-        else:
-            table_info_kwargs["get_col_comments"] = True
+            raise ValueError(msg)
+        table_info_kwargs["get_col_comments"] = True
 
     inputs = {
         "input": lambda x: x["question"] + "\nSQLQuery: ",

langchain/chains/structured_output/__init__.py

@@ -3,4 +3,4 @@ from langchain.chains.structured_output.base import (
     create_structured_output_runnable,
 )
 
-__all__ = ["create_structured_output_runnable", "create_openai_fn_runnable"]
+__all__ = ["create_openai_fn_runnable", "create_structured_output_runnable"]