I got this one to work:
from openai import OpenAI client = OpenAI(api_key=OPENAI_API_KEY) response = client.chat.completions.create(...
I hope it solves the issue.