scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(caching): remove cache census casts

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-28 11:52:12 +00:00
parent 39a69561db
commit 8f05215cee
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/integrations/anthropic_cache_control_hook.py
View file

@ -127,23 +127,22 @@ def _carries_cache_breakpoint(block: object) -> bool:
def _attribute_or_key(value: object, key: str) -> object | None:
if hasattr(value, key):
return cast(object, getattr(value, key))
return getattr(value, key)
if isinstance(value, Mapping):
value_mapping: Final = cast(Mapping[str, object], value)
return value_mapping.get(key)
return value.get(key)
return None
def _as_object_list(value: object | None) -> list[object] | None:
if not isinstance(value, list):
return None
return cast(list[object], value)
return _validated_object_list(value)
def _as_object_iterable(value: object | None) -> Iterable[object] | None:
if not isinstance(value, Iterable):
return None
return cast(Iterable[object], value)
return value
def _has_server_tool_result(tool_call_id: str, results: Iterable[object] | None) -> bool:
@ -164,12 +163,11 @@ def _tool_call_cache_control_is_forwarded(tool_call: object, message: object) ->
if not isinstance(provider_specific_fields, dict):
return True
provider_fields_mapping: Final = cast(Mapping[str, object], provider_specific_fields)
server_tool_result_keys: Final = ("web_search_results", "tool_results")
return not any(
_has_server_tool_result(
tool_call_id,
_as_object_iterable(provider_fields_mapping.get(result_key)),
_as_object_iterable(provider_specific_fields.get(result_key)),
)
for result_key in server_tool_result_keys
)