wave-code 0.11.7 → 0.12.1

package/src/components/ModelSelector.tsx

@@ -0,0 +1,109 @@
+import React, { useState } from "react";
+import { Box, Text, useInput } from "ink";
+
+export interface ModelSelectorProps {
+  onCancel: () => void;
+  currentModel: string;
+  configuredModels: string[];
+  onSelectModel: (model: string) => void;
+}
+
+export const ModelSelector: React.FC<ModelSelectorProps> = ({
+  onCancel,
+  currentModel,
+  configuredModels,
+  onSelectModel,
+}) => {
+  const [selectedIndex, setSelectedIndex] = useState(() => {
+    const index = configuredModels.indexOf(currentModel);
+    return index !== -1 ? index : 0;
+  });
+
+  useInput((_input, key) => {
+    if (key.return) {
+      if (configuredModels.length > 0) {
+        onSelectModel(configuredModels[selectedIndex]);
+      }
+      onCancel();
+      return;
+    }
+
+    if (key.escape) {
+      onCancel();
+      return;
+    }
+
+    if (key.upArrow) {
+      setSelectedIndex((prev) => Math.max(0, prev - 1));
+      return;
+    }
+
+    if (key.downArrow) {
+      setSelectedIndex((prev) =>
+        Math.min(configuredModels.length - 1, prev + 1),
+      );
+      return;
+    }
+  });
+
+  if (configuredModels.length === 0) {
+    return (
+      <Box
+        flexDirection="column"
+        borderStyle="single"
+        borderColor="cyan"
+        borderBottom={false}
+        borderLeft={false}
+        borderRight={false}
+        paddingTop={1}
+      >
+        <Text color="cyan" bold>
+          Select AI Model
+        </Text>
+        <Text>No models configured</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>
+          Select AI Model
+        </Text>
+      </Box>
+      <Text dimColor>Select a model to use for the current session</Text>
+
+      {configuredModels.map((model, index) => (
+        <Box key={model}>
+          <Text
+            color={index === selectedIndex ? "black" : "white"}
+            backgroundColor={index === selectedIndex ? "cyan" : undefined}
+          >
+            {index === selectedIndex ? "▶ " : "  "}
+            {model}
+            {model === currentModel ? (
+              <Text color="green"> (current)</Text>
+            ) : (
+              ""
+            )}
+          </Text>
+        </Box>
+      ))}
+
+      <Box marginTop={1}>
+        <Text dimColor>↑/↓ to select · Enter to confirm · Esc to cancel</Text>
+      </Box>
+    </Box>
+  );
+};

package/src/components/SlashDisplay.tsx

@@ -0,0 +1,70 @@
+import React from "react";
+import { Box, Text } from "ink";
+import type { SlashBlock } from "wave-agent-sdk";
+
+interface SlashDisplayProps {
+  block: SlashBlock;
+  isExpanded?: boolean;
+}
+
+export const SlashDisplay: React.FC<SlashDisplayProps> = ({
+  block,
+  isExpanded = false,
+}) => {
+  const { command, args, stage, shortResult, error, result } = block;
+
+  const getStatusColor = () => {
+    switch (stage) {
+      case "running":
+        return "yellow";
+      case "success":
+        return "green";
+      case "error":
+        return "red";
+      case "aborted":
+        return "gray";
+      default:
+        return "white";
+    }
+  };
+
+  return (
+    <Box flexDirection="column">
+      <Box>
+        <Text color={getStatusColor()}>/ </Text>
+        <Text color="white">{command}</Text>
+        {args && <Text color="gray"> {args}</Text>}
+      </Box>
+
+      {/* Display shortResult in collapsed state */}
+      {!isExpanded && shortResult && !error && (
+        <Box paddingLeft={2} flexDirection="column">
+          {shortResult.split("\n").map((line, index) => (
+            <Text key={index} color="gray">
+              {line}
+            </Text>
+          ))}
+        </Box>
+      )}
+
+      {/* Display complete result in expanded state */}
+      {isExpanded && result && (
+        <Box paddingLeft={2} flexDirection="column">
+          <Text color="cyan" bold>
+            Result:
+          </Text>
+          <Text color="white">{result}</Text>
+        </Box>
+      )}
+
+      {/* Error information always displayed */}
+      {error && (
+        <Box paddingLeft={2}>
+          <Text color="red">
+            Error: {typeof error === "string" ? error : String(error)}
+          </Text>
+        </Box>
+      )}
+    </Box>
+  );
+};

package/src/components/StatusCommand.tsx

@@ -86,6 +86,13 @@ export const StatusCommand: React.FC<StatusCommandProps> = ({ onCancel }) => {
         <Text color="white">{modelConfig.model}</Text>
       </Box>
 
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Fast model:</Text>
+        </Box>
+        <Text color="white">{modelConfig.fastModel}</Text>
+      </Box>
+
       <Box marginTop={1}>
         <Text dimColor>Esc to cancel</Text>
       </Box>

package/src/components/ToolDisplay.tsx

@@ -81,7 +81,7 @@ export const ToolDisplay: React.FC<ToolDisplayProps> = ({
           flexDirection="column"
         >
           {shortResult.split("\n").map((line, index) => (
-            <Text key={index} color="gray">
+            <Text key={index} color="gray" wrap="truncate-end">
               {line}
             </Text>
           ))}

package/src/constants/commands.ts

@@ -44,4 +44,10 @@ export const AVAILABLE_COMMANDS: SlashCommand[] = [
     description: "Ask a side question without tool use",
     handler: () => {}, // Handler here won't be used, actual processing is in the hook
   },
+  {
+    id: "model",
+    name: "model",
+    description: "Switch between configured AI models",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
 ];

package/src/contexts/useChat.tsx

@@ -54,6 +54,11 @@ export interface ChatContextType {
   askBtw: (question: string) => Promise<string>;
   abortMessage: () => void;
   latestTotalTokens: number;
+  // Model functionality
+  currentModel: string;
+  configuredModels: string[];
+  getConfiguredModels: () => string[];
+  setModel: (model: string) => void;
   // MCP functionality
   mcpServers: McpServerStatus[];
   connectMcpServer: (serverName: string) => Promise<boolean>;
@@ -72,9 +77,6 @@ export interface ChatContextType {
   // Slash Command functionality
   slashCommands: SlashCommand[];
   hasSlashCommand: (commandId: string) => boolean;
-  // Subagent messages
-  subagentMessages: Record<string, Message[]>;
-  subagentLatestTokens: Record<string, number>;
   // Permission functionality
   permissionMode: PermissionMode;
   setPermissionMode: (mode: PermissionMode) => void;
@@ -176,6 +178,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   const [sessionId, setSessionId] = useState("");
   const [isCommandRunning, setIsCommandRunning] = useState(false);
   const [isCompressing, setIsCompressing] = useState(false);
+  const [currentModel, setCurrentModelState] = useState("");
+  const [configuredModels, setConfiguredModels] = useState<string[]>([]);
 
   // MCP State
   const [mcpServers, setMcpServers] = useState<McpServerStatus[]>([]);
@@ -188,14 +192,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // Command state
   const [slashCommands, setSlashCommands] = useState<SlashCommand[]>([]);
 
-  // Subagent messages state
-  const [subagentMessages, setSubagentMessages] = useState<
-    Record<string, Message[]>
-  >({});
-  const [subagentLatestTokens, setSubagentLatestTokens] = useState<
-    Record<string, number>
-  >({});
-
   // Permission state
   const [permissionMode, setPermissionModeState] = useState<PermissionMode>(
     initialPermissionMode ||
@@ -269,20 +265,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     !!bypassPermissions || initialPermissionMode === "bypassPermissions";
 
   const agentRef = useRef<Agent | null>(null);
-  const taskUpdateTimerRef = useRef<NodeJS.Timeout | null>(null);
-
-  const debouncedSetTasks = useCallback(
-    (newTasks: Task[]) => {
-      if (taskUpdateTimerRef.current) {
-        clearTimeout(taskUpdateTimerRef.current);
-      }
-      taskUpdateTimerRef.current = setTimeout(() => {
-        setTasks([...newTasks]);
-        taskUpdateTimerRef.current = null;
-      }, 100);
-    },
-    [setTasks],
-  );
 
   // Permission confirmation methods with queue support
   const showConfirmation = useCallback(
@@ -336,27 +318,17 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           setBackgroundTasks([...tasks]);
         },
         onTasksChange: (tasks) => {
-          debouncedSetTasks(tasks);
-        },
-        onSubagentMessagesChange: (subagentId: string, messages: Message[]) => {
-          logger.debug("onSubagentMessagesChange", subagentId, messages.length);
-          setSubagentMessages((prev) => ({
-            ...prev,
-            [subagentId]: [...messages],
-          }));
-        },
-        onSubagentLatestTotalTokensChange: (
-          subagentId: string,
-          tokens: number,
-        ) => {
-          setSubagentLatestTokens((prev) => ({
-            ...prev,
-            [subagentId]: tokens,
-          }));
+          setTasks([...tasks]);
         },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
         },
+        onModelChange: (model) => {
+          setCurrentModelState(model);
+        },
+        onConfiguredModelsChange: (models) => {
+          setConfiguredModels(models);
+        },
       };
 
       try {
@@ -414,6 +386,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         setIsCompressing(agent.isCompressing);
         setPermissionModeState(agent.getPermissionMode());
         setWorkingDirectory(agent.workingDirectory);
+        setCurrentModelState(agent.getModelConfig().model);
+        setConfiguredModels(agent.getConfiguredModels());
 
         // Get initial MCP servers state
         const mcpServers = agent.getMcpServers?.() || [];
@@ -440,17 +414,12 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workdir,
     worktreeSession,
     model,
-    debouncedSetTasks,
     initialPermissionMode,
   ]);
 
   // Cleanup on unmount
   useEffect(() => {
     return () => {
-      if (taskUpdateTimerRef.current) {
-        clearTimeout(taskUpdateTimerRef.current);
-      }
-
       if (agentRef.current) {
         try {
           // Display usage summary before cleanup
@@ -672,6 +641,20 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     return agentRef.current.getModelConfig();
   }, []);
 
+  const getConfiguredModels = useCallback(() => {
+    if (!agentRef.current) {
+      return [];
+    }
+    return agentRef.current.getConfiguredModels();
+  }, []);
+
+  const setModel = useCallback((model: string) => {
+    if (agentRef.current) {
+      agentRef.current.setModel(model);
+      setCurrentModelState(model);
+    }
+  }, []);
+
   // Listen for Ctrl+O hotkey to toggle collapse/expand state and ESC to cancel confirmation
   useInput((input, key) => {
     if (key.ctrl && input === "o") {
@@ -729,6 +712,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     askBtw,
     abortMessage,
     latestTotalTokens,
+    currentModel,
+    configuredModels,
+    getConfiguredModels,
+    setModel,
     isCompressing,
     mcpServers,
     connectMcpServer,
@@ -739,8 +726,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     stopBackgroundTask,
     slashCommands,
     hasSlashCommand,
-    subagentMessages,
-    subagentLatestTokens,
     permissionMode,
     setPermissionMode,
     allowBypassInCycle,

package/src/hooks/useInputManager.ts

@@ -149,6 +149,10 @@ export const useInputManager = (
     callbacksRef.current.onPluginManagerStateChange?.(state.showPluginManager);
   }, [state.showPluginManager]);
 
+  useEffect(() => {
+    callbacksRef.current.onModelSelectorStateChange?.(state.showModelSelector);
+  }, [state.showModelSelector]);
+
   useEffect(() => {
     callbacksRef.current.onImagesStateChange?.(state.attachedImages);
   }, [state.attachedImages]);
@@ -343,6 +347,10 @@ export const useInputManager = (
     dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: show });
   }, []);
 
+  const setShowModelSelector = useCallback((show: boolean) => {
+    dispatch({ type: "SET_SHOW_MODEL_SELECTOR", payload: show });
+  }, []);
+
   const setPermissionMode = useCallback((mode: PermissionMode) => {
     dispatch({ type: "SET_PERMISSION_MODE", payload: mode });
     callbacksRef.current.onPermissionModeChange?.(mode);
@@ -452,6 +460,7 @@ export const useInputManager = (
     showHelp: state.showHelp,
     showStatusCommand: state.showStatusCommand,
     showPluginManager: state.showPluginManager,
+    showModelSelector: state.showModelSelector,
     permissionMode: state.permissionMode,
     attachedImages: state.attachedImages,
     btwState: state.btwState,
@@ -493,6 +502,7 @@ export const useInputManager = (
     setShowHelp,
     setShowStatusCommand,
     setShowPluginManager,
+    setShowModelSelector,
     setPermissionMode,
     setAllowBypassInCycle,
     setBtwState,

package/src/index.ts

@@ -1,6 +1,7 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import { startCli } from "./cli.js";
+import { logger } from "./utils/logger.js";
 import { Scope, generateRandomName, type PermissionMode } from "wave-agent-sdk";
 import { createWorktree, type WorktreeSession } from "./utils/worktree.js";
 import path from "path";
@@ -14,6 +15,18 @@ const version = packageJson.version;
 
 // Export main function for external use
 export async function main() {
+  // Start memory monitoring in debug mode
+  if (process.env.LOG_LEVEL === "DEBUG") {
+    setInterval(() => {
+      const usage = process.memoryUsage();
+      logger.debug(
+        `[Memory] RSS: ${Math.round(usage.rss / 1024 / 1024)}MB, ` +
+          `Heap: ${Math.round(usage.heapUsed / 1024 / 1024)}/${Math.round(usage.heapTotal / 1024 / 1024)}MB, ` +
+          `External: ${Math.round(usage.external / 1024 / 1024)}MB`,
+      );
+    }, 10000).unref();
+  }
+
   try {
     const originalCwd = process.cwd();
     const argv = await yargs(hideBin(process.argv))

package/src/managers/inputHandlers.ts

@@ -370,6 +370,8 @@ export const handleCommandSelect = (
           dispatch({ type: "SET_SHOW_STATUS_COMMAND", payload: true });
         } else if (command === "plugin") {
           dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
+        } else if (command === "model") {
+          dispatch({ type: "SET_SHOW_MODEL_SELECTOR", payload: true });
         } else if (command === "btw") {
           dispatch({
             type: "SET_BTW_STATE",
@@ -754,6 +756,7 @@ export const handleInput = async (
         state.showHelp ||
         state.showStatusCommand ||
         state.showPluginManager ||
+        state.showModelSelector ||
         state.btwState.isActive
       )
     ) {
@@ -782,6 +785,7 @@ export const handleInput = async (
     state.showHelp ||
     state.showStatusCommand ||
     state.showPluginManager ||
+    state.showModelSelector ||
     state.btwState.isActive
   ) {
     if (
@@ -791,6 +795,7 @@ export const handleInput = async (
       state.showHelp ||
       state.showStatusCommand ||
       state.showPluginManager ||
+      state.showModelSelector ||
       state.btwState.isActive
     ) {
       return true;

package/src/managers/inputReducer.ts

@@ -40,6 +40,7 @@ export interface InputManagerCallbacks {
   onHelpStateChange?: (show: boolean) => void;
   onStatusCommandStateChange?: (show: boolean) => void;
   onPluginManagerStateChange?: (show: boolean) => void;
+  onModelSelectorStateChange?: (show: boolean) => void;
   onImagesStateChange?: (images: AttachedImage[]) => void;
   onSendMessage?: (
     content: string,
@@ -84,6 +85,7 @@ export interface InputState {
   showHelp: boolean;
   showStatusCommand: boolean;
   showPluginManager: boolean;
+  showModelSelector: boolean;
   permissionMode: PermissionMode;
   allowBypassInCycle: boolean;
   selectorJustUsed: boolean;
@@ -120,6 +122,7 @@ export const initialState: InputState = {
   showHelp: false,
   showStatusCommand: false,
   showPluginManager: false,
+  showModelSelector: false,
   permissionMode: "default",
   allowBypassInCycle: false,
   selectorJustUsed: false,
@@ -163,6 +166,7 @@ export type InputAction =
   | { type: "SET_SHOW_HELP"; payload: boolean }
   | { type: "SET_SHOW_STATUS_COMMAND"; payload: boolean }
   | { type: "SET_SHOW_PLUGIN_MANAGER"; payload: boolean }
+  | { type: "SET_SHOW_MODEL_SELECTOR"; payload: boolean }
   | { type: "SET_PERMISSION_MODE"; payload: PermissionMode }
   | { type: "SET_ALLOW_BYPASS_IN_CYCLE"; payload: boolean }
   | { type: "SET_SELECTOR_JUST_USED"; payload: boolean }
@@ -358,6 +362,12 @@ export function inputReducer(
         showPluginManager: action.payload,
         selectorJustUsed: !action.payload ? true : state.selectorJustUsed,
       };
+    case "SET_SHOW_MODEL_SELECTOR":
+      return {
+        ...state,
+        showModelSelector: action.payload,
+        selectorJustUsed: !action.payload ? true : state.selectorJustUsed,
+      };
     case "SET_PERMISSION_MODE":
       return { ...state, permissionMode: action.payload };
     case "SET_ALLOW_BYPASS_IN_CYCLE":

package/src/utils/highlightUtils.ts

@@ -49,14 +49,6 @@ function nodeToAnsi(node: Node): string {
 
     for (const className of classes) {
       if (theme[className]) {
-        // If content has newlines, split it and apply style to each line
-        // to ensure ANSI codes are correctly applied when splitting the final string by lines.
-        if (content.includes("\n")) {
-          return content
-            .split("\n")
-            .map((line) => theme[className](line))
-            .join("\n");
-        }
         return theme[className](content);
       }
     }

package/src/utils/logger.ts

@@ -124,21 +124,23 @@ const formatArg = (arg: unknown): string => {
   if (arg === null) return "null";
   if (arg === undefined) return "undefined";
 
+  let result: string;
   if (arg instanceof Error) {
     // Special handling for Error objects, display stack or message
-    return arg.stack || arg.message || String(arg);
-  }
-
-  if (typeof arg === "object") {
+    result = arg.stack || arg.message || String(arg);
+  } else if (typeof arg === "object") {
     try {
-      return JSON.stringify(arg, null, 2);
+      result = JSON.stringify(arg);
     } catch {
       // If JSON.stringify fails, fallback to String()
-      return String(arg);
+      result = String(arg);
     }
+  } else {
+    result = String(arg);
   }
 
-  return String(arg);
+  // Ensure no newlines in the result to keep log entries single-line
+  return result.replace(/[\n\r]+/g, " ");
 };
 
 /**

package/src/utils/toolParameterTransforms.ts

@@ -3,7 +3,11 @@
  * Forces type judgment based on tool name using type assertions
  */
 
-import { type Change, type EditToolParameters } from "wave-agent-sdk";
+import {
+  type Change,
+  type WriteToolParameters,
+  type EditToolParameters,
+} from "wave-agent-sdk";
 import { logger } from "./logger.js";
 
 /**
@@ -22,6 +26,20 @@ function parseToolParameters(parameters: string): unknown {
   }
 }
 
+/**
+ * Transform Write tool parameters to changes
+ */
+export function transformWriteParameters(
+  parameters: WriteToolParameters,
+): Change[] {
+  return [
+    {
+      oldContent: "", // No previous content for write operations
+      newContent: parameters.content,
+    },
+  ];
+}
+
 /**
  * Transform Edit tool parameters to changes
  */
@@ -54,6 +72,10 @@ export function transformToolBlockToChanges(
 
     let changes: Change[] = [];
     switch (toolName) {
+      case "Write":
+        changes = transformWriteParameters(parsedParams as WriteToolParameters);
+        break;
+
       case "Edit":
         changes = transformEditParameters(parsedParams as EditToolParameters);
         break;

package/src/utils/worktree.ts

@@ -13,8 +13,6 @@ export interface WorktreeSession {
   isNew: boolean;
 }
 
-export const WORKTREE_DIR = ".wave/worktrees";
-
 /**
  * Create a new git worktree
  * @param name Worktree name
@@ -23,7 +21,13 @@ export const WORKTREE_DIR = ".wave/worktrees";
  */
 export function createWorktree(name: string, cwd: string): WorktreeSession {
   const repoRoot = getGitMainRepoRoot(cwd);
-  const worktreePath = path.join(repoRoot, WORKTREE_DIR, name);
+  const projectName = path.basename(repoRoot);
+  const worktreePath = path.join(
+    repoRoot,
+    "..",
+    `${projectName}.worktrees`,
+    name,
+  );
   const branchName = `worktree-${name}`;
   const baseBranch = getDefaultRemoteBranch(cwd);