get query description

Signed-off-by: Prithvi Kannan <[email protected]>
databricks · Oct 30, 2024 · 63dd12d · 63dd12d
1 parent 2bd27eb
commit 63dd12d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/databricks_ai_bridge/genie.py b/src/databricks_ai_bridge/genie.py
@@ -108,7 +108,10 @@ def poll_result():
                     headers=self.headers,
                 )
                 if resp["status"] == "EXECUTING_QUERY":
-                    sql = next(r for r in resp["attachments"] if "query" in r)["query"]["query"]
+                    query = next(r for r in resp["attachments"] if "query" in r)["query"]
+                    description = query["description"]
+                    sql = query["query"]
+                    logging.debug(f"Description: {description}")
                     logging.debug(f"SQL: {sql}")
                     return poll_query_results()
                 elif resp["status"] == "COMPLETED":