From 9315f87f533a8d74cdc066b839a11e434b5d01ba Mon Sep 17 00:00:00 2001
From: Niels Geens <niels.geens@gmail.com>
Date: Sat, 19 Apr 2025 12:15:23 +0000
Subject: [PATCH] Fix stop bug

---
 sia/web_agent.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sia/web_agent.py b/sia/web_agent.py
index 133ba21..4d5f9ff 100644
--- a/sia/web_agent.py
+++ b/sia/web_agent.py
@@ -124,6 +124,8 @@ class WebAgent(BaseAgent):
         if llm_name not in self._llms:
             raise ValueError(f"Unknown LLM: {llm_name}")
         self._stop_flags[llm_name] = True
+        with self._llm_lock:
+            self._set_llm_state(llm_name, LlmState.IDLE)
 
     def approve_response(self) -> None:
         """Process approved response from specified LLM"""