interrupt - activate - set context separately (#1194)

Author: pgrayy

src/strands/event_loop/event_loop.py

@@ -483,7 +483,8 @@ async def _handle_tool_execution(
 
     if interrupts:
         # Session state stored on AfterInvocationEvent.
-        agent._interrupt_state.activate(context={"tool_use_message": message, "tool_results": tool_results})
+        agent._interrupt_state.context = {"tool_use_message": message, "tool_results": tool_results}
+        agent._interrupt_state.activate()
 
         agent.event_loop_metrics.end_cycle(cycle_start_time, cycle_trace)
         yield EventLoopStopEvent(

src/strands/interrupt.py

@@ -53,13 +53,8 @@ class _InterruptState:
     context: dict[str, Any] = field(default_factory=dict)
     activated: bool = False
 
-    def activate(self, context: dict[str, Any] | None = None) -> None:
-        """Activate the interrupt state.
-
-        Args:
-            context: Context associated with the interrupt event.
-        """
-        self.context = context or {}
+    def activate(self) -> None:
+        """Activate the interrupt state."""
         self.activated = True
 
     def deactivate(self) -> None:

tests/strands/agent/test_agent.py

@@ -2001,8 +2001,9 @@ def test_agent__call__resume_interrupt(mock_model, tool_decorated, agenerator):
         reason="test reason",
     )
 
-    agent._interrupt_state.activate(context={"tool_use_message": tool_use_message, "tool_results": []})
+    agent._interrupt_state.context = {"tool_use_message": tool_use_message, "tool_results": []}
     agent._interrupt_state.interrupts[interrupt.id] = interrupt
+    agent._interrupt_state.activate()
 
     interrupt_response = {}
 

tests/strands/event_loop/test_event_loop.py

@@ -964,8 +964,9 @@ async def test_event_loop_cycle_interrupt_resume(agent, model, tool, tool_times_
         },
     ]
 
-    agent._interrupt_state.activate(context={"tool_use_message": tool_use_message, "tool_results": tool_results})
+    agent._interrupt_state.context = {"tool_use_message": tool_use_message, "tool_results": tool_results}
     agent._interrupt_state.interrupts[interrupt.id] = interrupt
+    agent._interrupt_state.activate()
 
     interrupt_response = {}
 

tests/strands/test_interrupt.py

@@ -27,14 +27,9 @@ def test_interrupt_to_dict(interrupt):
 def test_interrupt_state_activate():
     interrupt_state = _InterruptState()
 
-    interrupt_state.activate(context={"test": "context"})
-
+    interrupt_state.activate()
     assert interrupt_state.activated
 
-    tru_context = interrupt_state.context
-    exp_context = {"test": "context"}
-    assert tru_context == exp_context
-
 
 def test_interrupt_state_deactivate():
     interrupt_state = _InterruptState(context={"test": "context"}, activated=True)