@peakflo/agent-harness 0.2.10-rc.2 → 0.2.10-rc.3

package/dist/runtime/session-runner.js

@@ -44,6 +44,25 @@ const permission_mode_1 = require("./permission-mode");
 const projection_1 = require("./projection");
 const space_files_1 = require("./space-files");
 const task_files_1 = require("./task-files");
+/** The virtual key keeps spend out of URLs and logs. */
+async function readGatewaySpend(baseUrl, key) {
+    try {
+        const response = await fetch(new URL('/key/info', baseUrl), {
+            headers: { Authorization: `Bearer ${key}` },
+            signal: AbortSignal.timeout(5_000)
+        });
+        if (!response.ok)
+            return null;
+        const data = (await response.json());
+        const spend = data.info?.spend;
+        return typeof spend === 'number' && Number.isFinite(spend) && spend >= 0
+            ? spend
+            : null;
+    }
+    catch {
+        return null;
+    }
+}
 /** A missing Pi consumption event must not hold the lease open for its full RPC timeout. */
 const FINAL_STEER_ACK_WAIT_MS = 5_000;
 /** How often the runner asks the adapter for new output when nothing pushes. */
@@ -658,6 +677,7 @@ class SessionRunner {
         let partsProduced = 0;
         let stopReason = 'completed';
         let error;
+        let gatewaySpendBefore = null;
         let repositoryStatuses = [];
         /** True when the backend session was resumed rather than created. */
         let resumed = false;
@@ -689,6 +709,11 @@ class SessionRunner {
             const refusal = modelGatewayRefusal(lease);
             if (refusal)
                 throw new Error(refusal);
+            if (session.agent.model.gatewayUrl && lease.modelToken) {
+                gatewaySpendBefore = await readGatewaySpend(session.agent.model.gatewayUrl, lease.modelToken);
+                if (gatewaySpendBefore === null)
+                    this.logger.warn(`[coding-agent] Gateway spend start unavailable for turn ${session.turnId}; this turn cannot be debited from gateway usage`);
+            }
             // The catalogue is advisory. Fetch it while the independent workspace
             // is prepared. Pi can start with the selected lease model while this
             // request fills the cache for a later turn. OpenCode still waits for
@@ -1331,6 +1356,16 @@ class SessionRunner {
              * that cannot remember the previous sentence.
              */
         }
+        let costUsd;
+        if (gatewaySpendBefore !== null &&
+            session.agent.model.gatewayUrl &&
+            lease.modelToken) {
+            const spendAfter = await readGatewaySpend(session.agent.model.gatewayUrl, lease.modelToken);
+            if (spendAfter === null)
+                this.logger.warn(`[coding-agent] Gateway spend end unavailable for turn ${session.turnId}; this turn cannot be debited from gateway usage`);
+            else
+                costUsd = Math.max(0, spendAfter - gatewaySpendBefore);
+        }
         const turn = {
             turnId: session.turnId,
             startedAt,
@@ -1338,6 +1373,7 @@ class SessionRunner {
             model: session.agent.model.model,
             stopReason,
             error,
+            costUsd,
             repositories: repositoryStatuses.length > 0 ? repositoryStatuses : undefined,
             // The checkpoint travels on the final ingest because the final ingest is
             // the ONE message guaranteed to be sent — it is what releases the lease,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@peakflo/agent-harness",
-  "version": "0.2.10-rc.2",
+  "version": "0.2.10-rc.3",
   "description": "Headless Peakflo coding-agent harness. One codebase, three hosting modes: a Daytona cloud sandbox, a customer VM, and the 20x desktop.",
   "license": "MIT",
   "type": "commonjs",