Merge pull request #72 from web-arena-x/neubig/allow_openai_base_url

Update openai_utils.py to allow for base url
web-arena-x · Nov 9, 2024 · 89f5af2 · 89f5af2
2 parents 1589092 + f161b0b
commit 89f5af2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llms/providers/openai_utils.py b/llms/providers/openai_utils.py
@@ -12,8 +12,8 @@
 import openai
 from openai import AsyncOpenAI, OpenAI
 
-client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
-aclient = AsyncOpenAI(api_key=os.environ["OPENAI_API_KEY"])
+client = OpenAI(api_key=os.environ["OPENAI_API_KEY"], base_url=os.environ.get("OPENAI_BASE_URL"))
+aclient = AsyncOpenAI(api_key=os.environ["OPENAI_API_KEY"], base_url=os.environ.get("OPENAI_BASE_URL"))
 from tqdm.asyncio import tqdm_asyncio