Function correct_sql_self

correct_sql.py:11–40

(query, table_info, sql)

def correct_sql_self(query, table_info, sql):
    sql_prompt = "For the given question, use the Database scheme to fix the given SQLite QUERY for any issuses.\n" \
                 "If there are any problems, please fix them.\n" \
                 "If there are no issues, return SQLite QUERY as is.\n" \
                 "### There are some instructions for fixing the SQL query:\n" \
                 "1) In sql, just select columns that are explicitly requested in the query.\n" \
                 "2) Pay attention to the columns that are used for the SELECT clause. " \
                 "Fix possible ambiguous columns if there are the same columns in different table in the SELECT clause.\n" \
                 "3) Pay attention to the correspondence between tables and fields. " \
                 "Cannot use fields that are not in the table.\n" \
                 "4) Pay attention to the columns that are used for the JOIN." \
                 "The join table condition must be in the Foreign_keys.\n" \
                 "5) Pay attention to the use of the JOIN." \
                 "Don't use LEFT JOIN unless necessary.\n" \
                 "6) Only change the SELECT, GROUP BY and ORDER BY clause when necessary.\n" \
                 "7) " \
                 f"Database scheme: {table_info}\n" \
                 f"### Question: {query}\n" \
                 f"### SQLite SQL QUERY:\n" \
                 f"{sql}\n" \
                 f"### Fixed SQL QUERY:" \
                 f"SELECT\n"
    prompt_dict = {
        "query": query,
        "table_info": table_info,
        "sql": sql
    }
    sql_prompt = get_prompt_content(sql_prompt, prompt_dict)
    sql = ask_llm(sql_prompt)
    return sql


def correct_sql_by_case(query, table_info, sql):