fix(query-awaited): Try increasing timeout, not do threadsafe and add… (#30225)

Author: timgl

posthog/api/query.py

@@ -279,7 +279,6 @@ async def query_awaited(request: Request, *args, **kwargs) -> StreamingHttpRespo
         # This provides better handling of task cancellation than run_in_executor
         async_process_query_model = sync_to_async(
             process_query_model,
-            thread_sensitive=False,  # Set to False since this doesn't need to run in the same thread as the request
         )
 
         # Create a task from the async wrapper
@@ -294,7 +293,7 @@ async def query_awaited(request: Request, *args, **kwargs) -> StreamingHttpRespo
         )
 
         # YOLO give the task a moment to materialize (otherwise the task looks like it's been cancelled)
-        await asyncio.sleep(0.01)
+        await asyncio.sleep(0.5)
 
         async def event_stream():
             assert kwargs.get("team_id") is not None
@@ -316,6 +315,7 @@ async def event_stream():
                     if query_task.cancelled():
                         # Explicitly check for cancellation first
                         yield f"data: {json.dumps({'error': 'Query was cancelled', 'status_code': 499})}\n\n".encode()
+                        capture_exception(Exception("Query was cancelled"))
                         break
                     try:
                         result = query_task.result()