Spaces:
Sleeping
Sleeping
debug
Browse files
duckdb-nsql/eval/prompt_formatters.py
CHANGED
|
@@ -65,6 +65,7 @@ class RajkumarFormatter:
|
|
| 65 |
|
| 66 |
@classmethod
|
| 67 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
|
|
|
| 68 |
pattern = r"```(?:sql|mysql|duckdb)?\n?(.*?)```"
|
| 69 |
match = re.search(pattern, output_sql, re.DOTALL)
|
| 70 |
sql = match.group(1).strip() if match else output_sql.strip()
|
|
@@ -72,7 +73,8 @@ class RajkumarFormatter:
|
|
| 72 |
# Handle edge case where regex captured empty content
|
| 73 |
if not sql:
|
| 74 |
sql = output_sql.strip()
|
| 75 |
-
|
|
|
|
| 76 |
# Ensure single trailing semicolon
|
| 77 |
return sql.rstrip(';') + ';'
|
| 78 |
|
|
|
|
| 65 |
|
| 66 |
@classmethod
|
| 67 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
| 68 |
+
print(output_sql)
|
| 69 |
pattern = r"```(?:sql|mysql|duckdb)?\n?(.*?)```"
|
| 70 |
match = re.search(pattern, output_sql, re.DOTALL)
|
| 71 |
sql = match.group(1).strip() if match else output_sql.strip()
|
|
|
|
| 73 |
# Handle edge case where regex captured empty content
|
| 74 |
if not sql:
|
| 75 |
sql = output_sql.strip()
|
| 76 |
+
|
| 77 |
+
print(sql)
|
| 78 |
# Ensure single trailing semicolon
|
| 79 |
return sql.rstrip(';') + ';'
|
| 80 |
|