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

langchain/chains/structured_output/base.py

@@ -47,18 +47,18 @@ from pydantic import BaseModel
         """
             from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
-    
+
             class Joke(BaseModel):
                 setup: str = Field(description="The setup of the joke")
-                punchline: str = Field(description="The punchline to the joke") 
-    
+                punchline: str = Field(description="The punchline to the joke")
+
             # Or any other chat model that supports tools.
             # Please reference to to the documentation of structured_output
-            # to see an up to date list of which models support 
+            # to see an up to date list of which models support
             # with_structured_output.
             model = ChatAnthropic(model="claude-3-opus-20240229", temperature=0)
             structured_llm = model.with_structured_output(Joke)
-            structured_llm.invoke("Tell me a joke about cats. 
+            structured_llm.invoke("Tell me a joke about cats.
                 Make sure to call the Joke function.")
             """
     ),
@@ -134,7 +134,8 @@ def create_openai_fn_runnable(
                 # -> RecordDog(name="Harry", color="brown", fav_food="chicken")
     """  # noqa: E501
     if not functions:
-        raise ValueError("Need to pass in at least one function. Received zero.")
+        msg = "Need to pass in at least one function. Received zero."
+        raise ValueError(msg)
     openai_functions = [convert_to_openai_function(f) for f in functions]
     llm_kwargs_: dict[str, Any] = {"functions": openai_functions, **llm_kwargs}
     if len(openai_functions) == 1 and enforce_single_function_usage:
@@ -142,8 +143,7 @@ def create_openai_fn_runnable(
     output_parser = output_parser or get_openai_output_parser(functions)
     if prompt:
         return prompt | llm.bind(**llm_kwargs_) | output_parser
-    else:
-        return llm.bind(**llm_kwargs_) | output_parser
+    return llm.bind(**llm_kwargs_) | output_parser
 
 
 @deprecated(
@@ -168,15 +168,15 @@ def create_openai_fn_runnable(
 
             class Joke(BaseModel):
                 setup: str = Field(description="The setup of the joke")
-                punchline: str = Field(description="The punchline to the joke") 
+                punchline: str = Field(description="The punchline to the joke")
 
             # Or any other chat model that supports tools.
             # Please reference to to the documentation of structured_output
-            # to see an up to date list of which models support 
+            # to see an up to date list of which models support
             # with_structured_output.
             model = ChatAnthropic(model="claude-3-opus-20240229", temperature=0)
             structured_llm = model.with_structured_output(Joke)
-            structured_llm.invoke("Tell me a joke about cats. 
+            structured_llm.invoke("Tell me a joke about cats.
                 Make sure to call the Joke function.")
             """
     ),
@@ -190,7 +190,9 @@ def create_structured_output_runnable(
     enforce_function_usage: bool = True,
     return_single: bool = True,
     mode: Literal[
-        "openai-functions", "openai-tools", "openai-json"
+        "openai-functions",
+        "openai-tools",
+        "openai-json",
     ] = "openai-functions",
     **kwargs: Any,
 ) -> Runnable:
