@meowlynxsea/koi 0.1.5 → 0.1.6

package/dist/main.js

@@ -415526,7 +415526,7 @@ var InputBox = import_react20.forwardRef(function InputBox2({
         }
       }
     }
-    if (event.name === "tab" && event.shift && onModeSwitch) {
+    if (event.name === "tab" && event.shift && onModeSwitch && !isBusy) {
       event.preventDefault();
       event.stopPropagation();
       onModeSwitch();
@@ -415556,6 +415556,7 @@ var InputBox = import_react20.forwardRef(function InputBox2({
     { name: "return", shift: true, action: "newline" }
   ], []);
   const [wavePhase, setWavePhase] = import_react20.useState(0);
+  const [shimmerIndex, setShimmerIndex] = import_react20.useState(-1);
   import_react20.useEffect(() => {
     if (!isBusy)
       return;
@@ -415564,6 +415565,24 @@ var InputBox = import_react20.forwardRef(function InputBox2({
     }, 150);
     return () => clearInterval(interval);
   }, [isBusy]);
+  import_react20.useEffect(() => {
+    if (!isBusy) {
+      setShimmerIndex(-1);
+      return;
+    }
+    const modeText = MODE_PREFIX[mode];
+    const textLength = modeText.length;
+    let currentIdx = -1;
+    const interval = setInterval(() => {
+      currentIdx = (currentIdx + 1) % (textLength + 4);
+      if (currentIdx >= textLength) {
+        setShimmerIndex(-1);
+      } else {
+        setShimmerIndex(currentIdx);
+      }
+    }, 80);
+    return () => clearInterval(interval);
+  }, [isBusy, mode]);
   const getInkWaveChars = import_react20.useCallback(() => {
     const totalWidth = width ?? 80;
     const phaseData = INK_WAVE_PHASES[wavePhase];
@@ -415637,13 +415656,37 @@ var InputBox = import_react20.forwardRef(function InputBox2({
         height: 3,
         children: [
           /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+            flexDirection: "row",
             marginRight: 1,
             flexShrink: 0,
-            children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
-              fg: MODE_COLOR[mode],
-              attributes: createTextAttributes({ bold: true }),
-              children: MODE_PREFIX[mode]
-            }, undefined, false, undefined, this)
+            children: (() => {
+              const modeText = MODE_PREFIX[mode];
+              const highlightColor = mode === "build" ? "#86efac" : mode === "ask" ? "#fde68a" : "#93c5fd";
+              const nearColor = mode === "build" ? "#bbf7d0" : mode === "ask" ? "#fef08a" : "#bfdbfe";
+              return modeText.split("").map((char, i) => {
+                const isHighlighted = isBusy && shimmerIndex === i;
+                const isNearHighlight = isBusy && shimmerIndex !== -1 && (shimmerIndex === i - 1 || shimmerIndex === i + 1);
+                if (isHighlighted) {
+                  return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                    fg: highlightColor,
+                    attributes: createTextAttributes({ bold: true }),
+                    children: char
+                  }, i, false, undefined, this);
+                }
+                if (isNearHighlight) {
+                  return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                    fg: nearColor,
+                    attributes: createTextAttributes({ bold: true }),
+                    children: char
+                  }, i, false, undefined, this);
+                }
+                return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                  fg: MODE_COLOR[mode],
+                  attributes: createTextAttributes({ bold: true }),
+                  children: char
+                }, i, false, undefined, this);
+              });
+            })()
           }, undefined, false, undefined, this),
           /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
             flexGrow: 1,
@@ -484172,6 +484215,124 @@ var forkManager = new ForkManager;
 // src/agent/session-store.ts
 init_mcp();
 init_mode();
+
+// src/agent/tool-output-guard.ts
+var TOOL_LIMITS = {
+  bash: { maxLines: 3000, maxBytes: 100 * 1024, truncateFrom: "tail" },
+  read: { maxLines: 2000, maxBytes: 100 * 1024, truncateFrom: "head" },
+  grep: { maxLines: 500, maxBytes: 50 * 1024, truncateFrom: "tail" },
+  glob: { maxLines: 500, maxBytes: 20 * 1024, truncateFrom: "tail" },
+  ls: { maxLines: 500, maxBytes: 20 * 1024, truncateFrom: "tail" },
+  write: { maxLines: 0, maxBytes: 0, truncateFrom: "tail" },
+  mcp: { maxLines: 1000, maxBytes: 100 * 1024, truncateFrom: "tail" }
+};
+var DEFAULT_LIMITS = {
+  maxLines: DEFAULT_MAX_LINES,
+  maxBytes: DEFAULT_MAX_BYTES,
+  truncateFrom: "tail"
+};
+function extractToolName(toolName) {
+  if (toolName.startsWith("mcp__")) {
+    const parts = toolName.split("__");
+    if (parts.length >= 3) {
+      return "mcp";
+    }
+  }
+  return toolName.split(":").pop() ?? toolName;
+}
+function truncateTextBlock(text, limits) {
+  if (limits.maxLines === 0 && limits.maxBytes === 0) {
+    return { text, wasTruncated: false, truncatedBy: null };
+  }
+  const options4 = {
+    maxLines: limits.maxLines || DEFAULT_MAX_LINES,
+    maxBytes: limits.maxBytes || DEFAULT_MAX_BYTES
+  };
+  const result = limits.truncateFrom === "head" ? truncateHead(text, options4) : truncateTail(text, options4);
+  return {
+    text: result.content,
+    wasTruncated: result.truncated,
+    truncatedBy: result.truncatedBy
+  };
+}
+function truncateToolContent(content, limits) {
+  const textBlocks = [];
+  let hasTruncation = false;
+  let lastTruncatedBy = null;
+  for (const item of content) {
+    if (item.type === "text") {
+      const result = truncateTextBlock(item.text ?? "", limits);
+      textBlocks.push(result);
+      if (result.wasTruncated) {
+        hasTruncation = true;
+        lastTruncatedBy = result.truncatedBy;
+      }
+    } else if (item.type === "image") {
+      textBlocks.push({ text: `[Image: ${item.data?.length ?? 0} bytes]`, wasTruncated: false, truncatedBy: null });
+    }
+  }
+  if (!hasTruncation) {
+    return { content, wasTruncated: false };
+  }
+  const truncatedContent = [];
+  for (const block2 of textBlocks) {
+    if (block2.text) {
+      truncatedContent.push({ type: "text", text: block2.text });
+    }
+  }
+  return {
+    content: truncatedContent,
+    wasTruncated: true,
+    truncationInfo: {
+      truncatedBy: lastTruncatedBy ?? "unknown",
+      totalLines: 0,
+      outputLines: 0,
+      totalBytes: 0,
+      outputBytes: 0
+    }
+  };
+}
+function createToolOutputGuard(customLimits) {
+  const limits = { ...TOOL_LIMITS, ...customLimits };
+  return async (context) => {
+    const { result, toolCall } = context;
+    const toolName = extractToolName(toolCall.name);
+    const toolLimits = limits[toolName] ?? DEFAULT_LIMITS;
+    if (!toolLimits || toolLimits.maxLines === 0 && toolLimits.maxBytes === 0) {
+      return;
+    }
+    const { content, wasTruncated } = truncateToolContent(result.content, toolLimits);
+    if (!wasTruncated) {
+      return;
+    }
+    const truncateNote = buildTruncationNote(toolName);
+    const finalContent = [...content];
+    const lastTextBlock = [...finalContent].reverse().find((c2) => c2.type === "text");
+    if (lastTextBlock) {
+      const idx = finalContent.findIndex((c2) => c2.type === "text" && c2.text === lastTextBlock.text);
+      if (idx >= 0) {
+        finalContent[idx] = {
+          type: "text",
+          text: `${lastTextBlock.text.trimEnd()}
+
+${truncateNote}`
+        };
+      }
+    } else {
+      finalContent.push({ type: "text", text: truncateNote });
+    }
+    return {
+      content: finalContent
+    };
+  };
+}
+function buildTruncationNote(toolName) {
+  const toolLabel = toolName === "mcp" ? "MCP tool" : `"${toolName}"`;
+  return `[Output truncated to prevent context overflow. Set a higher limit or use filters to see more of the ${toolLabel} output.]`;
+}
+var MCP_TOOL_LIMITS = TOOL_LIMITS["mcp"];
+
+// src/agent/session-store.ts
 var CONFIG_DIR3 = path25.join(os16.homedir(), ".config", "koi");
 var KOI_SESSIONS_DIR2 = path25.join(CONFIG_DIR3, "sessions");
 var PI_AGENT_DIR2 = path25.join(CONFIG_DIR3, "pi");
