fix: Fix query awaited endpoint (#29225)

PostHog · Feb 26, 2025 · d301bcb · d301bcb
1 parent 739c08e
commit d301bcb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/frontend/src/queries/query.ts b/frontend/src/queries/query.ts
@@ -158,6 +158,7 @@ async function executeQuery<N extends DataNode>(
                     onerror(err) {
                         abortController.abort()
                         reject(err)
+                        throw err // make sure fetchEventSource doesn't attempt to retry
                     },
                 }).catch(reject)
             })

diff --git a/posthog/api/query.py b/posthog/api/query.py
@@ -314,7 +314,8 @@ async def event_stream():
                             status = await sync_to_async(manager.get_clickhouse_progresses)()
 
                             if isinstance(status, BaseModel):
-                                yield f"data: {status.model_dump_json(by_alias=True)}\n\n".encode()
+                                status_update = {"complete": False, **status.model_dump(by_alias=True)}
+                                yield f"data: {json.dumps(status_update)}\n\n".encode()
                                 last_update_time = current_time
                     # Just ignore errors when getting progress, shouldn't impact users
                     except Exception as e: