cascade-ai 0.10.1 → 0.10.4

package/dist/index.cjs

@@ -8060,6 +8060,19 @@ var TaskAnalyzer = class {
     this.lastSelectedModels.clear();
     void this.tracker.save();
   }
+  /**
+   * Record an explicit user rating (good/bad) for the last run's selected models.
+   * Explicit ratings carry 3× the weight of auto-detected outcomes.
+   * Does NOT clear lastSelectedModels — the auto record already did that.
+   */
+  recordExplicitRating(rating) {
+    if (!this.tracker || !this.lastProfile) return false;
+    const taskType = this.lastProfile.type;
+    for (const [, model] of this.lastSelectedModels) {
+      this.tracker.recordExplicit(model.id, taskType, rating, 0);
+    }
+    return this.lastSelectedModels.size > 0;
+  }
   scoreModel(model, profile) {
     const perf = this.tracker?.performanceScore(model.id, profile.type) ?? 0.5;
     const costEff = this.costEfficiency(model, profile.complexity);
@@ -8139,6 +8152,20 @@ var ModelPerformanceTracker = class {
       sampleCount: s.sampleCount + 1
     });
   }
+  /**
+   * Record an explicit user rating (good/bad). Counts as 3 automatic samples
+   * so user feedback carries significantly more weight than auto-detected outcomes.
+   */
+  recordExplicit(modelId, taskType, rating, costUsd = 0) {
+    const outcome = rating === "good" ? "success" : "failure";
+    this.record(modelId, taskType, outcome, 0, costUsd);
+    this.record(modelId, taskType, outcome, 0, 0);
+    this.record(modelId, taskType, outcome, 0, 0);
+  }
+  /** Returns all stats keyed by "modelId:taskType" — used by `cascade stats`. */
+  getAll() {
+    return new Map(this.stats);
+  }
   /**
    * Returns 0.05–1.0; defaults to 0.5 (neutral prior) when no history exists.
    * High retry counts penalise the score.
@@ -8743,6 +8770,18 @@ ${last.partialOutput}` : "");
     if (!prompt) return null;
     return this.run({ prompt });
   }
+  /**
+   * Record an explicit user rating for the last completed run.
+   * Explicit ratings carry 3× the weight of auto-detected outcomes so user
+   * feedback meaningfully shifts future routing decisions.
+   * Returns false when called before any task has run in this session.
+   */
+  rateLastRun(rating) {
+    if (!this.taskAnalyzer) return false;
+    const recorded = this.taskAnalyzer.recordExplicitRating(rating);
+    if (recorded) void this.perfTracker?.save();
+    return recorded;
+  }
   /**
    * Rough pre-execution cost estimate for a plan: ~3 T2 calls per section
    * plus ~4 T3 calls per subtask at typical token volumes. A ballpark for
@@ -10539,6 +10578,31 @@ var DashboardSocket = class {
         const { sessionId } = normalizeSessionSubscriptionPayload(payload);
         socket.leave(`session:${sessionId}`);
       });
+      socket.on("session:rate", (payload) => {
+        const sessionId = typeof payload?.sessionId === "string" ? payload.sessionId : "";
+        const rating = payload?.rating === "good" || payload?.rating === "bad" ? payload.rating : null;
+        if (sessionId && rating) {
+          this.io.emit("session:rate", { sessionId, rating });
+        }
+      });
+    });
+  }
+  onSessionRate(callback) {
+    this.io.on("connection", (socket) => {
+      socket.on("session:rate", (payload) => {
+        const sessionId = typeof payload?.sessionId === "string" ? payload.sessionId : "";
+        const rating = payload?.rating === "good" || payload?.rating === "bad" ? payload.rating : null;
+        if (sessionId && rating) callback(sessionId, rating);
+      });
+    });
+  }
+  onConfigUpdate(callback) {
+    this.io.on("connection", (socket) => {
+      socket.on("config:update", (payload) => {
+        if (typeof payload === "object" && payload !== null) {
+          callback(payload);
+        }
+      });
     });
   }
   close() {
@@ -10555,6 +10619,7 @@ var DashboardServer = class {
   store;
   globalStore = null;
   broadcastTimer = null;
+  activeSessions = /* @__PURE__ */ new Map();
   port;
   host;
   workspacePath;
@@ -10574,6 +10639,30 @@ var DashboardServer = class {
     });
     this.setupMiddleware();
     this.setupRoutes();
+    this.socket.onSessionRate((sessionId, rating) => {
+      this.activeSessions.get(sessionId)?.rateLastRun(rating);
+    });
+    this.socket.onConfigUpdate((data) => {
+      if (data.keys) {
+        for (const [type, apiKey] of Object.entries(data.keys)) {
+          if (!apiKey) continue;
+          const provider = this.config.providers.find((p) => p.type === type);
+          if (provider) provider.apiKey = apiKey;
+          else this.config.providers.push({ type, apiKey });
+        }
+      }
+      if (data.models) {
+        this.config.models = { ...this.config.models, ...data.models };
+      }
+      if (data.budget) {
+        if (typeof data.budget.maxCostPerRun === "number") {
+          this.config.budget.maxCostPerRunUsd = data.budget.maxCostPerRun;
+        }
+        if (data.budget.autoBias === "balanced" || data.budget.autoBias === "quality" || data.budget.autoBias === "cost") {
+          this.config.autoBias = data.budget.autoBias;
+        }
+      }
+    });
   }
   async start() {
     const isLoopback = this.host === "127.0.0.1" || this.host === "::1" || this.host === "localhost";
@@ -11027,6 +11116,7 @@ var DashboardServer = class {
       res.json({ sessionId, status: "ACTIVE" });
       void (async () => {
         const cascade = new Cascade(this.config, this.workspacePath, this.store);
+        this.activeSessions.set(sessionId, cascade);
         cascade.on("stream:token", (e) => {
           this.socket.broadcast("stream:token", { sessionId, tierId: e.tierId, text: e.text });
           this.socket.broadcastToRoom(`session:${sessionId}`, "stream:token", { sessionId, tierId: e.tierId, text: e.text });
@@ -11055,6 +11145,8 @@ var DashboardServer = class {
             sessionId,
             error: err instanceof Error ? err.message : String(err)
           });
+        } finally {
+          this.activeSessions.delete(sessionId);
         }
       })();
     });