From d213673a34a1ad5ee65afdaba5a5dddfd7bc2885 Mon Sep 17 00:00:00 2001
From: Aniket Kumar <86551851+aniketkrs@users.noreply.github.com>
Date: Sun, 18 Oct 2026 09:49:37 +0530
Subject: [PATCH] fix(autogen-core): count actually retained messages in head
 and tail skipped count

---
 .../model_context/_head_and_tail_chat_completion_context.py     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/packages/autogen-core/src/autogen_core/model_context/_head_and_tail_chat_completion_context.py b/python/packages/autogen-core/src/autogen_core/model_context/_head_and_tail_chat_completion_context.py
index 75493618e90d..29819eea84fe 100644
--- a/python/packages/autogen-core/src/autogen_core/model_context/_head_and_tail_chat_completion_context.py
+++ b/python/packages/autogen-core/src/autogen_core/model_context/_head_and_tail_chat_completion_context.py
@@ -57,7 +57,7 @@ async def get_messages(self) -> List[LLMMessage]:
             # Remove the first message from the tail.
             tail_messages = tail_messages[1:]
 
-        num_skipped = len(self._messages) - self._head_size - self._tail_size
+        num_skipped = len(self._messages) - len(head_messages) - len(tail_messages)
         if num_skipped <= 0:
             # If there are not enough messages to fill the head and tail,
             # return all messages.