Use builtins.dict instead of typing.Dict

Author: cbornet

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
@@ -366,7 +366,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:
         """Raise deprecation warning if ``callback_manager`` is used.
 
         Args:
@@ -393,7 +393,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 ---
@@ -743,8 +743,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]:
@@ -790,7 +790,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}
@@ -854,7 +854,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,
@@ -969,7 +969,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,
@@ -1542,20 +1542,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,
@@ -1574,11 +1572,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:

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
@@ -305,7 +305,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(
@@ -317,7 +317,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 ---
@@ -821,7 +821,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,
@@ -1083,7 +1083,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,
@@ -1285,7 +1285,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.
@@ -1334,7 +1334,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."""
@@ -1404,10 +1404,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