fix "keep_first_message" to make sure messages are in correct order, (#…

…4653) described in #3178 (comment) Co-authored-by: Eric Zhu <[email protected]>
microsoft · Dec 12, 2024 · e90b9aa · e90b9aa
1 parent 837b2e0
commit e90b9aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogen/agentchat/contrib/capabilities/transforms.py b/autogen/agentchat/contrib/capabilities/transforms.py
@@ -96,7 +96,7 @@ def apply_transform(self, messages: List[Dict]) -> List[Dict]:
             if remaining_count == 1:
                 # If there's only 1 slot left and it's a 'tools' message, ignore it.
                 if messages[i].get("role") != "tool":
-                    truncated_messages.insert(1, messages[i])
+                    truncated_messages.insert(1 if self._keep_first_message else 0, messages[i])
 
             remaining_count -= 1
             if remaining_count == 0: