Refactor session runtime persistence

2026-06-12 16:17:58 +08:00
parent 77d360c1e8
commit 953126e1d3
15 changed files with 717 additions and 253 deletions
@@ -1025,8 +1025,13 @@ class AgentRuntimeTests(unittest.TestCase):
            message for message in result.transcript
            if message.get('metadata', {}).get('kind') == 'compact_boundary'
        ]
-        self.assertEqual(len(transcript_compact_messages), 1)
-        compact_metadata = transcript_compact_messages[0].get('metadata', {})
+        self.assertEqual(len(transcript_compact_messages), 0)
+        model_compact_messages = [
+            message for message in (agent.last_session.messages if agent.last_session else [])
+            if message.metadata.get('kind') == 'compact_boundary'
+        ]
+        self.assertEqual(len(model_compact_messages), 1)
+        compact_metadata = model_compact_messages[0].metadata
        self.assertEqual(compact_metadata.get('compaction_depth'), 1)
        self.assertEqual(compact_metadata.get('nested_compaction_count'), 0)
        self.assertIn('preserved_tail_ids', compact_metadata)