langchain 0.3.22__py3-none-any.whl → 0.3.24__py3-none-any.whl

langchain/chains/qa_with_sources/base.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import inspect
 import re
 from abc import ABC, abstractmethod
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.callbacks import (
@@ -69,7 +69,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
             document_prompt=document_prompt,
             document_variable_name="summaries",
         )
-        reduce_documents_chain = ReduceDocumentsChain(  # type: ignore[misc]
+        reduce_documents_chain = ReduceDocumentsChain(
             combine_documents_chain=combine_results_chain
         )
         combine_documents_chain = MapReduceDocumentsChain(
@@ -103,7 +103,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
     )
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Expect input key.
 
         :meta private:
@@ -111,7 +111,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
         return [self.question_key]
 
     @property
-    def output_keys(self) -> List[str]:
+    def output_keys(self) -> list[str]:
         """Return output key.
 
         :meta private:
@@ -123,13 +123,13 @@ class BaseQAWithSourcesChain(Chain, ABC):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_naming(cls, values: Dict) -> Any:
+    def validate_naming(cls, values: dict) -> Any:
         """Fix backwards compatibility in naming."""
         if "combine_document_chain" in values:
             values["combine_documents_chain"] = values.pop("combine_document_chain")
         return values
 
-    def _split_sources(self, answer: str) -> Tuple[str, str]:
+    def _split_sources(self, answer: str) -> tuple[str, str]:
         """Split sources from answer."""
         if re.search(r"SOURCES?:", answer, re.IGNORECASE):
             answer, sources = re.split(
@@ -143,17 +143,17 @@ class BaseQAWithSourcesChain(Chain, ABC):
     @abstractmethod
     def _get_docs(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         *,
         run_manager: CallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs to run questioning over."""
 
     def _call(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> Dict[str, str]:
+    ) -> dict[str, str]:
         _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager()
         accepts_run_manager = (
             "run_manager" in inspect.signature(self._get_docs).parameters
@@ -167,7 +167,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
             input_documents=docs, callbacks=_run_manager.get_child(), **inputs
         )
         answer, sources = self._split_sources(answer)
-        result: Dict[str, Any] = {
+        result: dict[str, Any] = {
             self.answer_key: answer,
             self.sources_answer_key: sources,
         }
@@ -178,17 +178,17 @@ class BaseQAWithSourcesChain(Chain, ABC):
     @abstractmethod
     async def _aget_docs(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         *,
         run_manager: AsyncCallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs to run questioning over."""
 
     async def _acall(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         _run_manager = run_manager or AsyncCallbackManagerForChainRun.get_noop_manager()
         accepts_run_manager = (
             "run_manager" in inspect.signature(self._aget_docs).parameters
@@ -201,7 +201,7 @@ class BaseQAWithSourcesChain(Chain, ABC):
             input_documents=docs, callbacks=_run_manager.get_child(), **inputs
         )
         answer, sources = self._split_sources(answer)
-        result: Dict[str, Any] = {
+        result: dict[str, Any] = {
             self.answer_key: answer,
             self.sources_answer_key: sources,
         }
@@ -225,7 +225,7 @@ class QAWithSourcesChain(BaseQAWithSourcesChain):
     input_docs_key: str = "docs"  #: :meta private:
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Expect input key.
 
         :meta private:
@@ -234,19 +234,19 @@ class QAWithSourcesChain(BaseQAWithSourcesChain):
 
     def _get_docs(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         *,
         run_manager: CallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs to run questioning over."""
         return inputs.pop(self.input_docs_key)
 
     async def _aget_docs(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         *,
         run_manager: AsyncCallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs to run questioning over."""
         return inputs.pop(self.input_docs_key)
 

langchain/chains/qa_with_sources/loading.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Mapping, Optional, Protocol
+from collections.abc import Mapping
+from typing import Any, Optional, Protocol
 
 from langchain_core._api import deprecated
 from langchain_core.language_models import BaseLanguageModel

langchain/chains/qa_with_sources/retrieval.py

@@ -1,6 +1,6 @@
 """Question-answering with sources over an index."""
 
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
@@ -25,7 +25,7 @@ class RetrievalQAWithSourcesChain(BaseQAWithSourcesChain):
     """Restrict the docs to return from store based on tokens,
     enforced only for StuffDocumentChain and if reduce_k_below_max_tokens is to true"""
 
-    def _reduce_tokens_below_limit(self, docs: List[Document]) -> List[Document]:
+    def _reduce_tokens_below_limit(self, docs: list[Document]) -> list[Document]:
         num_docs = len(docs)
 
         if self.reduce_k_below_max_tokens and isinstance(
@@ -43,8 +43,8 @@ class RetrievalQAWithSourcesChain(BaseQAWithSourcesChain):
         return docs[:num_docs]
 
     def _get_docs(
-        self, inputs: Dict[str, Any], *, run_manager: CallbackManagerForChainRun
-    ) -> List[Document]:
+        self, inputs: dict[str, Any], *, run_manager: CallbackManagerForChainRun
+    ) -> list[Document]:
         question = inputs[self.question_key]
         docs = self.retriever.invoke(
             question, config={"callbacks": run_manager.get_child()}
@@ -52,8 +52,8 @@ class RetrievalQAWithSourcesChain(BaseQAWithSourcesChain):
         return self._reduce_tokens_below_limit(docs)
 
     async def _aget_docs(
-        self, inputs: Dict[str, Any], *, run_manager: AsyncCallbackManagerForChainRun
-    ) -> List[Document]:
+        self, inputs: dict[str, Any], *, run_manager: AsyncCallbackManagerForChainRun
+    ) -> list[Document]:
         question = inputs[self.question_key]
         docs = await self.retriever.ainvoke(
             question, config={"callbacks": run_manager.get_child()}

langchain/chains/qa_with_sources/vector_db.py

@@ -1,7 +1,7 @@
 """Question-answering with sources over a vector database."""
 
 import warnings
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
@@ -27,10 +27,10 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain):
     max_tokens_limit: int = 3375
     """Restrict the docs to return from store based on tokens,
     enforced only for StuffDocumentChain and if reduce_k_below_max_tokens is to true"""
-    search_kwargs: Dict[str, Any] = Field(default_factory=dict)
+    search_kwargs: dict[str, Any] = Field(default_factory=dict)
     """Extra search args."""
 
-    def _reduce_tokens_below_limit(self, docs: List[Document]) -> List[Document]:
+    def _reduce_tokens_below_limit(self, docs: list[Document]) -> list[Document]:
         num_docs = len(docs)
 
         if self.reduce_k_below_max_tokens and isinstance(
@@ -48,8 +48,8 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain):
         return docs[:num_docs]
 
     def _get_docs(
-        self, inputs: Dict[str, Any], *, run_manager: CallbackManagerForChainRun
-    ) -> List[Document]:
+        self, inputs: dict[str, Any], *, run_manager: CallbackManagerForChainRun
+    ) -> list[Document]:
         question = inputs[self.question_key]
         docs = self.vectorstore.similarity_search(
             question, k=self.k, **self.search_kwargs
@@ -57,13 +57,13 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain):
         return self._reduce_tokens_below_limit(docs)
 
     async def _aget_docs(
-        self, inputs: Dict[str, Any], *, run_manager: AsyncCallbackManagerForChainRun
-    ) -> List[Document]:
+        self, inputs: dict[str, Any], *, run_manager: AsyncCallbackManagerForChainRun
+    ) -> list[Document]:
         raise NotImplementedError("VectorDBQAWithSourcesChain does not support async")
 
     @model_validator(mode="before")
     @classmethod
-    def raise_deprecation(cls, values: Dict) -> Any:
+    def raise_deprecation(cls, values: dict) -> Any:
         warnings.warn(
             "`VectorDBQAWithSourcesChain` is deprecated - "
             "please use `from langchain.chains import RetrievalQAWithSourcesChain`"

langchain/chains/query_constructor/base.py

@@ -3,7 +3,8 @@
 from __future__ import annotations
 
 import json
-from typing import Any, Callable, List, Optional, Sequence, Tuple, Union, cast
+from collections.abc import Sequence
+from typing import Any, Callable, Optional, Union, cast
 
 from langchain_core._api import deprecated
 from langchain_core.exceptions import OutputParserException
@@ -172,7 +173,7 @@ def _format_attribute_info(info: Sequence[Union[AttributeInfo, dict]]) -> str:
     return json.dumps(info_dicts, indent=4).replace("{", "{{").replace("}", "}}")
 
 
-def construct_examples(input_output_pairs: Sequence[Tuple[str, dict]]) -> List[dict]:
+def construct_examples(input_output_pairs: Sequence[tuple[str, dict]]) -> list[dict]:
     """Construct examples from input-output pairs.
 
     Args:
@@ -267,7 +268,7 @@ def load_query_constructor_chain(
     llm: BaseLanguageModel,
     document_contents: str,
     attribute_info: Sequence[Union[AttributeInfo, dict]],
-    examples: Optional[List] = None,
+    examples: Optional[list] = None,
     allowed_comparators: Sequence[Comparator] = tuple(Comparator),
     allowed_operators: Sequence[Operator] = tuple(Operator),
     enable_limit: bool = False,

langchain/chains/query_constructor/parser.py

@@ -1,6 +1,7 @@
 import datetime
 import warnings
-from typing import Any, Literal, Optional, Sequence, Union
+from collections.abc import Sequence
+from typing import Any, Literal, Optional, Union
 
 from langchain_core.utils import check_package_version
 from typing_extensions import TypedDict
@@ -10,12 +11,12 @@ try:
     from lark import Lark, Transformer, v_args
 except ImportError:
 
-    def v_args(*args: Any, **kwargs: Any) -> Any:  # type: ignore
+    def v_args(*args: Any, **kwargs: Any) -> Any:  # type: ignore[misc]
         """Dummy decorator for when lark is not installed."""
         return lambda _: None
 
-    Transformer = object  # type: ignore
-    Lark = object  # type: ignore
+    Transformer = object  # type: ignore[assignment,misc]
+    Lark = object  # type: ignore[assignment,misc]
 
 from langchain_core.structured_query import (
     Comparator,

langchain/chains/question_answering/chain.py

@@ -1,6 +1,7 @@
 """Load question answering chains."""
 
-from typing import Any, Mapping, Optional, Protocol
+from collections.abc import Mapping
+from typing import Any, Optional, Protocol
 
 from langchain_core._api import deprecated
 from langchain_core.callbacks import BaseCallbackManager, Callbacks
@@ -155,7 +156,7 @@ def _load_map_reduce_chain(
             verbose=verbose,  # type: ignore[arg-type]
             callback_manager=callback_manager,
         )
-    reduce_documents_chain = ReduceDocumentsChain(  # type: ignore[misc]
+    reduce_documents_chain = ReduceDocumentsChain(
         combine_documents_chain=combine_documents_chain,
         collapse_documents_chain=collapse_chain,
         token_max=token_max,

langchain/chains/retrieval.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, Union
+from typing import Any, Union
 
 from langchain_core.retrievers import (
     BaseRetriever,
@@ -11,7 +11,7 @@ from langchain_core.runnables import Runnable, RunnablePassthrough
 
 def create_retrieval_chain(
     retriever: Union[BaseRetriever, Runnable[dict, RetrieverOutput]],
-    combine_docs_chain: Runnable[Dict[str, Any], str],
+    combine_docs_chain: Runnable[dict[str, Any], str],
 ) -> Runnable:
     """Create retrieval chain that retrieves documents and then passes them on.
 

langchain/chains/retrieval_qa/base.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import inspect
 import warnings
 from abc import abstractmethod
-from typing import Any, Dict, List, Optional
+from typing import Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.callbacks import (
@@ -54,7 +54,7 @@ class BaseRetrievalQA(Chain):
     )
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Input keys.
 
         :meta private:
@@ -62,7 +62,7 @@ class BaseRetrievalQA(Chain):
         return [self.input_key]
 
     @property
-    def output_keys(self) -> List[str]:
+    def output_keys(self) -> list[str]:
         """Output keys.
 
         :meta private:
@@ -123,14 +123,14 @@ class BaseRetrievalQA(Chain):
         question: str,
         *,
         run_manager: CallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get documents to do question answering over."""
 
     def _call(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Run get_relevant_text and llm on input query.
 
         If chain has 'return_source_documents' as 'True', returns
@@ -166,14 +166,14 @@ class BaseRetrievalQA(Chain):
         question: str,
         *,
         run_manager: AsyncCallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get documents to do question answering over."""
 
     async def _acall(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Run get_relevant_text and llm on input query.
 
         If chain has 'return_source_documents' as 'True', returns
@@ -266,7 +266,7 @@ class RetrievalQA(BaseRetrievalQA):
         question: str,
         *,
         run_manager: CallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs."""
         return self.retriever.invoke(
             question, config={"callbacks": run_manager.get_child()}
@@ -277,7 +277,7 @@ class RetrievalQA(BaseRetrievalQA):
         question: str,
         *,
         run_manager: AsyncCallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs."""
         return await self.retriever.ainvoke(
             question, config={"callbacks": run_manager.get_child()}
@@ -307,12 +307,12 @@ class VectorDBQA(BaseRetrievalQA):
     """Number of documents to query for."""
     search_type: str = "similarity"
     """Search type to use over vectorstore. `similarity` or `mmr`."""
-    search_kwargs: Dict[str, Any] = Field(default_factory=dict)
+    search_kwargs: dict[str, Any] = Field(default_factory=dict)
     """Extra search args."""
 
     @model_validator(mode="before")
     @classmethod
-    def raise_deprecation(cls, values: Dict) -> Any:
+    def raise_deprecation(cls, values: dict) -> Any:
         warnings.warn(
             "`VectorDBQA` is deprecated - "
             "please use `from langchain.chains import RetrievalQA`"
@@ -321,7 +321,7 @@ class VectorDBQA(BaseRetrievalQA):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_search_type(cls, values: Dict) -> Any:
+    def validate_search_type(cls, values: dict) -> Any:
         """Validate search type."""
         if "search_type" in values:
             search_type = values["search_type"]
@@ -334,7 +334,7 @@ class VectorDBQA(BaseRetrievalQA):
         question: str,
         *,
         run_manager: CallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs."""
         if self.search_type == "similarity":
             docs = self.vectorstore.similarity_search(
@@ -353,7 +353,7 @@ class VectorDBQA(BaseRetrievalQA):
         question: str,
         *,
         run_manager: AsyncCallbackManagerForChainRun,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Get docs."""
         raise NotImplementedError("VectorDBQA does not support async")
 

langchain/chains/router/base.py

@@ -3,7 +3,8 @@
 from __future__ import annotations
 
 from abc import ABC
-from typing import Any, Dict, List, Mapping, NamedTuple, Optional
+from collections.abc import Mapping
+from typing import Any, NamedTuple, Optional
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
@@ -17,17 +18,17 @@ from langchain.chains.base import Chain
 
 class Route(NamedTuple):
     destination: Optional[str]
-    next_inputs: Dict[str, Any]
+    next_inputs: dict[str, Any]
 
 
 class RouterChain(Chain, ABC):
     """Chain that outputs the name of a destination chain and the inputs to it."""
 
     @property
-    def output_keys(self) -> List[str]:
+    def output_keys(self) -> list[str]:
         return ["destination", "next_inputs"]
 
-    def route(self, inputs: Dict[str, Any], callbacks: Callbacks = None) -> Route:
+    def route(self, inputs: dict[str, Any], callbacks: Callbacks = None) -> Route:
         """
         Route inputs to a destination chain.
 
@@ -42,7 +43,7 @@ class RouterChain(Chain, ABC):
         return Route(result["destination"], result["next_inputs"])
 
     async def aroute(
-        self, inputs: Dict[str, Any], callbacks: Callbacks = None
+        self, inputs: dict[str, Any], callbacks: Callbacks = None
     ) -> Route:
         result = await self.acall(inputs, callbacks=callbacks)
         return Route(result["destination"], result["next_inputs"])
@@ -67,7 +68,7 @@ class MultiRouteChain(Chain):
     )
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Will be whatever keys the router chain prompt expects.
 
         :meta private:
@@ -75,7 +76,7 @@ class MultiRouteChain(Chain):
         return self.router_chain.input_keys
 
     @property
-    def output_keys(self) -> List[str]:
+    def output_keys(self) -> list[str]:
         """Will always return text key.
 
         :meta private:
@@ -84,9 +85,9 @@ class MultiRouteChain(Chain):
 
     def _call(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager()
         callbacks = _run_manager.get_child()
         route = self.router_chain.route(inputs, callbacks=callbacks)
@@ -109,9 +110,9 @@ class MultiRouteChain(Chain):
 
     async def _acall(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         _run_manager = run_manager or AsyncCallbackManagerForChainRun.get_noop_manager()
         callbacks = _run_manager.get_child()
         route = await self.router_chain.aroute(inputs, callbacks=callbacks)

langchain/chains/router/embedding_router.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List, Optional, Sequence, Tuple, Type
+from collections.abc import Sequence
+from typing import Any, Optional
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
@@ -18,7 +19,7 @@ class EmbeddingRouterChain(RouterChain):
     """Chain that uses embeddings to route between options."""
 
     vectorstore: VectorStore
-    routing_keys: List[str] = ["query"]
+    routing_keys: list[str] = ["query"]
 
     model_config = ConfigDict(
         arbitrary_types_allowed=True,
@@ -26,7 +27,7 @@ class EmbeddingRouterChain(RouterChain):
     )
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Will be whatever keys the LLM chain prompt expects.
 
         :meta private:
@@ -35,18 +36,18 @@ class EmbeddingRouterChain(RouterChain):
 
     def _call(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         _input = ", ".join([inputs[k] for k in self.routing_keys])
         results = self.vectorstore.similarity_search(_input, k=1)
         return {"next_inputs": inputs, "destination": results[0].metadata["name"]}
 
     async def _acall(
         self,
-        inputs: Dict[str, Any],
+        inputs: dict[str, Any],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         _input = ", ".join([inputs[k] for k in self.routing_keys])
         results = await self.vectorstore.asimilarity_search(_input, k=1)
         return {"next_inputs": inputs, "destination": results[0].metadata["name"]}
@@ -54,8 +55,8 @@ class EmbeddingRouterChain(RouterChain):
     @classmethod
     def from_names_and_descriptions(
         cls,
-        names_and_descriptions: Sequence[Tuple[str, Sequence[str]]],
-        vectorstore_cls: Type[VectorStore],
+        names_and_descriptions: Sequence[tuple[str, Sequence[str]]],
+        vectorstore_cls: type[VectorStore],
         embeddings: Embeddings,
         **kwargs: Any,
     ) -> EmbeddingRouterChain:
@@ -72,8 +73,8 @@ class EmbeddingRouterChain(RouterChain):
     @classmethod
     async def afrom_names_and_descriptions(
         cls,
-        names_and_descriptions: Sequence[Tuple[str, Sequence[str]]],
-        vectorstore_cls: Type[VectorStore],
+        names_and_descriptions: Sequence[tuple[str, Sequence[str]]],
+        vectorstore_cls: type[VectorStore],
         embeddings: Embeddings,
         **kwargs: Any,
     ) -> EmbeddingRouterChain: