cascade-ai 0.12.0 → 0.12.2

package/dist/index.cjs

@@ -10605,6 +10605,18 @@ var DashboardSocket = class {
       });
     });
   }
+  emitToSocket(socketId, event, data) {
+    this.io.sockets.sockets.get(socketId)?.emit(event, data);
+  }
+  onCascadeRun(callback) {
+    this.io.on("connection", (socket) => {
+      socket.on("cascade:run", (payload) => {
+        if (typeof payload?.prompt === "string" && payload.prompt.trim()) {
+          callback(payload.prompt.trim(), payload.model ?? "auto", socket.id);
+        }
+      });
+    });
+  }
   close() {
     this.io.close();
   }
@@ -10663,6 +10675,43 @@ var DashboardServer = class {
         }
       }
     });
+    this.socket.onCascadeRun(async (prompt, model, socketId) => {
+      const sessionId = crypto.randomUUID();
+      const cfg = model !== "auto" ? { ...this.config, models: { ...this.config.models, t1: model } } : this.config;
+      const cascade = new Cascade(cfg, this.workspacePath, this.store);
+      this.activeSessions.set(sessionId, cascade);
+      cascade.on("stream:token", (e) => {
+        this.socket.emitToSocket(socketId, "stream:token", { sessionId, tierId: e.tierId, text: e.text });
+        this.socket.broadcast("stream:token", { sessionId, tierId: e.tierId, text: e.text });
+      });
+      cascade.on("tier:status", (e) => {
+        this.socket.emitToSocket(socketId, "tier:status", { sessionId, ...e });
+        this.socket.broadcast("tier:status", { sessionId, ...e });
+      });
+      cascade.on("permission:user-required", (e) => {
+        this.socket.emitToSocket(socketId, "permission:user-required", { sessionId, ...e });
+      });
+      cascade.on("peer:message", (e) => {
+        this.socket.emitPeerMessage(e);
+      });
+      try {
+        const result = await cascade.run({ prompt });
+        this.socket.emitToSocket(socketId, "session:complete", { sessionId, result });
+        this.socket.broadcast("cost:update", {
+          sessionId,
+          totalTokens: result.usage.totalTokens,
+          totalCostUsd: result.usage.estimatedCostUsd
+        });
+        this.throttledBroadcast("workspace");
+      } catch (err) {
+        this.socket.emitToSocket(socketId, "session:error", {
+          sessionId,
+          error: err instanceof Error ? err.message : String(err)
+        });
+      } finally {
+        this.activeSessions.delete(sessionId);
+      }
+    });
   }
   async start() {
     const isLoopback = this.host === "127.0.0.1" || this.host === "::1" || this.host === "localhost";