Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
import os | |
from openai import OpenAI | |
from dotenv import load_dotenv | |
# Load environment variables | |
load_dotenv() | |
def test_openai_connection(): | |
try: | |
# Initialize OpenAI client | |
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY')) | |
# Make a simple request | |
response = client.chat.completions.create( | |
model="gpt-3.5-turbo", | |
messages=[ | |
{"role": "user", "content": "Say 'Hello World'"} | |
] | |
) | |
print("β OpenAI API connection successful!") | |
print(f"Response: {response.choices[0].message.content}") | |
return True | |
except Exception as e: | |
print("β OpenAI API connection failed!") | |
print(f"Error: {str(e)}") | |
return False | |
if __name__ == "__main__": | |
test_openai_connection() |