def get_join_prompt()

in nl2sql_src/nl2sql_generic.py [0:0]

• 26 lines of code
• 2 McCabe index (conditional complexity)

    def get_join_prompt(self,
                        dataset,
                        table_1_name,
                        table_2_name,
                        question,
                        sample_question=None,
                        sample_sql=None,
                        one_shot=False):
        """
            Crete the prompt for Joins
        """
        prompt = ""
        table_1 = self.table_details(table_1_name)
        table_2 = self.table_details(table_2_name)

        if one_shot:
            prompt = join_prompt_template_one_shot.format(
                data_set=dataset,
                table_1=table_1,
                table_2=table_2,
                sample_question=sample_question,
                sample_sql=sample_sql,
                question=question
            )
        else:
            prompt = join_prompt_template.format(data_set=dataset,
                                                 table_1=table_1,
                                                 table_2=table_2,
                                                 question=question)

        return prompt