verify the openai key (#56)

This commit is contained in:
Pao Sheng
2024-04-01 16:48:22 +08:00
committed by GitHub
parent 71f97a3b72
commit 8127af8e69
+9
View File
@@ -4,6 +4,7 @@ from typing import Dict, List, Optional
import requests
from dotenv import load_dotenv
from openai import OpenAI
def clean_generation_result(result: str) -> str:
@@ -28,9 +29,17 @@ def load_env_vars() -> str:
else:
load_dotenv(".env.prod", override=True)
_verify_env_vars()
return "dev" if is_dev_env else "prod"
def _verify_env_vars() -> None:
"""
this is a temporary solution to verify that the required environment variables are set
"""
OpenAI().models.list()
def classify_invalid_generation_results(
api_endpoint: str,
generation_results: List[Dict[str, str]],