Open AI API completion function

openai.alternatives.openai.py

import openai

openai.api_key = os.getenv('OPENAI_API_KEY_1')

def get_open_ai_completion(prompt, model="gpt-3.5-turbo"):
    messages = [{"role": "user", "content": prompt}]
    open_ai_response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=0.2,
    )
    return open_ai_response.choices[0].message["content"]