load API kay from .env

ku-nlp · Nov 28, 2023 · 4c28013 · 4c28013
1 parent 402c241
commit 4c28013
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/llm_judge/gen_gpt3.5_answer.py b/llm_judge/gen_gpt3.5_answer.py
@@ -7,6 +7,13 @@
 
 from typing import List
 
+from dotenv import load_dotenv
+
+load_dotenv()
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+openai.organization = os.getenv("OPENAI_ORGANIZATION")
+
 
 class GPT3_Demo(object):
     def __init__(