langchain 0.3.23__py3-none-any.whl → 0.3.25__py3-none-any.whl

langchain/indexes/vectorstore.py

@@ -1,6 +1,6 @@
 """Vectorstore stubs for the indexing api."""
 
-from typing import Any, Dict, List, Optional, Type
+from typing import Any, Optional
 
 from langchain_core.document_loaders import BaseLoader
 from langchain_core.documents import Document
@@ -33,7 +33,7 @@ class VectorStoreIndexWrapper(BaseModel):
         self,
         question: str,
         llm: Optional[BaseLanguageModel] = None,
-        retriever_kwargs: Optional[Dict[str, Any]] = None,
+        retriever_kwargs: Optional[dict[str, Any]] = None,
         **kwargs: Any,
     ) -> str:
         """Query the vectorstore using the provided LLM.
@@ -65,7 +65,7 @@ class VectorStoreIndexWrapper(BaseModel):
         self,
         question: str,
         llm: Optional[BaseLanguageModel] = None,
-        retriever_kwargs: Optional[Dict[str, Any]] = None,
+        retriever_kwargs: Optional[dict[str, Any]] = None,
         **kwargs: Any,
     ) -> str:
         """Asynchronously query the vectorstore using the provided LLM.
@@ -97,7 +97,7 @@ class VectorStoreIndexWrapper(BaseModel):
         self,
         question: str,
         llm: Optional[BaseLanguageModel] = None,
-        retriever_kwargs: Optional[Dict[str, Any]] = None,
+        retriever_kwargs: Optional[dict[str, Any]] = None,
         **kwargs: Any,
     ) -> dict:
         """Query the vectorstore and retrieve the answer along with sources.
@@ -129,7 +129,7 @@ class VectorStoreIndexWrapper(BaseModel):
         self,
         question: str,
         llm: Optional[BaseLanguageModel] = None,
-        retriever_kwargs: Optional[Dict[str, Any]] = None,
+        retriever_kwargs: Optional[dict[str, Any]] = None,
         **kwargs: Any,
     ) -> dict:
         """Asynchronously query the vectorstore and retrieve the answer and sources.
@@ -158,7 +158,7 @@ class VectorStoreIndexWrapper(BaseModel):
         return await chain.ainvoke({chain.question_key: question})
 
 
-def _get_in_memory_vectorstore() -> Type[VectorStore]:
+def _get_in_memory_vectorstore() -> type[VectorStore]:
     """Get the InMemoryVectorStore."""
     import warnings
 
@@ -179,7 +179,7 @@ def _get_in_memory_vectorstore() -> Type[VectorStore]:
 class VectorstoreIndexCreator(BaseModel):
     """Logic for creating indexes."""
 
-    vectorstore_cls: Type[VectorStore] = Field(
+    vectorstore_cls: type[VectorStore] = Field(
         default_factory=_get_in_memory_vectorstore
     )
     embedding: Embeddings
@@ -191,7 +191,7 @@ class VectorstoreIndexCreator(BaseModel):
         extra="forbid",
     )
 
-    def from_loaders(self, loaders: List[BaseLoader]) -> VectorStoreIndexWrapper:
+    def from_loaders(self, loaders: list[BaseLoader]) -> VectorStoreIndexWrapper:
         """Create a vectorstore index from a list of loaders.
 
         Args:
@@ -205,7 +205,7 @@ class VectorstoreIndexCreator(BaseModel):
             docs.extend(loader.load())
         return self.from_documents(docs)
 
-    async def afrom_loaders(self, loaders: List[BaseLoader]) -> VectorStoreIndexWrapper:
+    async def afrom_loaders(self, loaders: list[BaseLoader]) -> VectorStoreIndexWrapper:
         """Asynchronously create a vectorstore index from a list of loaders.
 
         Args:
@@ -220,7 +220,7 @@ class VectorstoreIndexCreator(BaseModel):
                 docs.append(doc)
         return await self.afrom_documents(docs)
 
