import openai

openai.api_key = "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"
upload_file_id = open("upload_key.txt", "rt").read().strip()
print(upload_file_id)


def query_openai():
    question = input("query: ")
    response = openai.Answer.create(
        search_model="ada",
        model="curie",
        question=question,
        file=upload_file_id,
        examples_context="In 2017, U.S. life expectancy was 78.6 years.",
        examples=[["What is human life expectancy in the United States?", "78 years."]],
        max_rerank=12,
        max_tokens=730,
        stop=["\n", "<|endoftext|>"],
    )

    for ix in range(len(response.answers)):
        print(f"response: {response.answers[ix]}\n")


while True:
    query_openai()