use re for barckets space fix

defog-ai · Sep 17, 2024 · a4664a1 · a4664a1
1 parent 7d97f64
commit a4664a1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/eval/api_runner.py b/eval/api_runner.py
@@ -13,6 +13,7 @@
 import requests
 from utils.reporting import upload_results
 import sqlparse
+import re
 
 
 def clean_generated_query(query: str):
@@ -34,9 +35,8 @@ def clean_generated_query(query: str):
     query = query.replace("/ NULLIF (", "/ NULLIF (1.0 * ")
 
     # remove extra spaces around brackets especially for MySQL
-    query = query.replace(" ( ", "(").replace(" )", ")")
-    query = query.replace(" (", "(").replace(") ", ")")
-    query = query.replace("( ", "(").replace(" )", ")")
+    query = re.sub(r"\s*\(\s*", "(", query)  # Remove spaces before and after '('
+    query = re.sub(r"\s*\)", ")", query)  # Remove spaces before ')'
 
     return query