Spaces:
Sleeping
Sleeping
Update duckdb-nsql/eval/prompt_formatters.py
Browse files
duckdb-nsql/eval/prompt_formatters.py
CHANGED
|
@@ -4,6 +4,7 @@ from random import shuffle
|
|
| 4 |
from manifest import Manifest
|
| 5 |
from schema import Table
|
| 6 |
import re
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
class RajkumarFormatter:
|
|
@@ -60,6 +61,7 @@ class RajkumarFormatter:
|
|
| 60 |
context_text: str,
|
| 61 |
) -> str | list[str]:
|
| 62 |
"""Get prompt format."""
|
|
|
|
| 63 |
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
|
| 64 |
|
| 65 |
@classmethod
|
|
|
|
| 4 |
from manifest import Manifest
|
| 5 |
from schema import Table
|
| 6 |
import re
|
| 7 |
+
import time
|
| 8 |
|
| 9 |
|
| 10 |
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
|