scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(lint): Partly fix TID251 lint issues

Browse source
This commit is contained in:
KnyazSh 2026-08-26 20:33:27 +00:00
parent 278fdbecb7
commit 6c36030e60
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/gigachat/chat/streaming.py
View file

@ -5,7 +5,7 @@ GigaChat Streaming Response Handler
import json
import uuid
from collections.abc import Mapping, Sequence
from typing import Any, Final, cast
from typing import Any, Final
from litellm.llms.gigachat.utils import convert_usage
from litellm.types.llms.openai import (
@ -76,7 +76,10 @@ class GigaChatModelResponseIterator:
if chunk_finish_reason == "stop":
usage_data: Final = chunk.get("usage") or {} # mutable-ok: empty dict default
if usage_data:
usage = convert_usage(cast("Mapping[str, int]", usage_data)) # cast-ok: dict from chunk has int values
validated_usage: Final = {
k: int(v) for k, v in dict(usage_data).items()
} # rebind-ok: dict comprehension
usage = convert_usage(validated_usage)
usage_block = ChatCompletionUsageBlock(
prompt_tokens=usage.prompt_tokens,
completion_tokens=usage.completion_tokens,
@ -90,7 +93,7 @@ class GigaChatModelResponseIterator:
)
return GenericStreamingChunk(
text=cast(str, text),
text=str(text),
tool_use=tool_use,
is_finished=chunk_finish_reason is not None,
finish_reason=finish_reason or "",