tdoehmen commited on
Commit
0b84806
·
verified ·
1 Parent(s): 9e806a7
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