ravensight-playtest 0.2.9 → 0.3.0

package/src/run/usage.js

@@ -29,6 +29,7 @@ export function createUsageLedger() {
   const byModel = new Map();
   const started = Date.now();
   let calls = 0;
+  let modelMs = 0;
   let serverTotalUsd = null;
   let serverCalls = null;
 
@@ -36,8 +37,10 @@ export function createUsageLedger() {
     /**
      * @param {{model: string|null, usage: Object}} turn
      */
-    record({ model, usage }) {
+    record({ model, usage, ms }) {
       calls += 1;
+      // Time spent waiting on the model, when the caller measured it.
+      if (Number.isFinite(Number(ms)) && Number(ms) > 0) modelMs += Number(ms);
       const key = model || 'unknown';
       const entry = byModel.get(key) || { model: key, input_tokens: 0, cache_read_tokens: 0, output_tokens: 0, usd: 0 };
       // The report schema has no cache-creation field, so creation tokens
@@ -64,6 +67,20 @@ export function createUsageLedger() {
       return calls;
     },
 
+    /**
+     * Wall time since the ledger was made, and time spent waiting on model
+     * calls, in seconds. What `complete` sends as `duration`, so the run card
+     * does not read 0s for a run that played for minutes.
+     *
+     * @returns {{wall_seconds: number, model_seconds: number}}
+     */
+    duration() {
+      return {
+        wall_seconds: Number(((Date.now() - started) / 1000).toFixed(3)),
+        model_seconds: Number((modelMs / 1000).toFixed(3))
+      };
+    },
+
     /**
      * The `usage` object as the report schema wants it.
      *
@@ -87,6 +104,7 @@ export function createUsageLedger() {
       const report = this.toReportUsage();
       return {
         ...report,
+        model_seconds: Number((modelMs / 1000).toFixed(3)),
         local_model_calls: calls,
         server_model_calls: serverCalls,
         usd_source: serverTotalUsd === null ? 'unavailable' : 'server',