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

langchain/chat_models/base.py

@@ -1,19 +1,13 @@
 from __future__ import annotations
 
 import warnings
+from collections.abc import AsyncIterator, Iterator, Sequence
 from importlib import util
 from typing import (
     Any,
-    AsyncIterator,
     Callable,
-    Dict,
-    Iterator,
-    List,
     Literal,
     Optional,
-    Sequence,
-    Tuple,
-    Type,
     Union,
     cast,
     overload,
@@ -47,7 +41,7 @@ __all__ = [
 
 
 @overload
-def init_chat_model(  # type: ignore[overload-overlap]
+def init_chat_model(
     model: str,
     *,
     model_provider: Optional[str] = None,
@@ -73,7 +67,7 @@ def init_chat_model(
     model: Optional[str] = None,
     *,
     model_provider: Optional[str] = None,
-    configurable_fields: Union[Literal["any"], List[str], Tuple[str, ...]] = ...,
+    configurable_fields: Union[Literal["any"], list[str], tuple[str, ...]] = ...,
     config_prefix: Optional[str] = None,
     **kwargs: Any,
 ) -> _ConfigurableModel: ...
@@ -87,7 +81,7 @@ def init_chat_model(
     *,
     model_provider: Optional[str] = None,
     configurable_fields: Optional[
-        Union[Literal["any"], List[str], Tuple[str, ...]]
+        Union[Literal["any"], list[str], tuple[str, ...]]
     ] = None,
     config_prefix: Optional[str] = None,
     **kwargs: Any,
@@ -139,6 +133,7 @@ def init_chat_model(
             - 'mistral...'                      -> 'mistralai'
             - 'deepseek...'                     -> 'deepseek'
             - 'grok...'                         -> 'xai'
+            - 'sonar...'                        -> 'perplexity'
         configurable_fields: Which model parameters are
             configurable:
 
@@ -352,7 +347,7 @@ def _init_chat_model_helper(
         _check_pkg("langchain_anthropic")
         from langchain_anthropic import ChatAnthropic
 
-        return ChatAnthropic(model=model, **kwargs)  # type: ignore[call-arg]
+        return ChatAnthropic(model=model, **kwargs)  # type: ignore[call-arg,unused-ignore]
     elif model_provider == "azure_openai":
         _check_pkg("langchain_openai")
         from langchain_openai import AzureChatOpenAI
@@ -407,7 +402,7 @@ def _init_chat_model_helper(
         _check_pkg("langchain_mistralai")
         from langchain_mistralai import ChatMistralAI
 
-        return ChatMistralAI(model=model, **kwargs)  # type: ignore[call-arg]
+        return ChatMistralAI(model=model, **kwargs)  # type: ignore[call-arg,unused-ignore]
     elif model_provider == "huggingface":
         _check_pkg("langchain_huggingface")
         from langchain_huggingface import ChatHuggingFace
@@ -510,11 +505,13 @@ def _attempt_infer_model_provider(model_name: str) -> Optional[str]:
         return "deepseek"
     elif model_name.startswith("grok"):
         return "xai"
+    elif model_name.startswith("sonar"):
+        return "perplexity"
     else:
         return None
 
 
-def _parse_model(model: str, model_provider: Optional[str]) -> Tuple[str, str]:
+def _parse_model(model: str, model_provider: Optional[str]) -> tuple[str, str]:
     if (
         not model_provider
         and ":" in model
@@ -554,12 +551,12 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
         self,
         *,
         default_config: Optional[dict] = None,
-        configurable_fields: Union[Literal["any"], List[str], Tuple[str, ...]] = "any",
+        configurable_fields: Union[Literal["any"], list[str], tuple[str, ...]] = "any",
         config_prefix: str = "",
-        queued_declarative_operations: Sequence[Tuple[str, Tuple, Dict]] = (),
+        queued_declarative_operations: Sequence[tuple[str, tuple, dict]] = (),
     ) -> None:
         self._default_config: dict = default_config or {}
-        self._configurable_fields: Union[Literal["any"], List[str]] = (
+        self._configurable_fields: Union[Literal["any"], list[str]] = (
             configurable_fields
             if configurable_fields == "any"
             else list(configurable_fields)
@@ -569,7 +566,7 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
             if config_prefix and not config_prefix.endswith("_")
             else config_prefix
         )
-        self._queued_declarative_operations: List[Tuple[str, Tuple, Dict]] = list(
+        self._queued_declarative_operations: list[tuple[str, tuple, dict]] = list(
             queued_declarative_operations
         )
 
@@ -670,7 +667,7 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
         return Union[
             str,
             Union[StringPromptValue, ChatPromptValueConcrete],
-            List[AnyMessage],
+            list[AnyMessage],
         ]
 
     def invoke(
@@ -708,12 +705,12 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
 
     def batch(
         self,
-        inputs: List[LanguageModelInput],
-        config: Optional[Union[RunnableConfig, List[RunnableConfig]]] = None,
+        inputs: list[LanguageModelInput],
+        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
         *,
         return_exceptions: bool = False,
         **kwargs: Optional[Any],
-    ) -> List[Any]:
+    ) -> list[Any]:
         config = config or None
         # If <= 1 config use the underlying models batch implementation.
         if config is None or isinstance(config, dict) or len(config) <= 1:
@@ -731,12 +728,12 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
 
     async def abatch(
         self,
-        inputs: List[LanguageModelInput],
-        config: Optional[Union[RunnableConfig, List[RunnableConfig]]] = None,
+        inputs: list[LanguageModelInput],
+        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
         *,
         return_exceptions: bool = False,
         **kwargs: Optional[Any],
-    ) -> List[Any]:
+    ) -> list[Any]:
         config = config or None
         # If <= 1 config use the underlying models batch implementation.
         if config is None or isinstance(config, dict) or len(config) <= 1:
@@ -759,7 +756,7 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
         *,
         return_exceptions: bool = False,
         **kwargs: Any,
-    ) -> Iterator[Tuple[int, Union[Any, Exception]]]:
+    ) -> Iterator[tuple[int, Union[Any, Exception]]]:
         config = config or None
         # If <= 1 config use the underlying models batch implementation.
         if config is None or isinstance(config, dict) or len(config) <= 1:
@@ -782,7 +779,7 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
         *,
         return_exceptions: bool = False,
         **kwargs: Any,
-    ) -> AsyncIterator[Tuple[int, Any]]:
+    ) -> AsyncIterator[tuple[int, Any]]:
         config = config or None
         # If <= 1 config use the underlying models batch implementation.
         if config is None or isinstance(config, dict) or len(config) <= 1:
@@ -808,8 +805,7 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
         config: Optional[RunnableConfig] = None,
         **kwargs: Optional[Any],
     ) -> Iterator[Any]:
-        for x in self._model(config).transform(input, config=config, **kwargs):
-            yield x
+        yield from self._model(config).transform(input, config=config, **kwargs)
 
     async def atransform(
         self,
@@ -915,13 +911,13 @@ class _ConfigurableModel(Runnable[LanguageModelInput, Any]):
     # Explicitly added to satisfy downstream linters.
     def bind_tools(
         self,
-        tools: Sequence[Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]],
+        tools: Sequence[Union[dict[str, Any], type[BaseModel], Callable, BaseTool]],
         **kwargs: Any,
     ) -> Runnable[LanguageModelInput, BaseMessage]:
         return self.__getattr__("bind_tools")(tools, **kwargs)
 
     # Explicitly added to satisfy downstream linters.
     def with_structured_output(
-        self, schema: Union[Dict, Type[BaseModel]], **kwargs: Any
-    ) -> Runnable[LanguageModelInput, Union[Dict, BaseModel]]:
+        self, schema: Union[dict, type[BaseModel]], **kwargs: Any
+    ) -> Runnable[LanguageModelInput, Union[dict, BaseModel]]:
         return self.__getattr__("with_structured_output")(schema, **kwargs)

langchain/embeddings/__init__.py

@@ -87,7 +87,7 @@ class HypotheticalDocumentEmbedder:
         )
         from langchain.chains.hyde.base import HypotheticalDocumentEmbedder as H
 
-        return H(*args, **kwargs)  # type: ignore
+        return H(*args, **kwargs)  # type: ignore[return-value]
 
     @classmethod
     def from_llm(cls, *args: Any, **kwargs: Any) -> Any:

langchain/embeddings/base.py

@@ -1,6 +1,6 @@
 import functools
 from importlib import util
-from typing import Any, List, Optional, Tuple, Union
+from typing import Any, Optional, Union
 
 from langchain_core._api import beta
 from langchain_core.embeddings import Embeddings
@@ -25,7 +25,7 @@ def _get_provider_list() -> str:
     )
 
 
-def _parse_model_string(model_name: str) -> Tuple[str, str]:
+def _parse_model_string(model_name: str) -> tuple[str, str]:
     """Parse a model string into provider and model name components.
 
     The model string should be in the format 'provider:model-name', where provider
@@ -78,7 +78,7 @@ def _parse_model_string(model_name: str) -> Tuple[str, str]:
 
 def _infer_model_and_provider(
     model: str, *, provider: Optional[str] = None
-) -> Tuple[str, str]:
+) -> tuple[str, str]:
     if not model.strip():
         raise ValueError("Model name cannot be empty")
     if provider is None and ":" in model:
@@ -122,7 +122,7 @@ def init_embeddings(
     *,
     provider: Optional[str] = None,
     **kwargs: Any,
-) -> Union[Embeddings, Runnable[Any, List[float]]]:
+) -> Union[Embeddings, Runnable[Any, list[float]]]:
     """Initialize an embeddings model from a model name and optional provider.
 
     **Note:** Must have the integration package corresponding to the model provider

langchain/embeddings/cache.py

@@ -12,8 +12,9 @@ from __future__ import annotations
 import hashlib
 import json
 import uuid
+from collections.abc import Sequence
 from functools import partial
-from typing import Callable, List, Optional, Sequence, Union, cast
+from typing import Callable, Optional, Union, cast
 
 from langchain_core.embeddings import Embeddings
 from langchain_core.stores import BaseStore, ByteStore
@@ -45,9 +46,9 @@ def _value_serializer(value: Sequence[float]) -> bytes:
     return json.dumps(value).encode()
 
 
-def _value_deserializer(serialized_value: bytes) -> List[float]:
+def _value_deserializer(serialized_value: bytes) -> list[float]:
     """Deserialize a value."""
-    return cast(List[float], json.loads(serialized_value.decode()))
+    return cast(list[float], json.loads(serialized_value.decode()))
 
 
 class CacheBackedEmbeddings(Embeddings):
@@ -88,10 +89,10 @@ class CacheBackedEmbeddings(Embeddings):
     def __init__(
         self,
         underlying_embeddings: Embeddings,
-        document_embedding_store: BaseStore[str, List[float]],
+        document_embedding_store: BaseStore[str, list[float]],
         *,
         batch_size: Optional[int] = None,
-        query_embedding_store: Optional[BaseStore[str, List[float]]] = None,
+        query_embedding_store: Optional[BaseStore[str, list[float]]] = None,
     ) -> None:
         """Initialize the embedder.
 
@@ -108,7 +109,7 @@ class CacheBackedEmbeddings(Embeddings):
         self.underlying_embeddings = underlying_embeddings
         self.batch_size = batch_size
 
-    def embed_documents(self, texts: List[str]) -> List[List[float]]:
+    def embed_documents(self, texts: list[str]) -> list[list[float]]:
         """Embed a list of texts.
 
         The method first checks the cache for the embeddings.
@@ -121,10 +122,10 @@ class CacheBackedEmbeddings(Embeddings):
         Returns:
             A list of embeddings for the given texts.
         """
-        vectors: List[Union[List[float], None]] = self.document_embedding_store.mget(
+        vectors: list[Union[list[float], None]] = self.document_embedding_store.mget(
             texts
         )
-        all_missing_indices: List[int] = [
+        all_missing_indices: list[int] = [
             i for i, vector in enumerate(vectors) if vector is None
         ]
 
@@ -138,10 +139,10 @@ class CacheBackedEmbeddings(Embeddings):
                 vectors[index] = updated_vector
 
         return cast(
-            List[List[float]], vectors
+            list[list[float]], vectors
         )  # Nones should have been resolved by now
 
-    async def aembed_documents(self, texts: List[str]) -> List[List[float]]:
+    async def aembed_documents(self, texts: list[str]) -> list[list[float]]:
         """Embed a list of texts.
 
         The method first checks the cache for the embeddings.
@@ -154,10 +155,10 @@ class CacheBackedEmbeddings(Embeddings):
         Returns:
             A list of embeddings for the given texts.
         """
-        vectors: List[
-            Union[List[float], None]
+        vectors: list[
+            Union[list[float], None]
         ] = await self.document_embedding_store.amget(texts)
-        all_missing_indices: List[int] = [
+        all_missing_indices: list[int] = [
             i for i, vector in enumerate(vectors) if vector is None
         ]
 
@@ -175,10 +176,10 @@ class CacheBackedEmbeddings(Embeddings):
                 vectors[index] = updated_vector
 
         return cast(
-            List[List[float]], vectors
+            list[list[float]], vectors
         )  # Nones should have been resolved by now
 
-    def embed_query(self, text: str) -> List[float]:
+    def embed_query(self, text: str) -> list[float]:
         """Embed query text.
 
         By default, this method does not cache queries. To enable caching, set the
@@ -201,7 +202,7 @@ class CacheBackedEmbeddings(Embeddings):
         self.query_embedding_store.mset([(text, vector)])
         return vector
 
-    async def aembed_query(self, text: str) -> List[float]:
+    async def aembed_query(self, text: str) -> list[float]:
         """Embed query text.
 
         By default, this method does not cache queries. To enable caching, set the
@@ -250,7 +251,7 @@ class CacheBackedEmbeddings(Embeddings):
         """
         namespace = namespace
         key_encoder = _create_key_encoder(namespace)
-        document_embedding_store = EncoderBackedStore[str, List[float]](
+        document_embedding_store = EncoderBackedStore[str, list[float]](
             document_embedding_cache,
             key_encoder,
             _value_serializer,
@@ -261,7 +262,7 @@ class CacheBackedEmbeddings(Embeddings):
         elif query_embedding_cache is False:
             query_embedding_store = None
         else:
-            query_embedding_store = EncoderBackedStore[str, List[float]](
+            query_embedding_store = EncoderBackedStore[str, list[float]](
                 query_embedding_cache,
                 key_encoder,
                 _value_serializer,

langchain/evaluation/agents/trajectory_eval_chain.py

@@ -6,13 +6,10 @@ chain (LLMChain) to generate the reasoning and scores.
 """
 
 import re
+from collections.abc import Sequence
 from typing import (
     Any,
-    Dict,
-    List,
     Optional,
-    Sequence,
-    Tuple,
     TypedDict,
     Union,
     cast,
@@ -145,7 +142,7 @@ class TrajectoryEvalChain(AgentTrajectoryEvaluator, LLMEvalChain):
         # 0
     """
 
-    agent_tools: Optional[List[BaseTool]] = None
+    agent_tools: Optional[list[BaseTool]] = None
     """A list of tools available to the agent."""
     eval_chain: LLMChain
     """The language model chain used for evaluation."""
@@ -184,7 +181,7 @@ Description: {tool.description}"""
 
     @staticmethod
     def get_agent_trajectory(
-        steps: Union[str, Sequence[Tuple[AgentAction, str]]],
+        steps: Union[str, Sequence[tuple[AgentAction, str]]],
     ) -> str:
         """Get the agent trajectory as a formatted string.
 
@@ -263,7 +260,7 @@ The following is the expected answer. Use this to measure correctness:
         )
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Get the input keys for the chain.
 
         Returns:
@@ -272,7 +269,7 @@ The following is the expected answer. Use this to measure correctness:
         return ["question", "agent_trajectory", "answer", "reference"]
 
     @property
-    def output_keys(self) -> List[str]:
+    def output_keys(self) -> list[str]:
         """Get the output keys for the chain.
 
         Returns:
@@ -280,16 +277,16 @@ The following is the expected answer. Use this to measure correctness:
         """
         return ["score", "reasoning"]
 
-    def prep_inputs(self, inputs: Union[Dict[str, Any], Any]) -> Dict[str, str]:
+    def prep_inputs(self, inputs: Union[dict[str, Any], Any]) -> dict[str, str]:
         """Validate and prep inputs."""
         inputs["reference"] = self._format_reference(inputs.get("reference"))
         return super().prep_inputs(inputs)
 
     def _call(
         self,
-        inputs: Dict[str, str],
+        inputs: dict[str, str],
         run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Run the chain and generate the output.
 
         Args:
@@ -311,9 +308,9 @@ The following is the expected answer. Use this to measure correctness:
 
     async def _acall(
         self,
-        inputs: Dict[str, str],
+        inputs: dict[str, str],
         run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Run the chain and generate the output.
 
         Args:
@@ -338,11 +335,11 @@ The following is the expected answer. Use this to measure correctness:
         *,
         prediction: str,
         input: str,
-        agent_trajectory: Sequence[Tuple[AgentAction, str]],
+        agent_trajectory: Sequence[tuple[AgentAction, str]],
         reference: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:
@@ -380,11 +377,11 @@ The following is the expected answer. Use this to measure correctness:
         *,
         prediction: str,
         input: str,
-        agent_trajectory: Sequence[Tuple[AgentAction, str]],
+        agent_trajectory: Sequence[tuple[AgentAction, str]],
         reference: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:

langchain/evaluation/comparison/eval_chain.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import logging
 import re
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
@@ -49,7 +49,7 @@ _SUPPORTED_CRITERIA = {
 
 
 def resolve_pairwise_criteria(
-    criteria: Optional[Union[CRITERIA_TYPE, str, List[CRITERIA_TYPE]]],
+    criteria: Optional[Union[CRITERIA_TYPE, str, list[CRITERIA_TYPE]]],
 ) -> dict:
     """Resolve the criteria for the pairwise evaluator.
 
@@ -95,7 +95,7 @@ def resolve_pairwise_criteria(
     return criteria_
 
 
-class PairwiseStringResultOutputParser(BaseOutputParser[dict]):  # type: ignore[override]
+class PairwiseStringResultOutputParser(BaseOutputParser[dict]):
     """A parser for the output of the PairwiseStringEvalChain.
 
     Attributes:
@@ -113,7 +113,7 @@ class PairwiseStringResultOutputParser(BaseOutputParser[dict]):  # type: ignore[
         """
         return "pairwise_string_result"
 
-    def parse(self, text: str) -> Dict[str, Any]:
+    def parse(self, text: str) -> dict[str, Any]:
         """Parse the output text.
 
         Args:
@@ -151,7 +151,7 @@ class PairwiseStringResultOutputParser(BaseOutputParser[dict]):  # type: ignore[
         }
 
 
-class PairwiseStringEvalChain(PairwiseStringEvaluator, LLMEvalChain, LLMChain):  # type: ignore[override]
+class PairwiseStringEvalChain(PairwiseStringEvaluator, LLMEvalChain, LLMChain):
     """A chain for comparing two outputs, such as the outputs
      of two models, prompts, or outputs of a single model on similar inputs.
 
@@ -314,8 +314,8 @@ Performance may be significantly worse with other models."
         input: Optional[str] = None,
         reference: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:
@@ -356,8 +356,8 @@ Performance may be significantly worse with other models."
         reference: Optional[str] = None,
         input: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:
@@ -391,7 +391,7 @@ Performance may be significantly worse with other models."
         return self._prepare_output(result)
 
 
-class LabeledPairwiseStringEvalChain(PairwiseStringEvalChain):  # type: ignore[override]
+class LabeledPairwiseStringEvalChain(PairwiseStringEvalChain):
     """A chain for comparing two outputs, such as the outputs
      of two models, prompts, or outputs of a single model on similar inputs,
      with labeled preferences.

langchain/evaluation/criteria/eval_chain.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
 
 import re
+from collections.abc import Mapping
 from enum import Enum
-from typing import Any, Dict, List, Mapping, Optional, Union
+from typing import Any, Optional, Union
 
 from langchain_core.callbacks.manager import Callbacks
 from langchain_core.language_models import BaseLanguageModel
@@ -68,7 +69,7 @@ class CriteriaResultOutputParser(BaseOutputParser[dict]):
     def _type(self) -> str:
         return "criteria_result"
 
-    def parse(self, text: str) -> Dict[str, Any]:
+    def parse(self, text: str) -> dict[str, Any]:
         """Parse the output text.
 
         Args:
@@ -121,7 +122,7 @@ CRITERIA_TYPE = Union[
 
 def resolve_criteria(
     criteria: Optional[Union[CRITERIA_TYPE, str]],
-) -> Dict[str, str]:
+) -> dict[str, str]:
     """Resolve the criteria to evaluate.
 
     Parameters
@@ -164,7 +165,7 @@ def resolve_criteria(
     return criteria_
 
 
-class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):  # type: ignore[override]
+class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):
     """LLM Chain for evaluating runs against criteria.
 
     Parameters
@@ -285,7 +286,7 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):  # type: ignor
     def resolve_criteria(
         cls,
         criteria: Optional[Union[CRITERIA_TYPE, str]],
-    ) -> Dict[str, str]:
+    ) -> dict[str, str]:
         """Resolve the criteria to evaluate.
 
         Parameters
@@ -404,8 +405,8 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):  # type: ignor
         reference: Optional[str] = None,
         input: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:
@@ -459,8 +460,8 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):  # type: ignor
         reference: Optional[str] = None,
         input: Optional[str] = None,
         callbacks: Callbacks = None,
-        tags: Optional[List[str]] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        tags: Optional[list[str]] = None,
+        metadata: Optional[dict[str, Any]] = None,
         include_run_info: bool = False,
         **kwargs: Any,
     ) -> dict:
@@ -508,7 +509,7 @@ class CriteriaEvalChain(StringEvaluator, LLMEvalChain, LLMChain):  # type: ignor
         return self._prepare_output(result)
 
 
-class LabeledCriteriaEvalChain(CriteriaEvalChain):  # type: ignore[override]
+class LabeledCriteriaEvalChain(CriteriaEvalChain):
     """Criteria evaluation chain that requires references."""
 
     @classmethod