wave-agent-sdk 1.1.4 → 1.1.5

package/dist/agent.d.ts

@@ -31,6 +31,8 @@ export declare class Agent {
     private taskManager;
     private foregroundTaskManager;
     private container;
+    /** Unregister module-level listeners registered during container setup; invoked at the end of destroy(). */
+    private teardown;
     private configurationService;
     private workdir;
     private systemPrompt?;

package/dist/agent.js

@@ -56,6 +56,8 @@ export class Agent {
         this.isAborting = false; // Transient guard: prevents tryDispatch from firing during abortMessage (reset when the abort completes)
         this.isDestroyed = false; // Terminal guard: set in destroy(), never reset — no dispatch may ever start after destroy
         this.dispatchAborted = false; // Set on abort while a dispatch is running: suppress the .finally re-check so preserved notifications don't get dispatched after abort
+        /** Unregister module-level listeners registered during container setup; invoked at the end of destroy(). */
+        this.teardown = () => { };
         this.sessionStartTime = Date.now();
         const { logger, workdir, systemPrompt, stream = true } = options;
         // Set working directory early as we need it for loading configuration
@@ -68,7 +70,7 @@ export class Agent {
         this.stream = stream; // Save streaming mode flag
         // Store options for dynamic configuration resolution
         this.options = options;
-        this.container = setupAgentContainer({
+        const { container, teardown } = setupAgentContainer({
             options,
             workdir: this.workdir,
             configurationService: this.configurationService,
@@ -92,6 +94,8 @@ export class Agent {
             addPermissionRule: (rule) => this.addPermissionRule(rule),
             addUsage: (usage) => this.messageManager.addUsage(usage),
         });
+        this.container = container;
+        this.teardown = teardown;
         // Retrieve managers from container
         this.foregroundTaskManager = this.container.get("ForegroundTaskManager");
         this.memoryRuleManager = this.container.get("MemoryRuleManager");
@@ -717,6 +721,14 @@ export class Agent {
         if (!drained) {
             this.logger?.error(`Async work did not drain: ${this.asyncWorkRegistry.size} live work item(s) remain after destroy`);
         }
+        // Unregister module-level listeners (remote settings hot-update, auth
+        // change) so the agent's object graph becomes collectable. Without this,
+        // the module-level callback arrays pin every created agent via the
+        // per-agent closure contexts, even after the host drops its references.
+        this.teardown();
+        // Break the DI container's internal references (services/factories) so no
+        // per-agent manager is retained through it after destroy.
+        this.container.clear();
     }
     /**
      * Trigger the rewind UI callback

package/dist/builtin/plugins.js

@@ -41,6 +41,9 @@ const guidance = [
   "- 边界模糊时也先写 spec 草稿请用户确认，不要直接改代码。",
   "- 规格编写技能（specify）由 AI 自动触发：对话中涉及新需求或需求变更时主动创建或更新规格文件，不需要用户手动调用（不出现在斜杠命令列表中）。",
   \`- 新增或修改 spec 后运行校验：\${specCount}（自动检测 docs/specs/，否则 specs/，否则退出）。\`,
+  "- 阶段衔接一律用 AskUserQuestion 单选让用户点击决策，不要求自然语言：spec 确认（确认通过/需要修改）、是否制作原型、是否制定技术方案（plan）均通过单选推进。",
+  "- 可选原型阶段（在规格之后、plan 之前）：仅实现前端界面，数据全部使用 mock；可选 plan 阶段：进入 plan 模式制定技术方案，批准后再编码。二者均可跳过——最短流程=规格+编码，最长=规格+原型+plan+编码。仅当需求涉及前端界面时才询问是否制作原型，无前端界面的需求（后端服务、CLI、算法库等）直接跳过原型询问。",
+  "- 用 task 工具追踪进度：规格、原型、plan、编码各阶段开始前用 TaskCreate 创建任务并标记进行中（TaskUpdate），完成/批准/确认后标记完成，让用户在任务列表中看到当前所处阶段。",
 ].join("\\n");
 
 // JSON form → parsed as hookSpecificOutput.additionalContext by the hook manager.
@@ -128,7 +131,7 @@ if (warnings.length) {
 `,
     "plugins/sdd/skills/specify/SKILL.md": `---
 name: specify
-description: 根据自然语言描述创建或更新功能规格说明，生成包含用户故事与验收场景的规格文件。
+description: 根据自然语言描述创建或更新功能规格说明，并通过单选衔接可选原型、技术方案（plan）与编码阶段。
 user-invocable: false
 ---
 
@@ -144,6 +147,8 @@ $ARGUMENTS
 
 根据对话中的功能描述，执行以下步骤：
 
+0. **创建进度任务**：用 TaskCreate 创建「编写功能规格」任务，并用 TaskUpdate 标记进行中。后续每个阶段（原型、plan、编码）同样在开始前创建任务、结束后更新状态，让用户在任务列表中看到当前进度。
+
 1. **确定规格文件路径**：
    - **确定规格根目录**：优先复用项目中已有的规格目录——若 \`docs/specs/\` 存在则用之，否则若 \`specs/\` 存在则用之，否则默认 \`specs/\`（并在完成报告中说明所选目录，便于用户纠正）。
    - **选择分组**：若规格目录下已有分组子目录，沿用其既有分组约定；否则默认扁平结构（直接放在规格根目录下）。
@@ -165,7 +170,26 @@ $ARGUMENTS
    - 将每个标记作为问题展示，附带建议答案
    - 等待用户回复后更新规格文件
 
-5. 报告完成，输出规格文件路径。
+5. **校验并确认规格**：
+   - 运行会话引导中给出的 spec-count 校验命令（自动检测 docs/specs/，否则 specs/，否则跳过）
+   - 输出规格文件路径，并通过 AskUserQuestion 单选请求确认（选项：确认通过 / 需要修改）
+   - 选「需要修改」→ 按用户反馈更新规格后重新校验，并再次单选确认；选「确认通过」→ 将「编写功能规格」任务标记完成
+
+6. **询问是否制作原型（可选阶段，位于规格之后、plan 之前）**：
+   - 仅当需求涉及前端界面时弹出询问；需求不涉及前端界面（如后端服务、CLI 工具、算法库）时，跳过本阶段直接进入下一步，不弹出原型选择
+   - 通过 AskUserQuestion 单选询问（选项：制作原型 / 跳过）
+   - 选「制作原型」→ 用 TaskCreate 创建「制作原型」任务并标记进行中；仅实现前端界面，数据全部使用 mock（不接后端、不接真实数据）；完成后展示可交互原型供用户查看，并将任务标记完成
+   - 选「跳过」→ 直接进入下一步
+
+7. **询问是否制定技术方案（可选 plan 阶段）**：
+   - 通过 AskUserQuestion 单选询问（选项：进入 plan 模式 / 跳过）
+   - 选「进入 plan 模式」→ 用 TaskCreate 创建「制定技术方案」任务并标记进行中；调用 EnterPlanMode 进入 plan 模式，制定技术方案（技术选型、架构设计、实现步骤）并写入计划文件；用 ExitPlanMode 请求批准——被拒绝则按反馈更新方案后重新请求，批准后标记任务完成
+   - 选「跳过」→ 直接进入下一步
+
+8. **编码阶段**：
+   - 用 TaskCreate 创建「实现功能」任务并标记进行中
+   - 按已确认的规格实现；若制作了原型则遵循其交互设计，若批准了技术方案则遵循其架构
+   - 实现完成后将任务标记完成
 
 ## 指南
 

package/dist/managers/aiManager.js

@@ -1353,6 +1353,9 @@ ${question}`;
                     // Add streaming callbacks only if streaming is enabled
                     if (this.stream) {
                         callAgentOptions.onContentUpdate = (content) => {
+                            // Agent may have been destroyed mid-stream; ignore in-flight updates.
+                            if (!this.messageManager)
+                                return;
                             // Create assistant message on first chunk if not already created
                             if (!assistantMessageCreated) {
                                 this.messageManager.addAssistantMessage();
@@ -1361,6 +1364,9 @@ ${question}`;
                             this.messageManager.updateCurrentMessageContent(content);
                         };
                         callAgentOptions.onToolUpdate = (toolCall) => {
+                            // Agent may have been destroyed mid-stream; ignore in-flight updates.
+                            if (!this.messageManager)
+                                return;
                             // Create assistant message on first tool update if not already created
                             if (!assistantMessageCreated) {
                                 this.messageManager.addAssistantMessage();
@@ -1383,6 +1389,9 @@ ${question}`;
                             });
                         };
                         callAgentOptions.onReasoningUpdate = (reasoning) => {
+                            // Agent may have been destroyed mid-stream; ignore in-flight updates.
+                            if (!this.messageManager)
+                                return;
                             // Create assistant message on first reasoning update if not already created
                             if (!assistantMessageCreated) {
                                 this.messageManager.addAssistantMessage();

package/dist/utils/container.d.ts

@@ -27,4 +27,10 @@ export declare class Container {
      * Check if a service exists
      */
     has(token: string): boolean;
+    /**
+     * Remove all registered services and factories. Call during teardown so the
+     * container does not retain references to per-agent services after the
+     * agent is destroyed (breaking the agent's object graph).
+     */
+    clear(): void;
 }

package/dist/utils/container.js

@@ -52,6 +52,15 @@ export class Container {
             this.factories.has(token) ||
             (this.parent ? this.parent.has(token) : false));
     }
+    /**
+     * Remove all registered services and factories. Call during teardown so the
+     * container does not retain references to per-agent services after the
+     * agent is destroyed (breaking the agent's object graph).
+     */
+    clear() {
+        this.services.clear();
+        this.factories.clear();
+    }
 }
 // Example usage for ToolManager:
 /*

package/dist/utils/containerSetup.d.ts

@@ -16,4 +16,14 @@ export interface AgentContainerSetupOptions {
     addPermissionRule: (rule: string) => Promise<void>;
     addUsage: (usage: Usage) => void;
 }
-export declare function setupAgentContainer(setupOptions: AgentContainerSetupOptions): Container;
+export interface AgentContainerSetupResult {
+    container: Container;
+    /**
+     * Unregister module-level listeners registered during setup (remote settings
+     * hot-update, auth change). Must be called when the agent is destroyed;
+     * without it the module-level callback arrays pin the agent's object graph
+     * (the callback closures capture per-agent managers).
+     */
+    teardown: () => void;
+}
+export declare function setupAgentContainer(setupOptions: AgentContainerSetupOptions): AgentContainerSetupResult;

package/dist/utils/containerSetup.js

@@ -38,6 +38,12 @@ export function setupAgentContainer(setupOptions) {
     const container = new Container();
     container.register("AgentOptions", options);
     container.register("Workdir", workdir);
+    // Module-level listener teardowns collected during setup and returned to the
+    // agent, which invokes them in destroy(). The remote-settings callback
+    // strongly captures the per-agent LiveConfigManager: without unsubscribing
+    // it in destroy(), the module-level callback array keeps every created
+    // agent's object graph alive even after the host drops its references.
+    const teardowns = [];
     if (options.worktreeName) {
         container.register("WorktreeName", options.worktreeName);
         container.register("MainRepoRoot", getGitMainRepoRoot(workdir));
@@ -107,14 +113,14 @@ export function setupAgentContainer(setupOptions) {
     });
     container.register("McpManager", mcpManager);
     // Wire up auth change callback to refresh/clear remote settings
-    authService.onAuthChange(async (event) => {
+    teardowns.push(authService.onAuthChange(async (event) => {
         if (event === "login") {
             await remoteSettingsService.refresh();
         }
         else if (event === "logout") {
             remoteSettingsService.clear();
         }
-    });
+    }));
     const lspManager = options.lspManager || new LspManager(container);
     container.register("LspManager", lspManager);
     const permissionManager = new PermissionManager(container, {
@@ -245,9 +251,12 @@ 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 () => {
+    // The callback strongly captures the per-agent LiveConfigManager, so it MUST
+    // be unsubscribed in destroy() via teardown — otherwise the module-level
+    // callback array pins the whole agent object graph.
+    teardowns.push(remoteSettingsService.onSettingsUpdate(async () => {
         await liveConfigManager.reload();
-    });
+    }));
     const subagentManager = new SubagentManager(container, {
         workdir,
         callbacks: {
@@ -287,5 +296,12 @@ export function setupAgentContainer(setupOptions) {
     cronManager.start();
     const workflowManager = new WorkflowManager(container);
     container.register("WorkflowManager", workflowManager);
-    return container;
+    return {
+        container,
+        teardown: () => {
+            for (const unsubscribe of teardowns) {
+                unsubscribe();
+            }
+        },
+    };
 }

package/package.json

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