@@ -484420,7 +484581,7 @@ async function createAgentSessionWithConfig(sessionManager, config4) {
   try {
     fs18.appendFileSync(logPath, logLine);
   } catch {}
-  return createAgentSession({
+  const result = await createAgentSession({
     cwd: process.cwd(),
     agentDir: PI_AGENT_DIR2,
     authStorage: config4.authStorage,
@@ -484432,6 +484593,9 @@ async function createAgentSessionWithConfig(sessionManager, config4) {
     sessionManager,
     resourceLoader
   });
+  const afterToolCall = createToolOutputGuard();
+  result.session.agent.afterToolCall = afterToolCall;
+  return result;
 }
 async function listSessions() {
   try {
@@ -486678,7 +486842,7 @@ function CommandPanel({ isActive, onClose, commands }) {
     }
   }, [isActive]);
   const query2 = filterText;
-  const { flatItems, cmdCount } = import_react29.useMemo(() => {
+  const { flatItems } = import_react29.useMemo(() => {
     let filtered = commands;
     if (query2) {
       const q3 = query2.toLowerCase();
@@ -486705,7 +486869,7 @@ function CommandPanel({ isActive, onClose, commands }) {
         cmdIdx++;
       }
     }
-    return { flatItems: items3, cmdCount: cmdIdx };
+    return { flatItems: items3 };
   }, [commands, query2]);
   const effectiveScrollOffset = scrollOffsetRef.current;
   useKeyboard((key) => {
@@ -486726,8 +486890,25 @@ function CommandPanel({ isActive, onClose, commands }) {
     if (key.name === "up" || key.name === "down") {
       key.preventDefault();
       key.stopPropagation();
-      const newIndex = key.name === "up" ? Math.max(0, selectedCmdIndex - 1) : Math.min(cmdCount - 1, selectedCmdIndex + 1);
-      const newFlatIndex = flatItems.findIndex((i) => i.type === "command" && i.cmdIndex === newIndex);
+      const enabledIndices = flatItems.filter((i) => i.type === "command" && !i.cmd?.disabled && i.cmdIndex !== undefined).map((i) => i.cmdIndex);
+      if (enabledIndices.length === 0)
+        return;
+      const currentIdx = enabledIndices.indexOf(selectedCmdIndex);
+      let targetIndex;
+      if (key.name === "up") {
+        if (currentIdx <= 0) {
+          targetIndex = enabledIndices[enabledIndices.length - 1];
+        } else {
+          targetIndex = enabledIndices[currentIdx - 1];
+        }
+      } else {
+        if (currentIdx < 0 || currentIdx >= enabledIndices.length - 1) {
+          targetIndex = enabledIndices[0];
+        } else {
+          targetIndex = enabledIndices[currentIdx + 1];
+        }
+      }
+      const newFlatIndex = flatItems.findIndex((i) => i.type === "command" && i.cmdIndex === targetIndex);
       const currentScroll = scrollOffsetRef.current;
       let newScrollOffset = currentScroll;
       if (newFlatIndex !== -1) {
@@ -486737,16 +486918,8 @@ function CommandPanel({ isActive, onClose, commands }) {
           newScrollOffset = newFlatIndex - listHeight + 1;
         }
       }
-      console.log("[CommandPanel] Key press:", {
-        key: key.name,
-        newIndex,
-        newFlatIndex,
-        currentScroll,
-        listHeight,
-        newScrollOffset
-      });
       scrollOffsetRef.current = newScrollOffset;
-      setSelectedCmdIndex(newIndex);
+      setSelectedCmdIndex(targetIndex);
       return;
     }
   });
@@ -486758,7 +486931,7 @@ function CommandPanel({ isActive, onClose, commands }) {
   };
   const handleSubmit = () => {
     const selectedItem = flatItems.find((i) => i.type === "command" && i.cmdIndex === selectedCmdIndex);
-    if (selectedItem?.cmd) {
+    if (selectedItem?.cmd && !selectedItem.cmd.disabled) {
       onClose();
       selectedItem.cmd.action();
     }
@@ -486824,14 +486997,15 @@ function CommandPanel({ isActive, onClose, commands }) {
                   }, undefined, false, undefined, this)
                 }, `h-${item.section}-${flatIndex}`, false, undefined, this);
               }
-              const isSelected = item.cmdIndex === selectedCmdIndex;
+              const isDisabled = item.cmd?.disabled;
+              const isSelected = !isDisabled && item.cmdIndex === selectedCmdIndex;
               return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
                 height: 1,
                 backgroundColor: isSelected ? "#44475a" : undefined,
                 paddingLeft: 2,
                 children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
-                  fg: isSelected ? "#ff79c6" : "#f8f8f2",
-                  children: `${item.cmd.id}  ${item.cmd.label}`
+                  fg: isDisabled ? "#666666" : isSelected ? "#ff79c6" : "#f8f8f2",
+                  children: `${item.cmd.id}  ${isDisabled ? item.cmd.label + " (busy)" : item.cmd.label}`
                 }, undefined, false, undefined, this)
               }, `c-${item.cmd.id}-${flatIndex}`, false, undefined, this);
             }),
