Use builtins.dict instead of typing.Dict

cbornet · cbornet · commit 21ee4417d6fd · 2025-09-09T11:39:24.000+02:00
diff --git a/libs/core/langchain_core/language_models/chat_models.py b/libs/core/langchain_core/language_models/chat_models.py
@@ -3,9 +3,9 @@
 from __future__ import annotations
 
 import asyncio
+import builtins
 import inspect
 import json
-import typing
 import warnings
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator, Iterator, Sequence
@@ -373,7 +373,7 @@ class BaseChatModel(BaseLanguageModel[AIMessage], ABC):
 
     @model_validator(mode="before")
     @classmethod
-    def raise_deprecation(cls, values: typing.Dict) -> Any:  # noqa: UP006
+    def raise_deprecation(cls, values: builtins.dict) -> Any:
         """Emit deprecation warning if ``callback_manager`` is used.
 
         Args:
@@ -397,7 +397,7 @@ def raise_deprecation(cls, values: typing.Dict) -> Any:  # noqa: UP006
     )
 
     @cached_property
-    def _serialized(self) -> typing.Dict[str, Any]:  # noqa: UP006
+    def _serialized(self) -> builtins.dict[str, Any]:
         return dumpd(self)
 
     # --- Runnable methods ---
@@ -747,8 +747,8 @@ async def astream(
 
     def _combine_llm_outputs(
         self,
-        llm_outputs: list[Optional[typing.Dict]],  # noqa: ARG002, UP006
-    ) -> typing.Dict:  # noqa: UP006
+        llm_outputs: list[Optional[builtins.dict]],  # noqa: ARG002
+    ) -> builtins.dict:
         return {}
 
     def _convert_cached_generations(self, cache_val: list) -> list[ChatGeneration]:
@@ -794,7 +794,7 @@ def _get_invocation_params(
         self,
         stop: Optional[list[str]] = None,
         **kwargs: Any,
-    ) -> typing.Dict:  # noqa: UP006
+    ) -> builtins.dict:
         params = self.asdict()
         params["stop"] = stop
         return {**params, **kwargs}
@@ -858,7 +858,7 @@ def generate(
         callbacks: Callbacks = None,
         *,
         tags: Optional[list[str]] = None,
-        metadata: Optional[typing.Dict[str, Any]] = None,  # noqa: UP006
+        metadata: Optional[builtins.dict[str, Any]] = None,
         run_name: Optional[str] = None,
         run_id: Optional[uuid.UUID] = None,
         **kwargs: Any,
@@ -974,7 +974,7 @@ async def agenerate(
         callbacks: Callbacks = None,
         *,
         tags: Optional[list[str]] = None,
-        metadata: Optional[typing.Dict[str, Any]] = None,  # noqa: UP006
+        metadata: Optional[builtins.dict[str, Any]] = None,
         run_name: Optional[str] = None,
         run_id: Optional[uuid.UUID] = None,
         **kwargs: Any,
@@ -1596,20 +1596,18 @@ def _llm_type(self) -> str:
 
     @deprecated("0.3.61", alternative="asdict", removal="1.0")
     @override
-    def dict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def dict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         return self.asdict()
 
-    def asdict(self) -> typing.Dict[str, Any]:  # noqa: UP006
+    def asdict(self) -> builtins.dict[str, Any]:
         """Return a dictionary of the LLM."""
         starter_dict = dict(self._identifying_params)
         starter_dict["_type"] = self._llm_type
         return starter_dict
 
     def bind_tools(
         self,
-        tools: Sequence[
-            Union[typing.Dict[str, Any], type, Callable, BaseTool]  # noqa: UP006
-        ],
+        tools: Sequence[Union[builtins.dict[str, Any], type, Callable, BaseTool]],
         *,
         tool_choice: Optional[Union[str]] = None,
         **kwargs: Any,
@@ -1628,11 +1626,11 @@ def bind_tools(
 
     def with_structured_output(
         self,
-        schema: Union[typing.Dict, type],  # noqa: UP006
+        schema: Union[builtins.dict, type],
         *,
         include_raw: bool = False,
         **kwargs: Any,
-    ) -> Runnable[LanguageModelInput, Union[typing.Dict, BaseModel]]:  # noqa: UP006
+    ) -> Runnable[LanguageModelInput, Union[builtins.dict, BaseModel]]:
         """Model wrapper that returns outputs formatted to match the given schema.
 
         Args:
diff --git a/libs/core/langchain_core/language_models/llms.py b/libs/core/langchain_core/language_models/llms.py
@@ -3,11 +3,11 @@
 from __future__ import annotations
 
 import asyncio
+import builtins
 import functools
 import inspect
 import json
 import logging
-import typing
 import warnings
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator, Iterator, Sequence
@@ -306,7 +306,7 @@ class BaseLLM(BaseLanguageModel[str], ABC):
 
     @model_validator(mode="before")
     @classmethod
-    def raise_deprecation(cls, values: typing.Dict[str, Any]) -> Any:  # noqa: UP006
+    def raise_deprecation(cls, values: builtins.dict[str, Any]) -> Any:
         """Raise deprecation warning if callback_manager is used."""
         if values.get("callback_manager") is not None:
             warnings.warn(
@@ -318,7 +318,7 @@ def raise_deprecation(cls, values: typing.Dict[str, Any]) -> Any:  # noqa: UP006
         return values
 
     @functools.cached_property
-    def _serialized(self) -> typing.Dict[str, Any]:  # noqa: UP006
+    def _serialized(self) -> builtins.dict[str, Any]:
         return dumpd(self)
 
     # --- Runnable methods ---
@@ -844,7 +844,7 @@ def generate(
         *,
         tags: Optional[Union[list[str], list[list[str]]]] = None,
         metadata: Optional[
-            Union[typing.Dict[str, Any], list[typing.Dict[str, Any]]]  # noqa: UP006
+            Union[builtins.dict[str, Any], list[builtins.dict[str, Any]]]
         ] = None,
         run_name: Optional[Union[str, list[str]]] = None,
         run_id: Optional[Union[uuid.UUID, list[Optional[uuid.UUID]]]] = None,
@@ -1112,7 +1112,7 @@ async def agenerate(
         *,
         tags: Optional[Union[list[str], list[list[str]]]] = None,
         metadata: Optional[
-            Union[typing.Dict[str, Any], list[typing.Dict[str, Any]]]  # noqa: UP006
+            Union[builtins.dict[str, Any], list[builtins.dict[str, Any]]]
         ] = None,
         run_name: Optional[Union[str, list[str]]] = None,
         run_id: Optional[Union[uuid.UUID, list[Optional[uuid.UUID]]]] = None,
@@ -1319,7 +1319,7 @@ def __call__(
         callbacks: Callbacks = None,
         *,
         tags: Optional[list[str]] = None,
-        metadata: Optional[typing.Dict[str, Any]] = None,  # noqa: UP006
+        metadata: Optional[builtins.dict[str, Any]] = None,
         **kwargs: Any,
     ) -> str:
         """Check Cache and run the LLM on the given prompt and input.
@@ -1368,7 +1368,7 @@ async def _call_async(
         callbacks: Callbacks = None,
         *,
         tags: Optional[list[str]] = None,
-        metadata: Optional[typing.Dict[str, Any]] = None,  # noqa: UP006
+        metadata: Optional[builtins.dict[str, Any]] = None,
         **kwargs: Any,
     ) -> str:
         """Check Cache and run the LLM on the given prompt and input."""
@@ -1438,10 +1438,10 @@ def _llm_type(self) -> str:
 
     @deprecated("0.3.61", alternative="asdict", removal="1.0")
     @override
-    def dict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def dict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         return self.asdict()
 
-    def asdict(self) -> typing.Dict[str, Any]:  # noqa: UP006
+    def asdict(self) -> builtins.dict[str, Any]:
         """Return a dictionary of the LLM."""
         starter_dict = dict(self._identifying_params)
         starter_dict["_type"] = self._llm_type
diff --git a/libs/core/langchain_core/output_parsers/base.py b/libs/core/langchain_core/output_parsers/base.py
@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+import builtins
 import contextlib
-import typing
 from abc import ABC, abstractmethod
 from typing import (
     TYPE_CHECKING,
@@ -332,10 +332,10 @@ def _type(self) -> str:
 
     @deprecated("0.3.61", alternative="asdict", removal="1.0")
     @override
-    def dict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def dict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         return self.asdict()
 
-    def asdict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def asdict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         """Return dictionary representation of output parser."""
         output_parser_dict = super().model_dump(**kwargs)
         with contextlib.suppress(NotImplementedError):
diff --git a/libs/core/langchain_core/prompts/base.py b/libs/core/langchain_core/prompts/base.py
@@ -2,9 +2,9 @@
 
 from __future__ import annotations
 
+import builtins
 import contextlib
 import json
-import typing
 from abc import ABC, abstractmethod
 from collections.abc import Mapping
 from functools import cached_property
@@ -55,7 +55,7 @@ class BasePromptTemplate(
     """optional_variables: A list of the names of the variables for placeholder
        or MessagePlaceholder that are optional. These variables are auto inferred
        from the prompt and user need not provide them."""
-    input_types: typing.Dict[str, Any] = Field(default_factory=dict, exclude=True)  # noqa: UP006
+    input_types: builtins.dict[str, Any] = Field(default_factory=dict, exclude=True)
     """A dictionary of the types of the variables the prompt template expects.
     If not provided, all variables are assumed to be strings."""
     output_parser: Optional[BaseOutputParser] = None
@@ -65,7 +65,7 @@ class BasePromptTemplate(
 
     Partial variables populate the template so that you don't need to
     pass them in every time you call the prompt."""
-    metadata: Optional[typing.Dict[str, Any]] = None  # noqa: UP006
+    metadata: Optional[builtins.dict[str, Any]] = None
     """Metadata to be used for tracing."""
     tags: Optional[list[str]] = None
     """Tags to be used for tracing."""
@@ -117,7 +117,7 @@ def is_lc_serializable(cls) -> bool:
     )
 
     @cached_property
-    def _serialized(self) -> typing.Dict[str, Any]:  # noqa: UP006
+    def _serialized(self) -> builtins.dict[str, Any]:
         return dumpd(self)
 
     @property
@@ -150,7 +150,7 @@ def get_input_schema(
             field_definitions={**required_input_variables, **optional_input_variables},
         )
 
-    def _validate_input(self, inner_input: Any) -> typing.Dict:  # noqa: UP006
+    def _validate_input(self, inner_input: Any) -> builtins.dict:
         if not isinstance(inner_input, dict):
             if len(self.input_variables) == 1:
                 var_name = self.input_variables[0]
@@ -186,22 +186,22 @@ def _validate_input(self, inner_input: Any) -> typing.Dict:  # noqa: UP006
 
     def _format_prompt_with_error_handling(
         self,
-        inner_input: typing.Dict,  # noqa: UP006
+        inner_input: builtins.dict,
     ) -> PromptValue:
         inner_input_ = self._validate_input(inner_input)
         return self.format_prompt(**inner_input_)
 
     async def _aformat_prompt_with_error_handling(
         self,
-        inner_input: typing.Dict,  # noqa: UP006
+        inner_input: builtins.dict,
     ) -> PromptValue:
         inner_input_ = self._validate_input(inner_input)
         return await self.aformat_prompt(**inner_input_)
 
     @override
     def invoke(
         self,
-        input: typing.Dict,  # noqa: UP006
+        input: builtins.dict,
         config: Optional[RunnableConfig] = None,
         **kwargs: Any,
     ) -> PromptValue:
@@ -230,7 +230,7 @@ def invoke(
     @override
     async def ainvoke(
         self,
-        input: typing.Dict,  # noqa: UP006
+        input: builtins.dict,
         config: Optional[RunnableConfig] = None,
         **kwargs: Any,
     ) -> PromptValue:
@@ -294,7 +294,9 @@ def partial(self, **kwargs: Union[str, Callable[[], str]]) -> BasePromptTemplate
         prompt_dict["partial_variables"] = {**self.partial_variables, **kwargs}
         return type(self)(**prompt_dict)
 
-    def _merge_partial_and_user_variables(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def _merge_partial_and_user_variables(
+        self, **kwargs: Any
+    ) -> builtins.dict[str, Any]:
         # Get partial params:
         partial_kwargs = {
             k: v if not callable(v) else v() for k, v in self.partial_variables.items()
@@ -344,10 +346,10 @@ def _prompt_type(self) -> str:
 
     @deprecated("0.3.61", alternative="asdict", removal="1.0")
     @override
-    def dict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def dict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         return self.asdict(**kwargs)
 
-    def asdict(self, **kwargs: Any) -> typing.Dict[str, Any]:  # noqa: UP006
+    def asdict(self, **kwargs: Any) -> builtins.dict[str, Any]:
         """Return dictionary representation of prompt.
 
         Args:
