From e390dfbb64160e3aa6a32a47ab597e8002a7d437 Mon Sep 17 00:00:00 2001 From: jesus Date: Tue, 15 Sep 2026 01:13:17 +0000 Subject: [PATCH] fix(headroom): format protected index assignment Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/compression/compress.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/compression/compress.py b/litellm/compression/compress.py index 99410a533f9..0af9618382f 100644 --- a/litellm/compression/compress.py +++ b/litellm/compression/compress.py @@ -238,7 +238,9 @@ def get_protected_indices(messages: Sequence[Mapping[str, object]]) -> tuple[int """ system_indices: Final = tuple(index for index, msg in enumerate(messages) if msg.get("role", "") == "system") last_user: Final = tuple(index for index, msg in enumerate(messages) if msg.get("role", "") == "user")[-1:] - last_assistant: Final = tuple(index for index, msg in enumerate(messages) if msg.get("role", "") == "assistant")[-1:] + last_assistant: Final = tuple(index for index, msg in enumerate(messages) if msg.get("role", "") == "assistant")[ + -1: + ] return tuple(dict.fromkeys(system_indices + last_user + last_assistant + _cached_prefix_indices(messages)))