diff --git a/changelog/3794.fixed.md b/changelog/3794.fixed.md new file mode 100644 index 000000000..e2b3c7c00 --- /dev/null +++ b/changelog/3794.fixed.md @@ -0,0 +1 @@ +- Added `LLMSpecificMessage` handling in `LLMContextSummarizationUtil` to skip provider-specific messages during context summarization. diff --git a/tests/test_context_summarization.py b/tests/test_context_summarization.py index 36559ed3f..3bb1246e9 100644 --- a/tests/test_context_summarization.py +++ b/tests/test_context_summarization.py @@ -634,8 +634,8 @@ class TestLLMSpecificMessageHandling(unittest.TestCase): result = LLMContextSummarizationUtil.get_messages_to_summarize(context, 2) - self.assertGreater(len(result.messages), 0) - self.assertGreater(result.last_summarized_index, 0) + self.assertEqual(len(result.messages), 4) + self.assertEqual(result.last_summarized_index, 4) def test_format_messages_skips_specific_messages(self): """Test that format_messages_for_summary skips LLMSpecificMessage objects."""