Support organization.

ossrs · Feb 2, 2024 · e44dcf1 · e44dcf1
1 parent 045a427
commit e44dcf1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools.py b/tools.py
@@ -24,6 +24,9 @@ def openai_init(key, proxy):
     else:
         raise Exception("OPENAI_API_KEY is not set")
 
+    if os.environ.get("OPENAI_ORGANIZATION") is not None:
+        openai.organization = os.environ.get("OPENAI_ORGANIZATION")
+
     if proxy is not None:
         openai.api_base = "http://" + proxy + "/v1/"
     elif os.environ.get("OPENAI_PROXY") is not None: