wave-code 1.0.6 → 1.0.7

package/dist/stdio/agentBridge.js

@@ -14,7 +14,7 @@
  * - Implement the canUseTool permission flow over the stdio protocol
  * - Handle config updates by destroying and recreating the Agent
  */
-import { Agent, listSessions, searchFiles, generateRandomName, getDefaultRemoteBranch, getMessageContent, PromptHistoryManager, AuthService, PluginCore, validateWorktreeRemovalPath, } from "wave-agent-sdk";
+import { Agent, listSessions, searchFiles, generateRandomName, getDefaultRemoteBranch, getMessageContent, PromptHistoryManager, AuthService, PluginCore, validateWorktreeRemovalPath, loadUserConfigEnv, } from "wave-agent-sdk";
 import { INVALID_PARAMS as PROTOCOL_INVALID_PARAMS, INTERNAL_ERROR as PROTOCOL_INTERNAL_ERROR, METHOD_NOT_FOUND as PROTOCOL_METHOD_NOT_FOUND, } from "./protocol.js";
 import { execFileSync } from "node:child_process";
 import { mkdirSync, existsSync, writeFileSync } from "node:fs";
@@ -32,6 +32,15 @@ export class AgentBridge {
         this.pendingPermissions = new Map();
         this.permissionCounter = 0;
         this.emit = options.emit;
+        // Mirror the user-level settings env WAVE_SERVER_URL into process.env
+        // before any agent initializes. getAuthStatus (webviewReady →
+        // pushInitialState) can run before the first agent, and AuthService falls
+        // back to the default URL otherwise — refreshing a custom-domain token
+        // against the wrong host 401s into a logged-out state.
+        const userEnv = loadUserConfigEnv();
+        if (userEnv.WAVE_SERVER_URL) {
+            process.env.WAVE_SERVER_URL = userEnv.WAVE_SERVER_URL;
+        }
     }
     // ── Public API ────────────────────────────────────────────────
     async handleRequest(method, params, sessionId) {
@@ -52,6 +61,10 @@ export class AgentBridge {
                 return this.listPendingPermissions();
             case "updateConfig":
                 return this.updateConfig(p, sessionId);
+            case "getConfiguredModels":
+                return this.getConfiguredModels(sessionId);
+            case "setModel":
+                return this.setModel(p.model, sessionId);
             // ── Messages ──
             case "sendMessage":
                 return this.sendMessage(p, sessionId);
@@ -92,6 +105,8 @@ export class AgentBridge {
             // ── Commands ──
             case "getSlashCommands":
                 return this.getSlashCommands(sessionId);
+            case "getSubagentConfigurations":
+                return this.getSubagentConfigurations(sessionId);
             // ── File / History (global — no session required) ──
             case "searchFiles":
                 return this.searchFiles(p, sessionId);
@@ -430,6 +445,22 @@ export class AgentBridge {
         });
         return { sessionId: agent.sessionId };
     }
+    getConfiguredModels(sessionId) {
+        const entry = this.requireSession(sessionId);
+        return {
+            models: entry.agent.getConfiguredModels(),
+            currentModel: entry.agent.getModelConfig().model,
+        };
+    }
+    async setModel(model, sessionId) {
+        const entry = this.requireSession(sessionId);
+        entry.agent.setModel(model);
+        // Keep storedConfig in sync: updateConfig recreates the agent from
+        // storedConfig, so without this a later config save would revert the
+        // model chosen here.
+        entry.storedConfig = { ...entry.storedConfig, model };
+        return null;
+    }
     // ── Messages ──────────────────────────────────────────────────
     async sendMessage(params, sessionId) {
         const entry = this.requireSession(sessionId);
@@ -592,6 +623,10 @@ export class AgentBridge {
         const entry = this.requireSession(sessionId);
         return { commands: entry.agent.getSlashCommands() };
     }
+    getSubagentConfigurations(sessionId) {
+        const entry = this.requireSession(sessionId);
+        return { configurations: entry.agent.getSubagentConfigurations() };
+    }
     // ── File / History (global) ───────────────────────────────────
     async searchFiles(params, sessionId) {
         const files = await searchFiles(params.query, {
@@ -666,13 +701,6 @@ export class AgentBridge {
     // ── Auth (global) ────────────────────────────────────────────
     async getAuthStatus() {
         const authService = AuthService.getInstance();
-        // A stale-but-refreshable token still means "logged in" — the daemon may
-        // have started with an expired access token (hourly expiry) and only
-        // refreshes lazily on the first API call. Without this proactive refresh a
-        // fresh client querying right after daemon start gets a false
-        // isAuthenticated and e.g. the desktop welcome page keeps showing the
-        // login button for an authenticated host. Mirrors the refresh that
-        // createAuthAwareFetch does before every real request.
         await authService.checkAndRefreshTokenIfNeeded();
         return {
             isAuthenticated: authService.isSSOAuthenticated(),
@@ -868,8 +896,8 @@ export class AgentBridge {
             onUpdateBangMessage: (command, output, messageId) => {
                 this.emit("bangMessageUpdated", { command, output, messageId }, ctx.registeredSessionId);
             },
-            onCompleteBangMessage: (command, exitCode, messageId) => {
-                this.emit("bangMessageCompleted", { command, exitCode, messageId }, ctx.registeredSessionId);
+            onCompleteBangMessage: (command, exitCode, messageId, output) => {
+                this.emit("bangMessageCompleted", { command, exitCode, messageId, output }, ctx.registeredSessionId);
             },
             onNotificationMessageAdded: (params) => {
                 const msg = ctx.agent?.messages.find((m) => m.role === "user" &&

package/dist/stdio/protocol.d.ts

@@ -34,7 +34,7 @@ export declare const INVALID_REQUEST = -32600;
 export declare const METHOD_NOT_FOUND = -32601;
 export declare const INVALID_PARAMS = -32602;
 export declare const INTERNAL_ERROR = -32603;
-export type RequestMethod = "initialize" | "destroy" | "restoreSession" | "listSessions" | "getSessionInfo" | "sendMessage" | "bang" | "askBtw" | "abortMessage" | "clearMessages" | "rewindToMessage" | "listRewindCheckpoints" | "deleteQueuedMessage" | "updateQueuedMessage" | "deleteQueuedMessageById" | "getMessages" | "getFullMessageThread" | "setPermissionMode" | "getPermissionMode" | "getMcpServers" | "connectMcpServer" | "disconnectMcpServer" | "getSlashCommands" | "searchFiles" | "writeArtifactFile" | "getPromptHistory" | "searchPromptHistory" | "updateConfig" | "listPendingPermissions" | "getAuthStatus" | "login" | "logout" | "listPlugins" | "installPlugin" | "uninstallPlugin" | "enablePlugin" | "disablePlugin" | "updatePlugin" | "listMarketplaces" | "addMarketplace" | "removeMarketplace" | "updateMarketplace" | "compact" | "getBackgroundTaskOutput" | "stopBackgroundTask" | "getWorkflowRuns" | "stopWorkflowRun" | "listGitBranches" | "createWorktree" | "removeWorktree";
+export type RequestMethod = "initialize" | "destroy" | "restoreSession" | "listSessions" | "getSessionInfo" | "sendMessage" | "bang" | "askBtw" | "abortMessage" | "clearMessages" | "rewindToMessage" | "listRewindCheckpoints" | "deleteQueuedMessage" | "updateQueuedMessage" | "deleteQueuedMessageById" | "getMessages" | "getFullMessageThread" | "setPermissionMode" | "getPermissionMode" | "getMcpServers" | "connectMcpServer" | "disconnectMcpServer" | "getSlashCommands" | "searchFiles" | "writeArtifactFile" | "getPromptHistory" | "searchPromptHistory" | "updateConfig" | "getConfiguredModels" | "setModel" | "listPendingPermissions" | "getAuthStatus" | "login" | "logout" | "listPlugins" | "installPlugin" | "uninstallPlugin" | "enablePlugin" | "disablePlugin" | "updatePlugin" | "listMarketplaces" | "addMarketplace" | "removeMarketplace" | "updateMarketplace" | "compact" | "getBackgroundTaskOutput" | "stopBackgroundTask" | "getWorkflowRuns" | "stopWorkflowRun" | "listGitBranches" | "createWorktree" | "removeWorktree";
 export type ClientNotificationMethod = "permissionResponse";
 export type ServerNotificationMethod = "userMessageAdded" | "assistantMessageAdded" | "assistantContentUpdated" | "assistantReasoningUpdated" | "toolBlockUpdated" | "errorBlockAdded" | "loadingChange" | "commandRunningChange" | "queuedMessagesChange" | "tasksChange" | "sessionIdChange" | "permissionModeChange" | "mcpServersChange" | "workdirChange" | "bangMessageAdded" | "bangMessageUpdated" | "bangMessageCompleted" | "notificationMessageAdded" | "permissionRequest" | "authUrl" | "compactBlockAdded" | "compactionStateChange" | "backgroundTasksChange" | "btwContent";
 export declare function isRequest(msg: unknown): msg is JsonRpcRequest;

package/dist/utils/usageSummary.d.ts

@@ -13,10 +13,6 @@ export interface TokenSummary {
     };
     cache_read_input_tokens?: number;
     cache_creation_input_tokens?: number;
-    cache_creation?: {
-        ephemeral_5m_input_tokens: number;
-        ephemeral_1h_input_tokens: number;
-    };
 }
 /**
  * Calculate token usage summary by model from usage array

package/dist/utils/usageSummary.js

@@ -34,20 +34,6 @@ export function calculateTokenSummary(usages) {
                 (summary.cache_creation_input_tokens || 0) +
                     usage.cache_creation_input_tokens;
         }
-        if (usage.cache_creation &&
-            (usage.cache_creation.ephemeral_5m_input_tokens > 0 ||
-                usage.cache_creation.ephemeral_1h_input_tokens > 0)) {
-            if (!summary.cache_creation) {
-                summary.cache_creation = {
-                    ephemeral_5m_input_tokens: 0,
-                    ephemeral_1h_input_tokens: 0,
-                };
-            }
-            summary.cache_creation.ephemeral_5m_input_tokens +=
-                usage.cache_creation.ephemeral_5m_input_tokens || 0;
-            summary.cache_creation.ephemeral_1h_input_tokens +=
-                usage.cache_creation.ephemeral_1h_input_tokens || 0;
-        }
         // Track operation types
         if (usage.operation_type === "agent") {
             summary.operations.agent_calls += 1;
@@ -86,8 +72,6 @@ export function displayUsageSummary(usages, sessionFilePath) {
     let totalCompactions = 0;
     let totalCacheRead = 0;
     let totalCacheCreation = 0;
-    let totalCache5m = 0;
-    let totalCache1h = 0;
     let hasCacheData = false;
     for (const [, summary] of Object.entries(summaries)) {
         console.log(`Model: ${summary.model}`);
@@ -96,8 +80,7 @@ export function displayUsageSummary(usages, sessionFilePath) {
         console.log(`  Total tokens: ${summary.total_tokens.toLocaleString()}`);
         // Display cache information if available
         if (summary.cache_read_input_tokens ||
-            summary.cache_creation_input_tokens ||
-            summary.cache_creation) {
+            summary.cache_creation_input_tokens) {
             hasCacheData = true;
             console.log("  Cache Usage:");
             if (summary.cache_read_input_tokens &&
@@ -110,16 +93,6 @@ export function displayUsageSummary(usages, sessionFilePath) {
                 console.log(`    Created cache: ${summary.cache_creation_input_tokens.toLocaleString()} tokens`);
                 totalCacheCreation += summary.cache_creation_input_tokens;
             }
-            if (summary.cache_creation) {
-                if (summary.cache_creation.ephemeral_5m_input_tokens > 0) {
-                    console.log(`    5m cache: ${summary.cache_creation.ephemeral_5m_input_tokens.toLocaleString()} tokens`);
-                    totalCache5m += summary.cache_creation.ephemeral_5m_input_tokens;
-                }
-                if (summary.cache_creation.ephemeral_1h_input_tokens > 0) {
-                    console.log(`    1h cache: ${summary.cache_creation.ephemeral_1h_input_tokens.toLocaleString()} tokens`);
-                    totalCache1h += summary.cache_creation.ephemeral_1h_input_tokens;
-                }
-            }
         }
         console.log(`  Operations: ${summary.operations.agent_calls} agent calls, ${summary.operations.compactions} compactions`);
         console.log();
@@ -142,12 +115,6 @@ export function displayUsageSummary(usages, sessionFilePath) {
             if (totalCacheCreation > 0) {
                 console.log(`    Created cache: ${totalCacheCreation.toLocaleString()} tokens`);
             }
-            if (totalCache5m > 0) {
-                console.log(`    5m cache: ${totalCache5m.toLocaleString()} tokens`);
-            }
-            if (totalCache1h > 0) {
-                console.log(`    1h cache: ${totalCache1h.toLocaleString()} tokens`);
-            }
         }
         console.log(`  Operations: ${totalAgentCalls} agent calls, ${totalCompactions} compactions`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "CLI-based code assistant powered by AI, built with React and Ink",
   "repository": {
     "type": "git",
@@ -43,7 +43,7 @@
     "wrap-ansi": "^10.0.0",
     "yargs": "^17.7.2",
     "zod": "^3.23.8",
-    "wave-agent-sdk": "1.0.6"
+    "wave-agent-sdk": "1.0.7"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",

package/src/components/AgentsManager.tsx

@@ -0,0 +1,290 @@
+import React, { useEffect, useMemo, useReducer } from "react";
+import { Box, Text, useInput, useStdout } from "ink";
+import type { SubagentConfiguration } from "wave-agent-sdk";
+import { Markdown } from "./Markdown.js";
+import {
+  agentsManagerReducer,
+  type AgentsManagerState,
+} from "../reducers/agentsManagerReducer.js";
+
+export interface AgentsManagerProps {
+  onCancel: () => void;
+  agentDefinitions: SubagentConfiguration[];
+}
+
+interface DisplayEntry {
+  kind: "header" | "definition" | "empty";
+  label: string;
+  sub?: string;
+  model?: string;
+  scope?: SubagentConfiguration["scope"];
+  selectableIndex: number; // -1 for non-selectable rows
+  definition?: SubagentConfiguration;
+}
+
+const SCOPE_LABELS: Record<SubagentConfiguration["scope"], string> = {
+  builtin: "Built-in agents",
+  user: "User agents",
+  project: "Project agents",
+  plugin: "Plugin agents",
+};
+
+const SCOPE_ORDER: SubagentConfiguration["scope"][] = [
+  "builtin",
+  "user",
+  "project",
+  "plugin",
+];
+
+const initialState: AgentsManagerState = {
+  selectedIndex: 0,
+  viewMode: "list",
+  pendingEffect: null,
+};
+
+export const AgentsManager: React.FC<AgentsManagerProps> = ({
+  onCancel,
+  agentDefinitions,
+}) => {
+  const [state, dispatch] = useReducer(agentsManagerReducer, initialState);
+  const { stdout } = useStdout();
+
+  // Handle pending effects
+  useEffect(() => {
+    if (!state.pendingEffect) return;
+    const effect = state.pendingEffect;
+    dispatch({ type: "CLEAR_PENDING_EFFECT" });
+    if (effect.type === "CANCEL") {
+      onCancel();
+    }
+  }, [state.pendingEffect, onCancel]);
+
+  // Flatten definitions (grouped by scope) into one navigable list. Headers
+  // and the empty-state line are non-selectable.
+  const entries = useMemo<DisplayEntry[]>(() => {
+    const result: DisplayEntry[] = [];
+    let selectableCount = 0;
+
+    result.push({ kind: "header", label: "AGENTS", selectableIndex: -1 });
+    let definitionCount = 0;
+    for (const scope of SCOPE_ORDER) {
+      const defs = agentDefinitions
+        .filter((d) => d.scope === scope)
+        .sort((a, b) => a.name.localeCompare(b.name));
+      if (defs.length === 0) continue;
+      result.push({
+        kind: "header",
+        label: SCOPE_LABELS[scope],
+        scope,
+        selectableIndex: -1,
+      });
+      for (const def of defs) {
+        result.push({
+          kind: "definition",
+          label: def.name,
+          model: def.model,
+          sub: def.description,
+          scope: def.scope,
+          selectableIndex: selectableCount++,
+          definition: def,
+        });
+        definitionCount++;
+      }
+    }
+    if (definitionCount === 0) {
+      result.push({
+        kind: "empty",
+        label: "No agents available",
+        selectableIndex: -1,
+      });
+    }
+
+    return result;
+  }, [agentDefinitions]);
+
+  const itemCount = entries.filter((e) => e.selectableIndex >= 0).length;
+
+  // Window slice: center the selected item within the visible area, clamping
+  // to the terminal's available rows (reusable pattern from
+  // BackgroundTaskManager).
+  const availableRows = stdout?.rows ?? 24;
+  const maxVisible = Math.max(3, Math.min(15, availableRows - 12));
+  const selectedFlatIndex = entries.findIndex(
+    (e) => e.selectableIndex === state.selectedIndex,
+  );
+  const startIndex = Math.max(
+    0,
+    Math.min(
+      selectedFlatIndex - Math.floor(maxVisible / 2),
+      Math.max(0, entries.length - maxVisible),
+    ),
+  );
+  const visibleEntries = entries.slice(startIndex, startIndex + maxVisible);
+
+  useInput((input, key) => {
+    dispatch({ type: "HANDLE_KEY", input, key, itemCount });
+  });
+
+  const selectedEntry = entries.find(
+    (e) => e.selectableIndex === state.selectedIndex,
+  );
+
+  // Detail view — body renders fully expanded with no height limit, no
+  // clipping and no scrolling (aligned with Claude Code's AgentDetail).
+  if (state.viewMode === "detail" && selectedEntry) {
+    const def = selectedEntry.definition;
+    return (
+      <Box
+        flexDirection="column"
+        borderStyle="single"
+        borderColor="cyan"
+        borderBottom={false}
+        borderLeft={false}
+        borderRight={false}
+        paddingTop={1}
+        gap={1}
+      >
+        <Box>
+          <Text color="cyan" bold>
+            Agent: {selectedEntry.label}
+          </Text>
+        </Box>
+
+        <Box flexDirection="column" gap={1}>
+          {def?.description && (
+            <Box>
+              <Text>
+                <Text color="blue">Description:</Text> {def.description}
+              </Text>
+            </Box>
+          )}
+          <Box>
+            <Text>
+              <Text color="blue">Model:</Text>{" "}
+              {def?.model || "default (not explicitly configured)"}
+            </Text>
+          </Box>
+          <Box>
+            <Text>
+              <Text color="blue">Scope:</Text>{" "}
+              {def ? SCOPE_LABELS[def.scope] : ""}
+            </Text>
+          </Box>
+          {def?.tools && def.tools.length > 0 && (
+            <Box>
+              <Text wrap="wrap">
+                <Text color="blue">Tools:</Text> {def.tools.join(", ")}
+              </Text>
+            </Box>
+          )}
+          {def?.filePath && (
+            <Box>
+              <Text wrap="wrap">
+                <Text color="blue">File:</Text> {def.filePath}
+              </Text>
+            </Box>
+          )}
+        </Box>
+
+        {def && (
+          <Box flexDirection="column" marginTop={1}>
+            <Text color="blue" bold>
+              System Prompt:
+            </Text>
+            <Box marginLeft={2} marginRight={2}>
+              <Markdown>{def.systemPrompt}</Markdown>
+            </Box>
+          </Box>
+        )}
+
+        <Box marginTop={1}>
+          <Text dimColor>Esc or Enter to go back</Text>
+        </Box>
+      </Box>
+    );
+  }
+
+  if (itemCount === 0) {
+    return (
+      <Box
+        flexDirection="column"
+        borderStyle="single"
+        borderColor="cyan"
+        borderBottom={false}
+        borderLeft={false}
+        borderRight={false}
+        paddingTop={1}
+      >
+        <Text color="cyan" bold>
+          Agents
+        </Text>
+        <Text>No agents available</Text>
+        <Text dimColor>Press Escape to close</Text>
+      </Box>
+    );
+  }
+
+  return (
+    <Box
+      flexDirection="column"
+      borderStyle="single"
+      borderColor="cyan"
+      borderBottom={false}
+      borderLeft={false}
+      borderRight={false}
+      paddingTop={1}
+      gap={1}
+    >
+      <Box>
+        <Text color="cyan" bold>
+          Agents
+        </Text>
+      </Box>
+      <Text dimColor>Select an agent to view details</Text>
+
+      <Box flexDirection="column">
+        {visibleEntries.map((entry, index) => {
+          const isSelected = entry.selectableIndex === state.selectedIndex;
+          if (entry.kind === "header") {
+            return (
+              <Text key={`${entry.kind}-${entry.label}-${index}`} dimColor bold>
+                {entry.label}
+              </Text>
+            );
+          }
+          if (entry.kind === "empty") {
+            return (
+              <Text key={`empty-${index}`} dimColor>
+                {entry.label}
+              </Text>
+            );
+          }
+          return (
+            <Text
+              key={`${entry.kind}-${entry.selectableIndex}`}
+              color={isSelected ? "black" : "white"}
+              backgroundColor={isSelected ? "cyan" : undefined}
+              wrap="truncate-end"
+            >
+              {isSelected ? "▶ " : "  "}
+              {entry.selectableIndex + 1}. {entry.label}
+              {entry.model ? (
+                <Text color={isSelected ? "black" : "gray"}>
+                  {" "}
+                  · {entry.model}
+                </Text>
+              ) : null}
+              {entry.sub ? ` · ${entry.sub}` : ""}
+            </Text>
+          );
+        })}
+      </Box>
+
+      <Box marginTop={1}>
+        <Text dimColor>
+          ↑/↓ to select · Enter to view details · Esc to close
+        </Text>
+      </Box>
+    </Box>
+  );
+};

package/src/components/ConfirmationSelector.tsx

@@ -1,4 +1,4 @@
-import React, { useEffect, useReducer } from "react";
+import React, { useEffect, useReducer, useRef } from "react";
 import { Box, Text, useInput } from "ink";
 import type {
   PermissionDecision,
@@ -13,6 +13,7 @@ import {
 } from "wave-agent-sdk";
 import { confirmationReducer } from "../reducers/confirmationReducer.js";
 import { questionReducer } from "../reducers/questionReducer.js";
+import { createBracketedPasteDetector } from "../utils/bracketedPaste.js";
 
 const getHeaderColor = (header: string) => {
   const colors = ["red", "green", "blue", "magenta", "cyan"] as const;
@@ -103,23 +104,37 @@ export const ConfirmationSelector: React.FC<ConfirmationSelectorProps> = ({
     return "Yes, and auto-accept edits";
   };
 
+  const pasteDetectorRef = useRef(createBracketedPasteDetector());
+
   useInput((input, key) => {
     if (key.escape) {
       onCancel();
       return;
     }
 
+    const result = pasteDetectorRef.current.process(input);
+    if (result.kind === "consume") {
+      // Content of an in-flight bracketed paste: hold it, never submit.
+      return;
+    }
+    let cleanInput: string;
+    if (result.kind === "paste") {
+      cleanInput = (result.leadingInput ?? "") + result.text;
+    } else {
+      cleanInput = result.input;
+    }
+
     if (toolName === ASK_USER_QUESTION_TOOL_NAME) {
       questionDispatch({
         type: "HANDLE_KEY",
-        input,
+        input: cleanInput,
         key,
         questions,
       });
     } else {
       dispatch({
         type: "HANDLE_KEY",
-        input,
+        input: cleanInput,
         key,
         toolName,
         toolInput,