feat: add support for secure wss connection

Arize-ai · Oct 18, 2024 · 243c6fd · 243c6fd
1 parent acce9d9
commit 243c6fd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/app/src/RelayEnvironment.ts b/app/src/RelayEnvironment.ts
@@ -13,7 +13,7 @@ import {
 } from "relay-runtime";
 
 import { authFetch } from "@phoenix/authFetch";
-import { BASE_URL } from "@phoenix/config";
+import { BASE_URL, WS_BASE_URL } from "@phoenix/config";
 
 const graphQLPath = BASE_URL + "/graphql";
 
@@ -56,8 +56,11 @@ const fetchRelay: FetchFunction = async (params, variables, _cacheConfig) => {
   return json;
 };
 
+/**
+ * Check whether or not we are running
+ */
 const wsClient = createClient({
-  url: "ws://localhost:6006/graphql",
+  url: `${WS_BASE_URL}/graphql`,
 });
 
 const subscribe: SubscribeFunction = (

diff --git a/app/src/config.ts b/app/src/config.ts
@@ -2,5 +2,9 @@ function getSanitizedPath(path: string): string {
   return path.endsWith("/") ? path.slice(0, -1) : path;
 }
 
-export const BASE_URL = `${window.location.protocol}//${window.location.host}${getSanitizedPath(window.Config.basename)}`;
+const sanitizedPath = getSanitizedPath(window.Config.basename);
+const isSecure = window.location.protocol === "https:";
+
+export const BASE_URL = `${window.location.protocol}//${window.location.host}${sanitizedPath}`;
+export const WS_BASE_URL = `${isSecure ? "wss" : "ws"}://${window.location.host}${sanitizedPath}`;
 export const VERSION = window.Config.platformVersion;