wave-agent-sdk 0.19.9 → 1.0.1

package/src/agent.ts

@@ -8,7 +8,6 @@ import { McpManager } from "./managers/mcpManager.js";
 import { LspManager } from "./managers/lspManager.js";
 import { BangManager } from "./managers/bangManager.js";
 import { CronManager } from "./managers/cronManager.js";
-import { GoalManager } from "./managers/goalManager.js";
 import { BackgroundTaskManager } from "./managers/backgroundTaskManager.js";
 import { MessageQueue, type QueuedMessage } from "./managers/messageQueue.js";
 import { SlashCommandManager } from "./managers/slashCommandManager.js";
@@ -38,9 +37,6 @@ import { LiveConfigManager } from "./managers/liveConfigManager.js";
 import { configValidator } from "./utils/configValidator.js";
 import { SkillManager } from "./managers/skillManager.js";
 import { TaskManager } from "./services/taskManager.js";
-import { btw } from "./services/aiService.js";
-import { convertMessagesForAPI } from "./utils/convertMessagesForAPI.js";
-import { supportsVision } from "./utils/modelCapabilities.js";
 import type { WorktreeSession } from "./utils/worktreeSession.js";
 import { parseTaskNotificationXml } from "./utils/notificationXml.js";
 import { InitializationService } from "./services/initializationService.js";
