🚀 feat: Add OPENAI_ORGANIZATION for all OpenAI Requests (danny-avil…

…a#1590)
aitok-ai · Jan 19, 2024 · e6746c4 · e6746c4
1 parent fc362f6
commit e6746c4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/app/clients/OpenAIClient.js b/app/clients/OpenAIClient.js
@@ -954,6 +954,10 @@ ${convo}
         opts.defaultHeaders = { ...opts.defaultHeaders, 'api-key': this.apiKey };
       }
 
+      if (process.env.OPENAI_ORGANIZATION) {
+        opts.organization = process.env.OPENAI_ORGANIZATION;
+      }
+
       let chatCompletion;
       const openai = new OpenAI({
         apiKey: this.apiKey,

diff --git a/server/services/ModelService.js b/server/services/ModelService.js
@@ -53,6 +53,10 @@ const fetchModels = async ({ apiKey, baseURL, name = 'OpenAI', azure = false })
       payload.httpsAgent = new HttpsProxyAgent(PROXY);
     }
 
+    if (process.env.OPENAI_ORGANIZATION && baseURL.includes('openai')) {
+      payload.headers['OpenAI-Organization'] = process.env.OPENAI_ORGANIZATION;
+    }
+
     const res = await axios.get(`${baseURL}${azure ? '' : '/models'}`, payload);
     models = res.data.data.map((item) => item.id);
   } catch (err) {