Add printout of query id

Eventual-Inc · Feb 6, 2025 · 45a4459 · 45a4459
1 parent 5e7e2bc
commit 45a4459
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/daft/dataframe/dataframe.py b/daft/dataframe/dataframe.py
@@ -190,15 +190,17 @@ def _explain_broadcast(self):
         plan_time_end = datetime.now(timezone.utc)
 
         try:
+            id = uuid4()
             requests.post(
                 f"http://{addr}:{port}",
                 json={
-                    "id": str(uuid4()),
+                    "id": str(id),
                     "mermaid-plan": mermaid_plan,
                     "plan-time-start": str(plan_time_start),
                     "plan-time-end": str(plan_time_end),
                 },
             )
+            print(f"Query ID: {id}")
         except requests.exceptions.ConnectionError as conn_error:
             warnings.warn(
                 "Unable to broadcast daft query plan over http."