From 24f779ff957c7e286aa6dcd628fb59617fc6ec93 Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
Date: Tue, 2 Dec 2025 17:41:31 +0800
Subject: [PATCH] fix: websocket connect failed in prod env

---
 web/src/app/infra/websocket/WebSocketClient.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/web/src/app/infra/websocket/WebSocketClient.ts b/web/src/app/infra/websocket/WebSocketClient.ts
index 32b1a793..7ebe418d 100644
--- a/web/src/app/infra/websocket/WebSocketClient.ts
+++ b/web/src/app/infra/websocket/WebSocketClient.ts
@@ -79,8 +79,10 @@ export class WebSocketClient {
         // 构建WebSocket URL
         const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
         // extract host from process.env.NEXT_PUBLIC_API_BASE_URL
+        // 如果环境变量未定义,使用当前页面的 host (适配生产环境)
         const host =
-          process.env.NEXT_PUBLIC_API_BASE_URL?.split('://')[1] || '';
+          process.env.NEXT_PUBLIC_API_BASE_URL?.split('://')[1] ||
+          window.location.host;
         const url = `${protocol}//${host}/api/v1/pipelines/${this.pipelineId}/ws/connect?session_type=${this.sessionType}`;
 
         this.ws = new WebSocket(url);