Fix colab path

ericfe-google · ericfe-google · commit 4ad1b71f330b · 2026-01-28T14:29:28.000-08:00
diff --git a/bigquery_magics/bigquery.py b/bigquery_magics/bigquery.py
@@ -633,8 +633,9 @@ def _handle_result(result, args):
 
 
 def _colab_query_callback(query: str, params: str):
+    query_results = json.loads(graph_server.graph_server.query_result.to_json())
     return IPython.core.display.JSON(
-        graph_server.convert_graph_data(query_results=json.loads(params))
+        graph_server.convert_graph_data(query_results=query_results)
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -633,8 +633,9 @@ def _handle_result(result, args):`
`633`	`633`
`634`	`634`
`635`	`635`	`def _colab_query_callback(query: str, params: str):`
	`636`	`+ query_results = json.loads(graph_server.graph_server.query_result.to_json())`
`636`	`637`	`return IPython.core.display.JSON(`
`637`		`- graph_server.convert_graph_data(query_results=json.loads(params))`
	`638`	`+ graph_server.convert_graph_data(query_results=query_results)`
`638`	`639`	`)`
`639`	`640`
`640`	`641`