test_prediction.py

from prediction import ModelPrediction
import utilities as us


def main():
    model = ModelPrediction()
    response = model.make_prediction(question='What is the name of Simone', db_schema='CREATE TABLE Player(Name, Age)', model_name="gpt-3.5", prompt='{question} {db_schema}')
    print(response)  # dict[response, response_parsed, cost]


if __name__ == "__main__":
    # main()
    us.check_and_create_dir('data/data_interface/')
    us.check_and_create_dir('data/data_results/')
    us.check_and_create_dir('data/databases/')