fix: prevent ScreenStackError when stopping agent from modal (#374)

This commit is contained in:
Ahmed Allam
2026-03-19 20:39:05 -07:00
committed by GitHub
parent 86341597c1
commit 9a0bc5e491

View File

@@ -252,10 +252,9 @@ class StopAgentScreen(ModalScreen): # type: ignore[misc]
event.prevent_default() event.prevent_default()
def on_button_pressed(self, event: Button.Pressed) -> None: def on_button_pressed(self, event: Button.Pressed) -> None:
self.app.pop_screen()
if event.button.id == "stop_agent": if event.button.id == "stop_agent":
self.app.action_confirm_stop_agent(self.agent_id) self.app.action_confirm_stop_agent(self.agent_id)
else:
self.app.pop_screen()
class VulnerabilityDetailScreen(ModalScreen): # type: ignore[misc] class VulnerabilityDetailScreen(ModalScreen): # type: ignore[misc]
@@ -1912,8 +1911,6 @@ class StrixTUIApp(App): # type: ignore[misc]
return agent_name, False return agent_name, False
def action_confirm_stop_agent(self, agent_id: str) -> None: def action_confirm_stop_agent(self, agent_id: str) -> None:
self.pop_screen()
try: try:
from strix.tools.agents_graph.agents_graph_actions import stop_agent from strix.tools.agents_graph.agents_graph_actions import stop_agent