@@ -486976,8 +487150,23 @@ function ConnectModal({ isActive, onClose }) {
   const [verifyResult, setVerifyResult] = import_react33.useState(null);
   const [spinnerFrame, setSpinnerFrame] = import_react33.useState(0);
   const [scrollOffset, setScrollOffset] = import_react33.useState(0);
+  const [filterText, setFilterText] = import_react33.useState("");
   const inputRef = import_react33.useRef(null);
+  const searchRef = import_react33.useRef(null);
   const listHeight = Math.min(10, Math.floor(height * 0.35));
+  const query2 = filterText;
+  const filteredProviders = import_react33.useMemo(() => {
+    if (!query2)
+      return providers;
+    const q3 = query2.toLowerCase();
+    return providers.filter((p) => p.toLowerCase().includes(q3));
+  }, [providers, query2]);
+  const handleSearchChange = () => {
+    const text = searchRef.current?.editBuffer.getText() ?? "";
+    setFilterText(text);
+    setSelectedProviderIndex(0);
+    setScrollOffset(0);
+  };
   import_react33.useEffect(() => {
     if (isActive) {
       setStep("provider");
@@ -486987,6 +487176,12 @@ function ConnectModal({ isActive, onClose }) {
       setVerifyResult(null);
       setSpinnerFrame(0);
       setScrollOffset(0);
+      setFilterText("");
+      const ta = searchRef.current;
+      if (ta) {
+        ta.editBuffer.replaceText("");
+        ta.focus();
+      }
     }
   }, [isActive]);
   import_react33.useEffect(() => {
@@ -487053,6 +487248,11 @@ function ConnectModal({ isActive, onClose }) {
       setScrollOffset(selectedProviderIndex - listHeight + 1);
     }
   }, [selectedProviderIndex, listHeight, scrollOffset]);
+  import_react33.useEffect(() => {
+    if (selectedProviderIndex >= filteredProviders.length && filteredProviders.length > 0) {
+      setSelectedProviderIndex(filteredProviders.length - 1);
+    }
+  }, [filteredProviders.length, selectedProviderIndex]);
   const handleSelectProvider = (provider) => {
     setSelectedProvider(provider);
     if (isProviderConfigured(provider)) {
@@ -487094,11 +487294,11 @@ function ConnectModal({ isActive, onClose }) {
         return;
       }
       if (key.name === "down") {
-        setSelectedProviderIndex((prev) => Math.max(0, Math.min(providers.length - 1, prev + 1)));
+        setSelectedProviderIndex((prev) => Math.max(0, Math.min(filteredProviders.length - 1, prev + 1)));
         return;
       }
       if (key.name === "return") {
-        const provider = providers[selectedProviderIndex];
+        const provider = filteredProviders[selectedProviderIndex];
         if (provider) {
           handleSelectProvider(provider);
         }
@@ -487143,7 +487343,7 @@ function ConnectModal({ isActive, onClose }) {
   const existingConfig = selectedProvider ? getProviderConfig(selectedProvider) : undefined;
   if (!isActive)
     return null;
-  const visibleProviders = providers.slice(scrollOffset, scrollOffset + listHeight);
+  const visibleProviders = filteredProviders.slice(scrollOffset, scrollOffset + listHeight);
   return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
     position: "absolute",
     top: 0,
@@ -487169,46 +487369,84 @@ function ConnectModal({ isActive, onClose }) {
               fg: "#ff79c6",
               children: "Select Provider"
             }, undefined, false, undefined, this),
+            /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+              height: 1,
+              marginTop: 1,
+              backgroundColor: "#16213e",
+              paddingX: 1,
+              children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("textarea", {
+                ref: searchRef,
+                initialValue: "",
+                focused: isActive,
+                showCursor: true,
+                height: 1,
+                wrapMode: "none",
+                textColor: "#f8f8f2",
+                backgroundColor: "#16213e",
+                onContentChange: handleSearchChange
+              }, undefined, false, undefined, this)
+            }, undefined, false, undefined, this),
+            /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+              height: 1,
+              marginTop: 1,
+              children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                fg: "#4a4a5a",
+                children: "\u2500".repeat(56)
+              }, undefined, false, undefined, this)
+            }, undefined, false, undefined, this),
             /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
               height: listHeight,
               flexDirection: "column",
               overflow: "hidden",
               marginTop: 1,
