File size: 293 Bytes
7dff9c2
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from openai import OpenAI

def generate_response(prompt):
    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[{"role": "user", "content": prompt}]
    )
    return response.choices[0].message.content