api.py

import os
from openai import OpenAI

def call_openai_api(model, messages):

    client = OpenAI(
    	# This is the default and can be omitted
    	api_key=os.environ.get("OPENAI_API_KEY")
    	)
    
    response = client.chat.completions.create(model=model, messages=messages)
    return response