fix: fixed issues with lost Context

aidial_analytics_realtime/utils/concurrency.py

@@ -1,4 +1,6 @@
 import asyncio
+import contextvars
+import functools
 from concurrent.futures import ThreadPoolExecutor
 from typing import Callable, ParamSpec, TypeVar
 
@@ -12,4 +14,6 @@ async def run_in_cpu_tasks_executor(
     func: Callable[_P, _T], *args: _P.args, **kwargs: _P.kwargs
 ) -> _T:
     loop = asyncio.get_event_loop()
-    return await loop.run_in_executor(cpu_task_executor, func, *args)  # type: ignore
+    ctx = contextvars.copy_context()
+    func_call = functools.partial(ctx.run, func, *args, **kwargs)
+    return await loop.run_in_executor(cpu_task_executor, func_call)  # type: ignore