handle llm stream returning false (#865)

livekit · Oct 8, 2024 · 3553810 · 3553810
1 parent eaddf66
commit 3553810
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.changeset/yellow-eggs-shout.md b/.changeset/yellow-eggs-shout.md
@@ -0,0 +1,5 @@
+---
+"livekit-agents": patch
+---
+
+Fixed bug where agent would freeze if before_llm_cb returned false
diff --git a/livekit-agents/livekit/agents/pipeline/pipeline_agent.py b/livekit-agents/livekit/agents/pipeline/pipeline_agent.py
@@ -552,6 +552,7 @@ async def _synthesize_answer_task(
 
         llm_stream = self._opts.before_llm_cb(self, copied_ctx)
         if llm_stream is False:
+            handle.interrupt()
             return
 
         if asyncio.iscoroutine(llm_stream):