tdoehmen commited on
Commit
64fb3b3
·
verified ·
1 Parent(s): 8719af2

Add new prompt to constants (#2)

Browse files

- Update duckdb-nsql/eval/constants.py (556178cc878ee6b5dd0ec16d2529e81058205c4e)

Files changed (1) hide show
  1. duckdb-nsql/eval/constants.py +2 -0
duckdb-nsql/eval/constants.py CHANGED
@@ -17,6 +17,7 @@ from prompt_formatters import (
17
  DuckDBInstFormatterPhiAzure,
18
  DuckDBInstFormatterLlamaSyntax,
19
  DuckDBInstFormatterCustom,
 
20
  )
21
 
22
  PROMPT_FORMATTERS = {
@@ -26,6 +27,7 @@ PROMPT_FORMATTERS = {
26
  "duckdbinst": DuckDBInstFormatter,
27
  "duckdbinstllamashort": DuckDBInstFormatterLlamaShort,
28
  "duckdbinstgraniteshort": DuckDBInstFormatterGraniteShort,
 
29
  "duckdbinstllama": DuckDBInstFormatterLlama,
30
  "duckdbinstgranite": DuckDBInstFormatterGranite,
31
  "duckdbinstnoshort": DuckDBInstNoShorthandFormatter,
 
17
  DuckDBInstFormatterPhiAzure,
18
  DuckDBInstFormatterLlamaSyntax,
19
  DuckDBInstFormatterCustom,
20
+ DuckDBInstFormatterGraniteBench
21
  )
22
 
23
  PROMPT_FORMATTERS = {
 
27
  "duckdbinst": DuckDBInstFormatter,
28
  "duckdbinstllamashort": DuckDBInstFormatterLlamaShort,
29
  "duckdbinstgraniteshort": DuckDBInstFormatterGraniteShort,
30
+ "duckdbinstgranitebench": DuckDBInstFormatterGraniteBench,
31
  "duckdbinstllama": DuckDBInstFormatterLlama,
32
  "duckdbinstgranite": DuckDBInstFormatterGranite,
33
  "duckdbinstnoshort": DuckDBInstNoShorthandFormatter,