Generate SQL for fewshot=query_sim to use
(self, input_dict, mode)
| 358 | return input_dict |
| 359 | |
| 360 | def step_get_pre_sql(self, input_dict, mode): |
| 361 | """Generate SQL for fewshot=query_sim to use""" |
| 362 | limitation = ",".join(input_dict["ner_results"]["limitation"]) |
| 363 | metric = ",".join(input_dict["ner_results"]["metric"]) |
| 364 | main_metric = input_dict["ner_results"]["query"] |
| 365 | few_shots = "" |
| 366 | query = input_dict["query"] |
| 367 | if self.question_type == QuestionType.EASY.value: |
| 368 | table_info_sql = input_dict["table_info"] |
| 369 | else: |
| 370 | table_info_sql = input_dict["add_table_info"] |
| 371 | |
| 372 | input_dict["suggestions"] = "" |
| 373 | sc_results = self.get_sql_by_llm(query, table_info_sql, limitation, metric, |
| 374 | main_metric, input_dict["suggestions"], few_shots) |
| 375 | input_dict["pre_sql_results"] = sc_results[0] |
| 376 | |
| 377 | if input_dict["pre_sql_results"].get("sql", "") != "": |
| 378 | pre_sql = input_dict["pre_sql_results"]["sql"] |
| 379 | elif list(input_dict["pre_sql_results"].values()): |
| 380 | pre_sql = list(input_dict["pre_sql_results"].values())[0] |
| 381 | elif input_dict["features"].get("sql", "") != "": |
| 382 | pre_sql = input_dict["features"]["sql"] |
| 383 | else: |
| 384 | pre_sql = "error" |
| 385 | |
| 386 | input_dict["pre_sql_results"]["sql"] = pre_sql |
| 387 | return input_dict |
| 388 | |
| 389 | def step_get_sql(self, input_dict, mode): |
| 390 | limitation = ",".join(input_dict["ner_results"]["limitation"]) |