wave-agent-sdk 0.18.3 → 0.18.4

package/src/tools/editTool.ts

@@ -109,13 +109,13 @@ Usage:
       };
     }
 
-    // Touch file to track it in context
-    context.messageManager?.touchFile(filePath);
+    // Trigger conditional rule loading for this file
+    context.messageManager?.triggerFileRead(filePath);
 
     // Enforce read-before-edit: the file must have been read first
     if (
       context.messageManager &&
-      !context.messageManager.hasFileInContext(filePath)
+      !context.messageManager.hasFileBeenRead(filePath)
     ) {
       return {
         success: false,

package/src/tools/readTool.ts

@@ -195,8 +195,8 @@ Usage:
       };
     }
 
-    // Touch file to track it in context
-    context.messageManager?.touchFile(filePath);
+    // Trigger conditional rule loading for this file
+    context.messageManager?.triggerFileRead(filePath);
 
     // Check for binary document formats
     if (isBinaryDocument(filePath)) {

package/src/tools/writeTool.ts

@@ -68,8 +68,8 @@ Usage:
       };
     }
 
-    // Touch file to track it in context
-    context.messageManager?.touchFile(filePath);
+    // Trigger conditional rule loading for this file
+    context.messageManager?.triggerFileRead(filePath);
 
     try {
       const resolvedPath = resolvePath(filePath, context.workdir);

package/src/utils/containerSetup.ts

@@ -117,6 +117,7 @@ export function setupAgentContainer(
             logger.error("Failed to sync task list with session:", error);
           });
         }
+        memoryService.clearCache();
         callbacks.onSessionIdChange?.(sessionId);
       },
     },

package/src/utils/openaiClient.ts

