import util

Authenticate and get the OpenAI module

openai = util.openai()

Define the prompt

prompt = 'Once upon an AI'

Set the maximum number of tokens to generate

max_tokens = 5

Create a completion object

completion = openai.Completion.create(prompt=prompt, max_tokens=max_tokens, echo=False)

Extract the generated text

generated_text = completion.choices[0].text

Print the generated text

print(generated_text)