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

langchain/llms/azureml_endpoint.py

@@ -38,11 +38,11 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "AzureMLEndpointClient",
+    "AzureMLOnlineEndpoint",
     "ContentFormatterBase",
+    "CustomOpenAIContentFormatter",
+    "DollyContentFormatter",
     "GPT2ContentFormatter",
-    "OSSContentFormatter",
     "HFContentFormatter",
-    "DollyContentFormatter",
-    "CustomOpenAIContentFormatter",
-    "AzureMLOnlineEndpoint",
+    "OSSContentFormatter",
 ]

langchain/llms/base.py

@@ -1,12 +1,20 @@
-# Backwards compatibility.
+"""
+This module provides backward-compatible exports of core language model classes.
+
+These classes are re-exported for compatibility with older versions of LangChain
+and allow users to import language model interfaces from a stable path.
+
+Exports:
+    - LLM: Abstract base class for all LLMs
+    - BaseLLM: Deprecated or foundational class for legacy LLMs
+    - BaseLanguageModel: Base class for core language model implementations
+"""
+
 from langchain_core.language_models import BaseLanguageModel
-from langchain_core.language_models.llms import (
-    LLM,
-    BaseLLM,
-)
+from langchain_core.language_models.llms import LLM, BaseLLM
 
 __all__ = [
-    "BaseLanguageModel",
-    "BaseLLM",
     "LLM",
+    "BaseLLM",
+    "BaseLanguageModel",
 ]

langchain/llms/bedrock.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "BedrockBase",
     "Bedrock",
+    "BedrockBase",
 ]

langchain/llms/cloudflare_workersai.py

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

langchain/llms/gradient_ai.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "TrainResult",
     "GradientLLM",
+    "TrainResult",
 ]

langchain/llms/loading.py

@@ -22,6 +22,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "load_llm_from_config",
     "load_llm",
+    "load_llm_from_config",
 ]

langchain/llms/openai.py

@@ -25,8 +25,8 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "AzureOpenAI",
     "BaseOpenAI",
     "OpenAI",
-    "AzureOpenAI",
     "OpenAIChat",
 ]

langchain/llms/sagemaker_endpoint.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "SagemakerEndpoint",
     "LLMContentHandler",
+    "SagemakerEndpoint",
 ]

langchain/load/dump.py

@@ -1,3 +1,3 @@
 from langchain_core.load.dump import default, dumpd, dumps
 
-__all__ = ["default", "dumps", "dumpd"]
+__all__ = ["default", "dumpd", "dumps"]

langchain/load/load.py

@@ -1,3 +1,3 @@
 from langchain_core.load.load import Reviver, load, loads
 
-__all__ = ["Reviver", "loads", "load"]
+__all__ = ["Reviver", "load", "loads"]

langchain/load/serializable.py

