@bike4mind/cli 0.2.29-feat-cli-websocket-streaming.18878 → 0.2.29-feat-cli-websocket-streaming.18880

package/dist/index.js

@@ -13484,9 +13484,10 @@ var WebSocketLlmBackend = class {
     this.apiClient = options.apiClient;
     this.currentModel = options.model;
     this.tokenGetter = options.tokenGetter;
+    this.wsCompletionUrl = options.wsCompletionUrl;
   }
   /**
-   * Make LLM completion request via WebSocket.
+   * Send completion request via HTTP POST, receive streaming response via WebSocket.
    * Collects all streamed chunks, then calls callback once at completion
    * with the full accumulated content.
    */
@@ -13500,10 +13501,6 @@ var WebSocketLlmBackend = class {
       throw new Error("WebSocket is not connected");
     }
     const requestId = uuidv411();
-    const token = await this.tokenGetter();
-    if (!token) {
-      throw new Error("No access token available");
-    }
     return new Promise((resolve3, reject) => {
       const isVerbose = process.env.B4M_VERBOSE === "1";
       const isUltraVerbose = process.env.B4M_DEBUG_STREAM === "1";
@@ -13519,10 +13516,16 @@ var WebSocketLlmBackend = class {
         if (settled) return;
         settled = true;
         this.wsManager.offRequest(requestId);
+        this.wsManager.offDisconnect(onDisconnect);
         action();
       };
       const settleResolve = () => settle(() => resolve3());
       const settleReject = (err) => settle(() => reject(err));
+      const onDisconnect = () => {
+        logger.debug("[WebSocketLlmBackend] Connection dropped during completion");
+        settleReject(new Error("WebSocket connection lost during completion"));
+      };
+      this.wsManager.onDisconnect(onDisconnect);
       if (options.abortSignal) {
         if (options.abortSignal.aborted) {
           settleResolve();
@@ -13578,10 +13581,10 @@ var WebSocketLlmBackend = class {
           settleReject(new Error(errorMsg));
         }
       });
-      try {
-        this.wsManager.send({
-          action: "cli_completion_request",
-          accessToken: token,
+      const axiosInstance = this.apiClient.getAxiosInstance();
+      axiosInstance.post(
+        this.wsCompletionUrl,
+        {
           requestId,
           model,
           messages,
@@ -13591,10 +13594,12 @@ var WebSocketLlmBackend = class {
             stream: true,
             tools: options.tools || []
           }
-        });
-      } catch (err) {
-        settleReject(err instanceof Error ? err : new Error(String(err)));
-      }
+        },
+        { signal: options.abortSignal }
+      ).catch((err) => {
+        const msg = err instanceof Error ? err.message : String(err);
+        settleReject(new Error(`HTTP request failed: ${msg}`));
+      });
     });
   }
   /**
@@ -13677,6 +13682,7 @@ var WebSocketConnectionManager = class {
     this.reconnectAttempts = 0;
     this.maxReconnectDelay = 3e4;
     this.handlers = /* @__PURE__ */ new Map();
+    this.disconnectHandlers = /* @__PURE__ */ new Set();
     this.connected = false;
     this.connecting = false;
     this.closed = false;
