diff --git a/wren-ai-service/src/utils.py b/wren-ai-service/src/utils.py index bad99cedc..8aa7fc59c 100644 --- a/wren-ai-service/src/utils.py +++ b/wren-ai-service/src/utils.py @@ -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]],