Revert "Disable managed identity for now"

This reverts commit 9b98534.

.env.local.example

@@ -6,7 +6,3 @@ OPENAI_API_KEY=YOUR_KEY
 # Google
 GOOGLE_API_KEY=YOUR_API_KEY
 GOOGLE_CSE_ID=YOUR_ENGINE_ID
-
-
-
-AZURE_USE_MANAGED_IDENTITY=false

instrumentation.ts

@@ -1,14 +1,14 @@
-// import * as Sentry from '@sentry/nextjs';
+import * as Sentry from '@sentry/nextjs';
 
-// export async function register() {
-//   if (process.env.NEXT_RUNTIME === 'nodejs') {
-//     await import('./sentry.server.config');
-//   }
+export async function register() {
+  if (process.env.NEXT_RUNTIME === 'nodejs') {
+    await import('./sentry.server.config');
+  }
 
-//   if (process.env.NEXT_RUNTIME === 'edge') {
-//     await import('./sentry.edge.config');
-//   }
-// }
+  if (process.env.NEXT_RUNTIME === 'edge') {
+    await import('./sentry.edge.config');
+  }
+}
 
 // export const onRequestError = Sentry.captureRequestError;
 export {}

pages/api/models.ts

@@ -26,7 +26,7 @@ const handler = async (req: Request): Promise<Response> => {
         ...(OPENAI_API_TYPE === 'openai' && {
           Authorization: `Bearer ${key ? key : process.env.OPENAI_API_KEY}`
         }),
-        ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY!="true" && {
+        ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY=="false" && {
           'api-key': `${key ? key : process.env.OPENAI_API_KEY}`
         }),
         ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY=="true" && {

utils/lib/azure.js

@@ -1,27 +1,19 @@
 // lib/azure.js
 
-//import { DefaultAzureCredential } from "@azure/identity";
-//import { log } from "console";
+import { DefaultAzureCredential } from "@azure/identity";
 //import { getCache, setCache } from "./cache";
 
 
 
 export async function getAuthToken() {
-  if (process.env.AZURE_USE_MANAGED_IDENTITY != "true") {
-    return "";
-  }
   let cachedToken = process.env.AUTH_TOKEN ? JSON.parse(process.env.AUTH_TOKEN) : '';
   if (!cachedToken || cachedToken.expiresOnTimestamp < Date.now()) {
     let cachedCredential = new DefaultAzureCredential();
     cachedToken = await cachedCredential.getToken("https://cognitiveservices.azure.com/.default");
-    //setCache("cachedToken", JSON.stringify(cachedToken));
-    //console.log("set the cachedToken in cache");
-    //let a = JSON.parse(getCache("cachedToken"));
-    //console.log("cachedToken from memory", a);
     process.env.AUTH_TOKEN = JSON.stringify(cachedToken);
     return cachedToken;
   }
-  console.log("cachedToken from memory", cachedToken);
+  // console.log("cachedToken from memory", cachedToken);
   return process.env.AUTH_TOKEN ? JSON.parse(process.env.AUTH_TOKEN) : '';;
 }
 

utils/server/index.ts

@@ -47,7 +47,7 @@ export const OpenAIStream = async (
     ...(OPENAI_API_TYPE === 'openai' && {
       Authorization: `Bearer ${key ? key : process.env.OPENAI_API_KEY}`
     }),
-    ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY!="true" && {
+    ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY=="false" && {
       'api-key': `${key ? key : process.env.OPENAI_API_KEY}`
     }),
     ...(OPENAI_API_TYPE === 'azure' && process.env.AZURE_USE_MANAGED_IDENTITY=="true" && {