@@ -73,7 +69,6 @@ export class Agent {
   private pluginManager: PluginManager; // Add plugin manager instance
   private skillManager: SkillManager; // Add skill manager instance
   private cronManager: CronManager; // Add cron manager instance
-  private goalManager: GoalManager; // Add goal manager instance
   private hookManager: HookManager; // Add hooks manager instance
   private reversionManager: ReversionManager;
   private messageQueue: MessageQueue; // Unified queue for messages, bang commands, and notifications
@@ -95,7 +90,10 @@ export class Agent {
 
   // Dynamic configuration getter methods
   public getGatewayConfig(): GatewayConfig {
-    return this.configurationService.resolveGatewayConfig();
+    return {
+      ...this.configurationService.resolveGatewayConfig(),
+      sessionId: this.messageManager.getSessionId(),
+    };
   }
 
   public getModelConfig(): ModelConfig {
@@ -205,7 +203,6 @@ export class Agent {
     this.pluginManager = this.container.get("PluginManager")!;
     this.bangManager = this.container.get("BangManager")!;
     this.cronManager = this.container.get("CronManager")!;
-    this.goalManager = this.container.get("GoalManager")!;
     this.messageQueue = this.container.get("MessageQueue")!;
 
     // Wire up CWD change callback from AIManager to sync Agent's workdir
@@ -235,16 +232,6 @@ export class Agent {
     if (options.permissionMode) {
       this.setPermissionMode(options.permissionMode);
     }
-
-    // Wire up goal state change callback
-    this.goalManager.setOnGoalStateChange((active, condition, elapsed) => {
-      this.options.callbacks?.onGoalStateChange?.(active, condition, elapsed);
-    });
-
-    // Wire up goal evaluating callback
-    this.goalManager.setOnGoalEvaluating((evaluating) => {
-      this.options.callbacks?.onGoalEvaluating?.(evaluating);
-    });
   }
 
   // Public getter methods
@@ -337,16 +324,6 @@ export class Agent {
       .filter((m) => m.type !== "notification");
   }
 
-  /** Get goal status string */
-  public get goalStatus(): string {
-    return this.goalManager.getStatusString();
-  }
-
-  /** Check if a goal is active */
-  public get isGoalActive(): boolean {
-    return this.goalManager.isGoalActive();
-  }
-
   /**
    * Remove a queued message by index
    * @param index - The index of the message to remove
@@ -706,9 +683,6 @@ export class Agent {
   public async clearMessages(): Promise<void> {
     this.aiManager.abortAIMessage();
 
-    // Clear any active goal
-    this.goalManager.clearGoal();
-
     // Capture old session info before clearing
     const oldSessionId = this.messageManager.getSessionId();
     const transcriptPath = this.messageManager.getTranscriptPath();
@@ -778,75 +752,6 @@ export class Agent {
     await this.messageManager.saveSession();
   }
 
-  /**
-   * Set an autonomous goal for the session
-   * @param condition - The goal condition to achieve
-   */
-  public async setGoal(condition: string): Promise<void> {
-    // Check plan mode
-    if (this.getPermissionMode() === "plan") {
-      this.messageManager.addUserMessage({
-        content:
-          "<system-reminder>Cannot set a goal in plan mode. Exit plan mode first.</system-reminder>",
-        isMeta: true,
-      });
-      return;
-    }
-
-    this.goalManager.setGoal(condition);
-    this.messageManager.addUserMessage({
-      content: `<system-reminder>Goal set: ${condition}. The agent will work autonomously until this goal is achieved.</system-reminder>`,
-      isMeta: true,
-    });
-    // Add the goal as a user directive to start working
-    this.messageManager.addUserMessage({
-      content: condition,
-    });
-    this.aiManager.sendAIMessage();
-
-    await this.messageManager.saveSession();
-  }
-
-  /**
-   * Clear the current autonomous goal
-   */
-  public async clearGoal(): Promise<void> {
-    if (this.goalManager.isGoalActive()) {
-      this.goalManager.clearGoal();
-      this.messageManager.addUserMessage({
-        content: "<system-reminder>Goal cleared.</system-reminder>",
-        isMeta: true,
-      });
-    } else {
-      this.messageManager.addUserMessage({
-        content: "<system-reminder>No active goal to clear.</system-reminder>",
-        isMeta: true,
-      });
-    }
-
-    await this.messageManager.saveSession();
-  }
-
-  /**
-   * Show the current goal status
-   */
-  public async showGoalStatus(): Promise<void> {
-    if (this.goalManager.isGoalActive()) {
-      this.messageManager.addUserMessage({
-        content: `<system-reminder>${this.goalManager.getStatusString()}</system-reminder>`,
-        isMeta: true,
-      });
-    } else {
-      this.messageManager.addUserMessage({
-        content:
-          "<system-reminder>No active goal. Use /goal <condition> to set one.</system-reminder>",
-        isMeta: true,
-      });
-    }
-
-    await this.messageManager.saveSession();
-  }
-
   /** Unified interrupt method, interrupts both AI messages and command execution */
   public abortMessage(): void {
     // Guard: prevent tryDispatch (triggered by abortAIMessage → setIsLoading(false))
@@ -924,7 +829,9 @@ export class Agent {
           cwd: this.workdir,
           worktreePath,
           env: Object.fromEntries(
-            Object.entries(process.env).filter((e) => e[1] !== undefined),
+            Object.entries(this.configurationService.getMergedEnv()).filter(
+              (e) => e[1] !== undefined,
+            ),
           ) as Record<string, string>,
         },
       );
@@ -989,6 +896,21 @@ export class Agent {
     this.subagentManager.cleanup();
     // Cleanup forked agent manager
     await this.forkedAgentManager.cleanup();
+    // Drain an in-flight auto-memory extraction fork so the process doesn't
+    // exit mid-extraction
+    try {
+      const autoMemoryService =
+        this.container.get<
+          import("./services/autoMemoryService.js").AutoMemoryService
+        >("AutoMemoryService");
+      if (autoMemoryService) {
+        await autoMemoryService.drain();
+      }
+    } catch (error) {
+      this.logger?.warn(
+        `Auto-memory extraction drain failed: ${(error as Error).message}`,
+      );
+    }
     // Cleanup skill manager
     await this.skillManager.destroy();
     // Cleanup live configuration reload
@@ -1013,21 +935,30 @@ export class Agent {
   }
 
   /**
-   * Ask a side question without tool use
+   * Ask a side question without interrupting the main agent.
+   *
+   * Runs a single-turn fork of the conversation that reuses the main loop's
+   * system prompt, tools, and memory injection (prompt-cache friendly) and
+   * never executes tools. The abort signal lets the UI cancel a pending
+   * question (loading dismiss); aborts propagate as thrown errors.
+   *
    * @param question - The side question to ask
+   * @param abortSignal - Optional signal to abort the pending request
    * @returns Promise that resolves to the AI's answer
    */
-  public async askBtw(question: string): Promise<string> {
-    const messages = convertMessagesForAPI(this.messageManager.getMessages(), {
-      supportsVision: supportsVision(this.getModelConfig().capabilities),
-    });
-    const result = await btw({
-      gatewayConfig: this.getGatewayConfig(),
-      modelConfig: this.getModelConfig(),
-      messages,
+  public async askBtw(
+    question: string,
+    abortSignal?: AbortSignal,
+    onContent?: (content: string) => void,
+    onReasoning?: (content: string) => void,
+  ): Promise<string> {
+    const result = await this.aiManager.runBtwFork(
       question,
-    });
-    return result.content;
+      abortSignal,
+      onContent,
+      onReasoning,
+    );
+    return result.content ?? result.error ?? "No response received";
   }
 
   /**
@@ -1061,7 +992,7 @@ export class Agent {
     images?: Array<{ path: string; mimeType: string }>,
   ): Promise<void> {
     // If the agent is busy, enqueue the message — unless it's an immediate
-    // slash command (e.g., /goal clear, /clear, /compact) that should execute
+    // slash command (e.g., /clear, /compact) that should execute
     // right away even while AI is processing
     if (this.aiManager.isLoading || this.isCommandRunning) {
       const trimmed = content.trim();

package/src/index.ts

@@ -29,6 +29,7 @@ export * from "./utils/tokenCalculation.js";
 export * from "./utils/gitUtils.js";
 export * from "./utils/nameGenerator.js";
 export * from "./utils/worktreeSession.js";
+export * from "./utils/worktreeUtils.js";
 export { loadMergedWaveConfig } from "./services/configurationService.js";
 export * from "./types/index.js";