From 2f6974eda8559a21bad5577f45c3cfaf02c28826 Mon Sep 17 00:00:00 2001
From: "Richard Kuo (Danswer)" <rkuo@onyx.app>
Date: Wed, 18 Dec 2024 11:21:05 -0800
Subject: [PATCH] fix some testing vars

---
 .../app/admin/connectors/[connector]/AddConnectorPage.tsx  | 7 ++++---
 web/src/lib/constants.ts                                   | 3 ++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/web/src/app/admin/connectors/[connector]/AddConnectorPage.tsx b/web/src/app/admin/connectors/[connector]/AddConnectorPage.tsx
index 6beb23f2ea2..534fae2814a 100644
--- a/web/src/app/admin/connectors/[connector]/AddConnectorPage.tsx
+++ b/web/src/app/admin/connectors/[connector]/AddConnectorPage.tsx
@@ -51,7 +51,7 @@ import { prepareOAuthAuthorizationRequest } from "@/lib/oauth_utils";
 import {
   EE_ENABLED,
   NEXT_PUBLIC_CLOUD_ENABLED,
-  TEST_ENV,
+  NEXT_PUBLIC_TEST_ENV,
 } from "@/lib/constants";
 import TemporaryLoadingModal from "@/components/TemporaryLoadingModal";
 import { getConnectorOauthRedirectUrl } from "@/lib/connectors/oauth";
@@ -131,7 +131,7 @@ export default function AddConnector({
       setCurrentPageUrl(window.location.href);
     }
 
-    if (EE_ENABLED && (NEXT_PUBLIC_CLOUD_ENABLED || TEST_ENV)) {
+    if (EE_ENABLED && (NEXT_PUBLIC_CLOUD_ENABLED || NEXT_PUBLIC_TEST_ENV)) {
       const sourceMetadata = getSourceMetadata(connector);
       if (sourceMetadata?.oauthSupported == true) {
         setIsAuthorizeVisible(true);
@@ -473,7 +473,8 @@ export default function AddConnector({
                         </button>
                         {/* Button to sign in via OAuth */}
                         {oauthSupportedSources.includes(connector) &&
-                          NEXT_PUBLIC_CLOUD_ENABLED && (
+                          (NEXT_PUBLIC_CLOUD_ENABLED ||
+                            NEXT_PUBLIC_TEST_ENV) && (
                             <button
                               onClick={handleAuthorize}
                               className="mt-6 text-sm bg-blue-500 px-2 py-1.5 flex text-text-200 flex-none rounded"
diff --git a/web/src/lib/constants.ts b/web/src/lib/constants.ts
index 83fbb1174c7..c4a6dd0b9f3 100644
--- a/web/src/lib/constants.ts
+++ b/web/src/lib/constants.ts
@@ -74,7 +74,8 @@ export const NEXT_PUBLIC_CLOUD_ENABLED =
 export const REGISTRATION_URL =
   process.env.INTERNAL_URL || "http://127.0.0.1:3001";
 
-export const TEST_ENV = process.env.TEST_ENV?.toLowerCase() === "true";
+export const NEXT_PUBLIC_TEST_ENV =
+  process.env.NEXT_PUBLIC_TEST_ENV?.toLowerCase() === "true";
 
 export const NEXT_PUBLIC_DELETE_ALL_CHATS_ENABLED =
   process.env.NEXT_PUBLIC_DELETE_ALL_CHATS_ENABLED?.toLowerCase() === "true";