@@ -13724,6 +13730,7 @@ var WebSocketConnectionManager = class {
       this.ws.onclose = () => {
         logger.debug("[WS] Connection closed");
         this.cleanup();
+        this.notifyDisconnect();
         if (!this.closed) {
           this.scheduleReconnect();
         }
@@ -13749,7 +13756,13 @@ var WebSocketConnectionManager = class {
     if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
       throw new Error("WebSocket is not connected");
     }
-    this.ws.send(JSON.stringify(data));
+    const payload = JSON.stringify(data);
+    const sizeKB = (payload.length / 1024).toFixed(1);
+    logger.debug(`[WS] Sending ${sizeKB} KB (action: ${data.action})`);
+    if (payload.length > 32e3) {
+      logger.warn(`[WS] Payload ${sizeKB} KB exceeds API Gateway 32 KB frame limit \u2014 connection will be closed`);
+    }
+    this.ws.send(payload);
   }
   /**
    * Register a handler for messages matching a specific requestId.
@@ -13763,6 +13776,18 @@ var WebSocketConnectionManager = class {
   offRequest(requestId) {
     this.handlers.delete(requestId);
   }
+  /**
+   * Register a handler that fires when the connection drops.
+   */
+  onDisconnect(handler) {
+    this.disconnectHandlers.add(handler);
+  }
+  /**
+   * Remove a disconnect handler.
+   */
+  offDisconnect(handler) {
+    this.disconnectHandlers.delete(handler);
+  }
   /**
    * Close the connection and stop all heartbeat/reconnect logic.
    */
@@ -13774,6 +13799,7 @@ var WebSocketConnectionManager = class {
       this.ws = null;
     }
     this.handlers.clear();
+    this.disconnectHandlers.clear();
   }
   startHeartbeat() {
     this.stopHeartbeat();
@@ -13798,6 +13824,14 @@ var WebSocketConnectionManager = class {
     this.connecting = false;
     this.stopHeartbeat();
   }
+  notifyDisconnect() {
+    for (const handler of this.disconnectHandlers) {
+      try {
+        handler();
+      } catch {
+      }
+    }
+  }
   scheduleReconnect() {
     if (this.closed) return;
     this.reconnectAttempts++;
@@ -13840,10 +13874,15 @@ var WebSocketToolExecutor = class {
         if (settled) return;
         settled = true;
         this.wsManager.offRequest(requestId);
+        this.wsManager.offDisconnect(onDisconnect);
         action();
       };
       const settleResolve = (result) => settle(() => resolve3(result));
       const settleReject = (err) => settle(() => reject(err));
+      const onDisconnect = () => {
+        settleReject(new Error("WebSocket connection lost during tool execution"));
+      };
+      this.wsManager.onDisconnect(onDisconnect);
       if (abortSignal) {
         if (abortSignal.aborted) {
           settleReject(new Error("Tool execution aborted"));
@@ -14014,7 +14053,7 @@ import { isAxiosError as isAxiosError2 } from "axios";
 // package.json
 var package_default = {
   name: "@bike4mind/cli",
-  version: "0.2.29-feat-cli-websocket-streaming.18878+661239950",
+  version: "0.2.29-feat-cli-websocket-streaming.18880+855c924c8",
   type: "module",
   description: "Interactive CLI tool for Bike4Mind with ReAct agents",
   license: "UNLICENSED",
@@ -14124,10 +14163,10 @@ var package_default = {
   },
   devDependencies: {
     "@bike4mind/agents": "0.1.0",
-    "@bike4mind/common": "2.50.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/mcp": "1.29.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/services": "2.48.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/utils": "2.5.1-feat-cli-websocket-streaming.18878+661239950",
+    "@bike4mind/common": "2.50.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/mcp": "1.29.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/services": "2.48.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/utils": "2.5.1-feat-cli-websocket-streaming.18880+855c924c8",
     "@types/better-sqlite3": "^7.6.13",
     "@types/diff": "^5.0.9",
     "@types/jsonwebtoken": "^9.0.4",
@@ -14144,7 +14183,7 @@ var package_default = {
   optionalDependencies: {
     "@vscode/ripgrep": "^1.17.0"
   },
-  gitHead: "661239950fcb26f7027596b319567a19d56fb1a7"
+  gitHead: "855c924c80996fa2e1ed1f16b536416809a8b87c"
 };
 
 // src/config/constants.ts
@@ -16079,9 +16118,12 @@ function CliApp() {
       let wsManager = null;
       let llm;
       try {
-        const serverConfig = await apiClient.get("/api/settings/serverConfig");
+        const serverConfig = await apiClient.get(
+          "/api/settings/serverConfig"
+        );
         const wsUrl = serverConfig?.websocketUrl;
-        if (wsUrl) {
+        const wsCompletionUrl = serverConfig?.wsCompletionUrl;
+        if (wsUrl && wsCompletionUrl) {
           wsManager = new WebSocketConnectionManager(wsUrl, tokenGetter);
           await wsManager.connect();
           const wsToolExecutor2 = new WebSocketToolExecutor(wsManager, tokenGetter);
@@ -16090,11 +16132,12 @@ function CliApp() {
             wsManager,
             apiClient,
             model: config.defaultModel,
-            tokenGetter
+            tokenGetter,
+            wsCompletionUrl
           });
           logger.debug("\u{1F50C} Using WebSocket transport (bypasses CloudFront timeout)");
         } else {
-          throw new Error("No websocketUrl in server config");
+          throw new Error("No websocketUrl or wsCompletionUrl in server config");
         }
       } catch (wsError) {
         logger.debug(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bike4mind/cli",
-  "version": "0.2.29-feat-cli-websocket-streaming.18878+661239950",
+  "version": "0.2.29-feat-cli-websocket-streaming.18880+855c924c8",
   "type": "module",
   "description": "Interactive CLI tool for Bike4Mind with ReAct agents",
   "license": "UNLICENSED",
@@ -110,10 +110,10 @@
   },
   "devDependencies": {
     "@bike4mind/agents": "0.1.0",
-    "@bike4mind/common": "2.50.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/mcp": "1.29.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/services": "2.48.1-feat-cli-websocket-streaming.18878+661239950",
-    "@bike4mind/utils": "2.5.1-feat-cli-websocket-streaming.18878+661239950",
+    "@bike4mind/common": "2.50.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/mcp": "1.29.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/services": "2.48.1-feat-cli-websocket-streaming.18880+855c924c8",
+    "@bike4mind/utils": "2.5.1-feat-cli-websocket-streaming.18880+855c924c8",
     "@types/better-sqlite3": "^7.6.13",
     "@types/diff": "^5.0.9",
     "@types/jsonwebtoken": "^9.0.4",
@@ -130,5 +130,5 @@
   "optionalDependencies": {
     "@vscode/ripgrep": "^1.17.0"
   },
-  "gitHead": "661239950fcb26f7027596b319567a19d56fb1a7"
+  "gitHead": "855c924c80996fa2e1ed1f16b536416809a8b87c"
 }