load API_TYPE etc.

ku-nlp · Feb 9, 2024 · cda14bd · cda14bd
1 parent 724d208
commit cda14bd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llm_judge/common.py b/llm_judge/common.py
@@ -17,6 +17,9 @@
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
 openai.organization = os.getenv("OPENAI_ORGANIZATION")
+openai.api_type = os.getenv("OPENAI_API_TYPE")
+openai.api_base = os.getenv("OPENAI_API_BASE")
+openai.api_version = os.getenv("OPENAI_API_VERSION")
 
 # Data paths
 JP_BENCH_DIR = Path(__file__).resolve().parent.parent / "data" / "jp_bench"