wave-agent-sdk 0.17.12 → 0.18.1

package/src/services/authService.ts

@@ -28,8 +28,9 @@ let _anonymousId: string | undefined;
 export class AuthService {
   private static instance: AuthService;
   private _serverUrl: string | undefined;
-  private onAuthChangeCallbacks: Array<(event: "login" | "logout") => void> =
-    [];
+  private onAuthChangeCallbacks: Array<
+    (event: "login" | "logout") => void | Promise<void>
+  > = [];
   private _refreshPromise: Promise<boolean> | null = null;
   private _authFileMtime: number = 0;
   private static readonly REFRESH_BUFFER_MS = 5 * 60 * 1000;
@@ -53,7 +54,9 @@ export class AuthService {
    * Register a callback for auth state changes.
    * Returns an unsubscribe function.
    */
-  onAuthChange(callback: (event: "login" | "logout") => void): () => void {
+  onAuthChange(
+    callback: (event: "login" | "logout") => void | Promise<void>,
+  ): () => void {
     this.onAuthChangeCallbacks.push(callback);
     return () => {
       this.onAuthChangeCallbacks = this.onAuthChangeCallbacks.filter(
@@ -62,10 +65,10 @@ export class AuthService {
     };
   }
 
-  private notifyAuthChange(event: "login" | "logout"): void {
+  private async notifyAuthChange(event: "login" | "logout"): Promise<void> {
     for (const cb of this.onAuthChangeCallbacks) {
       try {
-        cb(event);
+        await cb(event);
       } catch {
         // Don't let callback errors break auth flow
       }
@@ -112,7 +115,7 @@ export class AuthService {
     }
   }
 
-  clearAuth(): void {
+  async clearAuth(): Promise<void> {
     const config = this.loadAuth();
     delete config.SSO_TOKEN;
     delete config.SSO_REFRESH_TOKEN;
@@ -125,7 +128,7 @@ export class AuthService {
     } else {
       this.saveAuth(config);
     }
-    this.notifyAuthChange("logout");
+    await this.notifyAuthChange("logout");
   }
 
   getSSOToken(): string | undefined {
@@ -177,7 +180,7 @@ export class AuthService {
       user,
     });
 
-    this.notifyAuthChange("login");
+    await this.notifyAuthChange("login");
 
     return token;
   }
@@ -385,7 +388,7 @@ export class AuthService {
         logger.info(
           `[Auth] Refresh token rejected (${response.status}), clearing auth`,
         );
-        this.clearAuth();
+        await this.clearAuth();
         return false;
       }
 
@@ -412,7 +415,7 @@ export class AuthService {
       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) {
       // Network error — don't clear auth (might be transient)

package/src/services/remoteSettingsService.ts

@@ -18,6 +18,32 @@ const FETCH_TIMEOUT_MS = 10_000;
 
 let _cachedSettings: RemoteSettingsCache | null = null;
 let _pollingTimer: ReturnType<typeof setInterval> | null = null;
+let _onSettingsUpdateCallbacks: Array<() => void | Promise<void>> = [];
+
+/**
+ * Register a callback invoked when remote settings change (checksum differs).
+ * Returns an unsubscribe function.
+ */
+export function onSettingsUpdate(
+  callback: () => void | Promise<void>,
+): () => void {
+  _onSettingsUpdateCallbacks.push(callback);
+  return () => {
+    _onSettingsUpdateCallbacks = _onSettingsUpdateCallbacks.filter(
+      (cb) => cb !== callback,
+    );
+  };
+}
+
+async function notifySettingsUpdate(): Promise<void> {
+  for (const cb of _onSettingsUpdateCallbacks) {
+    try {
+      await cb();
+    } catch {
+      // Don't let callback errors break the fetch flow
+    }
+  }
+}
 
 function loadCacheFromDisk(): void {
   try {
@@ -77,6 +103,8 @@ async function fetchRemoteSettings(): Promise<RemoteSettingsFetchResult> {
     return { success: false, error: "Missing SSO token or server URL" };
   }
 
+  const oldChecksum = _cachedSettings?.checksum;
+
   const headers: Record<string, string> = {
     Authorization: `Bearer ${token}`,
   };
@@ -107,6 +135,9 @@ async function fetchRemoteSettings(): Promise<RemoteSettingsFetchResult> {
       logger.debug("remoteSettings: 404 not configured — clearing stale cache");
       _cachedSettings = null;
       removeCacheFromDisk();
+      if (oldChecksum) {
+        await notifySettingsUpdate();
+      }
       return { success: true, notConfigured: true, settings: null };
     }
 
@@ -134,6 +165,9 @@ async function fetchRemoteSettings(): Promise<RemoteSettingsFetchResult> {
     logger.debug("remoteSettings: fetched new settings", {
       checksum: data.checksum,
     });
+    if (data.checksum !== oldChecksum) {
+      await notifySettingsUpdate();
+    }
     return {
       success: true,
       settings: data.settings,
@@ -312,4 +346,5 @@ export const remoteSettingsService = {
   clear,
   shutdown,
   mergeRemoteSettings,
+  onSettingsUpdate,
 } as const;

package/src/utils/containerSetup.ts

@@ -161,9 +161,9 @@ export function setupAgentContainer(
   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();
     }
@@ -300,6 +300,12 @@ export function setupAgentContainer(
   });
   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/src/utils/messageOperations.ts

@@ -58,6 +58,14 @@ export type AgentToolBlockUpdateParams = Omit<
   "messages"
 >;
 
+// Callback payload type — SDK always fills messageId (required)
+export type ToolBlockUpdateCallbackParams = Omit<
+  AgentToolBlockUpdateParams,
+  "messageId"
+> & {
+  messageId: string;
+};
+
 export interface AddErrorBlockParams {
   messages: Message[];
   error: string;
@@ -289,7 +297,7 @@ export const updateToolBlockInMessage = ({
   compactParams,
   parametersChunk,
   isManuallyBackgrounded,
-}: UpdateToolBlockParams): Message[] => {
+}: UpdateToolBlockParams): { messages: Message[]; messageId?: string } => {
   const newMessages = [...messages];
 
   // If messageId is provided, target that specific message
@@ -321,7 +329,7 @@ export const updateToolBlockInMessage = ({
         }
       }
     }
-    return newMessages;
+    return { messages: newMessages, messageId };
   }
 
   // Find the last assistant or user message
@@ -350,7 +358,8 @@ export const updateToolBlockInMessage = ({
           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
         // This handles cases where we're streaming tool parameters before execution
@@ -370,11 +379,12 @@ export const updateToolBlockInMessage = ({
           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

package/dist/utils/microcompact.d.ts

@@ -1,7 +0,0 @@
-import type { Message } from "../types/messaging.js";
-export interface MicrocompactOptions {
-    timeThresholdMS: number;
-    recentResultsToKeep: number;
-}
-export declare function microcompactMessages(messages: Message[], options: MicrocompactOptions): Message[];
-//# sourceMappingURL=microcompact.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"microcompact.d.ts","sourceRoot":"","sources":["../../src/utils/microcompact.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAa,MAAM,uBAAuB,CAAC;AAEhE,MAAM,WAAW,mBAAmB;IAClC,eAAe,EAAE,MAAM,CAAC;IACxB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAID,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,EAAE,CAwFX"}

package/dist/utils/microcompact.js

@@ -1,78 +0,0 @@
-const CLEARED_RESULT = "[Old tool result content cleared]";
-export function microcompactMessages(messages, options) {
-    const { timeThresholdMS, recentResultsToKeep } = options;
-    // 1. Find the latest tool block timestamp across all assistant messages
-    let lastAssistantTime = 0;
-    for (const msg of messages) {
-        if (msg.role === "assistant") {
-            for (const block of msg.blocks) {
-                if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-                    if (block.timestamp > lastAssistantTime) {
-                        lastAssistantTime = block.timestamp;
-                    }
-                }
-            }
-        }
-    }
-    // 2. If no prior assistant messages with completed tools, return unchanged
-    if (lastAssistantTime === 0) {
-        return messages;
-    }
-    // 3. If within threshold, return unchanged
-    if (Date.now() - lastAssistantTime < timeThresholdMS) {
-        return messages;
-    }
-    const toolRefs = [];
-    for (let mi = 0; mi < messages.length; mi++) {
-        const msg = messages[mi];
-        if (msg.role === "assistant") {
-            for (let bi = 0; bi < msg.blocks.length; bi++) {
-                const block = msg.blocks[bi];
-                if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-                    toolRefs.push({
-                        msgIndex: mi,
-                        blockIndex: bi,
-                        timestamp: block.timestamp,
-                    });
-                }
-            }
-        }
-    }
-    toolRefs.sort((a, b) => b.timestamp - a.timestamp);
-    // 5. Mark the top N as "keep"
-    const keepSet = new Set();
-    for (let i = 0; i < Math.min(recentResultsToKeep, toolRefs.length); i++) {
-        const ref = toolRefs[i];
-        keepSet.add(`${ref.msgIndex}:${ref.blockIndex}`);
-    }
-    // 6. Deep-copy messages and clear result + shortResult on non-kept blocks
-    const result = messages.map((msg) => ({
-        ...msg,
-        blocks: msg.blocks.map((block) => {
-            if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-                return { ...block };
-            }
-            return block;
-        }),
-    }));
-    // Clear non-kept tool blocks
-    for (const ref of toolRefs) {
-        const key = `${ref.msgIndex}:${ref.blockIndex}`;
-        if (!keepSet.has(key)) {
-            result[ref.msgIndex] = {
-                ...result[ref.msgIndex],
-                blocks: result[ref.msgIndex].blocks.map((b, idx) => {
-                    if (idx === ref.blockIndex && b.type === "tool") {
-                        return {
-                            ...b,
-                            result: CLEARED_RESULT,
-                            shortResult: undefined,
-                        };
-                    }
-                    return b;
-                }),
-            };
-        }
-    }
-    return result;
-}

package/src/utils/microcompact.ts

@@ -1,101 +0,0 @@
-import type { Message, ToolBlock } from "../types/messaging.js";
-
-export interface MicrocompactOptions {
-  timeThresholdMS: number;
-  recentResultsToKeep: number;
-}
-
-const CLEARED_RESULT = "[Old tool result content cleared]";
-
-export function microcompactMessages(
-  messages: Message[],
-  options: MicrocompactOptions,
-): Message[] {
-  const { timeThresholdMS, recentResultsToKeep } = options;
-
-  // 1. Find the latest tool block timestamp across all assistant messages
-  let lastAssistantTime = 0;
-  for (const msg of messages) {
-    if (msg.role === "assistant") {
-      for (const block of msg.blocks) {
-        if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-          if (block.timestamp > lastAssistantTime) {
-            lastAssistantTime = block.timestamp;
-          }
-        }
-      }
-    }
-  }
-
-  // 2. If no prior assistant messages with completed tools, return unchanged
-  if (lastAssistantTime === 0) {
-    return messages;
-  }
-
-  // 3. If within threshold, return unchanged
-  if (Date.now() - lastAssistantTime < timeThresholdMS) {
-    return messages;
-  }
-
-  // 4. Collect all completed tool results with timestamps, sorted newest-first
-  type ToolRef = { msgIndex: number; blockIndex: number; timestamp: number };
-
-  const toolRefs: ToolRef[] = [];
-  for (let mi = 0; mi < messages.length; mi++) {
-    const msg = messages[mi];
-    if (msg.role === "assistant") {
-      for (let bi = 0; bi < msg.blocks.length; bi++) {
-        const block = msg.blocks[bi];
-        if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-          toolRefs.push({
-            msgIndex: mi,
-            blockIndex: bi,
-            timestamp: block.timestamp,
-          });
-        }
-      }
-    }
-  }
-
-  toolRefs.sort((a, b) => b.timestamp - a.timestamp);
-
-  // 5. Mark the top N as "keep"
-  const keepSet = new Set<string>();
-  for (let i = 0; i < Math.min(recentResultsToKeep, toolRefs.length); i++) {
-    const ref = toolRefs[i];
-    keepSet.add(`${ref.msgIndex}:${ref.blockIndex}`);
-  }
-
-  // 6. Deep-copy messages and clear result + shortResult on non-kept blocks
-  const result: Message[] = messages.map((msg) => ({
-    ...msg,
-    blocks: msg.blocks.map((block) => {
-      if (block.type === "tool" && block.stage === "end" && block.timestamp) {
-        return { ...block } as ToolBlock;
-      }
-      return block;
-    }),
-  }));
-
-  // Clear non-kept tool blocks
-  for (const ref of toolRefs) {
-    const key = `${ref.msgIndex}:${ref.blockIndex}`;
-    if (!keepSet.has(key)) {
-      result[ref.msgIndex] = {
-        ...result[ref.msgIndex],
-        blocks: result[ref.msgIndex].blocks.map((b, idx) => {
-          if (idx === ref.blockIndex && b.type === "tool") {
-            return {
-              ...b,
-              result: CLEARED_RESULT,
-              shortResult: undefined,
-            } as ToolBlock;
-          }
-          return b;
-        }),
-      };
-    }
-  }
-
-  return result;
-}