fix(FN-4396): support token budget fields in updateTask

Fusion-Task-Id: FN-4396
Fusion-Task-Lineage: 3839210b-b443-410d-a476-aa81128c739a

packages/core/src/__tests__/store-token-usage.test.ts

@@ -145,23 +145,16 @@ describe("TaskStore", () => {
       const created = await harness.store().createTask({
         description: "Token budget fields",
       });
-      (harness.store() as any).db.prepare(`
-        UPDATE tasks
-        SET tokenBudgetSoftAlertedAt = ?,
-            tokenBudgetHardAlertedAt = ?,
-            tokenBudgetOverride = ?
-        WHERE id = ?
-      `).run(
-        "2026-05-14T01:00:00.000Z",
-        "2026-05-14T01:05:00.000Z",
-        JSON.stringify({
+      await harness.store().updateTask(created.id, {
+        tokenBudgetSoftAlertedAt: "2026-05-14T01:00:00.000Z",
+        tokenBudgetHardAlertedAt: "2026-05-14T01:05:00.000Z",
+        tokenBudgetOverride: {
           soft: 1_000_000,
           hard: 2_000_000,
           raisedAt: "2026-05-14T01:06:00.000Z",
           reason: "manual override",
-        }),
-        created.id,
-      );
+        },
+      });
 
       const reloaded = await harness.store().getTask(created.id);
       expect(reloaded.tokenBudgetSoftAlertedAt).toBe("2026-05-14T01:00:00.000Z");