-    def from_documents(self, documents: List[Document]) -> VectorStoreIndexWrapper:
+    def from_documents(self, documents: list[Document]) -> VectorStoreIndexWrapper:
         """Create a vectorstore index from a list of documents.
 
         Args:
@@ -236,7 +236,7 @@ class VectorstoreIndexCreator(BaseModel):
         return VectorStoreIndexWrapper(vectorstore=vectorstore)
 
     async def afrom_documents(
-        self, documents: List[Document]
+        self, documents: list[Document]
     ) -> VectorStoreIndexWrapper:
         """Asynchronously create a vectorstore index from a list of documents.
 

langchain/llms/__init__.py

@@ -19,7 +19,7 @@ access to the large language model (**LLM**) APIs and services.
 """  # noqa: E501
 
 import warnings
-from typing import Any, Callable, Dict, Type
+from typing import Any, Callable
 
 from langchain_core._api import LangChainDeprecationWarning
 from langchain_core.language_models.llms import BaseLLM
@@ -557,7 +557,7 @@ def __getattr__(name: str) -> Any:
 
     if name == "type_to_cls_dict":
         # for backwards compatibility
-        type_to_cls_dict: Dict[str, Type[BaseLLM]] = {
+        type_to_cls_dict: dict[str, type[BaseLLM]] = {
             k: v() for k, v in get_type_to_cls_dict().items()
         }
         return type_to_cls_dict
@@ -650,7 +650,7 @@ __all__ = [
 ]
 
 
-def get_type_to_cls_dict() -> Dict[str, Callable[[], Type[BaseLLM]]]:
+def get_type_to_cls_dict() -> dict[str, Callable[[], type[BaseLLM]]]:
     return {
         "ai21": _import_ai21,
         "aleph_alpha": _import_aleph_alpha,

langchain/memory/buffer.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional
+from typing import Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.messages import BaseMessage, get_buffer_string
@@ -43,7 +43,7 @@ class ConversationBufferMemory(BaseChatMemory):
             else await self.abuffer_as_str()
         )
 
-    def _buffer_as_str(self, messages: List[BaseMessage]) -> str:
+    def _buffer_as_str(self, messages: list[BaseMessage]) -> str:
         return get_buffer_string(
             messages,
             human_prefix=self.human_prefix,
@@ -61,27 +61,27 @@ class ConversationBufferMemory(BaseChatMemory):
         return self._buffer_as_str(messages)
 
     @property
-    def buffer_as_messages(self) -> List[BaseMessage]:
+    def buffer_as_messages(self) -> list[BaseMessage]:
         """Exposes the buffer as a list of messages in case return_messages is False."""
         return self.chat_memory.messages
 
-    async def abuffer_as_messages(self) -> List[BaseMessage]:
+    async def abuffer_as_messages(self) -> list[BaseMessage]:
         """Exposes the buffer as a list of messages in case return_messages is False."""
         return await self.chat_memory.aget_messages()
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}
 
-    async def aload_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    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()
         return {self.memory_key: buffer}
@@ -117,7 +117,7 @@ class ConversationStringBufferMemory(BaseMemory):
     memory_key: str = "history"  #: :meta private:
 
     @pre_init
-    def validate_chains(cls, values: Dict) -> Dict:
+    def validate_chains(cls, values: dict) -> dict:
         """Validate that return messages is not True."""
         if values.get("return_messages", False):
             raise ValueError(
@@ -126,21 +126,21 @@ class ConversationStringBufferMemory(BaseMemory):
         return values
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}
 
-    async def aload_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
+    async def aload_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         """Return history buffer."""
         return self.load_memory_variables(inputs)
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Save context from this conversation to buffer."""
         if self.input_key is None:
             prompt_input_key = get_prompt_input_key(inputs, self.memory_variables)
