File size: 569 Bytes
18fb155
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import openai
import os

def check_openai_api_key():
    try:
        openai.api_key = os.get_env("OPENAI_API_KEY")
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": "Hello, OpenAI!"}
            ]
        )
        print(response)
        if response and "choices" in response:
            return True
        else:
            return False
    except Exception as e:
        return False