@@ -7,6 +7,28 @@ import {
 import { GatewayConfig } from "../types/config.js";
 import { logger } from "./globalLogger.js";
 
+const MAX_RETRIES = 10;
+const BASE_DELAY_MS = 500;
+const MAX_DELAY_MS = 32000;
+
+function getRetryDelay(
+  attempt: number,
+  retryAfterHeader: string | null,
+): number {
+  if (retryAfterHeader) {
+    const seconds = parseInt(retryAfterHeader, 10);
+    if (!isNaN(seconds)) {
+      return seconds * 1000;
+    }
+  }
+  const baseDelay = Math.min(
+    BASE_DELAY_MS * Math.pow(2, attempt - 1),
+    MAX_DELAY_MS,
+  );
+  const jitter = Math.random() * 0.25 * baseDelay;
+  return baseDelay + jitter;
+}
+
 type CreateParams =
   | ChatCompletionCreateParamsNonStreaming
   | ChatCompletionCreateParamsStreaming;
@@ -88,12 +110,11 @@ export class OpenAIClient {
 
     const fetchFn = (customFetch as typeof fetch) || fetch;
     let lastError: (Error & { status?: number; body?: unknown }) | undefined;
-    const maxRetries = 5;
-    const initialDelay = 2000;
+    let lastRetryAfter: string | null = null;
 
-    for (let attempt = 0; attempt <= maxRetries; attempt++) {
+    for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
       if (attempt > 0) {
-        const delay = initialDelay * Math.pow(2, attempt - 1);
+        const delay = getRetryDelay(attempt, lastRetryAfter);
         await new Promise((resolve) => setTimeout(resolve, delay));
       }
 
@@ -110,12 +131,13 @@ export class OpenAIClient {
         if (e instanceof Error && e.name === "AbortError") {
           throw e;
         }
-        if (attempt < maxRetries) {
+        if (attempt < MAX_RETRIES) {
           logger.warn("OpenAI API network error, retrying...", {
             attempt: attempt + 1,
             error: e,
           });
           lastError = e as Error;
+          lastRetryAfter = null;
           continue;
         }
         throw e;
@@ -179,10 +201,12 @@ export class OpenAIClient {
       const retryableStatus =
         response.status === 429 ||
         (response.status >= 500 && response.status !== 501);
-      if (retryableStatus && attempt < maxRetries) {
+      if (retryableStatus && attempt < MAX_RETRIES) {
+        lastRetryAfter = response.headers.get("retry-after");
         logger.warn("OpenAI API error, retrying...", {
           attempt: attempt + 1,
           status: response.status,
+          retryAfter: lastRetryAfter,
         });
         lastError = error;
         continue;
@@ -222,12 +246,16 @@ export class OpenAIClient {
 
           const data = trimmedLine.slice(5).trim();
           if (data === "[DONE]") return;
+          if (!data) continue; // Skip empty data lines (keepalive heartbeats)
 
           try {
             const json = JSON.parse(data);
             yield json as ChatCompletionChunk;
           } catch (e) {
-            logger.error("Failed to parse stream chunk", { error: e, data });
+            logger.error("Failed to parse stream chunk", {
+              error: e instanceof Error ? e.message : String(e),
+              data,
+            });
           }
         }
       }

package/src/utils/taskReminder.ts

@@ -1,18 +1,12 @@
 import type { Message } from "../types/messaging.js";
 import type { Task } from "../types/tasks.js";
-import type { ChatCompletionMessageParam } from "openai/resources.js";
 
 export const TASK_REMINDER_CONFIG = {
   TURNS_SINCE_WRITE: 10,
   TURNS_BETWEEN_REMINDERS: 10,
 };
 
-const TASK_MANAGEMENT_TOOLS = new Set([
-  "TaskCreate",
-  "TaskUpdate",
-  "TaskList",
-  "TaskGet",
-]);
+const TASK_MANAGEMENT_TOOLS = new Set(["TaskCreate", "TaskUpdate"]);
 const TASK_REMINDER_MARKER = "<!-- task-reminder -->";
 
 function isQualifyingAssistantMessage(message: Message): boolean {
@@ -66,43 +60,37 @@ export function getTaskReminderTurnCounts(messages: Message[]): {
 export function buildTaskReminderText(tasks: Task[]): string {
   let text = `${TASK_REMINDER_MARKER}\n`;
   text +=
-    "Here is a gentle reminder about the task list. The user may not have explicitly asked for the following to be done, but you should check the task list to see if any tasks need attention.\n";
+    "The task tools haven't been used recently. If you're working on tasks that would benefit from tracking progress, consider using TaskCreate to add new tasks and TaskUpdate to update task status (set to in_progress when starting, completed when done). Also consider cleaning up the task list if it has become stale. Only use these if relevant to the current work. This is just a gentle reminder - ignore if not applicable. Make sure that you never mention this reminder to the user";
 
   if (tasks.length > 0) {
-    text += "\n";
+    text += "\n\nHere are the existing tasks:\n";
     for (const task of tasks) {
       text += `#${task.id} [${task.status}] ${task.subject}\n`;
     }
-  } else {
-    text += "The task list is currently empty.\n";
   }
 
   return text;
 }
 
 export function maybeInjectTaskReminder(
-  messages: ChatCompletionMessageParam[],
   turnCounts: {
     turnsSinceLastTaskManagement: number;
     turnsSinceLastReminder: number;
   },
   tasks: Task[],
-): void {
+): string | null {
   if (
     turnCounts.turnsSinceLastTaskManagement <
     TASK_REMINDER_CONFIG.TURNS_SINCE_WRITE
   ) {
-    return;
+    return null;
   }
   if (
     turnCounts.turnsSinceLastReminder <
     TASK_REMINDER_CONFIG.TURNS_BETWEEN_REMINDERS
   ) {
-    return;
+    return null;
   }
 
-  messages.push({
-    role: "user",
-    content: buildTaskReminderText(tasks),
-  } as ChatCompletionMessageParam);
+  return buildTaskReminderText(tasks);
 }