pi-ui-extend 0.1.31 → 0.1.32

package/dist/app/model/model-usage-status.d.ts

@@ -8,6 +8,7 @@ export type ModelUsageDescriptor = BaseModelUsageDescriptor & ({
     readonly kind: "google-antigravity";
     readonly quotaModelKey: string;
     readonly account: AntigravityQuotaAccount;
+    readonly accounts?: readonly AntigravityQuotaAccount[];
 });
 export type ModelUsageLimitWindow = {
     readonly remainingPercent: number;

package/dist/app/model/model-usage-status.js

@@ -35,14 +35,16 @@ export function modelUsageDescriptor(model) {
     }
     if (ANTIGRAVITY_QUOTA_PROVIDERS.has(provider)) {
         const quotaModelKey = resolveAntigravityQuotaModelKey(model);
-        const account = readActiveAntigravityQuotaAccount();
+        const accounts = readAllAntigravityQuotaAccounts();
+        const account = readActiveAntigravityQuotaAccount(accounts);
         if (!quotaModelKey || !account)
             return undefined;
         return {
             kind: "google-antigravity",
-            modelKey: `${model.provider}/${model.id}@${account.cacheKey}`,
+            modelKey: `${model.provider}/${model.id}@all:${accounts.map((item) => item.cacheKey).join(",")}`,
             quotaModelKey,
             account,
+            accounts,
         };
     }
     return undefined;
@@ -384,15 +386,9 @@ export function resolveAntigravityQuotaModelKey(model) {
     return undefined;
 }
 export function googleAntigravityUsageStatusFromResponse(data, descriptor, now = Date.now()) {
-    const quotaInfo = data.models[descriptor.quotaModelKey]?.quotaInfo;
-    if (!quotaInfo)
+    const window = googleAntigravityWindowFromResponse(data, descriptor.quotaModelKey, now);
+    if (!window)
         return undefined;
-    const resetAt = parseResetTime(quotaInfo.resetTime, now);
-    const window = {
-        remainingPercent: quotaRemainingPercent(quotaInfo),
-        resetAt,
-        windowSeconds: Math.max(0, Math.round((resetAt - now) / 1000)),
-    };
     const weekly = window.windowSeconds >= DAY_SECONDS ? window : undefined;
     const hourly = weekly ? undefined : window;
     return {
@@ -404,10 +400,48 @@ export function googleAntigravityUsageStatusFromResponse(data, descriptor, now =
         ...(hourly ? { hourly } : {}),
     };
 }
+function googleAntigravityWindowFromResponse(data, quotaModelKey, now) {
+    const quotaInfo = data.models[quotaModelKey]?.quotaInfo;
+    if (!quotaInfo)
+        return undefined;
+    const resetAt = parseResetTime(quotaInfo.resetTime, now);
+    return {
+        remainingPercent: quotaRemainingPercent(quotaInfo),
+        resetAt,
+        windowSeconds: Math.max(0, Math.round((resetAt - now) / 1000)),
+    };
+}
 async function queryGoogleAntigravityModelUsage(descriptor) {
     const now = Date.now();
-    const response = await fetchGoogleAntigravityQuotaForAccount(descriptor.account, now);
-    return googleAntigravityUsageStatusFromResponse(response, descriptor, now);
+    const accounts = descriptor.accounts?.length ? descriptor.accounts : [descriptor.account];
+    const windows = (await Promise.all(accounts.map(async (account) => {
+        try {
+            const response = await fetchGoogleAntigravityQuotaForAccount(account, now);
+            return googleAntigravityWindowFromResponse(response, descriptor.quotaModelKey, now);
+        }
+        catch {
+            return undefined;
+        }
+    }))).filter((window) => window !== undefined);
+    if (windows.length === 0)
+        return undefined;
+    const resetAt = Math.min(...windows.map((window) => window.resetAt));
+    const windowSeconds = Math.max(0, Math.round((resetAt - now) / 1000));
+    const aggregateWindow = {
+        remainingPercent: clampPercent(Math.round(windows.reduce((sum, window) => sum + window.remainingPercent, 0) / windows.length)),
+        resetAt,
+        windowSeconds,
+    };
+    const weekly = aggregateWindow.windowSeconds >= DAY_SECONDS ? aggregateWindow : undefined;
+    const hourly = weekly ? undefined : aggregateWindow;
+    return {
+        modelKey: descriptor.modelKey,
+        provider: "google-antigravity",
+        updatedAt: now,
+        accountEmail: "Σ",
+        ...(weekly ? { weekly } : {}),
+        ...(hourly ? { hourly } : {}),
+    };
 }
 const GOOGLE_ACCOUNT_QUOTA_WINDOWS = [
     { label: "Claude Opus", quotaModelKey: "claude-opus-4-6-thinking" },
@@ -440,8 +474,7 @@ async function queryGoogleAntigravityAccountUsage(now) {
     }));
     return results;
 }
