@creature-ai/sdk 0.1.3 → 0.1.4

package/dist/server/index.js

@@ -13617,13 +13617,22 @@ var WebSocketConnectionInternal = class {
 };
 var WebSocketManager = class {
   wss = null;
+  httpServer = null;
+  isAttached = false;
   connections = /* @__PURE__ */ new Map();
   /**
-   * Attach the WebSocket server to an HTTP server.
+   * Store the HTTP server reference for lazy WebSocket attachment.
+   */
+  setServer(server) {
+    this.httpServer = server;
+  }
+  /**
+   * Lazily attach the WebSocket server when first needed.
    */
-  attach(server) {
+  ensureAttached() {
+    if (this.isAttached || !this.httpServer) return;
     this.wss = new WebSocketServer({ noServer: true });
-    server.on("upgrade", (request, socket, head) => {
+    this.httpServer.on("upgrade", (request, socket, head) => {
       const url = request.url || "";
       if (!url.startsWith("/ws/")) {
         socket.destroy();
@@ -13662,9 +13671,12 @@ var WebSocketManager = class {
         connection.removeClient(ws);
       });
     });
+    this.isAttached = true;
+    console.log(`WebSocket available at ws://localhost:${this.httpServer.address()?.port || "?"}/ws`);
   }
   /**
    * Create a WebSocket connection for an instance.
+   * Lazily initializes the WebSocket server on first call.
    *
    * @param instanceId - The instance ID
    * @param config - WebSocket configuration
@@ -13672,6 +13684,7 @@ var WebSocketManager = class {
    * @returns A WebSocketConnection for bidirectional communication
    */
   createWebSocket(instanceId, config, port) {
+    this.ensureAttached();
     const existing = this.connections.get(instanceId);
     if (existing) {
       existing.closeAll();
@@ -13723,6 +13736,7 @@ var WebSocketManager = class {
       this.wss.close();
       this.wss = null;
     }
+    this.isAttached = false;
   }
 };
 
@@ -13809,8 +13823,7 @@ var App = class {
     this.httpServer = expressApp.listen(port, () => {
       console.log(`MCP server ready on port ${port}`);
     });
-    this.websocketManager.attach(this.httpServer);
-    console.log(`WebSocket available at ws://localhost:${port}/ws`);
+    this.websocketManager.setServer(this.httpServer);
     this.registerShutdownHandlers();
   }
   /**