Method __init__

prompt/get_sql_prompt_for_fewshot.py:10–25

(self)

class SQLPromptError:
    def __init__(self):
        self.error_case = "Here are the questions you didn’t get right sql:\n" \
                          "Q: What is first names of the top 5 staff who have handled the greatest " \
                          "number of complaints?\n" \
                          "sql: SELECT t1.first_name FROM staff AS t1 JOIN complaints AS t2 ON t1.staff_id = "\
                          "t2.staff_id GROUP BY t2.staff_id ORDER BY count(*) LIMIT 5\n\n" \
                          "Q: What are the first name and major of the students who are able to consume soy?\n" \
                          "sql: SELECT fname, major FROM Student WHERE StuID NOT IN (SELECT StuID FROM " \
                          "Has_allergy WHERE Allergy = 'Soy')\n\n" \
                          "Q: What is the name of the instructor who advises the student with the  greatest number " \
                          "of total credits?\n" \
                          ": SELECT T2.name FROM advisor AS T1 JOIN instructor AS T2 ON T1.i_id = T2.id JOIN student " \
                          "AS T3 ON T1.s_id = T3.id ORDER BY T3.tot_cred DESC LIMIT 1\n\n" \
                          "Q: How many countries do not have an roller coaster longer than 3000?\n" \
                          "SELECT LastName FROM CUSTOMER EXCEPT SELECT T1.LastName FROM CUSTOMER AS T1 JOIN " \
                          "Invoice AS T2 ON T1.CustomerId = T2.CustomerId WHERE T2.total > 20\n\n"

    def get_multi_sql_prompt(self):
        sql_prompt = """{few_shots}### Database scheme: {table_info}