paseo-acp-agy 1.3.2 → 1.3.3

package/dist/acp-server.js

@@ -418,6 +418,7 @@ export class ACPServer extends EventEmitter {
                             }
                         }
                         if (!isNotification) {
+                            this.publishUsageUpdate(session);
                             this.sendSuccess(id, await this.sessionState(session));
                             this.publishCommands(session.id);
                             this.publishUsageUpdate(session);
@@ -465,6 +466,7 @@ export class ACPServer extends EventEmitter {
                             catch { }
                         }
                         if (!isNotification) {
+                            this.publishUsageUpdate(session);
                             this.sendSuccess(id, await this.sessionState(session));
                             this.publishCommands(session.id);
                             this.publishUsageUpdate(session);

package/dist/paseo-patcher.js

@@ -714,10 +714,18 @@ function toneFromUsedPct(usedPct) {
 
 function isStandaloneTelemetryPluginEnabled() {
     try {
-        const paseoHome = process.env.PASEO_HOME || path.join(os.homedir(), ".paseo");
+        let paseoHome = process.env.PASEO_HOME;
+        if (!paseoHome && Array.isArray(process.argv)) {
+            const homeIdx = process.argv.indexOf("--home");
+            if (homeIdx !== -1 && process.argv[homeIdx + 1]) {
+                paseoHome = process.argv[homeIdx + 1];
+            }
+        }
+        paseoHome = paseoHome || path.join(os.homedir(), ".paseo");
         const cfgPath = path.join(paseoHome, "config.json");
         if (!fs.existsSync(cfgPath)) return false;
         const cfg = JSON.parse(fs.readFileSync(cfgPath, "utf-8"));
+        if (cfg && cfg.pluginsEnabled === false) return false;
         const entry = cfg && cfg.plugins && cfg.plugins["antigravity-telemetry"];
         return Boolean(entry && entry.enabled !== false);
     } catch {
@@ -1265,6 +1273,28 @@ ${braceIndent}}`;
                     acpModified = true;
                 }
             }
+            // Patch handlePromptResponse so response.usage merges defined fields instead of wiping
+            // totalCostUsd / contextWindowMaxTokens / contextWindowUsedTokens from this.currentTurnUsage
+            const promptRespAssignRegex = /([ \t]*)this\.currentTurnUsage\s*=\s*mapACPUsage\(response\.usage\)\s*\?\?\s*this\.currentTurnUsage;/;
+            if (promptRespAssignRegex.test(acpCode) && !acpCode.includes("_mappedRespUsage")) {
+                acpCode = acpCode.replace(promptRespAssignRegex, (_m, ind) => `${ind}const _mappedRespUsage = mapACPUsage(response.usage);\n` +
+                    `${ind}if (_mappedRespUsage) {\n` +
+                    `${ind}    this.currentTurnUsage = {\n` +
+                    `${ind}        ...this.currentTurnUsage,\n` +
+                    `${ind}        ...Object.fromEntries(Object.entries(_mappedRespUsage).filter(([, v]) => v !== undefined)),\n` +
+                    `${ind}    };\n` +
+                    `${ind}}`);
+                acpModified = true;
+            }
+            // Patch subscribe(callback) so any usage_update received during newSession/loadSession
+            // (before subscribers attach, including history resumes) is replayed immediately
+            const subscribeMethodMatch = acpCode.match(/subscribe\s*\([^)]*\)\s*\{[\s\S]*?\n\s*\}/);
+            if (subscribeMethodMatch &&
+                acpCode.includes('type: "thread_started"') &&
+                !subscribeMethodMatch[0].includes("usage_updated")) {
+                acpCode = acpCode.replace(/(\bcallback\(\{\s*type:\s*"thread_started"[\s\S]*?\}\);\s*\})/, `$1\n        if (this.currentTurnUsage) {\n            callback({\n                type: "usage_updated",\n                provider: this.provider,\n                usage: this.currentTurnUsage,\n            });\n        }`);
+                acpModified = true;
+            }
             // Patch mapPermissionRequest for native QuestionFormCard support in Paseo UI
             if (!acpCode.includes("snapshot?.kind === \"question\" || params?.kind === \"question\"")) {
                 const oldMapPermRegex = /function\s+mapPermissionRequest\s*\([^)]*\)\s*\{[\s\S]*?const\s+kind\s*=\s*snapshot\.kind\s*===\s*"switch_mode"\s*\?\s*"mode"\s*:\s*"tool";/m;
@@ -1971,6 +2001,10 @@ export function isPaseoServerPatched(serverDir) {
             if (pluginDir && (!content.includes("usageSession()") || content.includes("!this.usageProvider"))) {
                 return false;
             }
+            if (content.includes("mapACPUsage(response.usage) ?? this.currentTurnUsage") &&
+                !content.includes("_mappedRespUsage")) {
+                return false;
+            }
         }
         return true;
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "paseo-acp-agy",
-  "version": "1.3.2",
+  "version": "1.3.3",
   "description": "ACP (Agent Client Protocol) provider for Google Antigravity in Paseo and Zed",
   "type": "module",
   "main": "dist/index.js",