diff --git a/src/control_backend/agents/llm/llm_agent.py b/src/control_backend/agents/llm/llm_agent.py index 49ca8bb..2e43ebe 100644 --- a/src/control_backend/agents/llm/llm_agent.py +++ b/src/control_backend/agents/llm/llm_agent.py @@ -185,6 +185,9 @@ class LLMAgent(BaseAgent): full_message = "" current_chunk = "" async for token in self._stream_query_llm(messages): + if self._interrupted: + return + full_message += token current_chunk += token