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"""