wave-agent-sdk 0.18.6 → 0.19.0

package/dist/utils/modelCapabilities.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Model Capability Detection
+ *
+ * Pure declarative capability accessors. Capabilities are configured per-model
+ * via the `capabilities` field in `ModelConfig` (settings.json), not via regex
+ * or environment variables.
+ */
+import type { ModelCapabilities } from "../types/config.js";
+/**
+ * Determines if a model supports prompt caching.
+ * @param capabilities - Declarative model capabilities
+ * @returns True if promptCaching is explicitly enabled (default: false)
+ */
+export declare function supportsPromptCaching(capabilities?: ModelCapabilities): boolean;
+/**
+ * Determines if a model supports image/vision recognition.
+ * Opt-out semantics: by default all models are assumed to support vision.
+ * @param capabilities - Declarative model capabilities
+ * @returns True unless vision is explicitly disabled (default: true)
+ */
+export declare function supportsVision(capabilities?: ModelCapabilities): boolean;
+//# sourceMappingURL=modelCapabilities.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"modelCapabilities.d.ts","sourceRoot":"","sources":["../../src/utils/modelCapabilities.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAE5D;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,YAAY,CAAC,EAAE,iBAAiB,GAC/B,OAAO,CAET;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,YAAY,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAExE"}

package/dist/utils/modelCapabilities.js

@@ -0,0 +1,24 @@
+/**
+ * Model Capability Detection
+ *
+ * Pure declarative capability accessors. Capabilities are configured per-model
+ * via the `capabilities` field in `ModelConfig` (settings.json), not via regex
+ * or environment variables.
+ */
+/**
+ * Determines if a model supports prompt caching.
+ * @param capabilities - Declarative model capabilities
+ * @returns True if promptCaching is explicitly enabled (default: false)
+ */
+export function supportsPromptCaching(capabilities) {
+    return capabilities?.promptCaching ?? false;
+}
+/**
+ * Determines if a model supports image/vision recognition.
+ * Opt-out semantics: by default all models are assumed to support vision.
+ * @param capabilities - Declarative model capabilities
+ * @returns True unless vision is explicitly disabled (default: true)
+ */
+export function supportsVision(capabilities) {
+    return capabilities?.vision ?? true;
+}

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