@@ -10,10 +10,10 @@ from langchain_core.load.serializable import (
 
 __all__ = [
     "BaseSerialized",
+    "Serializable",
     "SerializedConstructor",
-    "SerializedSecret",
     "SerializedNotImplemented",
-    "try_neq_default",
-    "Serializable",
+    "SerializedSecret",
     "to_json_not_implemented",
+    "try_neq_default",
 ]

langchain/memory/__init__.py

@@ -139,15 +139,15 @@ __all__ = [
     "ReadOnlySharedMemory",
     "RedisChatMessageHistory",
     "RedisEntityStore",
-    "SingleStoreDBChatMessageHistory",
     "SQLChatMessageHistory",
     "SQLiteEntityStore",
     "SimpleMemory",
+    "SingleStoreDBChatMessageHistory",
     "StreamlitChatMessageHistory",
+    "UpstashRedisChatMessageHistory",
+    "UpstashRedisEntityStore",
     "VectorStoreRetrieverMemory",
     "XataChatMessageHistory",
     "ZepChatMessageHistory",
     "ZepMemory",
-    "UpstashRedisEntityStore",
-    "UpstashRedisChatMessageHistory",
 ]

langchain/memory/buffer.py

@@ -4,6 +4,7 @@ from langchain_core._api import deprecated
 from langchain_core.memory import BaseMemory
 from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_core.utils import pre_init
+from typing_extensions import override
 
 from langchain.memory.chat_memory import BaseChatMemory
 from langchain.memory.utils import get_prompt_input_key
@@ -78,10 +79,12 @@ class ConversationBufferMemory(BaseChatMemory):
         """
         return [self.memory_key]
 
+    @override
     def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}
 
+    @override
     async def aload_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return key-value pairs given the text input to the chain."""
         buffer = await self.abuffer()
@@ -121,9 +124,8 @@ class ConversationStringBufferMemory(BaseMemory):
     def validate_chains(cls, values: dict) -> dict:
         """Validate that return messages is not True."""
         if values.get("return_messages", False):
-            raise ValueError(
-                "return_messages must be False for ConversationStringBufferMemory"
-            )
+            msg = "return_messages must be False for ConversationStringBufferMemory"
+            raise ValueError(msg)
         return values
 
     @property
@@ -133,6 +135,7 @@ class ConversationStringBufferMemory(BaseMemory):
         """
         return [self.memory_key]
 
+    @override
     def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}
@@ -149,16 +152,19 @@ class ConversationStringBufferMemory(BaseMemory):
             prompt_input_key = self.input_key
         if self.output_key is None:
             if len(outputs) != 1:
-                raise ValueError(f"One output key expected, got {outputs.keys()}")
-            output_key = list(outputs.keys())[0]
+                msg = f"One output key expected, got {outputs.keys()}"
+                raise ValueError(msg)
+            output_key = next(iter(outputs.keys()))
         else:
             output_key = self.output_key
         human = f"{self.human_prefix}: " + inputs[prompt_input_key]
         ai = f"{self.ai_prefix}: " + outputs[output_key]
-        self.buffer += "\n" + "\n".join([human, ai])
+        self.buffer += f"\n{human}\n{ai}"
 
     async def asave_context(
-        self, inputs: dict[str, Any], outputs: dict[str, str]
+        self,
+        inputs: dict[str, Any],
+        outputs: dict[str, str],
     ) -> None:
         """Save context from this conversation to buffer."""
         return self.save_context(inputs, outputs)
@@ -167,5 +173,6 @@ class ConversationStringBufferMemory(BaseMemory):
         """Clear memory contents."""
         self.buffer = ""
 
+    @override
     async def aclear(self) -> None:
         self.clear()

langchain/memory/buffer_window.py

@@ -2,6 +2,7 @@ from typing import Any, Union
 
 from langchain_core._api import deprecated
 from langchain_core.messages import BaseMessage, get_buffer_string
+from typing_extensions import override
 
 from langchain.memory.chat_memory import BaseChatMemory
 
@@ -55,6 +56,7 @@ class ConversationBufferWindowMemory(BaseChatMemory):
         """
         return [self.memory_key]
 
+    @override
     def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}

langchain/memory/chat_memory.py

