From 1da5e1119d68f4c5b122e2aad98e3084cda8ed3b Mon Sep 17 00:00:00 2001 From: soulteary Date: Wed, 27 Dec 2023 15:45:48 +0800 Subject: [PATCH] feat: allow custom openai base url --- agentverse/llms/openai.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/agentverse/llms/openai.py b/agentverse/llms/openai.py index c119dcd80..7897ef2fc 100644 --- a/agentverse/llms/openai.py +++ b/agentverse/llms/openai.py @@ -30,6 +30,10 @@ if os.environ.get("OPENAI_API_KEY") != None: openai.api_key = os.environ.get("OPENAI_API_KEY") is_openai_available = True + # set openai api base url if it is set + if os.environ.get("OPENAI_BASE_URL") != None: + openai.base_url = os.environ.get("OPENAI_BASE_URL") + print("use new openai base url", openai.base_url) elif os.environ.get("AZURE_OPENAI_API_KEY") != None: openai.api_type = "azure" openai.api_key = os.environ.get("AZURE_OPENAI_API_KEY")