@@ -157,7 +157,7 @@ class ConversationStringBufferMemory(BaseMemory):
         self.buffer += "\n" + "\n".join([human, 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)

langchain/memory/buffer_window.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Union
+from typing import Any, Union
 
 from langchain_core._api import deprecated
 from langchain_core.messages import BaseMessage, get_buffer_string
@@ -28,7 +28,7 @@ class ConversationBufferWindowMemory(BaseChatMemory):
     """Number of messages to store in buffer."""
 
     @property
-    def buffer(self) -> Union[str, List[BaseMessage]]:
+    def buffer(self) -> Union[str, list[BaseMessage]]:
         """String buffer of memory."""
         return self.buffer_as_messages if self.return_messages else self.buffer_as_str
 
@@ -43,18 +43,18 @@ class ConversationBufferWindowMemory(BaseChatMemory):
         )
 
     @property
-    def buffer_as_messages(self) -> List[BaseMessage]:
+    def buffer_as_messages(self) -> list[BaseMessage]:
         """Exposes the buffer as a list of messages in case return_messages is True."""
         return self.chat_memory.messages[-self.k * 2 :] if self.k > 0 else []
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    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

@@ -1,6 +1,6 @@
 import warnings
 from abc import ABC
-from typing import Any, Dict, Optional, Tuple
+from typing import Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.chat_history import (
@@ -41,8 +41,8 @@ class BaseChatMemory(BaseMemory, ABC):
     return_messages: bool = False
 
     def _get_input_output(
-        self, inputs: Dict[str, Any], outputs: Dict[str, str]
-    ) -> Tuple[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)
         else:
@@ -67,7 +67,7 @@ class BaseChatMemory(BaseMemory, ABC):
             output_key = self.output_key
         return inputs[prompt_input_key], outputs[output_key]
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(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)
         self.chat_memory.add_messages(
@@ -78,7 +78,7 @@ class BaseChatMemory(BaseMemory, ABC):
         )
 
     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)

langchain/memory/combined.py

@@ -1,5 +1,5 @@
 import warnings
-from typing import Any, Dict, List, Set
+from typing import Any
 
 from langchain_core.memory import BaseMemory
 from pydantic import field_validator
@@ -10,15 +10,15 @@ from langchain.memory.chat_memory import BaseChatMemory
 class CombinedMemory(BaseMemory):
     """Combining multiple memories' data together."""
 
-    memories: List[BaseMemory]
+    memories: list[BaseMemory]
     """For tracking all the memories that should be accessed."""
 
     @field_validator("memories")
     @classmethod
     def check_repeated_memory_variable(
-        cls, value: List[BaseMemory]
-    ) -> List[BaseMemory]:
-        all_variables: Set[str] = set()
+        cls, value: list[BaseMemory]
+    ) -> list[BaseMemory]:
+        all_variables: set[str] = set()
         for val in value:
             overlap = all_variables.intersection(val.memory_variables)
             if overlap:
@@ -32,7 +32,7 @@ class CombinedMemory(BaseMemory):
 
     @field_validator("memories")
     @classmethod
-    def check_input_key(cls, value: List[BaseMemory]) -> List[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):
@@ -45,7 +45,7 @@ class CombinedMemory(BaseMemory):
         return value
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """All the memory variables that this instance provides."""
         """Collected from the all the linked memories."""
 
@@ -56,9 +56,9 @@ class CombinedMemory(BaseMemory):
 
         return memory_variables
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         """Load all vars from sub-memories."""
-        memory_data: Dict[str, Any] = {}
+        memory_data: dict[str, Any] = {}
 
         # Collect vars from all sub-memories
         for memory in self.memories:
@@ -72,7 +72,7 @@ class CombinedMemory(BaseMemory):
 
         return memory_data
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Save context from this session for every memory."""
         # Save context for all sub-memories
         for memory in self.memories:

langchain/memory/entity.py

@@ -2,8 +2,9 @@
 
 import logging
 from abc import ABC, abstractmethod
+from collections.abc import Iterable
 from itertools import islice
-from typing import Any, Dict, Iterable, List, Optional
+from typing import Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.language_models import BaseLanguageModel
@@ -70,7 +71,7 @@ class BaseEntityStore(BaseModel, ABC):
 class InMemoryEntityStore(BaseEntityStore):
     """In-memory Entity store."""
 
-    store: Dict[str, Optional[str]] = {}
+    store: dict[str, Optional[str]] = {}
 
     def get(self, key: str, default: Optional[str] = None) -> Optional[str]:
         return self.store.get(key, default)
@@ -403,7 +404,7 @@ class ConversationEntityMemory(BaseChatMemory):
 
     # Cache of recently detected entity names, if any
     # It is updated when load_memory_variables is called:
-    entity_cache: List[str] = []
+    entity_cache: list[str] = []
 
     # Number of recent message pairs to consider when updating entities:
     k: int = 3
@@ -414,19 +415,19 @@ class ConversationEntityMemory(BaseChatMemory):
     entity_store: BaseEntityStore = Field(default_factory=InMemoryEntityStore)
 
     @property
-    def buffer(self) -> List[BaseMessage]:
+    def buffer(self) -> list[BaseMessage]:
         """Access chat memory messages."""
         return self.chat_memory.messages
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return ["entities", self.chat_history_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """
         Returns chat history and all generated entities with summaries if available,
         and updates or clears the recent entity cache.
@@ -491,7 +492,7 @@ class ConversationEntityMemory(BaseChatMemory):
             "entities": entity_summaries,
         }
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """
         Save context from this conversation history to the entity store.
 

langchain/memory/readonly.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.memory import BaseMemory
 
@@ -9,15 +9,15 @@ class ReadOnlySharedMemory(BaseMemory):
     memory: BaseMemory
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Return memory variables."""
         return self.memory.memory_variables
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         """Load memory variables from memory."""
         return self.memory.load_memory_variables(inputs)
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Nothing should be saved or changed"""
         pass
 

langchain/memory/simple.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.memory import BaseMemory
 
@@ -8,16 +8,16 @@ class SimpleMemory(BaseMemory):
     ever change between prompts.
     """
 
