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

langchain/memory/vectorstore.py

@@ -1,6 +1,7 @@
 """Class for a VectorStore-backed memory object."""
 
-from typing import Any, Dict, List, Optional, Sequence, Union
+from collections.abc import Sequence
+from typing import Any, Optional, Union
 
 from langchain_core._api import deprecated
 from langchain_core.documents import Document
@@ -40,20 +41,20 @@ class VectorStoreRetrieverMemory(BaseMemory):
     """Input keys to exclude in addition to memory key when constructing the document"""
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """The list of keys emitted from the load_memory_variables method."""
         return [self.memory_key]
 
-    def _get_prompt_input_key(self, inputs: Dict[str, Any]) -> str:
+    def _get_prompt_input_key(self, inputs: dict[str, Any]) -> str:
         """Get the input key for the prompt."""
         if self.input_key is None:
             return get_prompt_input_key(inputs, self.memory_variables)
         return self.input_key
 
     def _documents_to_memory_variables(
-        self, docs: List[Document]
-    ) -> Dict[str, Union[List[Document], str]]:
-        result: Union[List[Document], str]
+        self, docs: list[Document]
+    ) -> dict[str, Union[list[Document], str]]:
+        result: Union[list[Document], str]
         if not self.return_docs:
             result = "\n".join([doc.page_content for doc in docs])
         else:
@@ -61,8 +62,8 @@ class VectorStoreRetrieverMemory(BaseMemory):
         return {self.memory_key: result}
 
     def load_memory_variables(
-        self, inputs: Dict[str, Any]
-    ) -> Dict[str, Union[List[Document], str]]:
+        self, inputs: dict[str, Any]
+    ) -> dict[str, Union[list[Document], str]]:
         """Return history buffer."""
         input_key = self._get_prompt_input_key(inputs)
         query = inputs[input_key]
@@ -70,8 +71,8 @@ class VectorStoreRetrieverMemory(BaseMemory):
         return self._documents_to_memory_variables(docs)
 
     async def aload_memory_variables(
-        self, inputs: Dict[str, Any]
-    ) -> Dict[str, Union[List[Document], str]]:
+        self, inputs: dict[str, Any]
+    ) -> dict[str, Union[list[Document], str]]:
         """Return history buffer."""
         input_key = self._get_prompt_input_key(inputs)
         query = inputs[input_key]
@@ -79,8 +80,8 @@ class VectorStoreRetrieverMemory(BaseMemory):
         return self._documents_to_memory_variables(docs)
 
     def _form_documents(
-        self, inputs: Dict[str, Any], outputs: Dict[str, str]
-    ) -> List[Document]:
+        self, inputs: dict[str, Any], outputs: dict[str, str]
+    ) -> list[Document]:
         """Format context from this conversation to buffer."""
         # Each document should only include the current turn, not the chat history
         exclude = set(self.exclude_input_keys)
@@ -93,13 +94,13 @@ class VectorStoreRetrieverMemory(BaseMemory):
         page_content = "\n".join(texts)
         return [Document(page_content=page_content)]
 
-    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."""
         documents = self._form_documents(inputs, outputs)
         self.retriever.add_documents(documents)
 
     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."""
         documents = self._form_documents(inputs, outputs)

langchain/memory/vectorstore_token_buffer_memory.py

@@ -9,7 +9,7 @@ sessions.
 
 import warnings
 from datetime import datetime
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.messages import BaseMessage
 from langchain_core.prompts.chat import SystemMessagePromptTemplate
@@ -109,8 +109,8 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory):
     previous_history_template: str = DEFAULT_HISTORY_TEMPLATE
     split_chunk_size: int = 1000
 
-    _memory_retriever: VectorStoreRetrieverMemory = PrivateAttr(default=None)  # type: ignore
-    _timestamps: List[datetime] = PrivateAttr(default_factory=list)
+    _memory_retriever: VectorStoreRetrieverMemory = PrivateAttr(default=None)  # type: ignore[assignment]
+    _timestamps: list[datetime] = PrivateAttr(default_factory=list)
 
     @property
     def memory_retriever(self) -> VectorStoreRetrieverMemory:
@@ -120,7 +120,7 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory):
         self._memory_retriever = VectorStoreRetrieverMemory(retriever=self.retriever)
         return self._memory_retriever
 
-    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 and memory buffer."""
         try:
             with warnings.catch_warnings():
@@ -142,7 +142,7 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory):
         messages.extend(current_history[self.memory_key])
         return {self.memory_key: messages}
 
-    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."""
         BaseChatMemory.save_context(self, inputs, outputs)
         self._timestamps.append(datetime.now().astimezone())
