atoye1's picture
pushing for deployment
18fb155
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