-              children: visibleProviders.map((p, i) => {
-                const actualIndex = scrollOffset + i;
-                const configured = isProviderConfigured(p);
-                return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+              children: [
+                filteredProviders.length === 0 && /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
                   height: 1,
-                  backgroundColor: actualIndex === selectedProviderIndex ? "#44475a" : undefined,
-                  paddingLeft: 1,
-                  flexDirection: "row",
-                  onMouseUp: (e) => {
-                    e.stopPropagation();
-                    handleSelectProvider(p);
-                  },
-                  children: [
-                    /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
-                      fg: actualIndex === selectedProviderIndex ? "#ff79c6" : "#f8f8f2",
-                      children: [
-                        configured ? "\u25CF " : "  ",
-                        p
-                      ]
-                    }, undefined, true, undefined, this),
-                    configured && /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
-                      fg: "#00ff99",
-                      marginLeft: 1,
-                      children: "configured"
-                    }, undefined, false, undefined, this)
-                  ]
-                }, p, true, undefined, this);
-              })
-            }, undefined, false, undefined, this),
+                  children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                    fg: "#6c6c7c",
+                    children: [
+                      'No providers match "',
+                      filterText,
+                      '"'
+                    ]
+                  }, undefined, true, undefined, this)
+                }, undefined, false, undefined, this),
+                visibleProviders.map((p, i) => {
+                  const actualIndex = scrollOffset + i;
+                  const configured = isProviderConfigured(p);
+                  return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+                    height: 1,
+                    backgroundColor: actualIndex === selectedProviderIndex ? "#44475a" : undefined,
+                    paddingLeft: 1,
+                    flexDirection: "row",
+                    onMouseUp: (e) => {
+                      e.stopPropagation();
+                      handleSelectProvider(p);
+                    },
+                    children: [
+                      /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                        fg: actualIndex === selectedProviderIndex ? "#ff79c6" : "#f8f8f2",
+                        children: [
+                          configured ? "\u25CF " : "  ",
+                          p
+                        ]
+                      }, undefined, true, undefined, this),
+                      configured && /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                        fg: "#00ff99",
+                        marginLeft: 1,
+                        children: "configured"
+                      }, undefined, false, undefined, this)
+                    ]
+                  }, p, true, undefined, this);
+                })
+              ]
+            }, undefined, true, undefined, this),
             /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
               marginTop: 1,
               children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
                 fg: "#6c6c7c",
                 attributes: createTextAttributes({ dim: true }),
-                children: "\u2191\u2193 Navigate  Enter Select  Esc Cancel"
+                children: "\u2191\u2193 Navigate  Enter Select  Esc Cancel  Type to search"
               }, undefined, false, undefined, this)
             }, undefined, false, undefined, this)
           ]
