Skip to content

Commit d3feb07

Browse files
authored
Merge branch 'main' into interact
2 parents 4b59c7a + fd4f91f commit d3feb07

3 files changed

Lines changed: 5 additions & 2 deletions

File tree

datasets/bat/example_run_config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ setup_directory: datasets/bat/setup
3838
scorers:
3939
exact_match: null
4040
llmrater:
41-
model_config: datasets/model_configs/gemini_1.5-pro-002_model.yaml
41+
model_config: datasets/model_configs/gemini_2.5_pro_model.yaml
4242
returned_sql: null
4343
set_match: null
4444
executable_sql: null

evalbench/generators/prompts/sqlgenbase.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,10 @@ def setup(self):
175175
self.schema = self.db.get_ddl_from_db()
176176

177177
def generate(self, item):
178+
item["prompt"] = self.get_prompt(item)
179+
return item
180+
181+
def get_prompt(self, item):
178182
return self.base_prompt.format(
179183
SCHEMA=self.schema, USER_PROMPT=item["nl_prompt"]
180184
)

evalbench/work/promptgenwork.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ def run(self, work_config: Any = None) -> dict:
2020
Returns:
2121
2222
"""
23-
generated_prompt = None
2423
prompt_generator_error = None
2524
try:
2625
self.generator.generate(self.eval_result)

0 commit comments

Comments
 (0)