From c72a3a0362775fee7c6e7aeff8c2c21d76b6eda6 Mon Sep 17 00:00:00 2001
From: Danny Avila <110412045+danny-avila@users.noreply.github.com>
Date: Wed, 10 May 2023 23:56:24 -0400
Subject: [PATCH] refactor(titleConvo.js, endpoints.js): add support for
 AZURE_OPENAI_API_KEY environment variable (#235)

---
 api/app/titleConvo.js          | 3 ++-
 api/server/routes/endpoints.js | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/api/app/titleConvo.js b/api/app/titleConvo.js
index 7d624664b3..15092498e2 100644
--- a/api/app/titleConvo.js
+++ b/api/app/titleConvo.js
@@ -34,7 +34,9 @@ const titleConvo = async ({ endpoint, text, response }) => {
     ||>Title:`
     };
 
+    const azure = process.env.AZURE_OPENAI_API_KEY ? true : false;
     const options = {
+      azure,
       reverseProxyUrl: process.env.OPENAI_REVERSE_PROXY || null,
       proxy: process.env.PROXY || null
     };
@@ -48,7 +50,6 @@ const titleConvo = async ({ endpoint, text, response }) => {
       frequency_penalty: 0
     };
 
-    const azure = process.env.AZURE_OPENAI_API_KEY ? true : false;
     let apiKey = process.env.OPENAI_KEY;
 
     if (azure) {
diff --git a/api/server/routes/endpoints.js b/api/server/routes/endpoints.js
index 57835b24ba..31b9957268 100644
--- a/api/server/routes/endpoints.js
+++ b/api/server/routes/endpoints.js
@@ -17,7 +17,7 @@ const getChatGPTBrowserModels = () => {
 
 router.get('/', function (req, res) {
   const azureOpenAI = !!process.env.AZURE_OPENAI_KEY;
-  const openAI = process.env.OPENAI_KEY ? { availableModels: getOpenAIModels() } : false;
+  const openAI = process.env.OPENAI_KEY || process.env.AZURE_OPENAI_API_KEY ? { availableModels: getOpenAIModels() } : false;
   const bingAI = process.env.BINGAI_TOKEN
     ? { userProvide: process.env.BINGAI_TOKEN == 'user_provided' }
     : false;