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
|