Function get_initial_prompt

api/app/api/utils/sql_gen/prompts.py:127–143

Crates the initial prompt for the given scope formatted to the given dialect and schemas. Args: dialect (str): This is something I should know what it is schemas (str): The schemas of the tables (currently formatted as CREATE TABLE...) scope (str, optional): The pro

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

}

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

    Args:
        dialect (str): This is something I should know what it is
        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 INITIAL_PROMPTS:
        prompt = INITIAL_PROMPTS[scope]
    else: prompt = INITIAL_PROMPTS["USA"]
    prompt = prompt.format(dialect, schemas)
    return prompt

def get_retry_prompt(dialect: str, natural_language_query:str, scope: str="USA") -> str:
    """