Fix ChatFireworks typing

pull/11117/head
Bagatur 11 months ago
parent 12aad659dd
commit 657581dbdf

@ -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

@ -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 ""

@ -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 ""

@ -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 ""

Loading…
Cancel
Save