from test_context import create_data_summary_by_cluster, loopv2, chat2data


if __name__ == "__main__":
    cluster_id = "10037815453688240957"
    dbname = "test"

    result = create_data_summary_by_cluster(cluster_id, dbname)["result"]
    data_summary_id, job_id = result["data_summary_id"], result["job_id"]
    loopv2(job_id)
    raw_question = "any sql"

    while True:
        job_id = chat2data(data_summary_id, raw_question)
        loopv2(job_id)