fix: os.environ['OPENAI_API_KEY'] exception when no environment variable set

2023-07-28 16:51:58 +08:00 · 2023-07-28 16:51:58 +08:00 · 8b1cf7effd
parent 70c6870776
commit 8b1cf7effd
1 changed files with 1 additions and 1 deletions
--- a/configs/model_config.py
+++ b/configs/model_config.py
@ -70,7 +70,7 @@ llm_model_dict = {
    "openai-chatgpt-3.5": {
        "local_model_path": "gpt-3.5-turbo",
        "api_base_url": "https://api.openapi.com/v1",
-        "api_key": os.environ["OPENAI_API_KEY"]
+        "api_key": os.environ.get("OPENAI_API_KEY")
    },
 }