File size: 429 Bytes
d3a3bf1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# test_openai.py

import os
from openai import OpenAI

from dotenv import load_dotenv
load_dotenv()
# Set your API key directly or use an environment variable
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

# Simple test: ask ChatGPT something basic
response = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[{"role": "user", "content": "Say hello"}],
)

print(response.choices[0].message.content)