diff --git a/libs/langchain/langchain/chat_models/fireworks.py b/libs/langchain/langchain/chat_models/fireworks.py index 22d9384d58..25f933f956 100644 --- a/libs/langchain/langchain/chat_models/fireworks.py +++ b/libs/langchain/langchain/chat_models/fireworks.py @@ -6,6 +6,7 @@ from typing import ( Iterator, List, Optional, + Type, Union, ) @@ -36,7 +37,7 @@ from langchain.utils.env import get_from_dict_or_env def _convert_delta_to_message_chunk( - _dict: Any, default_class: type[BaseMessageChunk] + _dict: Any, default_class: Type[BaseMessageChunk] ) -> BaseMessageChunk: """Convert a delta response to a message chunk.""" role = _dict.role diff --git a/libs/langchain/langchain/chat_models/jinachat.py b/libs/langchain/langchain/chat_models/jinachat.py index 51f5a0b60c..7b885d6482 100644 --- a/libs/langchain/langchain/chat_models/jinachat.py +++ b/libs/langchain/langchain/chat_models/jinachat.py @@ -12,6 +12,7 @@ from typing import ( Mapping, Optional, Tuple, + Type, Union, ) @@ -91,7 +92,7 @@ async def acompletion_with_retry(llm: JinaChat, **kwargs: Any) -> Any: def _convert_delta_to_message_chunk( - _dict: Mapping[str, Any], default_class: type[BaseMessageChunk] + _dict: Mapping[str, Any], default_class: Type[BaseMessageChunk] ) -> BaseMessageChunk: role = _dict.get("role") content = _dict.get("content") or "" diff --git a/libs/langchain/langchain/chat_models/litellm.py b/libs/langchain/langchain/chat_models/litellm.py index f568cf2760..275b8cba79 100644 --- a/libs/langchain/langchain/chat_models/litellm.py +++ b/libs/langchain/langchain/chat_models/litellm.py @@ -12,6 +12,7 @@ from typing import ( Mapping, Optional, Tuple, + Type, Union, ) @@ -144,7 +145,7 @@ async def acompletion_with_retry( def _convert_delta_to_message_chunk( - _dict: Mapping[str, Any], default_class: type[BaseMessageChunk] + _dict: Mapping[str, Any], default_class: Type[BaseMessageChunk] ) -> BaseMessageChunk: role = _dict.get("role") content = _dict.get("content") or "" diff --git a/libs/langchain/langchain/chat_models/openai.py b/libs/langchain/langchain/chat_models/openai.py index 333c7fe37c..f22b55a103 100644 --- a/libs/langchain/langchain/chat_models/openai.py +++ b/libs/langchain/langchain/chat_models/openai.py @@ -14,6 +14,7 @@ from typing import ( Mapping, Optional, Tuple, + Type, Union, ) @@ -97,7 +98,7 @@ async def acompletion_with_retry( def _convert_delta_to_message_chunk( - _dict: Mapping[str, Any], default_class: type[BaseMessageChunk] + _dict: Mapping[str, Any], default_class: Type[BaseMessageChunk] ) -> BaseMessageChunk: role = _dict.get("role") content = _dict.get("content") or ""