@@ -166,7 +166,7 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory):
         while len(buffer) > 0:
             self._pop_and_store_interaction(buffer)
 
-    def _pop_and_store_interaction(self, buffer: List[BaseMessage]) -> None:
+    def _pop_and_store_interaction(self, buffer: list[BaseMessage]) -> None:
         input = buffer.pop(0)
         output = buffer.pop(0)
         timestamp = self._timestamps.pop(0).strftime(TIMESTAMP_FORMAT)
@@ -179,6 +179,6 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory):
                 {"AI": f"<{timestamp}/{index:02}> {chunk}"},
             )
 
-    def _split_long_ai_text(self, text: str) -> List[str]:
+    def _split_long_ai_text(self, text: str) -> list[str]:
         splitter = RecursiveCharacterTextSplitter(chunk_size=self.split_chunk_size)
         return [chunk.page_content for chunk in splitter.create_documents([text])]

langchain/model_laboratory.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import List, Optional, Sequence
+from collections.abc import Sequence
+from typing import Optional
 
 from langchain_core.language_models.llms import BaseLLM
 from langchain_core.prompts.prompt import PromptTemplate
@@ -15,7 +16,7 @@ from langchain.chains.llm import LLMChain
 class ModelLaboratory:
     """A utility to experiment with and compare the performance of different models."""
 
-    def __init__(self, chains: Sequence[Chain], names: Optional[List[str]] = None):
+    def __init__(self, chains: Sequence[Chain], names: Optional[list[str]] = None):
         """Initialize the ModelLaboratory with chains to experiment with.
 
         Args:
@@ -58,7 +59,7 @@ class ModelLaboratory:
 
     @classmethod
     def from_llms(
-        cls, llms: List[BaseLLM], prompt: Optional[PromptTemplate] = None
+        cls, llms: list[BaseLLM], prompt: Optional[PromptTemplate] = None
     ) -> ModelLaboratory:
         """Initialize the ModelLaboratory with LLMs and an optional prompt.
 

langchain/output_parsers/combining.py

@@ -1,22 +1,22 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.utils import pre_init
 
 
-class CombiningOutputParser(BaseOutputParser[Dict[str, Any]]):
+class CombiningOutputParser(BaseOutputParser[dict[str, Any]]):
     """Combine multiple output parsers into one."""
 
-    parsers: List[BaseOutputParser]
+    parsers: list[BaseOutputParser]
 
     @classmethod
     def is_lc_serializable(cls) -> bool:
         return True
 
     @pre_init
-    def validate_parsers(cls, values: Dict[str, Any]) -> Dict[str, Any]:
+    def validate_parsers(cls, values: dict[str, Any]) -> dict[str, Any]:
         """Validate the parsers."""
         parsers = values["parsers"]
         if len(parsers) < 2:
@@ -43,7 +43,7 @@ class CombiningOutputParser(BaseOutputParser[Dict[str, Any]]):
         )
         return f"{initial}\n{subsequent}"
 
-    def parse(self, text: str) -> Dict[str, Any]:
+    def parse(self, text: str) -> dict[str, Any]:
         """Parse the output of an LLM call."""
         texts = text.split("\n\n")
         output = dict()

langchain/output_parsers/datetime.py

@@ -1,6 +1,5 @@
 import random
 from datetime import datetime, timedelta
-from typing import List
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import BaseOutputParser
@@ -12,7 +11,7 @@ def _generate_random_datetime_strings(
     n: int = 3,
     start_date: datetime = datetime(1, 1, 1),
     end_date: datetime = datetime.now() + timedelta(days=3650),
-) -> List[str]:
+) -> list[str]:
     """Generates n random datetime strings conforming to the
     given pattern within the specified date range.
 

langchain/output_parsers/enum.py

@@ -1,5 +1,4 @@
 from enum import Enum
-from typing import Dict, List, Type
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import BaseOutputParser
@@ -9,18 +8,18 @@ from langchain_core.utils import pre_init
 class EnumOutputParser(BaseOutputParser[Enum]):
     """Parse an output that is one of a set of values."""
 
-    enum: Type[Enum]
+    enum: type[Enum]
     """The enum to parse. Its values must be strings."""
 
     @pre_init
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    def raise_deprecation(cls, values: dict) -> dict:
         enum = values["enum"]
         if not all(isinstance(e.value, str) for e in enum):
             raise ValueError("Enum values must be strings")
         return values
 
     @property
-    def _valid_values(self) -> List[str]:
+    def _valid_values(self) -> list[str]:
         return [e.value for e in self.enum]
 
     def parse(self, response: str) -> Enum:
@@ -36,5 +35,5 @@ class EnumOutputParser(BaseOutputParser[Enum]):
         return f"Select one of the following options: {', '.join(self._valid_values)}"
 
     @property
-    def OutputType(self) -> Type[Enum]:
+    def OutputType(self) -> type[Enum]:
         return self.enum

langchain/output_parsers/pandas_dataframe.py

@@ -1,5 +1,5 @@
 import re
-from typing import Any, Dict, List, Tuple, Union
+from typing import Any, Union
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers.base import BaseOutputParser
@@ -10,7 +10,7 @@ from langchain.output_parsers.format_instructions import (
 )
 
 
-class PandasDataFrameOutputParser(BaseOutputParser[Dict[str, Any]]):
+class PandasDataFrameOutputParser(BaseOutputParser[dict[str, Any]]):
     """Parse an output using Pandas DataFrame format."""
 
     """The Pandas DataFrame to parse."""
@@ -33,8 +33,8 @@ class PandasDataFrameOutputParser(BaseOutputParser[Dict[str, Any]]):
 
     def parse_array(
         self, array: str, original_request_params: str
-    ) -> Tuple[List[Union[int, str]], str]:
-        parsed_array: List[Union[int, str]] = []
+    ) -> tuple[list[Union[int, str]], str]:
+        parsed_array: list[Union[int, str]] = []
 
         # Check if the format is [1,3,5]
         if re.match(r"\[\d+(,\s*\d+)*\]", array):
@@ -78,7 +78,7 @@ class PandasDataFrameOutputParser(BaseOutputParser[Dict[str, Any]]):
 
         return parsed_array, original_request_params.split("[")[0]
 
-    def parse(self, request: str) -> Dict[str, Any]:
+    def parse(self, request: str) -> dict[str, Any]:
         stripped_request_params = None
         splitted_request = request.strip().split(":")
         if len(splitted_request) != 2:

langchain/output_parsers/regex.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import re
-from typing import Dict, List, Optional
+from typing import Optional
 
 from langchain_core.output_parsers import BaseOutputParser
 
 
-class RegexParser(BaseOutputParser[Dict[str, str]]):
+class RegexParser(BaseOutputParser[dict[str, str]]):
     """Parse the output of an LLM call using a regex."""
 
     @classmethod
@@ -15,7 +15,7 @@ class RegexParser(BaseOutputParser[Dict[str, str]]):
 
     regex: str
     """The regex to use to parse the output."""
-    output_keys: List[str]
+    output_keys: list[str]
     """The keys to use for the output."""
     default_output_key: Optional[str] = None
     """The default key to use for the output."""
@@ -25,7 +25,7 @@ class RegexParser(BaseOutputParser[Dict[str, str]]):
         """Return the type key."""
         return "regex_parser"
 
-    def parse(self, text: str) -> Dict[str, str]:
+    def parse(self, text: str) -> dict[str, str]:
         """Parse the output of an LLM call."""
         match = re.search(self.regex, text)
         if match:

langchain/output_parsers/regex_dict.py

@@ -1,17 +1,17 @@
 from __future__ import annotations
 
 import re
-from typing import Dict, Optional
+from typing import Optional
 
 from langchain_core.output_parsers import BaseOutputParser
 
 
-class RegexDictParser(BaseOutputParser[Dict[str, str]]):
+class RegexDictParser(BaseOutputParser[dict[str, str]]):
     """Parse the output of an LLM call into a Dictionary using a regex."""
 
     regex_pattern: str = r"{}:\s?([^.'\n']*)\.?"  # : :meta private:
     """The regex pattern to use to parse the output."""
-    output_key_to_format: Dict[str, str]
+    output_key_to_format: dict[str, str]
     """The keys to use for the output."""
     no_update_value: Optional[str] = None
     """The default key to use for the output."""
@@ -21,7 +21,7 @@ class RegexDictParser(BaseOutputParser[Dict[str, str]]):
         """Return the type key."""
         return "regex_dict_parser"
 
-    def parse(self, text: str) -> Dict[str, str]:
+    def parse(self, text: str) -> dict[str, str]:
         """Parse the output of an LLM call."""
         result = {}
         for output_key, expected_format in self.output_key_to_format.items():

langchain/output_parsers/retry.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, TypeVar, Union
+from typing import Annotated, Any, TypeVar, Union
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.language_models import BaseLanguageModel
@@ -9,7 +9,7 @@ from langchain_core.prompt_values import PromptValue
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
 from langchain_core.runnables import RunnableSerializable
 from pydantic import SkipValidation
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 NAIVE_COMPLETION_RETRY = """Prompt:
 {prompt}

langchain/output_parsers/structured.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.output_parsers.json import parse_and_check_json_markdown
@@ -31,15 +31,15 @@ def _get_sub_string(schema: ResponseSchema) -> str:
     )
 
 
-class StructuredOutputParser(BaseOutputParser[Dict[str, Any]]):
+class StructuredOutputParser(BaseOutputParser[dict[str, Any]]):
     """Parse the output of an LLM call to a structured output."""
 
-    response_schemas: List[ResponseSchema]
+    response_schemas: list[ResponseSchema]
     """The schemas for the response."""
 
     @classmethod
     def from_response_schemas(
-        cls, response_schemas: List[ResponseSchema]
+        cls, response_schemas: list[ResponseSchema]
     ) -> StructuredOutputParser:
         return cls(response_schemas=response_schemas)
 
@@ -92,7 +92,7 @@ class StructuredOutputParser(BaseOutputParser[Dict[str, Any]]):
         else:
             return STRUCTURED_FORMAT_INSTRUCTIONS.format(format=schema_str)
 
-    def parse(self, text: str) -> Dict[str, Any]:
+    def parse(self, text: str) -> dict[str, Any]:
         expected_keys = [rs.name for rs in self.response_schemas]
         return parse_and_check_json_markdown(text, expected_keys)
 

langchain/output_parsers/yaml.py

@@ -1,6 +1,6 @@
 import json
 import re
-from typing import Type, TypeVar
+from typing import TypeVar
 
 import yaml
 from langchain_core.exceptions import OutputParserException
@@ -15,7 +15,7 @@ T = TypeVar("T", bound=BaseModel)
 class YamlOutputParser(BaseOutputParser[T]):
     """Parse YAML output using a pydantic model."""
 