-    memories: Dict[str, Any] = dict()
+    memories: dict[str, Any] = dict()
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         return list(self.memories.keys())
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, str]:
         return self.memories
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Nothing should be saved or changed, my memory is set in stone."""
         pass
 

langchain/memory/summary.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List, Type
+from typing import Any
 
 from langchain_core._api import deprecated
 from langchain_core.caches import BaseCache as BaseCache  # For model_rebuild
@@ -32,10 +32,10 @@ class SummarizerMixin(BaseModel):
     ai_prefix: str = "AI"
     llm: BaseLanguageModel
     prompt: BasePromptTemplate = SUMMARY_PROMPT
-    summary_message_cls: Type[BaseMessage] = SystemMessage
+    summary_message_cls: type[BaseMessage] = SystemMessage
 
     def predict_new_summary(
-        self, messages: List[BaseMessage], existing_summary: str
+        self, messages: list[BaseMessage], existing_summary: str
     ) -> str:
         new_lines = get_buffer_string(
             messages,
@@ -47,7 +47,7 @@ class SummarizerMixin(BaseModel):
         return chain.predict(summary=existing_summary, new_lines=new_lines)
 
     async def apredict_new_summary(
-        self, messages: List[BaseMessage], existing_summary: str
+        self, messages: list[BaseMessage], existing_summary: str
     ) -> str:
         new_lines = get_buffer_string(
             messages,
@@ -95,14 +95,14 @@ class ConversationSummaryMemory(BaseChatMemory, SummarizerMixin):
         return obj
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         if self.return_messages:
             buffer: Any = [self.summary_message_cls(content=self.buffer)]
@@ -111,7 +111,7 @@ class ConversationSummaryMemory(BaseChatMemory, SummarizerMixin):
         return {self.memory_key: buffer}
 
     @pre_init
-    def validate_prompt_input_variables(cls, values: Dict) -> Dict:
+    def validate_prompt_input_variables(cls, values: dict) -> dict:
         """Validate that prompt input variables are consistent."""
         prompt_variables = values["prompt"].input_variables
         expected_keys = {"summary", "new_lines"}
@@ -122,7 +122,7 @@ class ConversationSummaryMemory(BaseChatMemory, SummarizerMixin):
             )
         return values
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Save context from this conversation to buffer."""
         super().save_context(inputs, outputs)
         self.buffer = self.predict_new_summary(

langchain/memory/summary_buffer.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Union
+from typing import Any, Union
 
 from langchain_core._api import deprecated
 from langchain_core.messages import BaseMessage, get_buffer_string
@@ -29,28 +29,28 @@ class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
     memory_key: str = "history"
 
     @property
-    def buffer(self) -> Union[str, List[BaseMessage]]:
+    def buffer(self) -> Union[str, list[BaseMessage]]:
         """String buffer of memory."""
         return self.load_memory_variables({})[self.memory_key]
 
-    async def abuffer(self) -> Union[str, List[BaseMessage]]:
+    async def abuffer(self) -> Union[str, list[BaseMessage]]:
         """Async memory buffer."""
         memory_variables = await self.aload_memory_variables({})
         return memory_variables[self.memory_key]
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         buffer = self.chat_memory.messages
         if self.moving_summary_buffer != "":
-            first_messages: List[BaseMessage] = [
+            first_messages: list[BaseMessage] = [
                 self.summary_message_cls(content=self.moving_summary_buffer)
             ]
             buffer = first_messages + buffer
@@ -62,11 +62,11 @@ class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
             )
         return {self.memory_key: final_buffer}
 
-    async def aload_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    async def aload_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Asynchronously return key-value pairs given the text input to the chain."""
         buffer = await self.chat_memory.aget_messages()
         if self.moving_summary_buffer != "":
-            first_messages: List[BaseMessage] = [
+            first_messages: list[BaseMessage] = [
                 self.summary_message_cls(content=self.moving_summary_buffer)
             ]
             buffer = first_messages + buffer
@@ -79,7 +79,7 @@ class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
         return {self.memory_key: final_buffer}
 
     @pre_init
-    def validate_prompt_input_variables(cls, values: Dict) -> Dict:
+    def validate_prompt_input_variables(cls, values: dict) -> dict:
         """Validate that prompt input variables are consistent."""
         prompt_variables = values["prompt"].input_variables
         expected_keys = {"summary", "new_lines"}
@@ -90,13 +90,13 @@ class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):
             )
         return values
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Save context from this conversation to buffer."""
         super().save_context(inputs, outputs)
         self.prune()
 
     async def asave_context(
-        self, inputs: Dict[str, Any], outputs: Dict[str, str]
+        self, inputs: dict[str, Any], outputs: dict[str, str]
     ) -> None:
         """Asynchronously save context from this conversation to buffer."""
         await super().asave_context(inputs, outputs)

langchain/memory/token_buffer.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core._api import deprecated
 from langchain_core.language_models import BaseLanguageModel
@@ -43,23 +43,23 @@ class ConversationTokenBufferMemory(BaseChatMemory):
         )
 
     @property
-    def buffer_as_messages(self) -> List[BaseMessage]:
+    def buffer_as_messages(self) -> list[BaseMessage]:
         """Exposes the buffer as a list of messages in case return_messages is True."""
         return self.chat_memory.messages
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Will always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer."""
         return {self.memory_key: self.buffer}
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
         """Save context from this conversation to buffer. Pruned."""
         super().save_context(inputs, outputs)
         # Prune buffer if it exceeds max token limit

langchain/memory/utils.py

@@ -1,7 +1,7 @@
-from typing import Any, Dict, List
+from typing import Any
 
 
-def get_prompt_input_key(inputs: Dict[str, Any], memory_variables: List[str]) -> str:
+def get_prompt_input_key(inputs: dict[str, Any], memory_variables: list[str]) -> str:
     """
     Get the prompt input key.