langchain 0.3.26__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

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

@@ -10,6 +10,7 @@ from langchain_core.callbacks import (
     CallbackManagerForChainRun,
 )
 from pydantic import Field
+from typing_extensions import override
 
 from langchain.chains.base import Chain
 
@@ -25,6 +26,7 @@ class TransformChain(Chain):
             from langchain.chains import TransformChain
             transform_chain = TransformChain(input_variables=["text"],
              output_variables["entities"], transform=func())
+
     """
 
     input_variables: list[str]
@@ -63,6 +65,7 @@ class TransformChain(Chain):
         """
         return self.output_variables
 
+    @override
     def _call(
         self,
         inputs: dict[str, str],
@@ -70,6 +73,7 @@ class TransformChain(Chain):
     ) -> dict[str, str]:
         return self.transform_cb(inputs)
 
+    @override
     async def _acall(
         self,
         inputs: dict[str, Any],
@@ -77,9 +81,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)