@@ -487411,7 +487649,9 @@ function ModelModal({
   const configuredProviders = getConfiguredProviders();
   const [selectedModelIndex, setSelectedModelIndex] = import_react35.useState(0);
   const [activeTab, setActiveTab] = import_react35.useState("primary");
+  const [filterText, setFilterText] = import_react35.useState("");
   const scrollOffsetRef = import_react35.useRef(0);
+  const inputRef = import_react35.useRef(null);
   const listHeight = Math.min(12, Math.floor(height * 0.4));
   const primaryModel = getCurrentModel();
   const auxiliaryModel2 = getAuxiliaryModel();
@@ -487419,33 +487659,73 @@ function ModelModal({
     if (isActive) {
       setSelectedModelIndex(0);
       setActiveTab("primary");
+      setFilterText("");
       scrollOffsetRef.current = 0;
+      const ta = inputRef.current;
+      if (ta) {
+        ta.editBuffer.replaceText("");
+        ta.focus();
+      }
     }
   }, [isActive]);
+  const query2 = filterText;
   const { flatItems, modelCount } = import_react35.useMemo(() => {
-    const items3 = [];
-    let mIdx = 0;
+    const allModels = [];
     for (const provider of configuredProviders) {
-      items3.push({ type: "header", provider });
       const models2 = getProviderModels(provider);
       for (const model of models2) {
-        items3.push({
-          type: "model",
+        allModels.push({
           provider,
           modelId: model.id,
-          modelName: model.name || model.id,
+          modelName: model.name || model.id
+        });
+      }
+    }
+    let filteredModels = allModels;
+    if (query2) {
+      const q3 = query2.toLowerCase();
+      filteredModels = allModels.map((m2) => {
+        const providerMatch = m2.provider.toLowerCase().includes(q3);
+        const modelIdMatch = m2.modelId.toLowerCase().includes(q3);
+        const modelNameMatch = m2.modelName.toLowerCase().includes(q3);
+        const priority = !providerMatch && !modelIdMatch && !modelNameMatch ? 3 : providerMatch ? 0 : modelIdMatch ? 1 : 2;
+        return { model: m2, priority };
+      }).filter((item) => item.priority < 3).sort((a, b2) => a.priority - b2.priority).map((item) => item.model);
+    }
+    const grouped = new Map;
+    for (const m2 of filteredModels) {
+      if (!grouped.has(m2.provider))
+        grouped.set(m2.provider, []);
+      grouped.get(m2.provider).push(m2);
+    }
+    const items3 = [];
+    let mIdx = 0;
+    for (const [provider, models2] of grouped) {
+      items3.push({ type: "header", provider });
+      for (const m2 of models2) {
+        items3.push({
+          type: "model",
+          provider: m2.provider,
+          modelId: m2.modelId,
+          modelName: m2.modelName,
           modelIndex: mIdx
         });
         mIdx++;
       }
     }
     return { flatItems: items3, modelCount: mIdx };
-  }, [configuredProviders]);
+  }, [configuredProviders, query2]);
   import_react35.useEffect(() => {
     if (selectedModelIndex >= modelCount && modelCount > 0) {
       setSelectedModelIndex(modelCount - 1);
     }
   }, [modelCount, selectedModelIndex]);
+  const handleContentChange = () => {
+    const text = inputRef.current?.editBuffer.getText() ?? "";
+    setFilterText(text);
+    setSelectedModelIndex(0);
+    scrollOffsetRef.current = 0;
+  };
   const effectiveScrollOffset = scrollOffsetRef.current;
   useKeyboard((key) => {
     if (!isActive)
@@ -487574,6 +487854,31 @@ function ModelModal({
             }, undefined, true, undefined, this)
           ]
         }, undefined, true, undefined, this),
+        /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+          height: 1,
+          marginTop: 1,
+          backgroundColor: "#16213e",
+          paddingX: 1,
+          children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("textarea", {
+            ref: inputRef,
+            initialValue: "",
+            focused: isActive,
+            showCursor: true,
+            height: 1,
+            wrapMode: "none",
+            textColor: "#f8f8f2",
+            backgroundColor: "#16213e",
+            onContentChange: handleContentChange
+          }, undefined, false, undefined, this)
+        }, undefined, false, undefined, this),
+        /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+          height: 1,
+          marginTop: 1,
+          children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+            fg: "#4a4a5a",
+            children: "\u2500".repeat(56)
+          }, undefined, false, undefined, this)
+        }, undefined, false, undefined, this),
         /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
           height: listHeight,
           flexDirection: "column",
@@ -487587,6 +487892,17 @@ function ModelModal({
                 children: "No providers configured. Use /connect to add one."
               }, undefined, false, undefined, this)
             }, undefined, false, undefined, this),
