wave-agent-sdk 1.0.5 → 1.0.6

package/dist/telemetry/instrumentation.js

@@ -8,13 +8,17 @@ import { logger } from "../utils/globalLogger.js";
 import * as fs from "node:fs";
 import { AuthService, getOrCreateAnonymousId, } from "../services/authService.js";
 // Lazy-loaded OTEL modules — only imported when telemetry is initialized
-let sdkNode;
+let sdkTraceBase;
+let otelResources;
 let api;
+let apiLogs;
 let sdkLogs;
 let exporterTraceOtlpHttp;
 let exporterLogsOtlpHttp;
-/** Internal SDK instance */
-let otelSdk;
+/** Internal trace provider instance */
+let tracerProvider;
+/** Internal log provider instance */
+let loggerProvider;
 /** Whether telemetry has been initialized */
 let initialized = false;
 /** Current resolved config */
@@ -34,14 +38,23 @@ function resolveTelemetryFilePath() {
 async function loadOTELModules() {
     if (api)
         return; // Already loaded
-    [sdkNode, api, sdkLogs, exporterTraceOtlpHttp, exporterLogsOtlpHttp] =
-        await Promise.all([
-            import("@opentelemetry/sdk-node"),
-            import("@opentelemetry/api"),
-            import("@opentelemetry/sdk-logs"),
-            import("@opentelemetry/exporter-trace-otlp-http"),
-            import("@opentelemetry/exporter-logs-otlp-http"),
-        ]);
+    [
+        sdkTraceBase,
+        otelResources,
+        api,
+        apiLogs,
+        sdkLogs,
+        exporterTraceOtlpHttp,
+        exporterLogsOtlpHttp,
+    ] = await Promise.all([
+        import("@opentelemetry/sdk-trace-base"),
+        import("@opentelemetry/resources"),
+        import("@opentelemetry/api"),
+        import("@opentelemetry/api-logs"),
+        import("@opentelemetry/sdk-logs"),
+        import("@opentelemetry/exporter-trace-otlp-http"),
+        import("@opentelemetry/exporter-logs-otlp-http"),
+    ]);
 }
 /**
  * Resolve the exporter target from environment variable or config.
@@ -281,33 +294,43 @@ export async function initializeTelemetry(config) {
             return;
         }
         const { tracesExporter, logsExporter } = currentConfig;
-        // Build resource attributes using new API (sdk-node 0.217+)
-        const { resources } = sdkNode;
-        const resource = resources.defaultResource().merge(resources.resourceFromAttributes({
+        // Build resource attributes (lightweight assembly: sdk-trace-base + resources)
+        const resource = otelResources.defaultResource().merge(otelResources.resourceFromAttributes({
             "service.name": "wave",
             "service.version": process.env.npm_package_version || "unknown",
             "os.type": process.platform,
             "host.arch": process.arch,
         }));
-        // Configure trace provider
-        const nodeSdkOptions = { resource };
+        const spanProcessors = [];
         if (tracesExporter === "otlp" && currentConfig.endpoint) {
-            nodeSdkOptions.traceExporter = createOTLPTraceExporter(currentConfig);
+            spanProcessors.push(new sdkTraceBase.BatchSpanProcessor(createOTLPTraceExporter(currentConfig)));
         }
         else if (tracesExporter === "jsonl") {
-            nodeSdkOptions.traceExporter = new JsonlSpanExporter();
+            spanProcessors.push(new sdkTraceBase.BatchSpanProcessor(new JsonlSpanExporter()));
         }
-        // Configure logs provider
+        const logProcessors = [];
         if (logsExporter === "otlp" && currentConfig.endpoint) {
-            nodeSdkOptions.logRecordProcessor = new sdkLogs.BatchLogRecordProcessor(createOTLPLogExporter(currentConfig));
+            logProcessors.push(new sdkLogs.BatchLogRecordProcessor(createOTLPLogExporter(currentConfig)));
         }
         else if (logsExporter === "jsonl") {
-            nodeSdkOptions.logRecordProcessor = new sdkLogs.BatchLogRecordProcessor(new JsonlLogExporter());
+            logProcessors.push(new sdkLogs.BatchLogRecordProcessor(new JsonlLogExporter()));
         }
-        // Only initialize SDK if at least one exporter is configured
-        if (nodeSdkOptions.traceExporter || nodeSdkOptions.logRecordProcessor) {
-            otelSdk = new sdkNode.NodeSDK(nodeSdkOptions);
-            await otelSdk.start();
+        // Only initialize providers if at least one exporter is configured
+        if (spanProcessors.length > 0 || logProcessors.length > 0) {
+            if (spanProcessors.length > 0) {
+                tracerProvider = new sdkTraceBase.BasicTracerProvider({
+                    resource,
+                    spanProcessors,
+                });
+                api.trace.setGlobalTracerProvider(tracerProvider);
+            }
+            if (logProcessors.length > 0) {
+                loggerProvider = new sdkLogs.LoggerProvider({
+                    resource,
+                    processors: logProcessors,
+                });
+                apiLogs.logs.setGlobalLoggerProvider(loggerProvider);
+            }
             initialized = true;
             logger?.info("OpenTelemetry initialized", {
                 tracesExporter,
@@ -329,7 +352,7 @@ export async function initializeTelemetry(config) {
  * Respects shutdownTimeoutMs config.
  */
 export async function shutdownTelemetry() {
-    if (!otelSdk || !initialized) {
+    if (!tracerProvider && !loggerProvider) {
         return;
     }
     try {
@@ -337,14 +360,20 @@ export async function shutdownTelemetry() {
         const timeoutPromise = new Promise((_, reject) => {
             setTimeout(() => reject(new Error("Telemetry shutdown timed out")), timeout);
         });
-        await Promise.race([otelSdk.shutdown(), timeoutPromise]);
+        const shutdowns = [];
+        if (tracerProvider)
+            shutdowns.push(tracerProvider.shutdown());
+        if (loggerProvider)
+            shutdowns.push(loggerProvider.shutdown());
+        await Promise.race([Promise.all(shutdowns), timeoutPromise]);
         logger?.debug("OpenTelemetry shut down successfully");
     }
     catch (error) {
         logger?.warn("OpenTelemetry shutdown failed:", error);
     }
     finally {
-        otelSdk = undefined;
+        tracerProvider = undefined;
+        loggerProvider = undefined;
         initialized = false;
         currentConfig = undefined;
     }

package/dist/tools/bashTool.js

@@ -152,6 +152,13 @@ The working directory persists between commands. Try to maintain your current wo
                 error: "Command parameter is required and must be a string",
             };
         }
+        // If the session cwd was auto-recovered (e.g. the previous worktree
+        // directory was removed manually via `git worktree remove`), surface
+        // the notice in this tool's result so the model is aware of the switch.
+        const cwdRecovery = context.aiManager?.consumeCwdRecovery?.();
+        const recoveryNotice = cwdRecovery
+            ? `Note: working directory ${cwdRecovery.from} no longer exists; session working directory recovered to ${cwdRecovery.to}.\n`
+            : "";
         // Resolve shell path: on Windows, use Git Bash; on macOS/Linux, use bash or zsh
         const shellPath = resolveShellPath();
         if (!shellPath) {
@@ -221,7 +228,7 @@ The working directory persists between commands. Try to maintain your current wo
             ].join("\n");
             return {
                 success: true,
-                content: backgroundMsg,
+                content: recoveryNotice + backgroundMsg,
                 shortResult: `Background process ${taskId} started${outputPath ? ` → ${outputPath}` : ""}`,
             };
         }
@@ -387,9 +394,11 @@ The working directory persists between commands. Try to maintain your current wo
                     const processedOutput = processToolResult(outputBuffer + (errorBuffer ? "\n" + errorBuffer : ""), BASH_MAX_OUTPUT_CHARS, "bash");
                     resolve({
                         success: false,
-                        content: processedOutput
-                            ? `${processedOutput}\n\n${reason}`
-                            : reason,
+                        content: recoveryNotice
+                            ? recoveryNotice + (processedOutput || reason)
+                            : processedOutput
+                                ? `${processedOutput}\n\n${reason}`
+                                : reason,
                         error: reason,
                     });
                 }
@@ -460,7 +469,8 @@ The working directory persists between commands. Try to maintain your current wo
                     const exitCode = code ?? 0;
                     const combinedOutput = outputBuffer + (errorBuffer ? "\n" + errorBuffer : "");
                     // Prepend CWD change message to output if present
-                    const finalOutput = (cwdMessage ? cwdMessage + "\n" : "") +
+                    const finalOutput = recoveryNotice +
+                        (cwdMessage ? cwdMessage + "\n" : "") +
                         (combinedOutput || `Command executed with exit code: ${exitCode}`);
                     const content = processToolResult(finalOutput, BASH_MAX_OUTPUT_CHARS, "bash");
                     const lines = combinedOutput.trim().split("\n");
@@ -490,7 +500,7 @@ The working directory persists between commands. Try to maintain your current wo
                     resolve({
                         success: false,
                         content: "",
-                        error: `Failed to execute command: ${error.message}`,
+                        error: `${recoveryNotice}Failed to execute command: ${error.message}`.trim(),
                     });
                 }
             });

package/dist/tools/grepTool.js

@@ -131,6 +131,13 @@ export const grepTool = {
         }
         try {
             const workdir = context.workdir;
+            // If the session cwd was auto-recovered (e.g. the previous worktree
+            // directory was removed manually via `git worktree remove`), surface
+            // the notice in this tool's result so the model is aware of the switch.
+            const cwdRecovery = context.aiManager?.consumeCwdRecovery?.();
+            const recoveryNotice = cwdRecovery
+                ? `Note: working directory ${cwdRecovery.from} no longer exists; session working directory recovered to ${cwdRecovery.to}.\n`
+                : "";
             const rgArgs = [
                 "--color=never",
                 "--max-columns=500",
@@ -207,7 +214,7 @@ export const grepTool = {
                 return {
                     success: false,
                     content: "",
-                    error: `ripgrep failed: ${result.stderr}`,
+                    error: `${recoveryNotice}ripgrep failed: ${result.stderr}`.trim(),
                 };
             }
             if (result.exitCode !== 0 && result.exitCode !== 1) {
@@ -217,7 +224,8 @@ export const grepTool = {
             if (!output) {
                 return {
                     success: true,
-                    content: "No matches found. Suggestion: specify the 'path' field to search in ignored or other directories (e.g., 'node_modules'), as the default search path is the current working directory and respects .gitignore.",
+                    content: recoveryNotice +
+                        "No matches found. Suggestion: specify the 'path' field to search in ignored or other directories (e.g., 'node_modules'), as the default search path is the current working directory and respects .gitignore.",
                     shortResult: "No matches found",
                     metadata: {
                         numMatches: 0,
@@ -256,7 +264,7 @@ export const grepTool = {
             }
             return {
                 success: true,
-                content: finalOutput,
+                content: recoveryNotice + finalOutput,
                 shortResult,
                 metadata: {
                     numMatches: totalMatches,

package/dist/types/index.d.ts

@@ -35,3 +35,6 @@ export * from "./cron.js";
 export * from "./telemetry.js";
 export * from "./auth.js";
 export * from "./workflow.js";
+export type { SessionMetadata, SessionData } from "../services/session.js";
+export type { ToolBlockUpdateCallbackParams } from "../utils/messageOperations.js";
+export type { QueuedMessage } from "../managers/messageQueue.js";

package/dist/types/messaging.d.ts

@@ -88,4 +88,6 @@ export interface TaskNotificationBlock {
     status: "completed" | "failed" | "killed" | "aborted";
     summary: string;
     outputFile?: string;
+    /** Full final response of the task (e.g. a fork subagent's result). */
+    result?: string;
 }

package/dist/utils/containerSetup.js

@@ -20,7 +20,6 @@ import { WorkflowManager } from "../managers/workflowManager.js";
 import { MemoryRuleManager } from "../managers/MemoryRuleManager.js";
 import { ReversionManager } from "../managers/reversionManager.js";
 import { SubagentManager } from "../managers/subagentManager.js";
-import { ForkedAgentManager } from "../managers/forkedAgentManager.js";
 import { LiveConfigManager } from "../managers/liveConfigManager.js";
 import { ReversionService } from "../services/reversionService.js";
 import { cleanupMetaOnlySessions } from "../services/session.js";
@@ -245,8 +244,6 @@ export function setupAgentContainer(setupOptions) {
         stream,
     });
     container.register("SubagentManager", subagentManager);
-    const forkedAgentManager = new ForkedAgentManager(container);
-    container.register("ForkedAgentManager", forkedAgentManager);
     const aiManager = new AIManager(container, {
         callbacks: {
             ...callbacks,

package/dist/utils/messageOperations.d.ts

@@ -132,5 +132,6 @@ export interface AddNotificationMessageParams {
     status: "completed" | "failed" | "killed" | "aborted";
     summary: string;
     outputFile?: string;
+    result?: string;
 }
-export declare const addNotificationMessageToMessages: ({ messages, taskId, taskType, status, summary, outputFile, }: AddNotificationMessageParams) => Message[];
+export declare const addNotificationMessageToMessages: ({ messages, taskId, taskType, status, summary, outputFile, result, }: AddNotificationMessageParams) => Message[];

package/dist/utils/messageOperations.js

@@ -452,7 +452,7 @@ export function getMessageContent(message) {
     }
     return "";
 }
-export const addNotificationMessageToMessages = ({ messages, taskId, taskType, status, summary, outputFile, }) => {
+export const addNotificationMessageToMessages = ({ messages, taskId, taskType, status, summary, outputFile, result, }) => {
     const block = {
         type: "task_notification",
         taskId,
@@ -460,6 +460,7 @@ export const addNotificationMessageToMessages = ({ messages, taskId, taskType, s
         status,
         summary,
         ...(outputFile !== undefined && { outputFile }),
+        ...(result !== undefined && { result }),
     };
     const notificationMessage = {
         id: generateMessageId(),

package/dist/utils/notificationXml.js

@@ -5,6 +5,9 @@ export function taskNotificationToXml(block) {
     if (block.outputFile) {
         xml += `<output-file>${block.outputFile}</output-file>\n`;
     }
+    if (block.result) {
+        xml += `<result>${block.result}</result>\n`;
+    }
     xml += `<status>${block.status}</status>\n`;
     xml += `<summary>${block.summary}</summary>\n`;
     xml += `</task-notification>`;
@@ -25,6 +28,7 @@ export function parseTaskNotificationXml(xml) {
             return null;
         }
         const outputFile = extractTag(xml, "output-file") || undefined;
+        const result = extractTag(xml, "result") || undefined;
         return {
             type: "task_notification",
             taskId,
@@ -32,6 +36,7 @@ export function parseTaskNotificationXml(xml) {
             status,
             summary,
             ...(outputFile && { outputFile }),
+            ...(result && { result }),
         };
     }
     catch {

package/dist/utils/ripgrep.d.ts

@@ -1,6 +1,6 @@
 /**
- * Path to the ripgrep binary bundled in the vendor directory.
- * Uses findUpSync to locate the package root, working correctly even
- * when bundled (e.g. esbuild into a VS Code extension).
+ * Path to the ripgrep binary, provided by the @vscode/ripgrep dependency.
+ * The wrapper resolves the platform-specific binary package at import time
+ * via optionalDependencies + os/cpu filtering.
  */
-export declare const rgPath: string;
+export { rgPath } from "@vscode/ripgrep";

package/dist/utils/ripgrep.js

@@ -1,29 +1,6 @@
-import path from "path";
-import process from "process";
-import { getPackageRoot } from "./configPaths.js";
 /**
- * Detect the current platform and architecture to find the correct bundled ripgrep binary.
+ * Path to the ripgrep binary, provided by the @vscode/ripgrep dependency.
+ * The wrapper resolves the platform-specific binary package at import time
+ * via optionalDependencies + os/cpu filtering.
  */
-function getPlatformKey() {
-    const platform = process.platform;
-    const arch = process.arch;
-    if (platform === "darwin") {
-        return arch === "arm64" ? "macos-aarch64" : "macos-x86_64";
-    }
-    else if (platform === "linux") {
-        return arch === "arm64" ? "linux-aarch64" : "linux-x86_64";
-    }
-    else if (platform === "win32") {
-        return arch === "arm64" ? "windows-aarch64" : "windows-x86_64";
-    }
-    throw new Error(`Unsupported platform: ${platform}-${arch}`);
-}
-const platformKey = getPlatformKey();
-const isWindows = platformKey.startsWith("windows");
-const binaryName = isWindows ? "rg.exe" : "rg";
-/**
- * Path to the ripgrep binary bundled in the vendor directory.
- * Uses findUpSync to locate the package root, working correctly even
- * when bundled (e.g. esbuild into a VS Code extension).
- */
-export const rgPath = path.join(getPackageRoot(), `vendor/ripgrep/${platformKey}/${binaryName}`);
+export { rgPath } from "@vscode/ripgrep";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",
@@ -13,6 +13,22 @@
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./types": {
+      "types": "./dist/types/index.d.ts",
+      "default": "./dist/types/index.js"
+    },
+    "./constants": {
+      "types": "./dist/constants/tools.d.ts",
+      "default": "./dist/constants/tools.js"
+    },
+    "./package.json": "./package.json",
+    "./*": "./*"
+  },
   "repository": {
     "type": "git",
     "url": "https://github.com/netease-lcap/wave-agent.git",
@@ -20,10 +36,6 @@
   },
   "files": [
     "dist",
-    "src",
-    "bin",
-    "vendor",
-    "scripts",
     "builtin",
     "README.md"
   ],
@@ -33,9 +45,10 @@
     "@opentelemetry/api-logs": "^0.217.0",
     "@opentelemetry/exporter-logs-otlp-http": "^0.217.0",
     "@opentelemetry/exporter-trace-otlp-http": "^0.217.0",
+    "@opentelemetry/resources": "^2.7.1",
     "@opentelemetry/sdk-logs": "^0.217.0",
-    "@opentelemetry/sdk-node": "^0.217.0",
-    "@opentelemetry/sdk-trace-node": "^2.7.1",
+    "@opentelemetry/sdk-trace-base": "^2.7.1",
+    "@vscode/ripgrep": "^1.18.0",
     "chokidar": "^4.0.3",
     "cron-parser": "^5.5.0",
     "find-up": "^8.0.0",
@@ -59,7 +72,6 @@
   },
   "license": "MIT",
   "scripts": {
-    "postinstall": "node scripts/postinstall.js",
     "build": "rimraf dist && tsc -p tsconfig.build.json && tsc-alias -p tsconfig.build.json",
     "type-check": "tsc --noEmit --incremental",
     "watch": "tsc -p tsconfig.build.json --watch & tsc-alias -p tsconfig.build.json --watch",

package/bin/rg

@@ -1,79 +0,0 @@
-#!/usr/bin/env dotslash
-
-{
-  "name": "rg",
-  "platforms": {
-    "macos-aarch64": {
-      "size": 1787248,
-      "hash": "blake3",
-      "digest": "8d9942032585ea8ee805937634238d9aee7b210069f4703c88fbe568e26fb78a",
-      "format": "tar.gz",
-      "path": "ripgrep-14.1.1-aarch64-apple-darwin/rg",
-      "providers": [
-        {
-          "url": "https://github.com/BurntSushi/ripgrep/releases/download/14.1.1/ripgrep-14.1.1-aarch64-apple-darwin.tar.gz"
-        }
-      ]
-    },
-    "linux-aarch64": {
-      "size": 2047405,
-      "hash": "blake3",
-      "digest": "0b670b8fa0a3df2762af2fc82cc4932f684ca4c02dbd1260d4f3133fd4b2a515",
-      "format": "tar.gz",
-      "path": "ripgrep-14.1.1-aarch64-unknown-linux-gnu/rg",
-      "providers": [
-        {
-          "url": "https://github.com/BurntSushi/ripgrep/releases/download/14.1.1/ripgrep-14.1.1-aarch64-unknown-linux-gnu.tar.gz"
-        }
-      ]
-    },
-    "macos-x86_64": {
-      "size": 2082672,
-      "hash": "blake3",
-      "digest": "e9b862fc8da3127f92791f0ff6a799504154ca9d36c98bf3e60a81c6b1f7289e",
-      "format": "tar.gz",
-      "path": "ripgrep-14.1.1-x86_64-apple-darwin/rg",
-      "providers": [
-        {
-          "url": "https://github.com/BurntSushi/ripgrep/releases/download/14.1.1/ripgrep-14.1.1-x86_64-apple-darwin.tar.gz"
-        }
-      ]
-    },
-    "linux-x86_64": {
-      "size": 2566310,
-      "hash": "blake3",
-      "digest": "f73cca4e54d78c31f832c7f6e2c0b4db8b04fa3eaa747915727d570893dbee76",
-      "format": "tar.gz",
-      "path": "ripgrep-14.1.1-x86_64-unknown-linux-musl/rg",
-      "providers": [
-        {
-          "url": "https://github.com/BurntSushi/ripgrep/releases/download/14.1.1/ripgrep-14.1.1-x86_64-unknown-linux-musl.tar.gz"
-        }
-      ]
-    },
-    "windows-x86_64": {
-      "size": 2058893,
-      "hash": "blake3",
-      "digest": "a8ce1a6fed4f8093ee997e57f33254e94b2cd18e26358b09db599c89882eadbd",
-      "format": "zip",
-      "path": "ripgrep-14.1.1-x86_64-pc-windows-msvc/rg.exe",
-      "providers": [
-        {
-          "url": "https://github.com/BurntSushi/ripgrep/releases/download/14.1.1/ripgrep-14.1.1-x86_64-pc-windows-msvc.zip"
-        }
-      ]
-    },
-    "windows-aarch64": {
-      "size": 1667740,
-      "hash": "blake3",
-      "digest": "47b971a8c4fca1d23a4e7c19bd4d88465ebc395598458133139406d3bf85f3fa",
-      "format": "zip",
-      "path": "rg.exe",
-      "providers": [
-        {
-          "url": "https://github.com/microsoft/ripgrep-prebuilt/releases/download/v13.0.0-13/ripgrep-v13.0.0-13-aarch64-pc-windows-msvc.zip"
-        }
-      ]
-    }
-  }
-}

package/dist/managers/forkedAgentManager.d.ts

@@ -1,50 +0,0 @@
-import * as fs from "fs";
-import type { Message } from "../types/index.js";
-import { Container } from "../utils/container.js";
-import { type SubagentInstance } from "./subagentManager.js";
-import type { PermissionMode } from "../types/permissions.js";
-export interface ForkedAgentEntry {
-    id: string;
-    instance: SubagentInstance;
-    logPath: string;
-    logStream?: fs.WriteStream;
-    status: "running" | "completed" | "failed";
-}
-export interface ForkedAgentManagerCallbacks {
-    onForkedAgentStatusChange?: (entries: ForkedAgentEntry[]) => void;
-}
-export declare class ForkedAgentManager {
-    private container;
-    private activeForks;
-    private inFlightPromises;
-    private callbacks;
-    constructor(container: Container, options?: {
-        callbacks?: ForkedAgentManagerCallbacks;
-    });
-    private get subagentManager();
-    /**
-     * Creates a forked subagent with conversation history and executes it asynchronously (fire-and-forget).
-     * Does NOT interact with BackgroundTaskManager.
-     */
-    forkAndExecute(subagentType: string, messages: Message[], parameters: {
-        description: string;
-        allowedTools?: string[];
-        model?: string;
-        permissionModeOverride?: PermissionMode;
-        maxTurns?: number;
-    }, prompt: string): Promise<string>;
-    private executeFork;
-    /**
-     * Abort a running forked agent.
-     */
-    stop(id: string): boolean;
-    /**
-     * Stop all running forked agents, wait for in-flight executions to settle, and clear the map.
-     */
-    cleanup(): Promise<void>;
-    /**
-     * Returns list of active forked agents.
-     */
-    getActiveForks(): ForkedAgentEntry[];
-    private notifyChange;
-}