Set api base from config if not specified from env

2023-08-26 08:58:34 +02:00
parent f740f307e2
commit d1ae51ae5d
1 changed files with 2 additions and 1 deletions
--- a/examples/discord/main.py
+++ b/examples/discord/main.py
@@ -14,7 +14,8 @@ OPENAI_API_KEY = config["openai"][str("api_key")]

 if OPENAI_API_KEY == "":
    OPENAI_API_KEY = "foo"
-os.environ["OPENAI_API_BASE"] = config["agent"]["api_base"]
+if "OPENAI_API_BASE" not in os.environ:
+    os.environ["OPENAI_API_BASE"] = config["agent"]["api_base"]
 os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 import openai