1 files changed, 38 insertions, 0 deletions
diff --git a/server/continuedev/libs/util/create_async_task.py b/server/continuedev/libs/util/create_async_task.py
new file mode 100644
index 00000000..232d3fa1
--- /dev/null
+++ b/server/continuedev/libs/util/create_async_task.py
@@ -0,0 +1,38 @@
+import asyncio
+import traceback
+from typing import Callable, Coroutine, Optional
+
+import nest_asyncio
+
+from .logging import logger
+from .telemetry import posthog_logger
+
+nest_asyncio.apply()
+
+
+def create_async_task(
+    coro: Coroutine, on_error: Optional[Callable[[Exception], Coroutine]] = None
+):
+    """asyncio.create_task and log errors by adding a callback"""
+    task = asyncio.create_task(coro)
+
+    def callback(future: asyncio.Future):
+        try:
+            future.result()
+        except Exception as e:
+            formatted_tb = "\n".join(traceback.format_exception(e))
+            logger.critical(f"Exception caught from async task: {formatted_tb}")
+            posthog_logger.capture_event(
+                "async_task_error",
+                {
+                    "error_title": e.__str__() or e.__repr__(),
+                    "error_message": "\n".join(traceback.format_exception(e)),
+                },
+            )
+
+            # Log the error to the GUI
+            if on_error is not None:
+                asyncio.create_task(on_error(e))
+
+    task.add_done_callback(callback)
+    return task