(improvement)(test) Fix Python exemplar file name error and fix chat unit tests. (#661)

This commit is contained in:
lexluo09
2024-01-19 18:12:01 +08:00
committed by GitHub
parent 026cf2056d
commit 36edc0c1b4
12 changed files with 189 additions and 193 deletions

View File

@@ -51,7 +51,7 @@ text2sql_agent.reload_setting(sql_ids, sql_exemplars, TEXT2DSL_EXAMPLE_NUM, TEXT
if text2sql_agent_autoCoT.count_examples()==0:
source_dir_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
example_dir_path = os.path.join(source_dir_path, 'few_shot_example')
example_json_file = os.path.join(example_dir_path, 's2ql_examplar3_transformed.json')
example_json_file = os.path.join(example_dir_path, 's2sql_exemplar3_transformed.json.json')
with open(example_json_file, 'r', encoding='utf-8') as f:
transformed_sql_examplar_list = json.load(f)