+            configuredProviders.length > 0 && modelCount === 0 && /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
+              height: 1,
+              children: /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
+                fg: "#6c6c7c",
+                children: [
+                  'No models match "',
+                  filterText,
+                  '"'
+                ]
+              }, undefined, true, undefined, this)
+            }, undefined, false, undefined, this),
             visibleItems.map((item, idx) => {
               const flatIndex = effectiveScrollOffset + idx;
               if (item.type === "header") {
@@ -487632,7 +487948,7 @@ function ModelModal({
             /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("text", {
               fg: "#6c6c7c",
               attributes: createTextAttributes({ dim: true }),
-              children: "Tab Switch"
+              children: "Type to search  Tab Switch"
             }, undefined, false, undefined, this)
           ]
         }, undefined, true, undefined, this)
@@ -489785,7 +490101,8 @@ function App({ onExit }) {
       return;
     }
     if (text.trim() === "/exit" || text.trim() === "/quit") {
-      setShowExitModal(true);
+      saveCurrentState();
+      onExit();
       return;
     }
     const skillInvocation = detectSkillInvocation(text);
@@ -489897,27 +490214,33 @@ function App({ onExit }) {
     }));
   }, [skills, session, prompt]);
   const commands = import_react48.useMemo(() => [
-    { id: "/new", label: "Start a new session", section: "Session", action: () => void handleNewSession() },
-    { id: "/fork", label: "Fork current session", section: "Session", action: () => setShowForkModal(true) },
+    { id: "/new", label: "Start a new session", section: "Session", action: () => void handleNewSession(), disabled: isStreaming },
+    { id: "/fork", label: "Fork current session", section: "Session", action: () => setShowForkModal(true), disabled: isStreaming },
     { id: "/sessions", label: "Browse sessions", section: "Session", action: async () => {
       await refreshSessionList();
       setShowSessionModal(true);
-    } },
+    }, disabled: isStreaming },
     { id: "/compact", label: "Compact current session", section: "Session", action: () => {
       session?.compact().catch(() => {});
-    } },
+    }, disabled: isStreaming },
     { id: "/rename", label: "Rename session", section: "Session", action: () => setShowRenameModal(true) },
-    { id: "/exit", label: "Exit Koi", section: "Session", action: () => setShowExitModal(true) },
-    { id: "/quit", label: "Exit Koi (alias)", section: "Session", action: () => setShowExitModal(true) },
+    { id: "/exit", label: "Exit Koi", section: "Session", action: () => {
+      saveCurrentState();
+      onExit();
+    } },
+    { id: "/quit", label: "Exit Koi (alias)", section: "Session", action: () => {
+      saveCurrentState();
+      onExit();
+    } },
     { id: "/yolo", label: "Toggle YOLO mode (auto-approve all permissions)", section: "Mode", action: () => setYoloMode2((prev) => !prev) },
     { id: "/mode", label: `Cycle agent mode (${agentMode})`, section: "Mode", action: () => handleModeSwitch() },
-    { id: "/plan", label: "Switch to plan mode (read-only, no file modifications)", section: "Mode", action: () => applyAgentMode("plan") },
+    { id: "/plan", label: "Switch to plan mode (read-only, no file modifications)", section: "Mode", action: () => applyAgentMode("plan"), disabled: isStreaming },
     { id: "/connect", label: "Connect to a provider", section: "Model", action: () => setShowConnectModal(true) },
-    { id: "/model", label: "Select a model", section: "Model", action: () => setShowModelModal(true) },
+    { id: "/model", label: "Select a model", section: "Model", action: () => setShowModelModal(true), disabled: isStreaming },
     { id: "/mcp", label: "Open MCP settings", section: "Extensions", action: () => setShowMCPSettings(true) },
     { id: "/skills", label: "List and manage skills", section: "Extensions", action: () => setShowSkillsModal(true) },
     ...skillCommands
-  ], [session, handleNewSession, refreshSessionList, agentMode, handleModeSwitch, applyAgentMode, skillCommands]);
+  ], [isStreaming, session, handleNewSession, refreshSessionList, agentMode, handleModeSwitch, applyAgentMode, skillCommands]);
   useKeyboard((key) => {
     if (anyModalOpen && !permissionModalOpen)
       return;
@@ -489968,18 +490291,24 @@ function App({ onExit }) {
       return;
     }
     if (key.ctrl && key.name === "s") {
-      (async () => {
-        await refreshSessionList();
-        setShowSessionModal(true);
-      })();
+      if (!isStreaming) {
+        (async () => {
+          await refreshSessionList();
+          setShowSessionModal(true);
+        })();
+      }
       return;
     }
     if (key.ctrl && key.name === "f") {
-      setShowForkModal(true);
+      if (!isStreaming) {
+        setShowForkModal(true);
+      }
       return;
     }
     if (key.name === "tab" && key.shift) {
-      handleModeSwitch();
+      if (!isStreaming) {
+        handleModeSwitch();
+      }
       return;
     }
     if (key.ctrl && key.name === "o") {