wave-agent-sdk 0.17.12 → 0.18.1

package/dist/services/authService.js

@@ -42,10 +42,10 @@ export class AuthService {
             this.onAuthChangeCallbacks = this.onAuthChangeCallbacks.filter((cb) => cb !== callback);
         };
     }
-    notifyAuthChange(event) {
+    async notifyAuthChange(event) {
         for (const cb of this.onAuthChangeCallbacks) {
             try {
-                cb(event);
+                await cb(event);
             }
             catch {
                 // Don't let callback errors break auth flow
@@ -92,7 +92,7 @@ export class AuthService {
             // ignore stat errors
         }
     }
-    clearAuth() {
+    async clearAuth() {
         const config = this.loadAuth();
         delete config.SSO_TOKEN;
         delete config.SSO_REFRESH_TOKEN;
@@ -106,7 +106,7 @@ export class AuthService {
         else {
             this.saveAuth(config);
         }
-        this.notifyAuthChange("logout");
+        await this.notifyAuthChange("logout");
     }
     getSSOToken() {
         const config = this.loadAuth();
@@ -139,7 +139,7 @@ export class AuthService {
                 : undefined,
             user,
         });
-        this.notifyAuthChange("login");
+        await this.notifyAuthChange("login");
         return token;
     }
     /**
@@ -303,7 +303,7 @@ export class AuthService {
             if (response.status === 400 || response.status === 401) {
                 // Refresh token revoked — clear auth
                 logger.info(`[Auth] Refresh token rejected (${response.status}), clearing auth`);
-                this.clearAuth();
+                await this.clearAuth();
                 return false;
             }
             if (!response.ok) {
@@ -324,7 +324,7 @@ export class AuthService {
                     : config.user,
             });
             logger.info(`[Auth] Token refreshed successfully, new token expires at ${newExpiresAt ? new Date(newExpiresAt).toISOString() : "never"}`);
-            this.notifyAuthChange("login");
+            await this.notifyAuthChange("login");
             return true;
         }
         catch (err) {

package/dist/services/remoteSettingsService.d.ts

@@ -1,5 +1,10 @@
 import type { RemoteSettingsFetchResult } from "../types/configuration.js";
 import type { WaveConfiguration } from "../types/configuration.js";
+/**
+ * Register a callback invoked when remote settings change (checksum differs).
+ * Returns an unsubscribe function.
+ */
+export declare function onSettingsUpdate(callback: () => void | Promise<void>): () => void;
 export declare function initialize(): void;
 export declare function getRemoteSettingsSync(): WaveConfiguration | null;
 export declare function refresh(): Promise<RemoteSettingsFetchResult>;
@@ -17,5 +22,6 @@ export declare const remoteSettingsService: {
     readonly clear: typeof clear;
     readonly shutdown: typeof shutdown;
     readonly mergeRemoteSettings: typeof mergeRemoteSettings;
+    readonly onSettingsUpdate: typeof onSettingsUpdate;
 };
 //# sourceMappingURL=remoteSettingsService.d.ts.map

package/dist/services/remoteSettingsService.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"remoteSettingsService.d.ts","sourceRoot":"","sources":["../../src/services/remoteSettingsService.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,yBAAyB,EAE1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AA2JnE,wBAAgB,UAAU,IAAI,IAAI,CASjC;AAED,wBAAgB,qBAAqB,IAAI,iBAAiB,GAAG,IAAI,CAEhE;AAED,wBAAsB,OAAO,IAAI,OAAO,CAAC,yBAAyB,CAAC,CAKlE;AAED,wBAAgB,KAAK,IAAI,IAAI,CAO5B;AAED,wBAAgB,QAAQ,IAAI,IAAI,CAK/B;AAqCD,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,iBAAiB,EAC9B,MAAM,EAAE,iBAAiB,GACxB,iBAAiB,CA4DnB;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB;;;;;;;CAOxB,CAAC"}
+{"version":3,"file":"remoteSettingsService.d.ts","sourceRoot":"","sources":["../../src/services/remoteSettingsService.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,yBAAyB,EAE1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAYnE;;;GAGG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,GACnC,MAAM,IAAI,CAOZ;AAoKD,wBAAgB,UAAU,IAAI,IAAI,CASjC;AAED,wBAAgB,qBAAqB,IAAI,iBAAiB,GAAG,IAAI,CAEhE;AAED,wBAAsB,OAAO,IAAI,OAAO,CAAC,yBAAyB,CAAC,CAKlE;AAED,wBAAgB,KAAK,IAAI,IAAI,CAO5B;AAED,wBAAgB,QAAQ,IAAI,IAAI,CAK/B;AAqCD,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,iBAAiB,EAC9B,MAAM,EAAE,iBAAiB,GACxB,iBAAiB,CA4DnB;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB;;;;;;;;CAQxB,CAAC"}

package/dist/services/remoteSettingsService.js

@@ -8,6 +8,27 @@ const POLLING_INTERVAL_MS = 60 * 60 * 1000; // 60 minutes
 const FETCH_TIMEOUT_MS = 10000;
 let _cachedSettings = null;
 let _pollingTimer = null;
+let _onSettingsUpdateCallbacks = [];
+/**
+ * Register a callback invoked when remote settings change (checksum differs).
+ * Returns an unsubscribe function.
+ */
+export function onSettingsUpdate(callback) {
+    _onSettingsUpdateCallbacks.push(callback);
+    return () => {
+        _onSettingsUpdateCallbacks = _onSettingsUpdateCallbacks.filter((cb) => cb !== callback);
+    };
+}
+async function notifySettingsUpdate() {
+    for (const cb of _onSettingsUpdateCallbacks) {
+        try {
+            await cb();
+        }
+        catch {
+            // Don't let callback errors break the fetch flow
+        }
+    }
+}
 function loadCacheFromDisk() {
     try {
         if (!fs.existsSync(CACHE_FILE)) {
@@ -64,6 +85,7 @@ async function fetchRemoteSettings() {
     if (!token || !serverUrl) {
         return { success: false, error: "Missing SSO token or server URL" };
     }
+    const oldChecksum = _cachedSettings?.checksum;
     const headers = {
         Authorization: `Bearer ${token}`,
     };
@@ -91,6 +113,9 @@ async function fetchRemoteSettings() {
             logger.debug("remoteSettings: 404 not configured — clearing stale cache");
             _cachedSettings = null;
             removeCacheFromDisk();
+            if (oldChecksum) {
+                await notifySettingsUpdate();
+            }
             return { success: true, notConfigured: true, settings: null };
         }
         if (!response.ok) {
@@ -116,6 +141,9 @@ async function fetchRemoteSettings() {
         logger.debug("remoteSettings: fetched new settings", {
             checksum: data.checksum,
         });
+        if (data.checksum !== oldChecksum) {
+            await notifySettingsUpdate();
+        }
         return {
             success: true,
             settings: data.settings,
@@ -277,4 +305,5 @@ export const remoteSettingsService = {
     clear,
     shutdown,
     mergeRemoteSettings,
+    onSettingsUpdate,
 };

package/dist/utils/containerSetup.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"containerSetup.d.ts","sourceRoot":"","sources":["../../src/utils/containerSetup.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AA0B3C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAM3E,OAAO,KAAK,EAAE,YAAY,EAAmB,MAAM,mBAAmB,CAAC;AACvE,OAAO,KAAK,EACV,cAAc,EACd,KAAK,EACL,IAAI,EACJ,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAM3B,MAAM,WAAW,0BAA0B;IACzC,OAAO,EAAE,YAAY,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,OAAO,CAAC;IAGhB,uBAAuB,EAAE,CAAC,KAAK,EAAE,cAAc,EAAE,KAAK,IAAI,CAAC;IAC3D,aAAa,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC;IACvC,sBAAsB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACvD,wBAAwB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACzD,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAClD,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,QAAQ,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED,wBAAgB,mBAAmB,CACjC,YAAY,EAAE,0BAA0B,GACvC,SAAS,CAuSX"}
+{"version":3,"file":"containerSetup.d.ts","sourceRoot":"","sources":["../../src/utils/containerSetup.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AA0B3C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAM3E,OAAO,KAAK,EAAE,YAAY,EAAmB,MAAM,mBAAmB,CAAC;AACvE,OAAO,KAAK,EACV,cAAc,EACd,KAAK,EACL,IAAI,EACJ,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAM3B,MAAM,WAAW,0BAA0B;IACzC,OAAO,EAAE,YAAY,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,OAAO,CAAC;IAGhB,uBAAuB,EAAE,CAAC,KAAK,EAAE,cAAc,EAAE,KAAK,IAAI,CAAC;IAC3D,aAAa,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC;IACvC,sBAAsB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACvD,wBAAwB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACzD,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAClD,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,QAAQ,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED,wBAAgB,mBAAmB,CACjC,YAAY,EAAE,0BAA0B,GACvC,SAAS,CA6SX"}

package/dist/utils/containerSetup.js

@@ -101,9 +101,9 @@ export function setupAgentContainer(setupOptions) {
     });
     container.register("McpManager", mcpManager);
     // Wire up auth change callback to refresh/clear remote settings
-    authService.onAuthChange((event) => {
+    authService.onAuthChange(async (event) => {
         if (event === "login") {
-            remoteSettingsService.refresh();
+            await remoteSettingsService.refresh();
         }
         else if (event === "logout") {
             remoteSettingsService.clear();
@@ -215,6 +215,11 @@ export function setupAgentContainer(setupOptions) {
         },
     });
     container.register("LiveConfigManager", liveConfigManager);
+    // Wire up remote settings hot-update: when polling detects changed settings,
+    // reload configuration so admin changes propagate to the running agent.
+    remoteSettingsService.onSettingsUpdate(async () => {
+        await liveConfigManager.reload();
+    });
     const subagentManager = new SubagentManager(container, {
         workdir,
         callbacks: {

package/dist/utils/messageOperations.d.ts

@@ -44,6 +44,9 @@ export interface UpdateToolBlockParams {
     timestamp?: number;
 }
 export type AgentToolBlockUpdateParams = Omit<UpdateToolBlockParams, "messages">;
+export type ToolBlockUpdateCallbackParams = Omit<AgentToolBlockUpdateParams, "messageId"> & {
+    messageId: string;
+};
 export interface AddErrorBlockParams {
     messages: Message[];
     error: string;
@@ -83,7 +86,10 @@ export declare const addToolBlockToMessageInMessages: (messages: Message[], mess
     messages: Message[];
     toolBlockId: string;
 };
-export declare const updateToolBlockInMessage: ({ messages, id, messageId, parameters, result, success, error, stage, name, shortResult, startLineNumber, images, compactParams, parametersChunk, isManuallyBackgrounded, }: UpdateToolBlockParams) => Message[];
+export declare const updateToolBlockInMessage: ({ messages, id, messageId, parameters, result, success, error, stage, name, shortResult, startLineNumber, images, compactParams, parametersChunk, isManuallyBackgrounded, }: UpdateToolBlockParams) => {
+    messages: Message[];
+    messageId?: string;
+};
 export declare const addErrorBlockToMessage: ({ messages, error, }: AddErrorBlockParams) => Message[];
 export declare const addBangMessage: ({ messages, command, }: AddBangParams) => Message[];
 export declare const updateBangInMessage: ({ messages, command, output, }: UpdateBangParams) => Message[];

package/dist/utils/messageOperations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"messageOperations.d.ts","sourceRoot":"","sources":["../../src/utils/messageOperations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,OAAO,EACP,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAI5E,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnD,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAGD,MAAM,WAAW,oBAAqB,SAAQ,iBAAiB;IAC7D,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAGD,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,qBAAqB,EACrB,UAAU,CACX,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;GAIG;AACH,eAAO,MAAM,oBAAoB,GAAI,WAAW,MAAM,KAAG,MAmCxD,CAAC;AAEF,eAAO,MAAM,iBAAiB,QAAO,MAA+B,CAAC;AAGrE,eAAO,MAAM,wBAAwB,GAAI,0EAQtC,oBAAoB,KAAG,OAAO,EA8BhC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B,GACtC,UAAU,OAAO,EAAE,EACnB,IAAI,MAAM,EACV,QAAQ,OAAO,CAAC,iBAAiB,CAAC,KACjC,OAAO,EAwBT,CAAC;AAGF,eAAO,MAAM,6BAA6B,GACxC,UAAU,OAAO,EAAE,EACnB,UAAU,MAAM,EAChB,YAAY,qCAAqC,EAAE,EACnD,QAAQ,KAAK,EACb,mBAAmB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KACzC,OAAO,EAgCT,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,+BAA+B,GAC1C,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,EACjB,QAAQ,IAAI,CAAC,0BAA0B,EAAE,IAAI,CAAC,KAC7C;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAuB5C,CAAC;AAGF,eAAO,MAAM,wBAAwB,GAAI,6KAgBtC,qBAAqB,KAAG,OAAO,EAsFjC,CAAC;AAGF,eAAO,MAAM,sBAAsB,GAAI,sBAGpC,mBAAmB,KAAG,OAAO,EAiC/B,CAAC;AAGF,eAAO,MAAM,cAAc,GAAI,wBAG5B,aAAa,KAAG,OAAO,EAiBzB,CAAC;AAGF,eAAO,MAAM,mBAAmB,GAAI,gCAIjC,gBAAgB,KAAG,OAAO,EAmB5B,CAAC;AAGF,eAAO,MAAM,qBAAqB,GAAI,0CAKnC,kBAAkB,KAAG,OAAO,EAuB9B,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAU3D;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAAI,UAAU,OAAO,EAAE,KAAG,OAAO,EASlE,CAAC;AAEF;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAoBtD;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,MAAM,CAUR;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAkB1D;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,CAAC;IACzC,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,CAAC;IACtD,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,gCAAgC,GAAI,8DAO9C,4BAA4B,KAAG,OAAO,EAkBxC,CAAC"}
+{"version":3,"file":"messageOperations.d.ts","sourceRoot":"","sources":["../../src/utils/messageOperations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,OAAO,EACP,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAI5E,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnD,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAGD,MAAM,WAAW,oBAAqB,SAAQ,iBAAiB;IAC7D,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAGD,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,qBAAqB,EACrB,UAAU,CACX,CAAC;AAGF,MAAM,MAAM,6BAA6B,GAAG,IAAI,CAC9C,0BAA0B,EAC1B,WAAW,CACZ,GAAG;IACF,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;GAIG;AACH,eAAO,MAAM,oBAAoB,GAAI,WAAW,MAAM,KAAG,MAmCxD,CAAC;AAEF,eAAO,MAAM,iBAAiB,QAAO,MAA+B,CAAC;AAGrE,eAAO,MAAM,wBAAwB,GAAI,0EAQtC,oBAAoB,KAAG,OAAO,EA8BhC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B,GACtC,UAAU,OAAO,EAAE,EACnB,IAAI,MAAM,EACV,QAAQ,OAAO,CAAC,iBAAiB,CAAC,KACjC,OAAO,EAwBT,CAAC;AAGF,eAAO,MAAM,6BAA6B,GACxC,UAAU,OAAO,EAAE,EACnB,UAAU,MAAM,EAChB,YAAY,qCAAqC,EAAE,EACnD,QAAQ,KAAK,EACb,mBAAmB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KACzC,OAAO,EAgCT,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,+BAA+B,GAC1C,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,EACjB,QAAQ,IAAI,CAAC,0BAA0B,EAAE,IAAI,CAAC,KAC7C;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAuB5C,CAAC;AAGF,eAAO,MAAM,wBAAwB,GAAI,6KAgBtC,qBAAqB,KAAG;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAwFnE,CAAC;AAGF,eAAO,MAAM,sBAAsB,GAAI,sBAGpC,mBAAmB,KAAG,OAAO,EAiC/B,CAAC;AAGF,eAAO,MAAM,cAAc,GAAI,wBAG5B,aAAa,KAAG,OAAO,EAiBzB,CAAC;AAGF,eAAO,MAAM,mBAAmB,GAAI,gCAIjC,gBAAgB,KAAG,OAAO,EAmB5B,CAAC;AAGF,eAAO,MAAM,qBAAqB,GAAI,0CAKnC,kBAAkB,KAAG,OAAO,EAuB9B,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAU3D;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAAI,UAAU,OAAO,EAAE,KAAG,OAAO,EASlE,CAAC;AAEF;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAoBtD;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,MAAM,CAUR;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAkB1D;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,CAAC;IACzC,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,CAAC;IACtD,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,gCAAgC,GAAI,8DAO9C,4BAA4B,KAAG,OAAO,EAkBxC,CAAC"}

package/dist/utils/messageOperations.js

@@ -193,7 +193,7 @@ export const updateToolBlockInMessage = ({ messages, id, messageId, parameters,
                 }
             }
         }
-        return newMessages;
+        return { messages: newMessages, messageId };
     }
     // Find the last assistant or user message
     for (let i = newMessages.length - 1; i >= 0; i--) {
@@ -225,7 +225,8 @@ export const updateToolBlockInMessage = ({ messages, id, messageId, parameters,
                     if (isManuallyBackgrounded !== undefined)
                         toolBlock.isManuallyBackgrounded = isManuallyBackgrounded;
                 }
-                break; // Found and updated, stop searching
+                const foundMessageId = newMessages[i].id;
+                return { messages: newMessages, messageId: foundMessageId };
             }
             else if (newMessages[i].role === "assistant") {
                 // If existing block not found in assistant message, create new one
@@ -246,11 +247,12 @@ export const updateToolBlockInMessage = ({ messages, id, messageId, parameters,
                     parametersChunk: parametersChunk,
                     isManuallyBackgrounded: isManuallyBackgrounded,
                 });
-                break; // Created and added, stop searching
+                const foundMessageId = newMessages[i].id;
+                return { messages: newMessages, messageId: foundMessageId };
             }
         }
     }
-    return newMessages;
+    return { messages: newMessages };
 };
 // Add Error Block to the last assistant message
 export const addErrorBlockToMessage = ({ messages, error, }) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.17.12",
+  "version": "0.18.1",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",

package/src/managers/aiManager.ts

@@ -1,7 +1,6 @@
 import { type CallAgentOptions } from "../services/aiService.js";
 import * as aiService from "../services/aiService.js";
 import { convertMessagesForAPI } from "../utils/convertMessagesForAPI.js";
-import { microcompactMessages } from "../utils/microcompact.js";
 import { parseTaskNotificationXml } from "../utils/notificationXml.js";
 import { calculateComprehensiveTotalTokens } from "../utils/tokenCalculation.js";
 import {
@@ -819,13 +818,9 @@ export class AIManager {
     // Add plan mode reminder as persistent meta message before getting messages
     this.maybeAddPlanModeMessage(currentMode);
 
-    // Get recent message history with microcompact applied
+    // Get recent message history
     const rawMessages = this.messageManager.getMessages();
-    const microcompactedMessages = microcompactMessages(rawMessages, {
-      timeThresholdMS: 30 * 60 * 1000, // 30 minutes
-      recentResultsToKeep: 3,
-    });
-    const recentMessages = convertMessagesForAPI(microcompactedMessages);
+    const recentMessages = convertMessagesForAPI(rawMessages);
 
     try {
       // Get combined memory content

package/src/managers/messageManager.ts

@@ -12,6 +12,7 @@ import {
   addNotificationMessageToMessages,
   UserMessageParams,
   type AgentToolBlockUpdateParams,
+  type ToolBlockUpdateCallbackParams,
   type AddNotificationMessageParams,
   generateMessageId,
 } from "../utils/messageOperations.js";
@@ -40,12 +41,22 @@ export interface MessageManagerCallbacks {
   // Incremental callback
   onUserMessageAdded?: (params: UserMessageParams) => void;
   // MODIFIED: Remove arguments for separation of concerns
-  onAssistantMessageAdded?: () => void;
+  onAssistantMessageAdded?: (messageId: string) => void;
   // NEW: Streaming content callback - FR-001: receives chunk and accumulated content
-  onAssistantContentUpdated?: (chunk: string, accumulated: string) => void;
+  onAssistantContentUpdated?: (params: {
+    messageId: string;
+    chunk: string;
+    accumulated: string;
+    stage: "streaming" | "end";
+  }) => void;
   // NEW: Streaming reasoning callback
-  onAssistantReasoningUpdated?: (chunk: string, accumulated: string) => void;
-  onToolBlockUpdated?: (params: AgentToolBlockUpdateParams) => void;
+  onAssistantReasoningUpdated?: (params: {
+    messageId: string;
+    chunk: string;
+    accumulated: string;
+    stage: "streaming" | "end";
+  }) => void;
+  onToolBlockUpdated?: (params: ToolBlockUpdateCallbackParams) => void;
   onErrorBlockAdded?: (error: string) => void;
   onCompactBlockAdded?: (content: string) => void;
   onCompactionStateChange?: (isCompacting: boolean) => void;
@@ -419,7 +430,8 @@ export class MessageManager {
       additionalFieldsRecord,
     );
     this.setMessages(newMessages);
-    this.callbacks.onAssistantMessageAdded?.();
+    const messageId = this.messages[this.messages.length - 1]?.id;
+    this.callbacks.onAssistantMessageAdded?.(messageId);
 
     // Note: Subagent-specific callbacks are now handled by SubagentManager
   }
@@ -459,13 +471,22 @@ export class MessageManager {
 
   public updateToolBlock(params: AgentToolBlockUpdateParams): void {
     // Finalize any streaming text/reasoning blocks before adding/updating a tool block
-    this.finalizeCurrentStreamingBlocks();
-    const newMessages = updateToolBlockInMessage({
-      messages: this.messages,
-      ...params,
-    });
+    this.finalizeStreamingBlocks();
+    const { messages: newMessages, messageId: resolvedMessageId } =
+      updateToolBlockInMessage({
+        messages: this.messages,
+        ...params,
+      });
     this.setMessages(newMessages);
-    this.callbacks.onToolBlockUpdated?.(params);
+    const callbackParams: ToolBlockUpdateCallbackParams = {
+      ...params,
+      messageId:
+        params.messageId ||
+        resolvedMessageId ||
+        this.messages[this.messages.length - 1]?.id ||
+        "",
+    };
+    this.callbacks.onToolBlockUpdated?.(callbackParams);
 
     // Note: Subagent-specific callbacks are now handled by SubagentManager
   }
@@ -478,11 +499,15 @@ export class MessageManager {
     params: Omit<AgentToolBlockUpdateParams, "id">,
   ): string {
     // Finalize any streaming text/reasoning blocks before adding a tool block
-    this.finalizeCurrentStreamingBlocks();
+    this.finalizeStreamingBlocks();
     const { messages: newMessages, toolBlockId } =
       addToolBlockToMessageInMessages(this.messages, messageId, params);
     this.setMessages(newMessages);
-    this.callbacks.onToolBlockUpdated?.({ ...params, id: toolBlockId });
+    this.callbacks.onToolBlockUpdated?.({
+      ...params,
+      id: toolBlockId,
+      messageId,
+    });
     return toolBlockId;
   }
 
@@ -658,6 +683,49 @@ export class MessageManager {
     this.callbacks.onUsagesChange?.([...this._usages]);
   }
 
+  /**
+   * Finalize a streaming block of the given type by setting its stage to "end".
+   * Fires the corresponding incremental callback with chunk="" to signal finalization.
+   * Does NOT call onMessagesChange — the caller is responsible for that.
+   * Returns true if a block was finalized.
+   */
+  private finalizeStreamingBlock(
+    lastMessage: Message,
+    type: "text" | "reasoning",
+  ): boolean {
+    const index = lastMessage.blocks.findIndex(
+      (block) =>
+        block.type === type &&
+        (block as { stage?: string }).stage === "streaming",
+    );
+    if (index < 0) return false;
+
+    const block = lastMessage.blocks[index] as {
+      type: "text" | "reasoning";
+      content: string;
+      stage?: string;
+    };
+    lastMessage.blocks[index] = {
+      type: block.type,
+      content: block.content,
+      stage: "end" as const,
+    };
+
+    // Fire incremental callback to signal finalization
+    const callbackParams = {
+      messageId: lastMessage.id,
+      chunk: "",
+      accumulated: block.content,
+      stage: "end" as const,
+    };
+    if (type === "text") {
+      this.callbacks.onAssistantContentUpdated?.(callbackParams);
+    } else {
+      this.callbacks.onAssistantReasoningUpdated?.(callbackParams);
+    }
+    return true;
+  }
+
   /**
    * Update the current assistant message content during streaming
    * This method updates the last assistant message's content without creating a new message
@@ -670,23 +738,7 @@ export class MessageManager {
     if (lastMessage.role !== "assistant") return;
 
     // Finalize any streaming reasoning blocks before text content arrives
-    const reasoningIndex = lastMessage.blocks.findIndex(
-      (block) =>
-        block.type === "reasoning" &&
-        (block as { stage?: string }).stage === "streaming",
-    );
-    if (reasoningIndex >= 0) {
-      const reasoningBlock = lastMessage.blocks[reasoningIndex] as {
-        type: "reasoning";
-        content: string;
-        stage?: string;
-      };
-      lastMessage.blocks[reasoningIndex] = {
-        type: "reasoning" as const,
-        content: reasoningBlock.content,
-        stage: "end" as const,
-      };
-    }
+    this.finalizeStreamingBlock(lastMessage, "reasoning");
 
     // Get the current content to calculate the chunk
     const textBlockIndex = lastMessage.blocks.findIndex(
@@ -722,7 +774,12 @@ export class MessageManager {
     }
 
     // FR-001: Trigger callbacks with chunk and accumulated content
-    this.callbacks.onAssistantContentUpdated?.(chunk, newAccumulatedContent);
+    this.callbacks.onAssistantContentUpdated?.({
+      messageId: lastMessage.id,
+      chunk,
+      accumulated: newAccumulatedContent,
+      stage: "streaming",
+    });
 
     // Note: Subagent-specific callbacks are now handled by SubagentManager
 
@@ -740,23 +797,7 @@ export class MessageManager {
     if (lastMessage.role !== "assistant") return;
 
     // Finalize any streaming text blocks before reasoning content arrives
-    const textIndex = lastMessage.blocks.findIndex(
-      (block) =>
-        block.type === "text" &&
-        (block as { stage?: string }).stage === "streaming",
-    );
-    if (textIndex >= 0) {
-      const textBlock = lastMessage.blocks[textIndex] as {
-        type: "text";
-        content: string;
-        stage?: string;
-      };
-      lastMessage.blocks[textIndex] = {
-        type: "text" as const,
-        content: textBlock.content,
-        stage: "end" as const,
-      };
-    }
+    this.finalizeStreamingBlock(lastMessage, "text");
 
     // Get the current reasoning content to calculate the chunk
     const reasoningBlockIndex = lastMessage.blocks.findIndex(
@@ -792,45 +833,33 @@ export class MessageManager {
     }
 
     // Trigger callbacks with chunk and accumulated reasoning content
-    this.callbacks.onAssistantReasoningUpdated?.(
+    this.callbacks.onAssistantReasoningUpdated?.({
+      messageId: lastMessage.id,
       chunk,
-      newAccumulatedReasoning,
-    );
+      accumulated: newAccumulatedReasoning,
+      stage: "streaming",
+    });
 
     this.callbacks.onMessagesChange?.([...this.messages]); // Still need to notify of changes
   }
 
-  /**
-   * Public wrapper for finalizeCurrentStreamingBlocks.
-   * Finalizes text/reasoning blocks after streaming completes (e.g. final response with no tools).
-   */
-  public finalizeStreamingBlocks(): void {
-    this.finalizeCurrentStreamingBlocks();
-  }
-
   /**
    * Finalize streaming text/reasoning blocks by setting their stage to "end".
-   * Called when a new block (e.g. tool) is appended during streaming.
+   * Called when a new block (e.g. tool) is appended during streaming, or when streaming completes.
+   * Fires incremental callbacks for each finalized block.
    */
-  private finalizeCurrentStreamingBlocks(): void {
+  public finalizeStreamingBlocks(): void {
     if (this.messages.length === 0) return;
     const lastMessage = this.messages[this.messages.length - 1];
     if (lastMessage.role !== "assistant") return;
 
-    const newBlocks = lastMessage.blocks.map((block) => {
-      if (
-        (block.type === "text" || block.type === "reasoning") &&
-        block.stage === "streaming"
-      ) {
-        return { ...block, stage: "end" as const };
-      }
-      return block;
-    });
+    const textFinalized = this.finalizeStreamingBlock(lastMessage, "text");
+    const reasoningFinalized = this.finalizeStreamingBlock(
+      lastMessage,
+      "reasoning",
+    );
 
-    // Only update if something changed
-    const changed = newBlocks.some((b, i) => b !== lastMessage.blocks[i]);
-    if (changed) {
-      lastMessage.blocks = newBlocks;
+    if (textFinalized || reasoningFinalized) {
       this.callbacks.onMessagesChange?.([...this.messages]);
     }
   }

package/src/managers/subagentManager.ts

@@ -17,7 +17,7 @@ import { NotificationQueue } from "./notificationQueue.js";
 import { logger } from "../utils/globalLogger.js";
 import {
   UserMessageParams,
-  type AgentToolBlockUpdateParams,
+  type ToolBlockUpdateCallbackParams,
 } from "../utils/messageOperations.js";
 
 import { Container } from "../utils/container.js";
@@ -26,30 +26,37 @@ import type { PermissionMode } from "../types/permissions.js";
 import { ConfigurationService } from "../services/configurationService.js";
 
 export interface SubagentManagerCallbacks {
-  // Granular subagent message callbacks (015-subagent-message-callbacks)
+  // Granular subagent message callbacks (009-subagent)
   /** Triggered when subagent adds user message */
   onSubagentUserMessageAdded?: (
     subagentId: string,
     params: UserMessageParams,
   ) => void;
   /** Triggered when subagent creates assistant message */
-  onSubagentAssistantMessageAdded?: (subagentId: string) => void;
-  /** Triggered during subagent content streaming updates */
-  onSubagentAssistantContentUpdated?: (
+  onSubagentAssistantMessageAdded?: (
     subagentId: string,
-    chunk: string,
-    accumulated: string,
+    messageId: string,
   ) => void;
+  /** Triggered during subagent content streaming updates */
+  onSubagentAssistantContentUpdated?: (params: {
+    subagentId: string;
+    messageId: string;
+    chunk: string;
+    accumulated: string;
+    stage: "streaming" | "end";
+  }) => void;
   /** Triggered during subagent reasoning streaming updates */
-  onSubagentAssistantReasoningUpdated?: (
-    subagentId: string,
-    chunk: string,
-    accumulated: string,
-  ) => void;
+  onSubagentAssistantReasoningUpdated?: (params: {
+    subagentId: string;
+    messageId: string;
+    chunk: string;
+    accumulated: string;
+    stage: "streaming" | "end";
+  }) => void;
   /** Triggered when subagent tool block is updated */
   onSubagentToolBlockUpdated?: (
     subagentId: string,
-    params: AgentToolBlockUpdateParams,
+    params: ToolBlockUpdateCallbackParams,
   ) => void;
   /** Triggered when subagent messages change */
   onSubagentMessagesChange?: (subagentId: string, messages: Message[]) => void;
@@ -777,35 +784,43 @@ export class SubagentManager {
         }
       },
 
-      onAssistantMessageAdded: () => {
+      onAssistantMessageAdded: (messageId: string) => {
         // Forward assistant message events to parent via SubagentManager callbacks
         if (this.callbacks?.onSubagentAssistantMessageAdded) {
-          this.callbacks.onSubagentAssistantMessageAdded(subagentId);
+          this.callbacks.onSubagentAssistantMessageAdded(subagentId, messageId);
         }
       },
 
-      onAssistantContentUpdated: (chunk: string, accumulated: string) => {
+      onAssistantContentUpdated: (params: {
+        messageId: string;
+        chunk: string;
+        accumulated: string;
+        stage: "streaming" | "end";
+      }) => {
         // Forward assistant content updates to parent via SubagentManager callbacks
         if (this.callbacks?.onSubagentAssistantContentUpdated) {
-          this.callbacks.onSubagentAssistantContentUpdated(
+          this.callbacks.onSubagentAssistantContentUpdated({
+            ...params,
             subagentId,
-            chunk,
-            accumulated,
-          );
+          });
         }
       },
-      onAssistantReasoningUpdated: (chunk: string, accumulated: string) => {
+      onAssistantReasoningUpdated: (params: {
+        messageId: string;
+        chunk: string;
+        accumulated: string;
+        stage: "streaming" | "end";
+      }) => {
         // Forward assistant reasoning updates to parent via SubagentManager callbacks
         if (this.callbacks?.onSubagentAssistantReasoningUpdated) {
-          this.callbacks.onSubagentAssistantReasoningUpdated(
+          this.callbacks.onSubagentAssistantReasoningUpdated({
+            ...params,
             subagentId,
-            chunk,
-            accumulated,
-          );
+          });
         }
       },
 
-      onToolBlockUpdated: (params: AgentToolBlockUpdateParams) => {
+      onToolBlockUpdated: (params: ToolBlockUpdateCallbackParams) => {
         const instance = this.instances.get(subagentId);
         if (instance) {
           // Log tool execution to file only when finalized