-    pydantic_object: Type[T]
+    pydantic_object: type[T]
     """The pydantic model to parse."""
     pattern: re.Pattern = re.compile(
         r"^```(?:ya?ml)?(?P<yaml>[^`]*)", re.MULTILINE | re.DOTALL
@@ -65,5 +65,5 @@ class YamlOutputParser(BaseOutputParser[T]):
         return "yaml"
 
     @property
-    def OutputType(self) -> Type[T]:
+    def OutputType(self) -> type[T]:
         return self.pydantic_object

langchain/pydantic_v1/__init__.py

@@ -12,12 +12,7 @@ from langchain_core._api import warn_deprecated
 # * Creating namespaces for pydantic v1 and v2 should allow us to write code that
 #   unambiguously uses either v1 or v2 API.
 # * This change is easier to roll out and roll back.
-
-try:
-    from pydantic.v1 import *  # noqa: F403
-except ImportError:
-    from pydantic import *  # type: ignore # noqa: F403
-
+from pydantic.v1 import *  # noqa: F403
 
 try:
     _PYDANTIC_MAJOR_VERSION: int = int(metadata.version("pydantic").split(".")[0])

langchain/pydantic_v1/dataclasses.py

@@ -1,9 +1,5 @@
 from langchain_core._api import warn_deprecated
-
-try:
-    from pydantic.v1.dataclasses import *  # noqa: F403
-except ImportError:
-    from pydantic.dataclasses import *  # type: ignore # noqa: F403
+from pydantic.v1.dataclasses import *  # noqa: F403
 
 warn_deprecated(
     "0.3.0",

langchain/pydantic_v1/main.py

@@ -1,9 +1,5 @@
 from langchain_core._api import warn_deprecated
-
-try:
-    from pydantic.v1.main import *  # noqa: F403
-except ImportError:
-    from pydantic.main import *  # type: ignore # noqa: F403
+from pydantic.v1.main import *  # noqa: F403
 
 warn_deprecated(
     "0.3.0",

langchain/retrievers/contextual_compression.py

@@ -1,4 +1,4 @@
-from typing import Any, List
+from typing import Any
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForRetrieverRun,
@@ -32,7 +32,7 @@ class ContextualCompressionRetriever(BaseRetriever):
         *,
         run_manager: CallbackManagerForRetrieverRun,
         **kwargs: Any,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get documents relevant for a query.
 
         Args:
@@ -58,7 +58,7 @@ class ContextualCompressionRetriever(BaseRetriever):
         *,
         run_manager: AsyncCallbackManagerForRetrieverRun,
         **kwargs: Any,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get documents relevant for a query.
 
         Args:

langchain/retrievers/document_compressors/base.py

@@ -1,5 +1,6 @@
+from collections.abc import Sequence
 from inspect import signature
-from typing import List, Optional, Sequence, Union
+from typing import Optional, Union
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.documents import (
@@ -13,7 +14,7 @@ from pydantic import ConfigDict
 class DocumentCompressorPipeline(BaseDocumentCompressor):
     """Document compressor that uses a pipeline of Transformers."""
 
-    transformers: List[Union[BaseDocumentTransformer, BaseDocumentCompressor]]
+    transformers: list[Union[BaseDocumentTransformer, BaseDocumentCompressor]]
     """List of document filters that are chained together and run in sequence."""
 
     model_config = ConfigDict(

langchain/retrievers/document_compressors/chain_extract.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Callable, Dict, Optional, Sequence, cast
+from collections.abc import Sequence
+from typing import Any, Callable, Optional, cast
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.documents import Document
@@ -19,7 +20,7 @@ from langchain.retrievers.document_compressors.chain_extract_prompt import (
 )
 
 
-def default_get_input(query: str, doc: Document) -> Dict[str, Any]:
+def default_get_input(query: str, doc: Document) -> dict[str, Any]:
     """Return the compression chain input."""
     return {"question": query, "context": doc.page_content}
 
@@ -98,7 +99,7 @@ class LLMChainExtractor(BaseDocumentCompressor):
             if len(outputs[i]) == 0:
                 continue
             compressed_docs.append(
-                Document(page_content=outputs[i], metadata=doc.metadata)  # type: ignore[arg-type]
+                Document(page_content=outputs[i], metadata=doc.metadata)
             )
         return compressed_docs
 

langchain/retrievers/document_compressors/chain_filter.py

@@ -1,6 +1,7 @@
 """Filter that uses an LLM to drop documents that aren't relevant to the query."""
 
-from typing import Any, Callable, Dict, Optional, Sequence
+from collections.abc import Sequence
+from typing import Any, Callable, Optional
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.documents import Document
@@ -27,7 +28,7 @@ def _get_default_chain_prompt() -> PromptTemplate:
     )
 
 
-def default_get_input(query: str, doc: Document) -> Dict[str, Any]:
+def default_get_input(query: str, doc: Document) -> dict[str, Any]:
     """Return the compression chain input."""
     return {"question": query, "context": doc.page_content}
 

langchain/retrievers/document_compressors/cohere_rerank.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
+from collections.abc import Sequence
 from copy import deepcopy
-from typing import Any, Dict, List, Optional, Sequence, Union
+from typing import Any, Optional, Union
 
 from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks.manager import Callbacks
@@ -37,7 +38,7 @@ class CohereRerank(BaseDocumentCompressor):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_environment(cls, values: Dict) -> Any:
+    def validate_environment(cls, values: dict) -> Any:
         """Validate that api key and python package exists in environment."""
         if not values.get("client"):
             try:
@@ -62,7 +63,7 @@ class CohereRerank(BaseDocumentCompressor):
         model: Optional[str] = None,
         top_n: Optional[int] = -1,
         max_chunks_per_doc: Optional[int] = None,
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Returns an ordered list of documents ordered by their relevance to the provided query.
 
         Args:

langchain/retrievers/document_compressors/cross_encoder.py

@@ -1,12 +1,11 @@
 from abc import ABC, abstractmethod
-from typing import List, Tuple
 
 
 class BaseCrossEncoder(ABC):
     """Interface for cross encoder models."""
 
     @abstractmethod
-    def score(self, text_pairs: List[Tuple[str, str]]) -> List[float]:
+    def score(self, text_pairs: list[tuple[str, str]]) -> list[float]:
         """Score pairs' similarity.
 
         Args:

langchain/retrievers/document_compressors/cross_encoder_rerank.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
 import operator
-from typing import Optional, Sequence
+from collections.abc import Sequence
+from typing import Optional
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import BaseDocumentCompressor, Document

langchain/retrievers/document_compressors/embeddings_filter.py

@@ -1,4 +1,5 @@
-from typing import Callable, Dict, Optional, Sequence
+from collections.abc import Sequence
+from typing import Callable, Optional
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.documents import Document
@@ -45,7 +46,7 @@ class EmbeddingsFilter(BaseDocumentCompressor):
     )
 
     @pre_init
-    def validate_params(cls, values: Dict) -> Dict:
+    def validate_params(cls, values: dict) -> dict:
         """Validate similarity parameters."""
         if values["k"] is None and values["similarity_threshold"] is None:
             raise ValueError("Must specify one of `k` or `similarity_threshold`.")