@@ -1 +1 @@
-{"version":3,"file":"skillParser.d.ts","sourceRoot":"","sources":["../../src/utils/skillParser.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EAEjB,aAAa,EACd,MAAM,mBAAmB,CAAC;AAE3B;;GAEG;AACH,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,iBAAsB,GAC9B,eAAe,CAkGjB;AAyDD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,EAAE,CAwCvE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAOtD;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAa5E"}
+{"version":3,"file":"skillParser.d.ts","sourceRoot":"","sources":["../../src/utils/skillParser.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EAEjB,aAAa,EACd,MAAM,mBAAmB,CAAC;AAE3B;;GAEG;AACH,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,iBAAsB,GAC9B,eAAe,CAoGjB;AAyDD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,EAAE,CAwCvE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAOtD;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAa5E"}

package/dist/utils/skillParser.js

@@ -38,7 +38,9 @@ export function parseSkillFile(filePath, options = {}) {
         const skillType = skillPath.includes("/.wave/skills") ||
             skillPath.includes("\\.wave\\skills") ||
             skillPath.includes("/.claude/skills") ||
-            skillPath.includes("\\.claude\\skills")
+            skillPath.includes("\\.claude\\skills") ||
+            skillPath.includes("/.agents/skills") ||
+            skillPath.includes("\\.agents\\skills")
             ? "project"
             : "personal";
         // Extract allowed tools

package/dist/utils/tokenEstimate.d.ts

@@ -0,0 +1,24 @@
+/**
+ * CJK-aware token estimation without external dependencies.
+ *
+ * The naive `length / 4` heuristic under-estimates CJK text by ~4x because
+ * each Chinese/Japanese/Korean character is typically 1-2 tokens, not 0.25.
+ * This function separates CJK characters from other text and applies
+ * different ratios:
+ *
+ * - CJK characters: 1 char ≈ 1 token
+ * - Other characters: 4 chars ≈ 1 token (2 for JSON/JSONL/JSONC)
+ *
+ * The estimate intentionally leans high for CJK (safe direction for limit
+ * checks — better to reject than to let oversized content through).
+ */
+/**
+ * Estimate token count for a string, with CJK-awareness.
+ *
+ * @param content - The text content to estimate
+ * @param ext - File extension (without dot), e.g. "ts", "json". JSON/JSONL/JSONC
+ *              files use a tighter ratio (2 chars/token) for non-CJK text.
+ * @returns Estimated token count
+ */
+export declare function estimateTokens(content: string, ext?: string): number;
+//# sourceMappingURL=tokenEstimate.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tokenEstimate.d.ts","sourceRoot":"","sources":["../../src/utils/tokenEstimate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAMH;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,GAAG,MAAM,CAMpE"}

package/dist/utils/tokenEstimate.js

@@ -0,0 +1,30 @@
+/**
+ * CJK-aware token estimation without external dependencies.
+ *
+ * The naive `length / 4` heuristic under-estimates CJK text by ~4x because
+ * each Chinese/Japanese/Korean character is typically 1-2 tokens, not 0.25.
+ * This function separates CJK characters from other text and applies
+ * different ratios:
+ *
+ * - CJK characters: 1 char ≈ 1 token
+ * - Other characters: 4 chars ≈ 1 token (2 for JSON/JSONL/JSONC)
+ *
+ * The estimate intentionally leans high for CJK (safe direction for limit
+ * checks — better to reject than to let oversized content through).
+ */
+// CJK Unified Ideographs + Extension A + Hiragana + Katakana + Hangul Syllables
+const CJK_REGEX = /[\u4e00-\u9fff\u3400-\u4dbf\u3040-\u309f\u30a0-\u30ff\uac00-\ud7af]/g;
+/**
+ * Estimate token count for a string, with CJK-awareness.
+ *
+ * @param content - The text content to estimate
+ * @param ext - File extension (without dot), e.g. "ts", "json". JSON/JSONL/JSONC
+ *              files use a tighter ratio (2 chars/token) for non-CJK text.
+ * @returns Estimated token count
+ */
+export function estimateTokens(content, ext) {
+    const cjkCount = (content.match(CJK_REGEX) || []).length;
+    const otherCount = content.length - cjkCount;
+    const bytesPerToken = ext === "json" || ext === "jsonl" || ext === "jsonc" ? 2 : 4;
+    return Math.ceil(cjkCount + otherCount / bytesPerToken);
+}

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

@@ -1 +1 @@
-{"version":3,"file":"worktreeUtils.d.ts","sourceRoot":"","sources":["../../src/utils/worktreeUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,mFAAmF;IACnF,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAmBvD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CA6B7C;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAKjD;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,YAAY,CA6JtE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAmEvD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,MAAM,GAAG,SAAS,GACrC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAsClD"}
+{"version":3,"file":"worktreeUtils.d.ts","sourceRoot":"","sources":["../../src/utils/worktreeUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAYH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,mFAAmF;IACnF,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAmBvD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CA6B7C;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAKjD;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,YAAY,CAgKtE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAmEvD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,MAAM,GAAG,SAAS,GACrC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAsClD"}

package/dist/utils/worktreeUtils.js

@@ -5,7 +5,7 @@
 import { execFileSync } from "node:child_process";
 import * as path from "node:path";
 import * as fs from "node:fs";
-import { getGitMainRepoRoot, getDefaultRemoteBranch } from "./gitUtils.js";
+import { getGitMainRepoRoot, getDefaultRemoteBranch, ensureWaveRuntimeFilesExcluded, } from "./gitUtils.js";
 import { logger } from "./globalLogger.js";
 /**
  * Validate a worktree name to prevent path traversal and invalid characters.
@@ -79,6 +79,8 @@ export function createWorktree(name, cwd) {
     const worktreePath = path.join(repoRoot, ".wave", "worktrees", name);
     const branchName = `worktree-${name}`;
     const baseBranch = getDefaultRemoteBranch(cwd);
+    // Ensure Wave runtime files are git-excluded in this repo
+    ensureWaveRuntimeFilesExcluded(cwd);
     // Ensure parent directory exists
     const parentDir = path.dirname(worktreePath);
     if (!fs.existsSync(parentDir)) {

package/package.json

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

package/src/agent.ts

@@ -41,6 +41,7 @@ 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 { parseTaskNotificationXml } from "./utils/notificationXml.js";
 import { InitializationService } from "./services/initializationService.js";
 import { InteractionService } from "./services/interactionService.js";
@@ -598,17 +599,20 @@ export class Agent {
       options,
     );
 
-    // Initialize OpenTelemetry (non-blocking, graceful degradation)
+    // Initialize OpenTelemetry (awaited to ensure ALS context is ready before
+    // startInteractionSpan is called, preventing trace context fragmentation)
     const telemetryConfig = this.configurationService.resolveTelemetryConfig();
-    initializeTelemetry(telemetryConfig).catch((error) => {
+    try {
+      await initializeTelemetry(telemetryConfig);
+    } catch (error) {
       this.logger?.warn("Telemetry initialization failed:", error);
-    });
+    }
 
     // Log session_start event
     try {
       const modelConfig = this.getModelConfig();
       if (modelConfig.model) {
-        logOTelEvent("session_start", {
+        await logOTelEvent("session_start", {
           sessionId: this.messageManager.getSessionId(),
           model: modelConfig.model,
           workdir: this.workdir,
@@ -992,7 +996,9 @@ export class Agent {
    * @returns Promise that resolves to the AI's answer
    */
   public async askBtw(question: string): Promise<string> {
-    const messages = convertMessagesForAPI(this.messageManager.getMessages());
+    const messages = convertMessagesForAPI(this.messageManager.getMessages(), {
+      supportsVision: supportsVision(this.getModelConfig().capabilities),
+    });
     const result = await btw({
       gatewayConfig: this.getGatewayConfig(),
       modelConfig: this.getModelConfig(),

package/src/index.ts

@@ -14,6 +14,7 @@ export * from "./managers/messageQueue.js";
 // Export all utilities
 export * from "./utils/bashParser.js";
 export * from "./utils/convertMessagesForAPI.js";
+export * from "./utils/modelCapabilities.js";
 export * from "./utils/fileSearch.js";
 export * from "./utils/globalLogger.js";
 export * from "./utils/mcpUtils.js";

package/src/managers/aiManager.ts

@@ -1,8 +1,10 @@
 import { type CallAgentOptions } from "../services/aiService.js";
 import * as aiService from "../services/aiService.js";
 import { convertMessagesForAPI } from "../utils/convertMessagesForAPI.js";
+import { supportsVision } from "../utils/modelCapabilities.js";
 import { parseTaskNotificationXml } from "../utils/notificationXml.js";
 import { calculateComprehensiveTotalTokens } from "../utils/tokenCalculation.js";
+import { estimateTokens } from "../utils/tokenEstimate.js";
 import {
   getTaskReminderTurnCounts,
   maybeInjectTaskReminder,
@@ -44,6 +46,7 @@ import {
   endInteractionSpan,
   startLLMRequestSpan,
   endLLMRequestSpan,
+  resetTracingState,
 } from "../telemetry/sessionTracing.js";
 import { logOTelEvent } from "../telemetry/events.js";
 
@@ -491,7 +494,7 @@ export class AIManager {
       );
 
       // 7. Execute message reconstruction
-      this.messageManager.compactMessagesAndUpdateSession(
+      await this.messageManager.compactMessagesAndUpdateSession(
         enhancedSummary,
         compactUsage,
       );
@@ -522,6 +525,9 @@ export class AIManager {
 
       this.consecutiveCompactionFailures = 0;
 
+      // Reset incremental tracing state after compaction
+      resetTracingState();
+
       // 9. Log OTEL event
       logOTelEvent("compaction", {
         beforeTokens: String(messagesToCompact.length),
@@ -606,7 +612,7 @@ export class AIManager {
     );
     let usedTokens = 0;
     for (const file of recentFiles) {
-      const fileTokens = Math.ceil(file.content.length / 4);
+      const fileTokens = estimateTokens(file.content);
       if (usedTokens + fileTokens > POST_COMPACT_MAX_TOKENS_PER_FILE) continue;
       if (fileTokens > 0) usedTokens += fileTokens;
       contextParts.push(`\n\n## ${file.path}\n\`\`\`\n${file.content}\n\`\`\``);
@@ -639,7 +645,7 @@ export class AIManager {
               skillContent.slice(0, maxSkillChars) + "\n\n...[truncated]...";
           }
 
-          const skillTokens = Math.ceil(skillContent.length / 4);
+          const skillTokens = estimateTokens(skillContent);
           if (skillsUsedTokens + skillTokens > POST_COMPACT_SKILLS_TOKEN_BUDGET)
             break;
           skillsUsedTokens += skillTokens;
@@ -813,6 +819,7 @@ export class AIManager {
       let turnDepth = turnOffset;
 
       inner: while (true) {
+        let llmSpan: import("@opentelemetry/api").Span | undefined;
         try {
           // Save session in each iteration to ensure message persistence
           await this.messageManager.saveSession();
@@ -836,7 +843,10 @@ export class AIManager {
 
           // Get recent message history
           const rawMessages = this.messageManager.getMessages();
-          const recentMessages = convertMessagesForAPI(rawMessages);
+          const currentModelConfig = this.getModelConfig();
+          const recentMessages = convertMessagesForAPI(rawMessages, {
+            supportsVision: supportsVision(currentModelConfig.capabilities),
+          });
 
           // Track if assistant message has been created
           let assistantMessageCreated = false;
@@ -949,15 +959,33 @@ export class AIManager {
             };
           }
 
-          startLLMRequestSpan(model || this.getModelConfig().model || "");
+          llmSpan = startLLMRequestSpan(
+            model || this.getModelConfig().model || "",
+            {
+              context: "interaction",
+              systemPrompt:
+                typeof callAgentOptions.systemPrompt === "string"
+                  ? callAgentOptions.systemPrompt
+                  : undefined,
+              inputMessages: callAgentOptions.messages,
+              toolsSchema: callAgentOptions.tools
+                ? JSON.stringify(callAgentOptions.tools)
+                : undefined,
+            },
+          );
 
           const result = await aiService.callAgent(callAgentOptions);
 
           // End LLM span with usage data
-          endLLMRequestSpan({
+          endLLMRequestSpan(llmSpan, {
             model: model || this.getModelConfig().model || "",
             success: true,
             hasToolCall: !!(result.tool_calls && result.tool_calls.length > 0),
+            inputTokens: result.usage?.prompt_tokens,
+            outputTokens: result.usage?.completion_tokens,
+            cacheReadTokens: result.usage?.cache_read_input_tokens,
+            cacheCreationTokens: result.usage?.cache_creation_input_tokens,
+            modelOutput: result.content,
           });
 
           const createdByStreaming = assistantMessageCreated;
@@ -1238,7 +1266,7 @@ export class AIManager {
           break inner;
         } catch (error) {
           // End LLM span with error
-          endLLMRequestSpan({
+          endLLMRequestSpan(llmSpan, {
             model: model || this.getModelConfig().model || "",
             success: false,
             error: error instanceof Error ? error.message : String(error),
@@ -1653,6 +1681,7 @@ export class AIManager {
         shortResult: toolResult.shortResult,
         isManuallyBackgrounded: toolResult.isManuallyBackgrounded,
         startLineNumber: toolResult.startLineNumber,
+        images: toolResult.images,
         timestamp: Date.now(),
       });
 

package/src/managers/cronManager.ts

@@ -15,6 +15,7 @@ import {
   releaseSchedulerLock,
   registerSchedulerLockCleanup,
 } from "../utils/cronTasksLock.js";
+import { ensureWaveRuntimeFilesExcluded } from "../utils/gitUtils.js";
 
 export class CronManager {
   private jobs = new Map<string, CronJob>();
@@ -100,6 +101,7 @@ export class CronManager {
   }
 
   private tryLock(workdir: string): void {
+    ensureWaveRuntimeFilesExcluded(workdir);
     tryAcquireSchedulerLock({ dir: workdir, sessionId: this.sessionId })
       .then((acquired) => {
         if (acquired) {

package/src/managers/liveConfigManager.ts

@@ -17,7 +17,6 @@ import type { HookManager } from "./hookManager.js";
 import type { PermissionManager } from "./permissionManager.js";
 import { isValidHookEvent } from "../types/hooks.js";
 import { ConfigurationService } from "../services/configurationService.js";
-import { ensureGlobalGitIgnore } from "../utils/fileUtils.js";
 import { Container } from "../utils/container.js";
 
 import type {
@@ -97,9 +96,6 @@ export class LiveConfigManager {
       if (projectPaths) {
         for (const projectPath of projectPaths) {
           if (existsSync(projectPath)) {
-            if (projectPath.endsWith("settings.local.json")) {
-              await ensureGlobalGitIgnore("**/.wave/settings.local.json");
-            }
             await this.fileWatcher.watchFile(projectPath, (event) =>
               this.handleFileChange(event, "project"),
             );
@@ -364,14 +360,6 @@ export class LiveConfigManager {
 
       // Handle file creation or modification
       if (event.type === "change" || event.type === "create") {
-        if (
-          source === "project" &&
-          event.path.endsWith("settings.local.json") &&
-          event.type === "create"
-        ) {
-          await ensureGlobalGitIgnore("**/.wave/settings.local.json");
-        }
-
         // Add small delay to ensure file write is complete
         await new Promise((resolve) => setTimeout(resolve, 50));
 

package/src/managers/mcpManager.ts

@@ -725,11 +725,17 @@ export class McpManager {
     }
   }
 
-  // Get all tools from connected servers
+  // Get all tools from connected servers.
+  // "reconnecting" servers retain their last-known tool snapshot so that
+  // transient SSE disconnects don't churn the tool list (which would bust
+  // the prompt cache). Tools are only dropped once the server truly fails
+  // (status "error"/"disconnected").
   getAllConnectedTools(): McpTool[] {
     const allTools: McpTool[] = [];
     for (const server of this.servers.values()) {
-      if (server.status === "connected" && server.tools) {
+      const usable =
+        server.status === "connected" || server.status === "reconnecting";
+      if (usable && server.tools) {
         allTools.push(...server.tools);
       }
     }
@@ -848,9 +854,16 @@ export class McpManager {
         textContent.push(String(result.content));
       }
 
+      const textContentStr =
+        textContent.length > 0
+          ? textContent.join("\n")
+          : images.length > 0
+            ? `Tool returned ${images.length} image(s).`
+            : "No content";
+
       return {
         success: true,
-        content: textContent.length > 0 ? textContent.join("\n") : "No content",
+        content: textContentStr,
         images: images.length > 0 ? images : undefined,
         serverName,
       };