(self)
| 137 | return results |
| 138 | |
| 139 | def get_generate_sql_prompt(self): |
| 140 | if self.few_shot_mode in ["random", "ques_sim", "masked_ques_sim", "query_sim"] and args.reduce_ql: |
| 141 | return self.sql_prompt_fewshot.get_reduce_ql_sql_prompt() |
| 142 | if self.few_shot_mode in ["random", "ques_sim", "masked_ques_sim", "query_sim"] and not args.has_error_case: |
| 143 | if self.question_type == QuestionType.EASY.value: |
| 144 | sql_prompt = self.sql_prompt_fewshot.get_single_sql_prompt() |
| 145 | elif self.question_type == QuestionType.NESTED.value: |
| 146 | sql_prompt = self.sql_prompt_fewshot.get_nested_sql_prompt() |
| 147 | elif self.question_type == QuestionType.JOIN_NESTED.value: |
| 148 | sql_prompt = self.sql_prompt_fewshot.get_join_nested_sql_prompt() |
| 149 | else: |
| 150 | sql_prompt = self.sql_prompt_fewshot.get_multi_sql_prompt() |
| 151 | return sql_prompt |
| 152 | |
| 153 | if args.has_error_case: |
| 154 | if self.question_type == QuestionType.EASY.value: |
| 155 | sql_prompt = self.sql_prompt_error.get_single_sql_prompt() |
| 156 | |
| 157 | elif self.question_type == QuestionType.NESTED.value: |
| 158 | sql_prompt = self.sql_prompt_error.get_nested_sql_prompt() |
| 159 | elif self.question_type == QuestionType.JOIN_NESTED.value: |
| 160 | sql_prompt = self.sql_prompt_error.get_join_nested_sql_prompt() |
| 161 | else: |
| 162 | sql_prompt = self.sql_prompt_error.get_multi_sql_prompt() |
| 163 | return sql_prompt |
| 164 | |
| 165 | if self.question_type == QuestionType.EASY.value: |
| 166 | sql_prompt = get_single_sql_prompt() |
| 167 | elif self.question_type == QuestionType.NESTED.value: |
| 168 | sql_prompt = get_nested_sql_prompt() |
| 169 | elif self.question_type == QuestionType.JOIN_NESTED.value: |
| 170 | sql_prompt = get_join_nested_sql_prompt() |
| 171 | else: |
| 172 | sql_prompt = get_multi_sql_prompt() |
| 173 | return sql_prompt |
| 174 | |
| 175 | @self_consistency(args.sc_nums, ["sql"], "each") |
| 176 | def get_sql_by_llm(self, query, table_info, limitation, metric, main_metric, suggestion, few_shots, |
no test coverage detected