from flushai.models.llms import OpenAI

llm = OpenAI(model_name="gpt-4", api_key="YOUR_OPENAI_API_KEY")

output = llm.generate("What year was Donald Trump born in?")

print(output)
# Donald Trump was born in 1946.