scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix(responses): cast content block before reading prompt_cache_breakpoint

Browse source 
basedpyright flags the raw dict.get read as reportUnknownArgumentType
(+2 against the error budget); cast the isinstance-narrowed block to
dict[str, object] first, matching the strip helpers' cast-ok idiom.
This commit is contained in:
Yucheng 2026-10-05 00:13:24 -07:00
parent 77e21ae37a
commit f6a0299e91
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/completion_extras/litellm_responses_transformation/transformation.py
View file

@ -1117,7 +1117,9 @@ class LiteLLMResponsesTransformationHandler(CompletionTransformationBridge):
if original_type == "text":
converted = with_prompt_cache_breakpoint(
self._convert_content_str_to_input_text(item.get("text", ""), role),
item.get("prompt_cache_breakpoint"),
cast(dict[str, object], item).get( # cast-ok: isinstance confirms the content block is a mapping
"prompt_cache_breakpoint"
),
)
result.append(converted)
verbose_logger.debug("Chat provider: text -> %s", converted)
@ -1130,7 +1132,9 @@ class LiteLLMResponsesTransformationHandler(CompletionTransformationBridge):
role,
)
),
item.get("prompt_cache_breakpoint"),
cast(dict[str, object], item).get( # cast-ok: isinstance confirms the content block is a mapping
"prompt_cache_breakpoint"
),
)
result.append(converted)
verbose_logger.debug("Chat provider: image_url -> %s", converted)
@ -1146,7 +1150,9 @@ class LiteLLMResponsesTransformationHandler(CompletionTransformationBridge):
_input_file_from_file_value(
cast("ChatCompletionFileObject", item).get("file"), # cast-ok: type tag checked
),
item.get("prompt_cache_breakpoint"),
cast(dict[str, object], item).get( # cast-ok: isinstance confirms the content block is a mapping
"prompt_cache_breakpoint"
),
)
result.append(converted)
verbose_logger.debug("Chat provider: file -> %s", converted)