fix: unblock goose2 push checks

Signed-off-by: Matt Toohey <contact@matttoohey.com>

ui/goose2/src/features/chat/hooks/__tests__/useChatSessionController.test.ts

@@ -203,6 +203,7 @@ describe("useChatSessionController", () => {
         "session-1",
         "anthropic",
         "/tmp/project",
+        { personaId: undefined },
       );
     });
 
@@ -245,7 +246,6 @@ describe("useChatSessionController", () => {
           archivedAt: null,
           createdAt: "2026-04-20T00:00:00.000Z",
           updatedAt: "2026-04-20T00:00:00.000Z",
-          artifactsDir: "/tmp/project/artifacts",
         },
       ],
       loading: false,
@@ -430,6 +430,7 @@ describe("useChatSessionController", () => {
         "session-2",
         "anthropic",
         "/tmp/project",
+        { personaId: undefined },
       );
     });
 

ui/goose2/src/shared/api/acpNotificationHandler.ts

@@ -556,18 +556,15 @@ function handleShared(
 
     case "usage_update": {
       const usage = update as SessionUpdate & { sessionUpdate: "usage_update" };
-      if (!getLocalSessionId(gooseSessionId)) {
-        pendingUsageUpdates.set(gooseSessionId, {
-          accumulatedTotal: usage.used,
-          contextLimit: usage.size,
-        });
-        break;
-      }
-
-      useChatStore.getState().updateTokenState(sessionId, {
+      const tokenState = {
         accumulatedTotal: usage.used,
         contextLimit: usage.size,
-      });
+      };
+      if (!getLocalSessionId(gooseSessionId)) {
+        pendingUsageUpdates.set(gooseSessionId, tokenState);
+      }
+
+      useChatStore.getState().updateTokenState(sessionId, tokenState);
       break;
     }