Function get_retry_prompt

api/app/api/utils/sql_gen/prompts.py:145–164

Crates the retry prompt for the given scope formatted to the given dialect and schemas. Args: dialect (str): This is something I should know what it is natural_language_query (str): The natural language query that the user is trying to answer schemas (str): The sche

(dialect: str, natural_language_query:str, scope: str="USA")

    return prompt

def get_retry_prompt(dialect: str, natural_language_query:str, scope: str="USA") -> str:
    """
    Crates the retry prompt for the given scope formatted to the given dialect and schemas.

    Args:
        dialect (str): This is something I should know what it is
        natural_language_query (str): The natural language query that the user is trying to answer
        schemas (str): The schemas of the tables (currently formatted as CREATE TABLE...)
        scope (str, optional): The project the prompt belongs to (e.g. USA, SF, etc.), defaults to 'USA'

    Returns:
        str: The formatted prompt
    """

    if scope in RETRY_PROMPTS2:
        prompt = RETRY_PROMPTS2[scope]
    else: prompt = RETRY_PROMPTS2["USA"]
    prompt = prompt.format(dialect,natural_language_query, natural_language_query)
    prompt = 'generation_id: ' + uuid.uuid4().hex + '\n' + prompt
    return prompt