firegoat 0.31.0 → 0.32.0

package/dist/cli.js

@@ -6947,7 +6947,7 @@ var require_public_api = __commonJS(function(exports) {
 // package.json
 var package_default = {
   name: "firegoat",
-  version: "0.31.0",
+  version: "0.32.0",
   repository: {
     type: "git",
     url: "git+https://github.com/firegoat-ai/firegoat-cli.git"
@@ -10716,6 +10716,30 @@ function runModel(r) {
     reported: label(r.reportedModel, "Reported model")
   };
 }
+var MAX_INT = 2147483647;
+function count(v, what) {
+  if (v === null)
+    return null;
+  if (!Number.isInteger(v) || v < 0 || v > MAX_INT)
+    throw new Error(`${what} is ${v}, the upload takes an integer from 0 to ${MAX_INT}`);
+  return v;
+}
+function runTokens(t) {
+  return {
+    input: count(t.input, "Input tokens"),
+    output: count(t.output, "Output tokens"),
+    cacheRead: count(t.cacheRead, "Cache read tokens"),
+    cacheWrite: count(t.cacheWrite, "Cache write tokens"),
+    total: count(t.total, "Total tokens")
+  };
+}
+function runPoints(r) {
+  const rewardPoints = count(r.rewardPoints, "Reward points");
+  const rewardPointsMax = count(r.rewardPointsMax, "Reward points max");
+  if (rewardPoints !== null && rewardPointsMax !== null && rewardPoints > rewardPointsMax)
+    throw new Error(`Reward points ${rewardPoints} are over the run's max ${rewardPointsMax}`);
+  return { rewardPoints, rewardPointsMax };
+}
 function runGit(g) {
   if (g.commit !== null && !COMMIT.test(g.commit))
     throw new Error(`Git commit ${printable(g.commit, 80)} is not 4-64 lowercase hex characters`);
@@ -10780,9 +10804,13 @@ async function uploadRun(t, r) {
   let project;
   let model;
   let git;
+  let tokens;
+  let points;
   try {
     project = projectName(t.project);
     model = runModel(r);
+    tokens = runTokens(r.tokens);
+    points = runPoints(r);
     git = t.git ? runGit(t.git) : undefined;
   } catch (e) {
     return { status: "failed", error: e.message };
@@ -10806,6 +10834,8 @@ async function uploadRun(t, r) {
       duration_ms: r.durationMs,
       harness: r.harness,
       model,
+      tokens,
+      ...points,
       ...git && { git }
     },
     files: Object.fromEntries(FILES.map((f) => [f, bytes[f].byteLength]))

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "firegoat",
-	"version": "0.31.0",
+	"version": "0.32.0",
 	"repository": {
 		"type": "git",
 		"url": "git+https://github.com/firegoat-ai/firegoat-cli.git"