-function readActiveAntigravityQuotaAccount() {
-    const accounts = readAllAntigravityQuotaAccounts();
+function readActiveAntigravityQuotaAccount(accounts = readAllAntigravityQuotaAccounts()) {
     const credential = readPiAuthSync().antigravity;
     return accounts[clampAccountIndex(credential?.activeIndex, accounts.length)];
 }
@@ -564,9 +597,9 @@ function googleQuotaResponseFromCachedQuota(cachedQuota, cachedQuotaUpdatedAt, n
     return Object.keys(models).length > 0 ? { models } : undefined;
 }
 function addCachedQuotaModels(models, quota, quotaModelKeys, cachedQuotaUpdatedAt, now) {
-    if (!quota || !Number.isFinite(quota.remainingFraction))
+    if (!quota)
         return;
-    const remainingFraction = quota.remainingFraction;
+    const remainingFraction = Number.isFinite(quota.remainingFraction) ? quota.remainingFraction : 0;
     const resetTime = cachedQuotaResetTimeForDisplay(quota.resetTime, cachedQuotaUpdatedAt, now);
     for (const quotaModelKey of quotaModelKeys) {
         models[quotaModelKey] = {

package/external/pi-tools-suite/src/dcp/config.ts

@@ -24,8 +24,8 @@ export interface DcpConfig {
     modelMaxContextLimits?: Record<string, number | string>
     modelMinContextLimits?: Record<string, number | string>
     summaryBuffer: boolean
-    nudgeFrequency: number // inject nudge every N context events (default: 5)
-    iterationNudgeThreshold: number // nudge after N tool calls since last user msg (default: 15)
+    nudgeFrequency: number // inject nudge every N context events (default: 2)
+    iterationNudgeThreshold: number // nudge after N tool calls since last user msg (default: 8)
     nudgeForce: "strong" | "soft"
     protectedTools: string[] // these tool outputs always protected from pruning
     protectTags: boolean
@@ -81,27 +81,27 @@ const DEFAULT_CONFIG: DcpConfig = {
     automaticStrategies: true,
   },
   compress: {
-    maxContextPercent: 0.8,
-    minContextPercent: 0.4,
+    maxContextPercent: 0.65,
+    minContextPercent: 0.25,
     modelMaxContextPercent: {},
     modelMinContextPercent: {},
     summaryBuffer: true,
-    nudgeFrequency: 5,
-    iterationNudgeThreshold: 15,
+    nudgeFrequency: 2,
+    iterationNudgeThreshold: 8,
     nudgeForce: "soft",
     protectedTools: ["compress", "write", "edit"],
     protectTags: false,
     protectUserMessages: false,
     autoCandidates: {
       enabled: true,
-      minContextPercent: 0.4,
+      minContextPercent: 0.25,
       keepRecentTurns: 2,
       minMessages: 6,
       minTokens: 1500,
     },
     messageMode: {
       enabled: true,
-      minContextPercent: 0.4,
+      minContextPercent: 0.25,
       keepRecentTurns: 2,
       mediumTokens: 500,
       highTokens: 5000,

package/external/pi-tools-suite/src/dcp/prompts.ts

@@ -174,6 +174,7 @@ You are at or beyond the configured max context threshold. This is an emergency
 You MUST use the \`compress\` tool now. Do not continue normal exploration until compression is handled.
 
 If you are in the middle of a critical atomic operation, finish that atomic step first, then compress immediately.
+If a completed implementation+verification slice exists, compress it before replying or starting another task.
 
 RANGE STRATEGY (MANDATORY)
 Prioritize one large, closed, high-yield compression range first.
@@ -201,6 +202,8 @@ ACTION REQUIRED: Context usage is high.
 Before doing more exploration, look for a closed, self-contained range that no longer needs to stay raw and compress it now.
 
 Do not treat this as optional housekeeping. If any completed research, implementation, verification, CI-log inspection, or dead-end debugging slice is present, call the \`compress\` tool before continuing normal work.
+If a completed implementation+verification slice exists, compress it before replying or starting another task.
+High-priority stale tool outputs must be compressed once no exact raw text is needed.
 
 RANGE SELECTION
 Prefer older, resolved history. Avoid the newest active working slice unless it is clearly done.
@@ -222,6 +225,7 @@ If any range is cleanly closed and unlikely to be needed again, use the \`compre
 If direction has shifted, compress earlier ranges that are now less relevant.
 
 Do not defer this across another batch of searches, reads, CI log fetches, or tests. The next safe action should be compression whenever a closed slice exists.
+High-priority stale tool outputs must be compressed once no exact raw text is needed.
 
 Prefer small, closed-range compressions over one broad compression.
 Use message-mode compression for isolated large stale messages.
@@ -238,6 +242,7 @@ ACTION REQUIRED: You've been iterating for a while after the last user message.
 Pause before the next non-atomic tool call. If there is a closed portion that is unlikely to be referenced immediately (for example, finished research before implementation, completed CI-log triage, a verified fix, or a dead-end investigation), use the \`compress\` tool on it now.
 
 Do not keep accumulating tool outputs while a completed slice remains raw. If a range is closed, compression is the next safe action.
+If a completed implementation+verification slice exists, compress it before replying or starting another task.
 
 Prefer multiple short, closed ranges over one large range when several independent slices are ready.
 Use message-mode compression for isolated large stale messages.

package/external/pi-tools-suite/src/dcp/pruner-metadata.ts

@@ -149,11 +149,11 @@ export function resolveContextThresholds(
     minContextPercent: min ??
       resolveThresholdValue(config.compress.minContextLimit) ??
       resolveThresholdValue(config.compress.minContextPercent) ??
-      0.4,
+      0.25,
     maxContextPercent: max ??
       resolveThresholdValue(config.compress.maxContextLimit) ??
       resolveThresholdValue(config.compress.maxContextPercent) ??
-      0.8,
+      0.65,
   };
 }
 

package/external/pi-tools-suite/src/dcp/pruner-nudge.ts

@@ -291,11 +291,11 @@ export function getNudgeType(
     config.compress;
   const minContextPercent = coercePercentThreshold(
     thresholds.minContextPercent ?? config.compress.minContextPercent,
-    0.4,
+    0.25,
   );
   const maxContextPercent = coercePercentThreshold(
     thresholds.maxContextPercent ?? config.compress.maxContextPercent,
-    0.8,
+    0.65,
   );
   const cadence = Math.max(1, Math.floor(nudgeFrequency));
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.31",
+  "version": "0.1.32",
   "private": false,
   "type": "module",
   "bin": {