test_prediction.py

from prediction import ModelPrediction


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()
    # do something with prompt