Update duckdb-nsql/eval/prompt_formatters.py
Browse files
duckdb-nsql/eval/prompt_formatters.py
CHANGED
@@ -61,12 +61,12 @@ class RajkumarFormatter:
|
|
61 |
context_text: str,
|
62 |
) -> str | list[str]:
|
63 |
"""Get prompt format."""
|
64 |
-
time.sleep(10)
|
65 |
return f"""{table_text}\n\n\n-- Using valid DuckDB SQL, answer the following question for the tables provided above.{context_text}\n\n-- {instruction}\n""" # noqa: E501
|
66 |
|
67 |
@classmethod
|
68 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
69 |
"""Format model output."""
|
|
|
70 |
clean_sql = (output_sql
|
71 |
.replace('sql\n', '')
|
72 |
.replace('```sql\n', '')
|
|
|
61 |
context_text: str,
|
62 |
) -> str | list[str]:
|
63 |
"""Get prompt format."""
|
|
|
64 |
return f"""{table_text}\n\n\n-- Using valid DuckDB SQL, answer the following question for the tables provided above.{context_text}\n\n-- {instruction}\n""" # noqa: E501
|
65 |
|
66 |
@classmethod
|
67 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
68 |
"""Format model output."""
|
69 |
+
time.sleep(10)
|
70 |
clean_sql = (output_sql
|
71 |
.replace('sql\n', '')
|
72 |
.replace('```sql\n', '')
|