fix

Author: seratch

src/agents/memory/openai_responses_compaction_session.py

@@ -190,6 +190,8 @@ async def get_items(self, limit: int | None = None) -> list[TResponseInputItem]:
         return await self.underlying_session.get_items(limit)
 
     async def _defer_compaction(self, response_id: str) -> None:
+        if self._deferred_response_id is not None:
+            return
         compaction_candidate_items, session_items = await self._ensure_compaction_candidates()
         should_compact = self.should_trigger_compaction(
             {
@@ -200,8 +202,6 @@ async def _defer_compaction(self, response_id: str) -> None:
         )
         if should_compact:
             self._deferred_response_id = response_id
-        else:
-            self._deferred_response_id = None
 
     def _get_deferred_compaction_response_id(self) -> str | None:
         return self._deferred_response_id

tests/memory/test_openai_responses_compaction_session.py

@@ -357,6 +357,50 @@ def should_trigger_compaction(context: dict[str, Any]) -> bool:
 
         mock_client.responses.compact.assert_awaited_once()
 
+    @pytest.mark.asyncio
+    async def test_deferred_compaction_persists_across_tool_turns(self) -> None:
+        underlying = SimpleListSession()
+        compacted = SimpleNamespace(
+            output=[{"type": "compaction", "summary": "compacted"}],
+        )
+        mock_client = MagicMock()
+        mock_client.responses.compact = AsyncMock(return_value=compacted)
+
+        should_compact_calls = {"count": 0}
+
+        def should_trigger_compaction(context: dict[str, Any]) -> bool:
+            should_compact_calls["count"] += 1
+            return should_compact_calls["count"] == 1
+
+        session = OpenAIResponsesCompactionSession(
+            session_id="demo",
+            underlying_session=underlying,
+            client=mock_client,
+            should_trigger_compaction=should_trigger_compaction,
+        )
+
+        tool = get_function_tool(name="do_thing", return_value="done")
+        model = FakeModel()
+        model.add_multiple_turn_outputs(
+            [
+                [get_function_tool_call("do_thing")],
+                [get_function_tool_call("do_thing")],
+                [get_text_message("ok")],
+            ]
+        )
+        agent = Agent(
+            name="assistant",
+            model=model,
+            tools=[tool],
+            tool_use_behavior="stop_on_first_tool",
+        )
+
+        await Runner.run(agent, "hello", session=session)
+        await Runner.run(agent, "again", session=session)
+        await Runner.run(agent, "final", session=session)
+
+        mock_client.responses.compact.assert_awaited_once()
+
 
 class TestTypeGuard:
     def test_is_compaction_aware_session_true(self) -> None: