From 552f9a40e2d3b2029c3d2a8f5a67c12e5f9c17bc Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Thu, 11 Jun 2026 16:06:43 +0000 Subject: [PATCH] Update AIC usage parser to read ai_credits_this_response Co-authored-by: pelikhan <4175913+pelikhan@users.noreply.github.com> --- actions/setup/js/check_daily_aic_workflow_guardrail.test.cjs | 3 ++- actions/setup/js/daily_aic_workflow_helpers.cjs | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/actions/setup/js/check_daily_aic_workflow_guardrail.test.cjs b/actions/setup/js/check_daily_aic_workflow_guardrail.test.cjs index 4678888a504..72576511a98 100644 --- a/actions/setup/js/check_daily_aic_workflow_guardrail.test.cjs +++ b/actions/setup/js/check_daily_aic_workflow_guardrail.test.cjs @@ -57,13 +57,14 @@ describe("check_daily_aic_workflow_guardrail", () => { JSON.stringify({ aic: 9, usage: { aic: 0.25 } }), JSON.stringify({ ai_credits: 8, usage: { ai_credits: 0.1 } }), JSON.stringify({ aiCredits: 7, usage: { aiCredits: 0.15 } }), + JSON.stringify({ usage: { ai_credits_this_response: 0.25 } }), JSON.stringify({ aiCredits: 0.2, usage: { aiCredits: "" } }), JSON.stringify({ aic: 0.3, usage: { aic: "" } }), ].join("\n"), "utf8" ); - expect(exports.sumAICFromUsageJSONLFiles(exports.findJSONLFiles(tmpDir))).toBe(5); + expect(exports.sumAICFromUsageJSONLFiles(exports.findJSONLFiles(tmpDir))).toBeCloseTo(5.25, 10); }); it("computes aggregate AIC statistics for prior runs", () => { diff --git a/actions/setup/js/daily_aic_workflow_helpers.cjs b/actions/setup/js/daily_aic_workflow_helpers.cjs index 67ba5c036ea..2e2798c4b6f 100644 --- a/actions/setup/js/daily_aic_workflow_helpers.cjs +++ b/actions/setup/js/daily_aic_workflow_helpers.cjs @@ -204,7 +204,7 @@ function sumAICFromUsageJSONLFiles(filePaths) { } const usage = normalizeUsageRecord(parsed.usage); - const explicitAICredits = getNumericAliasField(usage, parsed, ["ai_credits", "aiCredits"]); + const explicitAICredits = getNumericAliasField(usage, parsed, ["ai_credits_this_response", "aiCreditsThisResponse", "ai_credits", "aiCredits"]); if (explicitAICredits > 0) { total += explicitAICredits; continue;