Spaces:
Sleeping
Sleeping
| import requests | |
| from typing import Dict | |
| def validate_openai_api_key(token: str) -> Dict[str, str]: | |
| api_endpoint = "https://api.openai.com/v1/chat/completions" | |
| api_key = token | |
| headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"} | |
| messages = [{"role": "user", "content": "Say this is a test!"}] | |
| data = {"model": "gpt-3.5-turbo", "messages": messages} | |
| response = requests.post(api_endpoint, json=data, headers=headers) | |
| if response.status_code == 200: | |
| return {"status": "success", "message": "API key is valid"} | |
| else: | |
| return {"status": "error", "message": response.json()["error"]["message"]} | |