blob: 232d3fa1aa52eda1f9321e1b7498975a7c7a1a99 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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
|
