Function build_prompt

explorer/assistant/utils.py:156–179

(db_connection, assistant_request, included_tables, query_error=None, sql=None)

def build_prompt(db_connection, assistant_request, included_tables, query_error=None, sql=None):
    included_tables = [t.lower() for t in included_tables]

    error_chunk = f"## Query Error ##\n{query_error}" if query_error else None
    sql_chunk = f"## Existing User-Written SQL ##\n{sql}" if sql else None
    request_chunk = f"## User's Request to Assistant ##\n{assistant_request}"
    table_chunks = [
        TablePromptData(
            name=t,
            schema=table_schema(db_connection, t),
            sample=sample_rows_from_table(db_connection.as_django_connection(), t),
            annotation=get_relevant_annotation(db_connection, t)
        ).render()
        for t in included_tables
    ]
    few_shot_chunk = get_few_shot_chunk(db_connection, included_tables)

    chunks = [error_chunk, sql_chunk, *table_chunks, few_shot_chunk, request_chunk]

    prompt = {
        "system": build_system_prompt(db_connection.as_django_connection().vendor),
        "user": "\n\n".join([c for c in chunks if c]),
    }
    return prompt