bug fix

aniket-s-kulkarni · aniket-s-kulkarni · commit 601323ba324a · 2025-05-15T16:42:03.000-04:00
diff --git a/src/dremioai/api/dremio/sql.py b/src/dremioai/api/dremio/sql.py
@@ -202,7 +202,10 @@ async def get_results(
         )
         raise RuntimeError(f"Job {qs.id} failed: {emsg}")
 
-    limit = 500
+    if job.row_count == 0:
+        return pd.DataFrame() if use_df else JobResultsWrapper([])
+
+    limit = min(500, job.row_count)
 
     results = await run_in_parallel(
         [

Original file line number	Diff line number	Diff line change
`@@ -202,7 +202,10 @@ async def get_results(`
`202`	`202`	`)`
`203`	`203`	`raise RuntimeError(f"Job {qs.id} failed: {emsg}")`
`204`	`204`
`205`		`- limit = 500`
	`205`	`+ if job.row_count == 0:`
	`206`	`+ return pd.DataFrame() if use_df else JobResultsWrapper([])`
	`207`	`+`
	`208`	`+ limit = min(500, job.row_count)`
`206`	`209`
`207`	`210`	`results = await run_in_parallel(`
`208`	`211`	`[`