from openai import OpenAI

import logging

client = OpenAI()

def openai_request(messages, model='gpt-3.5-turbo', temperature=1, top_n=1, max_trials=100):
    if openai.api_key is None:
        raise ValueError(
            "You need to set OpenAI API key manually. `opalai.api_key = [your key]`")

    for _ in range(max_trials):
        try:
            results = client.completions.create(
                model=model, messages=messages, temperature=temperature, n=top_n
            )

            assert len(results.choices) == top_n
            return results
        except Exception as e:
            logging.warning("OpenAI API call failed. Retrying...")