@@ -34,14 +34,16 @@ class BaseChatMemory(BaseMemory, ABC):
     """
 
     chat_memory: BaseChatMessageHistory = Field(
-        default_factory=InMemoryChatMessageHistory
+        default_factory=InMemoryChatMessageHistory,
     )
     output_key: Optional[str] = None
     input_key: Optional[str] = None
     return_messages: bool = False
 
     def _get_input_output(
-        self, inputs: dict[str, Any], outputs: dict[str, str]
+        self,
+        inputs: dict[str, Any],
+        outputs: dict[str, str],
     ) -> tuple[str, str]:
         if self.input_key is None:
             prompt_input_key = get_prompt_input_key(inputs, self.memory_variables)
@@ -49,20 +51,22 @@ class BaseChatMemory(BaseMemory, ABC):
             prompt_input_key = self.input_key
         if self.output_key is None:
             if len(outputs) == 1:
-                output_key = list(outputs.keys())[0]
+                output_key = next(iter(outputs.keys()))
             elif "output" in outputs:
                 output_key = "output"
                 warnings.warn(
                     f"'{self.__class__.__name__}' got multiple output keys:"
                     f" {outputs.keys()}. The default 'output' key is being used."
-                    f" If this is not desired, please manually set 'output_key'."
+                    f" If this is not desired, please manually set 'output_key'.",
+                    stacklevel=3,
                 )
             else:
-                raise ValueError(
+                msg = (
                     f"Got multiple output keys: {outputs.keys()}, cannot "
                     f"determine which to store in memory. Please set the "
                     f"'output_key' explicitly."
                 )
+                raise ValueError(msg)
         else:
             output_key = self.output_key
         return inputs[prompt_input_key], outputs[output_key]
@@ -74,11 +78,13 @@ class BaseChatMemory(BaseMemory, ABC):
             [
                 HumanMessage(content=input_str),
                 AIMessage(content=output_str),
-            ]
+            ],
         )
 
     async def asave_context(
-        self, inputs: dict[str, Any], outputs: dict[str, str]
+        self,
+        inputs: dict[str, Any],
+        outputs: dict[str, str],
     ) -> None:
         """Save context from this conversation to buffer."""
         input_str, output_str = self._get_input_output(inputs, outputs)
@@ -86,7 +92,7 @@ class BaseChatMemory(BaseMemory, ABC):
             [
                 HumanMessage(content=input_str),
                 AIMessage(content=output_str),
-            ]
+            ],
         )
 
     def clear(self) -> None:

langchain/memory/chat_message_histories/__init__.py

@@ -75,8 +75,8 @@ __all__ = [
     "PostgresChatMessageHistory",
     "RedisChatMessageHistory",
     "RocksetChatMessageHistory",
-    "SingleStoreDBChatMessageHistory",
     "SQLChatMessageHistory",
+    "SingleStoreDBChatMessageHistory",
     "StreamlitChatMessageHistory",
     "UpstashRedisChatMessageHistory",
     "XataChatMessageHistory",

langchain/memory/chat_message_histories/astradb.py

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

langchain/memory/chat_message_histories/cassandra.py

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

langchain/memory/chat_message_histories/cosmos_db.py

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

langchain/memory/chat_message_histories/dynamodb.py

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

langchain/memory/chat_message_histories/elasticsearch.py

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

langchain/memory/chat_message_histories/file.py

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

langchain/memory/chat_message_histories/firestore.py

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

langchain/memory/chat_message_histories/momento.py

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

langchain/memory/chat_message_histories/mongodb.py

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

langchain/memory/chat_message_histories/neo4j.py

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

langchain/memory/chat_message_histories/postgres.py

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

langchain/memory/chat_message_histories/redis.py

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

langchain/memory/chat_message_histories/rocksetdb.py

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

langchain/memory/chat_message_histories/singlestoredb.py

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

langchain/memory/chat_message_histories/streamlit.py

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

langchain/memory/chat_message_histories/upstash_redis.py

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

langchain/memory/chat_message_histories/xata.py

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

langchain/memory/chat_message_histories/zep.py

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

langchain/memory/combined.py

@@ -15,17 +15,19 @@ class CombinedMemory(BaseMemory):
 
     @field_validator("memories")
     @classmethod
-    def check_repeated_memory_variable(
-        cls, value: list[BaseMemory]
+    def _check_repeated_memory_variable(
+        cls,
+        value: list[BaseMemory],
     ) -> list[BaseMemory]:
         all_variables: set[str] = set()
         for val in value:
             overlap = all_variables.intersection(val.memory_variables)
             if overlap:
-                raise ValueError(
+                msg = (
                     f"The same variables {overlap} are found in multiple"
                     "memory object, which is not allowed by CombinedMemory."
                 )
+                raise ValueError(msg)
             all_variables |= set(val.memory_variables)
 
         return value
@@ -35,13 +37,13 @@ class CombinedMemory(BaseMemory):
     def check_input_key(cls, value: list[BaseMemory]) -> list[BaseMemory]:
         """Check that if memories are of type BaseChatMemory that input keys exist."""
         for val in value:
-            if isinstance(val, BaseChatMemory):
-                if val.input_key is None:
-                    warnings.warn(
-                        "When using CombinedMemory, "
-                        "input keys should be so the input is known. "
-                        f" Was not set on {val}"
-                    )
+            if isinstance(val, BaseChatMemory) and val.input_key is None:
+                warnings.warn(
+                    "When using CombinedMemory, "
+                    "input keys should be so the input is known. "
+                    f" Was not set on {val}",
+                    stacklevel=5,
+                )
         return value
 
     @property
@@ -65,9 +67,8 @@ class CombinedMemory(BaseMemory):
             data = memory.load_memory_variables(inputs)
             for key, value in data.items():
                 if key in memory_data:
-                    raise ValueError(
-                        f"The variable {key} is repeated in the CombinedMemory."
-                    )
+                    msg = f"The variable {key} is repeated in the CombinedMemory."
+                    raise ValueError(msg)
                 memory_data[key] = value
 
         return memory_data