Spaces:
Sleeping
Sleeping
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', '')
|