Catch concurrent future cancelled exception instead of asyncio

trypromptly · Nov 4, 2024 · 1e3b63d · 1e3b63d
1 parent 3dda9c9
commit 1e3b63d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/llmstack/play/utils.py b/llmstack/play/utils.py
@@ -2,6 +2,7 @@
 import logging
 import re
 import threading
+from concurrent.futures import CancelledError
 from urllib.parse import quote
 
 from diff_match_patch import diff_match_patch
@@ -82,8 +83,8 @@ def start_loop(loop):
     def _done_callback(future):
         try:
             future.result()
-        except asyncio.CancelledError:
-            pass
+        except CancelledError:
+            logger.info("Task cancelled")
         except Exception as e:
             logger.exception(f"Task failed with error: {e}")
         finally: