feat: allow custom openai base url (#108)

OpenBMB · Dec 28, 2023 · 5fb1f77 · 5fb1f77
1 parent 33e61d1
commit 5fb1f77
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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")