@@ -201,39 +203,39 @@ def create_structured_output_runnable(
             is passed in, it's assumed to already be a valid JsonSchema.
             For best results, pydantic.BaseModels should have docstrings describing what
             the schema represents and descriptions for the parameters.
-        llm: Language model to use. Assumed to support the OpenAI function-calling API 
-            if mode is 'openai-function'. Assumed to support OpenAI response_format 
+        llm: Language model to use. Assumed to support the OpenAI function-calling API
+            if mode is 'openai-function'. Assumed to support OpenAI response_format
             parameter if mode is 'openai-json'.
-        prompt: BasePromptTemplate to pass to the model. If mode is 'openai-json' and 
-            prompt has input variable 'output_schema' then the given output_schema 
+        prompt: BasePromptTemplate to pass to the model. If mode is 'openai-json' and
+            prompt has input variable 'output_schema' then the given output_schema
             will be converted to a JsonSchema and inserted in the prompt.
         output_parser: Output parser to use for parsing model outputs. By default
             will be inferred from the function types. If pydantic.BaseModel is passed
-            in, then the OutputParser will try to parse outputs using the pydantic 
+            in, then the OutputParser will try to parse outputs using the pydantic
             class. Otherwise model outputs will be parsed as JSON.
-        mode: How structured outputs are extracted from the model. If 'openai-functions' 
-            then OpenAI function calling is used with the deprecated 'functions', 
-            'function_call' schema. If 'openai-tools' then OpenAI function 
-            calling with the latest 'tools', 'tool_choice' schema is used. This is 
-            recommended over 'openai-functions'. If 'openai-json' then OpenAI model 
+        mode: How structured outputs are extracted from the model. If 'openai-functions'
+            then OpenAI function calling is used with the deprecated 'functions',
+            'function_call' schema. If 'openai-tools' then OpenAI function
+            calling with the latest 'tools', 'tool_choice' schema is used. This is
+            recommended over 'openai-functions'. If 'openai-json' then OpenAI model
             with response_format set to JSON is used.
-        enforce_function_usage: Only applies when mode is 'openai-tools' or 
-            'openai-functions'. If True, then the model will be forced to use the given 
-            output schema. If False, then the model can elect whether to use the output 
+        enforce_function_usage: Only applies when mode is 'openai-tools' or
+            'openai-functions'. If True, then the model will be forced to use the given
+            output schema. If False, then the model can elect whether to use the output
             schema.
-        return_single: Only applies when mode is 'openai-tools'. Whether to a list of 
-            structured outputs or a single one. If True and model does not return any 
-            structured outputs then chain output is None. If False and model does not 
+        return_single: Only applies when mode is 'openai-tools'. Whether to a list of
+            structured outputs or a single one. If True and model does not return any
+            structured outputs then chain output is None. If False and model does not
             return any structured outputs then chain output is an empty list.
         kwargs: Additional named arguments.
 
     Returns:
-        A runnable sequence that will return a structured output(s) matching the given 
+        A runnable sequence that will return a structured output(s) matching the given
             output_schema.
-    
+
     OpenAI tools example with Pydantic schema (mode='openai-tools'):
         .. code-block:: python
-        
+
                 from typing import Optional
 
                 from langchain.chains import create_structured_output_runnable
@@ -251,23 +253,23 @@ def create_structured_output_runnable(
                 llm = ChatOpenAI(model="gpt-3.5-turbo-0125", temperature=0)
                 prompt = ChatPromptTemplate.from_messages(
                     [
-                        ("system", "You are an extraction algorithm. Please extract every possible instance"), 
+                        ("system", "You are an extraction algorithm. Please extract every possible instance"),
                         ('human', '{input}')
                     ]
                 )
                 structured_llm = create_structured_output_runnable(
-                    RecordDog, 
-                    llm, 
-                    mode="openai-tools", 
-                    enforce_function_usage=True, 
+                    RecordDog,
+                    llm,
+                    mode="openai-tools",
+                    enforce_function_usage=True,
                     return_single=True
                 )
                 structured_llm.invoke({"input": "Harry was a chubby brown beagle who loved chicken"})
                 # -> RecordDog(name="Harry", color="brown", fav_food="chicken")
-                
+
     OpenAI tools example with dict schema (mode="openai-tools"):
         .. code-block:: python
-        
+
                 from typing import Optional
 
                 from langchain.chains import create_structured_output_runnable
@@ -303,15 +305,15 @@ def create_structured_output_runnable(
 
                 llm = ChatOpenAI(model="gpt-3.5-turbo-0125", temperature=0)
                 structured_llm = create_structured_output_runnable(
-                    dog_schema, 
-                    llm, 
-                    mode="openai-tools", 
-                    enforce_function_usage=True, 
+                    dog_schema,
+                    llm,
+                    mode="openai-tools",
+                    enforce_function_usage=True,
                     return_single=True
                 )
                 structured_llm.invoke("Harry was a chubby brown beagle who loved chicken")
                 # -> {'name': 'Harry', 'color': 'brown', 'fav_food': 'chicken'}
-    
+
     OpenAI functions example (mode="openai-functions"):
         .. code-block:: python
 
@@ -332,7 +334,7 @@ def create_structured_output_runnable(
                 structured_llm = create_structured_output_runnable(Dog, llm, mode="openai-functions")
                 structured_llm.invoke("Harry was a chubby brown beagle who loved chicken")
                 # -> Dog(name="Harry", color="brown", fav_food="chicken")
-                
+
     OpenAI functions with prompt example:
         .. code-block:: python
 
@@ -361,7 +363,7 @@ def create_structured_output_runnable(
                 # -> Dog(name="Harry", color="brown", fav_food="chicken")
     OpenAI json response format example (mode="openai-json"):
         .. code-block:: python
-        
+
                 from typing import Optional
 
                 from langchain.chains import create_structured_output_runnable
@@ -379,9 +381,9 @@ def create_structured_output_runnable(
                 llm = ChatOpenAI(model="gpt-3.5-turbo-0125", temperature=0)
                 structured_llm = create_structured_output_runnable(Dog, llm, mode="openai-json")
                 system = '''You are a world class assistant for extracting information in structured JSON formats. \
-                
+
                 Extract a valid JSON blob from the user input that matches the following JSON Schema:
-                
+
                 {output_schema}'''
                 prompt = ChatPromptTemplate.from_messages(
                     [("system", system), ("human", "{input}"),]
@@ -391,7 +393,8 @@ def create_structured_output_runnable(
     """  # noqa: E501
     # for backwards compatibility
     force_function_usage = kwargs.get(
-        "enforce_single_function_usage", enforce_function_usage
+        "enforce_single_function_usage",
+        enforce_function_usage,
     )
 
     if mode == "openai-tools":
@@ -400,9 +403,8 @@ def create_structured_output_runnable(
         # Backwards compatibility keys
         unrecognized_keys = keys_in_kwargs - {"enforce_single_function_usage"}
         if unrecognized_keys:
-            raise TypeError(
-                f"Got an unexpected keyword argument(s): {unrecognized_keys}."
-            )
+            msg = f"Got an unexpected keyword argument(s): {unrecognized_keys}."
+            raise TypeError(msg)
 
         return _create_openai_tools_runnable(
             output_schema,
@@ -413,7 +415,7 @@ def create_structured_output_runnable(
             first_tool_only=return_single,
         )
 
-    elif mode == "openai-functions":
+    if mode == "openai-functions":
         return _create_openai_functions_structured_output_runnable(
             output_schema,
             llm,
@@ -422,19 +424,24 @@ def create_structured_output_runnable(
             enforce_single_function_usage=force_function_usage,
             **kwargs,  # llm-specific kwargs
         )
-    elif mode == "openai-json":
+    if mode == "openai-json":
         if force_function_usage:
-            raise ValueError(
+            msg = (
                 "enforce_single_function_usage is not supported for mode='openai-json'."
             )
+            raise ValueError(msg)
         return _create_openai_json_runnable(
-            output_schema, llm, prompt=prompt, output_parser=output_parser, **kwargs
-        )
-    else:
-        raise ValueError(
-            f"Invalid mode {mode}. Expected one of 'openai-tools', 'openai-functions', "
-            f"'openai-json'."
+            output_schema,
+            llm,
+            prompt=prompt,
+            output_parser=output_parser,
+            **kwargs,
         )
+    msg = (
+        f"Invalid mode {mode}. Expected one of 'openai-tools', 'openai-functions', "
+        f"'openai-json'."
+    )
+    raise ValueError(msg)
 
 
 def _create_openai_tools_runnable(
@@ -454,12 +461,12 @@ def _create_openai_tools_runnable(
             "function": {"name": oai_tool["function"]["name"]},
         }
     output_parser = output_parser or _get_openai_tool_output_parser(
-        tool, first_tool_only=first_tool_only
+        tool,
+        first_tool_only=first_tool_only,
     )
     if prompt:
         return prompt | llm.bind(**llm_kwargs) | output_parser
-    else:
-        return llm.bind(**llm_kwargs) | output_parser
+    return llm.bind(**llm_kwargs) | output_parser
 
 
 def _get_openai_tool_output_parser(
@@ -474,7 +481,8 @@ def _get_openai_tool_output_parser(
     else:
         key_name = convert_to_openai_tool(tool)["function"]["name"]
         output_parser = JsonOutputKeyToolsParser(
-            first_tool_only=first_tool_only, key_name=key_name
+            first_tool_only=first_tool_only,
+            key_name=key_name,
         )
     return output_parser
 
@@ -533,8 +541,7 @@ def _create_openai_json_runnable(
             prompt = prompt.partial(output_schema=json.dumps(schema_as_dict, indent=2))
 
         return prompt | llm | output_parser
-    else:
-        return llm | output_parser
+    return llm | output_parser
 
 
 def _create_openai_functions_structured_output_runnable(
@@ -563,7 +570,8 @@ def _create_openai_functions_structured_output_runnable(
 
         function = _OutputFormatter
         output_parser = output_parser or PydanticAttrOutputFunctionsParser(
-            pydantic_schema=_OutputFormatter, attr_name="output"
+            pydantic_schema=_OutputFormatter,
+            attr_name="output",
         )
     return create_openai_fn_runnable(
         [function],

langchain/chains/summarize/chain.py

@@ -20,13 +20,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_stuff_chain(
     llm: BaseLanguageModel,
+    *,
     prompt: BasePromptTemplate = stuff_prompt.PROMPT,
     document_variable_name: str = "text",
     verbose: Optional[bool] = None,
@@ -44,6 +47,7 @@ def _load_stuff_chain(
 
 def _load_map_reduce_chain(
     llm: BaseLanguageModel,
+    *,
     map_prompt: BasePromptTemplate = map_reduce_prompt.PROMPT,
     combine_prompt: BasePromptTemplate = map_reduce_prompt.PROMPT,
     combine_document_variable_name: str = "text",
@@ -54,7 +58,6 @@ def _load_map_reduce_chain(
     verbose: Optional[bool] = None,
     token_max: int = 3000,
     callbacks: Callbacks = None,
-    *,
     collapse_max_retries: Optional[int] = None,
     **kwargs: Any,
 ) -> MapReduceDocumentsChain:
@@ -81,10 +84,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(
@@ -116,6 +120,7 @@ def _load_map_reduce_chain(
 
 def _load_refine_chain(
     llm: BaseLanguageModel,
+    *,
     question_prompt: BasePromptTemplate = refine_prompts.PROMPT,
     refine_prompt: BasePromptTemplate = refine_prompts.REFINE_PROMPT,
     document_variable_name: str = "text",
@@ -140,7 +145,7 @@ def _load_refine_chain(
 def load_summarize_chain(
     llm: BaseLanguageModel,
     chain_type: str = "stuff",
-    verbose: Optional[bool] = None,
+    verbose: Optional[bool] = None,  # noqa: FBT001
     **kwargs: Any,
 ) -> BaseCombineDocumentsChain:
     """Load summarizing chain.
@@ -161,8 +166,9 @@ def load_summarize_chain(
         "refine": _load_refine_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, **kwargs)

langchain/chains/summarize/map_reduce_prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts import PromptTemplate
 
 prompt_template = """Write a concise summary of the following:

langchain/chains/summarize/stuff_prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts import PromptTemplate
 
 prompt_template = """Write a concise summary of the following:

langchain/chains/transform.py

@@ -77,9 +77,8 @@ class TransformChain(Chain):
     ) -> dict[str, Any]:
         if self.atransform_cb is not None:
             return await self.atransform_cb(inputs)
-        else:
-            self._log_once(
-                "TransformChain's atransform is not provided, falling"
-                " back to synchronous transform"
-            )
-            return self.transform_cb(inputs)
+        self._log_once(
+            "TransformChain's atransform is not provided, falling"
+            " back to synchronous transform",
+        )
+        return self.transform_cb(inputs)

langchain/chat_loaders/facebook_messenger.py

@@ -29,4 +29,4 @@ def __getattr__(name: str) -> Any:
     return import_lookup(name)
 
 
-__all__ = ["SingleFileFacebookMessengerChatLoader", "FolderFacebookMessengerChatLoader"]
+__all__ = ["FolderFacebookMessengerChatLoader", "SingleFileFacebookMessengerChatLoader"]

langchain/chat_loaders/langsmith.py

@@ -25,6 +25,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "LangSmithRunChatLoader",
     "LangSmithDatasetChatLoader",
+    "LangSmithRunChatLoader",
 ]

langchain/chat_loaders/utils.py

@@ -29,8 +29,8 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "merge_chat_runs_in_session",
-    "merge_chat_runs",
-    "map_ai_messages_in_session",
     "map_ai_messages",
+    "map_ai_messages_in_session",
+    "merge_chat_runs",
+    "merge_chat_runs_in_session",
 ]

langchain/chat_models/__init__.py

@@ -36,6 +36,7 @@ def __getattr__(name: str) -> None:
             "Please import from langchain-community instead:\n\n"
             f"`from langchain_community.chat_models import {name}`.\n\n"
             "To install langchain-community run `pip install -U langchain-community`.",
+            stacklevel=2,
             category=LangChainDeprecationWarning,
         )
 
@@ -43,35 +44,35 @@ def __getattr__(name: str) -> None:
 
 
 __all__ = [
-    "init_chat_model",
-    "ChatOpenAI",
-    "BedrockChat",
     "AzureChatOpenAI",
-    "FakeListChatModel",
-    "PromptLayerChatOpenAI",
-    "ChatDatabricks",
-    "ChatEverlyAI",
+    "BedrockChat",
     "ChatAnthropic",
+    "ChatAnyscale",
+    "ChatBaichuan",
     "ChatCohere",
+    "ChatDatabricks",
+    "ChatEverlyAI",
+    "ChatFireworks",
     "ChatGooglePalm",
-    "ChatMlflow",
+    "ChatHunyuan",
+    "ChatJavelinAIGateway",
+    "ChatKonko",
+    "ChatLiteLLM",
     "ChatMLflowAIGateway",
+    "ChatMlflow",
     "ChatOllama",
+    "ChatOpenAI",
     "ChatVertexAI",
-    "JinaChat",
+    "ChatYandexGPT",
+    "ErnieBotChat",
+    "FakeListChatModel",
+    "GigaChat",
     "HumanInputChatModel",
+    "JinaChat",
     "MiniMaxChat",
-    "ChatAnyscale",
-    "ChatLiteLLM",
-    "ErnieBotChat",
-    "ChatJavelinAIGateway",
-    "ChatKonko",
     "PaiEasChatEndpoint",
+    "PromptLayerChatOpenAI",
     "QianfanChatEndpoint",
-    "ChatFireworks",
-    "ChatYandexGPT",
-    "ChatBaichuan",
-    "ChatHunyuan",
-    "GigaChat",
     "VolcEngineMaasChat",
+    "init_chat_model",
 ]

langchain/chat_models/anthropic.py

@@ -25,6 +25,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "convert_messages_to_prompt_anthropic",
     "ChatAnthropic",
+    "convert_messages_to_prompt_anthropic",
 ]

langchain/chat_models/azureml_endpoint.py

@@ -25,6 +25,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "LlamaContentFormatter",
     "AzureMLChatOnlineEndpoint",
+    "LlamaContentFormatter",
 ]

langchain/chat_models/baidu_qianfan_endpoint.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "QianfanChatEndpoint": "langchain_community.chat_models.baidu_qianfan_endpoint"
+    "QianfanChatEndpoint": "langchain_community.chat_models.baidu_qianfan_endpoint",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)