code-ollama 0.17.0 → 0.18.0

package/dist/assets/{tui-CCHcdC7V.js → tui-4cX-bKvd.js}

@@ -1,4 +1,4 @@
-import { A as PLAN_GENERATION_INSTRUCTION, C as withSystemMessage, D as ASSISTANT, E as getTheme, F as APPROVE, I as REJECT, L as VERSION, M as LABEL, N as PLAN, O as SYSTEM, P as SAFE, R as LIST, S as resetSystemMessage, T as LIST$1, _ as setClearHandler, a as tick, b as loadConfig, c as appendMessage, d as deleteSessionIfEmpty, f as listSessions, g as reset, h as clear, i as WRITE_TOOLS, j as AUTO, k as USER, l as createSession, m as updateSessionModel, n as READ_TOOLS, o as color, p as loadSession, r as TOOLS, s as write, t as executeTool, u as deleteSession, v as listModels, w as HEADER_PREFIX, x as saveConfig, y as streamChat } from "../cli.js";
+import { A as SYSTEM, B as REJECT, C as resetSystemMessage, D as LIST$1, E as WARNING, F as AUTO, H as LIST, I as LABEL, L as PLAN, M as PLAN_GENERATION_INSTRUCTION, N as BACK, O as getTheme, P as CATALOG, R as SAFE, S as saveConfig, T as HEADER_PREFIX, V as VERSION, _ as deleteModel, a as color, b as streamChat, c as createSession, d as listSessions, f as loadSession, g as setClearHandler, h as reset, i as WRITE_TOOLS, j as USER, k as ASSISTANT, l as deleteSession, m as clear, n as READ_TOOLS, o as write, p as updateSessionModel, r as TOOLS, s as appendMessage, t as executeTool, u as deleteSessionIfEmpty, v as listModels, w as withSystemMessage, x as loadConfig, y as pullModel, z as APPROVE } from "../cli.js";
 import { readdirSync } from "node:fs";
 import { homedir } from "node:os";
 import { join, relative } from "node:path";
@@ -6,7 +6,7 @@ import { exec } from "node:child_process";
 import { Box, Static, Text, render, useApp, useInput, useStdout } from "ink";
 import { memo, useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { Fragment, jsx, jsxs } from "react/jsx-runtime";
-import { Select, Spinner } from "@inkjs/ui";
+import { ProgressBar, Select, Spinner } from "@inkjs/ui";
 import { Marked } from "marked";
 import { markedTerminal } from "marked-terminal";
 //#region src/components/CodeBlock/CodeBlock.tsx
@@ -630,9 +630,13 @@ function ToolApproval({ toolCall, onDecision, theme = getTheme() }) {
 			onChange: handleChange,
 			onCancel: handleEscape,
 			children: [
-				/* @__PURE__ */ jsx(Text, {
+				/* @__PURE__ */ jsxs(Text, {
 					color: theme.colors.warning,
-					children: "Tool requires approval ⚠️ "
+					children: [
+						"Tool requires approval ",
+						WARNING,
+						" "
+					]
 				}),
 				/* @__PURE__ */ jsxs(Box, {
 					flexDirection: "column",
@@ -817,8 +821,57 @@ function CommandMenu({ input, onSubmit }) {
 	});
 }
 //#endregion
+//#region src/components/Suggestions.tsx
+var DEFAULT_MAX_VISIBLE_OPTIONS = 5;
+function Suggestions({ options, isDisabled = false, maxVisibleOptions = DEFAULT_MAX_VISIBLE_OPTIONS, resetKey, onHighlight, onSelect }) {
+	const [focusedIndex, setFocusedIndex] = useState(0);
+	useEffect(() => {
+		setFocusedIndex(0);
+	}, [resetKey]);
+	useEffect(() => {
+		if (!options.length) {
+			setFocusedIndex(0);
+			onHighlight?.(null);
+			return;
+		}
+		setFocusedIndex((currentIndex) => Math.min(currentIndex, options.length - 1));
+	}, [onHighlight, options]);
+	useEffect(() => {
+		onHighlight?.(options[focusedIndex] ?? null);
+	}, [
+		focusedIndex,
+		onHighlight,
+		options
+	]);
+	useInput((_, key) => {
+		if (isDisabled || !options.length) return;
+		if (key.downArrow) {
+			setFocusedIndex((currentIndex) => Math.min(currentIndex + 1, options.length - 1));
+			return;
+		}
+		if (key.upArrow) {
+			setFocusedIndex((currentIndex) => Math.max(currentIndex - 1, 0));
+			return;
+		}
+		if (key.tab || key.return) onSelect(options[focusedIndex]);
+	});
+	if (!options.length) return null;
+	const visibleStart = Math.min(Math.max(0, focusedIndex - maxVisibleOptions + 1), Math.max(0, options.length - maxVisibleOptions));
+	return /* @__PURE__ */ jsx(Box, {
+		flexDirection: "column",
+		children: options.slice(visibleStart, visibleStart + maxVisibleOptions).map((option, index) => {
+			return /* @__PURE__ */ jsx(Box, {
+				marginLeft: 2,
+				children: /* @__PURE__ */ jsx(Text, {
+					color: visibleStart + index === focusedIndex ? "cyan" : void 0,
+					children: option.label
+				})
+			}, option.label);
+		})
+	});
+}
+//#endregion
 //#region src/components/Chat/FileSuggestions.tsx
-var MAX_VISIBLE_OPTIONS = 5;
 var MENTION_PATTERN = /(^|.)@(\S+)/;
 var RIPGREP_MAX_BUFFER = 10 * 1024 * 1024;
 function normalizePath(filePath) {
@@ -897,7 +950,6 @@ async function listProjectFiles(rootDir) {
 }
 function FileSuggestions({ input, isDisabled = false, onChange, onSelect }) {
 	const [filePaths, setFilePaths] = useState([]);
-	const [focusedIndex, setFocusedIndex] = useState(0);
 	useEffect(() => {
 		async function loadProjectFiles() {
 			setFilePaths(await listProjectFiles(process.cwd()));
@@ -910,55 +962,29 @@ function FileSuggestions({ input, isDisabled = false, onChange, onSelect }) {
 		const normalizedQuery = mentionMatch.query.toLowerCase();
 		return filePaths.filter((filePath) => filePath.toLowerCase().includes(normalizedQuery));
 	}, [filePaths, mentionMatch]);
-	useEffect(() => {
-		setFocusedIndex(0);
-	}, [input]);
-	useEffect(() => {
-		if (!options.length) {
-			setFocusedIndex(0);
-			return;
-		}
-		setFocusedIndex((currentIndex) => Math.min(currentIndex, options.length - 1));
-	}, [options]);
 	useEffect(() => {
 		if (!onChange) return;
-		if (!mentionMatch || !options.length) {
-			onChange(null);
-			return;
-		}
-		onChange(buildNextInput(input, options[focusedIndex]).value);
+		if (!mentionMatch || !options.length) onChange(null);
 	}, [
-		focusedIndex,
-		input,
 		mentionMatch,
 		onChange,
 		options
 	]);
-	useInput((_, key) => {
-		if (isDisabled || !options.length) return;
-		if (key.downArrow) {
-			setFocusedIndex((currentIndex) => Math.min(currentIndex + 1, options.length - 1));
-			return;
-		}
-		if (key.upArrow) {
-			setFocusedIndex((currentIndex) => Math.max(currentIndex - 1, 0));
-			return;
-		}
-		if (key.tab || key.return) onSelect(buildNextInput(input, options[focusedIndex]));
-	});
 	if (!mentionMatch || !options.length) return null;
-	const visibleStart = Math.min(Math.max(0, focusedIndex - MAX_VISIBLE_OPTIONS + 1), Math.max(0, options.length - MAX_VISIBLE_OPTIONS));
-	return /* @__PURE__ */ jsx(Box, {
-		flexDirection: "column",
-		children: options.slice(visibleStart, visibleStart + MAX_VISIBLE_OPTIONS).map((option, index) => {
-			return /* @__PURE__ */ jsx(Box, {
-				marginLeft: 2,
-				children: /* @__PURE__ */ jsx(Text, {
-					color: visibleStart + index === focusedIndex ? "cyan" : void 0,
-					children: option
-				})
-			}, option);
-		})
+	return /* @__PURE__ */ jsx(Suggestions, {
+		isDisabled,
+		options: options.map((option) => ({
+			label: option,
+			value: option
+		})),
+		resetKey: input,
+		onHighlight: (option) => {
+			// v8 ignore next
+			onChange?.(option ? buildNextInput(input, option.value).value : null);
+		},
+		onSelect: (option) => {
+			onSelect(buildNextInput(input, option.value));
+		}
 	});
 }
 //#endregion
@@ -1485,7 +1511,7 @@ function Chat({ initialMessages, model, onCommand, onMessagesChange, mode, onMod
 				marginBottom: 1,
 				children: /* @__PURE__ */ jsx(Text, {
 					color: theme.colors.error,
-					children: interruptReason === INTERRUPT_REASON.REJECTED ? "❗ Tool call rejected." : "❗ Execution interrupted."
+					children: interruptReason === INTERRUPT_REASON.REJECTED ? `❗ Tool call rejected.` : `❗ Execution interrupted.`
 				})
 			}),
 			!pendingPlan && !pendingToolCall && /* @__PURE__ */ jsx(Box, {
@@ -1588,7 +1614,7 @@ function Header({ model, onLoad, theme = getTheme() }) {
 						/* @__PURE__ */ jsxs(Text, {
 							color: theme.colors.secondary,
 							dimColor: true,
-							children: [" ", "to switch"]
+							children: [" ", "to manage"]
 						})
 					]
 				}),
@@ -1602,48 +1628,608 @@ function Header({ model, onLoad, theme = getTheme() }) {
 	});
 }
 //#endregion
-//#region src/components/ModelPicker.tsx
-function ModelPicker({ currentModel, onSelect, onClose, theme = getTheme() }) {
-	const [options, setOptions] = useState([]);
-	const [error, setError] = useState(null);
-	const handleChange = useCallback((model) => {
-		onSelect({ model });
-	}, [onSelect]);
-	useInput(async (_input, key) => {
-		if (!error && options.length && key.return) {
-			await tick();
-			onClose();
+//#region src/components/ModelManager/ModelSuggestions.tsx
+function rankCatalogMatch(entry, normalizedInput) {
+	const normalizedValue = entry.value.toLowerCase();
+	const normalizedLabel = entry.label.toLowerCase();
+	// v8 ignore start
+	switch (true) {
+		case normalizedValue.startsWith(normalizedInput): return 0;
+		case normalizedLabel.startsWith(normalizedInput): return 1;
+		case normalizedValue.includes(normalizedInput): return 2;
+		case normalizedLabel.includes(normalizedInput): return 3;
+		default: return Number.MAX_SAFE_INTEGER;
+	}
+	// v8 ignore stop
+}
+function ModelSuggestions({ catalog, input, isDisabled = false, onHighlight, onSelect }) {
+	const normalizedInput = input.trim().toLowerCase();
+	return /* @__PURE__ */ jsx(Suggestions, {
+		isDisabled,
+		options: useMemo(() => {
+			if (!normalizedInput) return [];
+			return catalog.filter((entry) => entry.value.toLowerCase().includes(normalizedInput) || entry.label.toLowerCase().includes(normalizedInput)).sort((left, right) => rankCatalogMatch(left, normalizedInput) - rankCatalogMatch(right, normalizedInput) || left.label.localeCompare(right.label)).map((entry) => ({
+				label: entry.value,
+				value: entry.value
+			}));
+		}, [catalog, normalizedInput]),
+		resetKey: input,
+		/* v8 ignore next */
+		onHighlight: (option) => onHighlight?.(option?.value ?? null),
+		onSelect: (option) => {
+			onSelect(option.value);
+		}
+	});
+}
+//#endregion
+//#region src/components/ModelManager/types.ts
+var View = /* @__PURE__ */ function(View) {
+	View["Menu"] = "menu";
+	View["Switch"] = "switch";
+	View["Download"] = "download";
+	View["CustomDownload"] = "custom-download";
+	View["Downloading"] = "downloading";
+	View["Delete"] = "delete";
+	View["DeleteConfirm"] = "delete-confirm";
+	return View;
+}({});
+var MenuAction = /* @__PURE__ */ function(MenuAction) {
+	MenuAction["Switch"] = "switch";
+	MenuAction["Download"] = "download";
+	MenuAction["Delete"] = "delete";
+	MenuAction["Cancel"] = "cancel";
+	return MenuAction;
+}({});
+var DownloadAction = /* @__PURE__ */ function(DownloadAction) {
+	DownloadAction["Custom"] = "custom";
+	return DownloadAction;
+}({});
+var ConfirmDeleteAction = /* @__PURE__ */ function(ConfirmDeleteAction) {
+	ConfirmDeleteAction["Delete"] = "delete";
+	return ConfirmDeleteAction;
+}({});
+//#endregion
+//#region src/components/ModelManager/utils.ts
+function buildMenuOptions() {
+	return [
+		{
+			label: "Switch model",
+			value: MenuAction.Switch
+		},
+		{
+			label: "Download model",
+			value: MenuAction.Download
+		},
+		{
+			label: "Delete model",
+			value: MenuAction.Delete
+		},
+		{
+			label: "Cancel",
+			value: MenuAction.Cancel
 		}
+	];
+}
+function buildInstalledModelOptions(models, currentModel) {
+	const nextModels = [...models];
+	if (nextModels.includes(currentModel)) {
+		nextModels.splice(nextModels.indexOf(currentModel), 1);
+		nextModels.unshift(currentModel);
+	}
+	return nextModels.map((model) => ({
+		label: model === currentModel ? `${model} (current model)` : model,
+		value: model
+	}));
+}
+function buildDownloadOptions(installedModels) {
+	const installedModelSet = new Set(installedModels);
+	const availableCatalog = CATALOG.filter(({ value, alias }) => !installedModelSet.has(value) && !(alias && installedModelSet.has(alias)));
+	return [
+		{
+			label: "Enter custom model...",
+			value: DownloadAction.Custom
+		},
+		...availableCatalog.map(({ label, value }) => ({
+			label,
+			value
+		})),
+		BACK
+	];
+}
+function getNoticeColor(tone, theme) {
+	switch (tone) {
+		case "error": return theme.colors.error;
+		case "success": return theme.colors.status;
+		default: return theme.colors.secondary;
+	}
+}
+function formatBytes(bytes) {
+	if (!Number.isFinite(bytes) || bytes <= 0) return "0 B";
+	const units = [
+		"B",
+		"KB",
+		"MB",
+		"GB",
+		"TB"
+	];
+	let value = bytes;
+	let index = 0;
+	while (value >= 1024 && index < units.length - 1) {
+		value /= 1024;
+		index += 1;
+	}
+	const fractionDigits = value >= 10 || index === 0 ? 0 : 1;
+	return `${value.toFixed(fractionDigits)} ${units[index]}`;
+}
+function isAbortError(error) {
+	return error instanceof Error && error.name === "AbortError";
+}
+function mergeDownloadProgress(previous, model, status, completed, total) {
+	const nextCompleted = typeof completed === "number" && Number.isFinite(completed) && completed >= 0 ? completed : null;
+	const nextTotal = typeof total === "number" && Number.isFinite(total) && total > 0 ? total : null;
+	if (nextTotal !== null && nextCompleted !== null) return {
+		model,
+		status,
+		completed: nextCompleted,
+		total: nextTotal
+	};
+	const hasPreviousProgress = previous?.model === model && previous.total > 0;
+	return {
+		model,
+		status,
+		completed: hasPreviousProgress ? previous.completed : 0,
+		total: hasPreviousProgress ? previous.total : 0
+	};
+}
+//#endregion
+//#region src/components/ModelManager/ModelCustomDownloadView.tsx
+function ModelCustomDownloadView({ downloadDraft, notice, theme, onDraftChange, onHighlight, onSelectSuggestion, onSubmit }) {
+	const renderNotice = () => notice ? /* @__PURE__ */ jsx(Text, {
+		color: getNoticeColor(notice.tone, theme),
+		children: notice.text
+	}) : null;
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [
+			/* @__PURE__ */ jsx(Text, { children: "Enter an Ollama model name to download." }),
+			/* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, { children: "> " }), /* @__PURE__ */ jsx(TextInput, {
+				value: downloadDraft,
+				placeholder: "name:tag",
+				wrapIndent: 2,
+				onChange: onDraftChange,
+				onSubmit
+			})] }),
+			/* @__PURE__ */ jsx(ModelSuggestions, {
+				catalog: [],
+				input: downloadDraft,
+				onHighlight,
+				onSelect: onSelectSuggestion
+			}),
+			renderNotice(),
+			/* @__PURE__ */ jsx(Text, {
+				color: theme.colors.secondary,
+				dimColor: true,
+				children: "Press Enter to download, Esc or Ctrl+C to go back."
+			})
+		]
+	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelDeleteConfirmView.tsx
+function ModelDeleteConfirmView({ deleteCandidate, isDeleting, notice, theme, onCancel, onConfirm }) {
+	if (isDeleting) return /* @__PURE__ */ jsx(Spinner, { label: `Deleting model ${deleteCandidate}...` });
+	const renderNotice = () => notice ? /* @__PURE__ */ jsx(Text, {
+		color: getNoticeColor(notice.tone, theme),
+		children: notice.text
+	}) : null;
+	return /* @__PURE__ */ jsxs(SelectPrompt, {
+		options: [{
+			label: `Yes, delete ${deleteCandidate}`,
+			value: ConfirmDeleteAction.Delete
+		}, {
+			...BACK,
+			label: "No"
+		}],
+		onCancel,
+		onChange: onConfirm,
+		children: [
+			/* @__PURE__ */ jsxs(Text, { children: [
+				WARNING,
+				" Delete model",
+				" ",
+				/* @__PURE__ */ jsx(Text, {
+					color: theme.colors.model,
+					children: deleteCandidate
+				}),
+				"?"
+			] }),
+			renderNotice(),
+			/* @__PURE__ */ jsx(SelectPromptHint, { message: "This action cannot be undone" })
+		]
+	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelDeleteView.tsx
+function ModelDeleteView({ currentModel, installedModels, isLoading, notice, theme, onCancel, onSelect }) {
+	if (isLoading) return /* @__PURE__ */ jsx(Spinner, { label: "Loading models..." });
+	return /* @__PURE__ */ jsxs(SelectPrompt, {
+		options: [...buildInstalledModelOptions(installedModels.filter((model) => model !== currentModel), currentModel), BACK],
+		onCancel,
+		onChange: (value) => {
+			if (value === BACK.value) onCancel();
+			else onSelect(value);
+		},
+		children: [
+			/* @__PURE__ */ jsxs(Text, { children: [
+				"Delete an installed model (current model",
+				" ",
+				/* @__PURE__ */ jsx(Text, {
+					color: theme.colors.model,
+					children: currentModel
+				}),
+				" cannot be deleted)."
+			] }),
+			notice && /* @__PURE__ */ jsx(Text, {
+				color: getNoticeColor(notice.tone, theme),
+				children: notice.text
+			}),
+			/* @__PURE__ */ jsx(SelectPromptHint, { message: "Delete models" })
+		]
 	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelDownloadingView.tsx
+function ModelDownloadingView({ progress, theme, onCancel }) {
+	const percent = progress.total > 0 && Number.isFinite(progress.completed) && Number.isFinite(progress.total) ? Math.round(progress.completed / progress.total * 100) : null;
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [
+			/* @__PURE__ */ jsxs(Text, { children: [
+				"Downloading model:",
+				" ",
+				/* @__PURE__ */ jsx(Text, {
+					color: theme.colors.model,
+					children: progress.model
+				})
+			] }),
+			/* @__PURE__ */ jsx(Text, { children: progress.status }),
+			percent !== null ? /* @__PURE__ */ jsxs(Fragment, { children: [/* @__PURE__ */ jsxs(Text, { children: [
+				percent,
+				"% (",
+				formatBytes(progress.completed),
+				" /",
+				" ",
+				formatBytes(progress.total),
+				")"
+			] }), /* @__PURE__ */ jsx(ProgressBar, { value: Math.max(0, Math.min(100, percent)) })] }) : /* @__PURE__ */ jsx(Text, {
+				color: theme.colors.secondary,
+				dimColor: true,
+				children: "Progress details unavailable. Waiting for Ollama updates..."
+			}),
+			/* @__PURE__ */ jsx(SelectPrompt, {
+				options: [{
+					label: "Cancel download",
+					value: "cancel-download"
+				}],
+				onCancel,
+				onChange: onCancel,
+				children: /* @__PURE__ */ jsx(SelectPromptHint, { message: "Press Enter, Esc, or Ctrl+C to cancel" })
+			})
+		]
+	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelDownloadView.tsx
+function ModelDownloadView({ installedModels, notice, theme, onCancel, onChange }) {
+	return /* @__PURE__ */ jsxs(SelectPrompt, {
+		options: buildDownloadOptions(installedModels),
+		onCancel,
+		onChange: (value) => {
+			switch (value) {
+				case "custom":
+					onChange("custom");
+					break;
+				case "back":
+					onCancel();
+					break;
+				default:
+					onChange(value);
+					break;
+			}
+		},
+		children: [
+			/* @__PURE__ */ jsx(Text, { children: "Choose a model to download or use a custom model name." }),
+			notice && /* @__PURE__ */ jsx(Text, {
+				color: getNoticeColor(notice.tone, theme),
+				children: notice.text
+			}),
+			/* @__PURE__ */ jsx(SelectPromptHint, { message: "Download models" })
+		]
+	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelSwitchView.tsx
+function ModelSwitchView({ currentModel, installedModels, isLoading, onCancel, onSelect }) {
+	if (isLoading) return /* @__PURE__ */ jsx(Spinner, { label: "Loading models..." });
+	return /* @__PURE__ */ jsx(SelectPrompt, {
+		options: [...buildInstalledModelOptions(installedModels, currentModel), BACK],
+		onCancel,
+		onChange: (value) => {
+			if (value === BACK.value) onCancel();
+			else onSelect(value);
+		},
+		children: /* @__PURE__ */ jsx(SelectPromptHint, { message: "Switch models" })
+	});
+}
+//#endregion
+//#region src/components/ModelManager/ModelManager.tsx
+function ModelManager({ currentModel, onSelect, onClose, theme = getTheme() }) {
+	const [view, setView] = useState(View.Menu);
+	const [installedModels, setInstalledModels] = useState([]);
+	const [isLoadingModels, setIsLoadingModels] = useState(true);
+	const [loadError, setLoadError] = useState(null);
+	const [notice, setNotice] = useState(null);
+	const [downloadDraft, setDownloadDraft] = useState("");
+	const [highlightedSuggestion, setHighlightedSuggestion] = useState(null);
+	const [downloadProgress, setDownloadProgress] = useState(null);
+	const [deleteCandidate, setDeleteCandidate] = useState(null);
+	const [isDeleting, setIsDeleting] = useState(false);
+	const isDeletingRef = useRef(false);
+	const pullRef = useRef(null);
+	const loadInstalledModels = useCallback(async () => {
+		setIsLoadingModels(true);
+		setLoadError(null);
+		try {
+			setInstalledModels(await listModels());
+		} catch (error) {
+			setLoadError(error instanceof Error ? error.message : 			/* v8 ignore next */ String(error));
+		} finally {
+			setIsLoadingModels(false);
+		}
+	}, []);
 	useEffect(() => {
-		async function load() {
-			try {
-				const models = await listModels();
-				if (models.includes(currentModel)) {
-					models.splice(models.indexOf(currentModel), 1);
-					models.unshift(currentModel);
-				}
-				setOptions(models.map((model) => ({
-					label: model,
-					value: model
-				})));
-			} catch (error) {
-				setError(error instanceof Error ? error.message : String(error));
+		loadInstalledModels();
+	}, [loadInstalledModels]);
+	const resetDownloadState = useCallback(() => {
+		setDownloadDraft("");
+		setHighlightedSuggestion(null);
+		setDownloadProgress(null);
+		pullRef.current = null;
+	}, []);
+	const handleBackToMenu = useCallback(() => {
+		setNotice(null);
+		setDeleteCandidate(null);
+		setIsDeleting(false);
+		isDeletingRef.current = false;
+		resetDownloadState();
+		setView(View.Menu);
+	}, [resetDownloadState]);
+	const cancelActivePull = useCallback(() => {
+		pullRef.current?.abort();
+	}, []);
+	useInput((input, key) => {
+		if (view === View.CustomDownload && (key.escape || key.ctrl && input === "c")) {
+			setNotice(null);
+			setHighlightedSuggestion(null);
+			setView(View.Download);
+			return;
+		}
+		if (view === View.Downloading && (key.escape || key.ctrl && input === "c")) cancelActivePull();
+	});
+	const handleMenuChange = useCallback((value) => {
+		setNotice(null);
+		switch (value) {
+			case "switch":
+				setView(View.Switch);
+				break;
+			case "download":
+				setView(View.Download);
+				break;
+			case "delete":
+				setView(View.Delete);
+				break;
+			default: onClose();
+		}
+	}, [onClose]);
+	const handleSwitchChange = useCallback((model) => {
+		onSelect({ model });
+	}, [onSelect]);
+	const startPull = useCallback(async (model) => {
+		const normalizedModel = model.trim();
+		if (!normalizedModel) {
+			setNotice({
+				tone: "error",
+				text: `❗ Enter a model name to download`
+			});
+			return;
+		}
+		if (installedModels.includes(normalizedModel)) {
+			setNotice({
+				tone: "info",
+				text: `${normalizedModel} is already installed`
+			});
+			return;
+		}
+		setNotice(null);
+		setDownloadProgress({
+			model: normalizedModel,
+			status: "Starting download...",
+			completed: 0,
+			total: 0
+		});
+		setView(View.Downloading);
+		try {
+			const pull = await pullModel(normalizedModel);
+			pullRef.current = pull;
+			for await (const update of pull) setDownloadProgress((previous) => {
+				return mergeDownloadProgress(previous, normalizedModel, update.status, update.completed, update.total);
+			});
+			pullRef.current = null;
+			resetDownloadState();
+			await loadInstalledModels();
+			setNotice({
+				tone: "success",
+				text: `✅ ${normalizedModel} downloaded successfully`
+			});
+			setView(View.Menu);
+		} catch (error) {
+			pullRef.current = null;
+			if (isAbortError(error)) {
+				setNotice({
+					tone: "error",
+					text: `❌ Download canceled for ${normalizedModel}`
+				});
+				setDownloadProgress(null);
+				setView(View.Download);
+				return;
 			}
+			setNotice({
+				tone: "error",
+				text: `❗ Error downloading model: ${error instanceof Error ? error.message : 				/* v8 ignore next */ String(error)}`
+			});
+			setDownloadProgress(null);
+			setView(View.CustomDownload);
 		}
-		load();
-	}, [currentModel]);
-	if (error) return /* @__PURE__ */ jsxs(Text, {
-		color: theme.colors.error,
-		children: ["Error loading models: ", error]
+	}, [
+		installedModels,
+		loadInstalledModels,
+		resetDownloadState
+	]);
+	const handleDownloadChange = useCallback((value) => {
+		if (value === "custom") {
+			setNotice(null);
+			setView(View.CustomDownload);
+			return;
+		}
+		// v8 ignore next 3
+		if (value === "back") {
+			handleBackToMenu();
+			return;
+		}
+		setDownloadDraft(value);
+		startPull(value);
+	}, [handleBackToMenu, startPull]);
+	const handleCustomDownloadSubmit = useCallback((value) => {
+		const nextValue = highlightedSuggestion ?? value.trim();
+		setDownloadDraft(nextValue);
+		startPull(nextValue);
+	}, [highlightedSuggestion, startPull]);
+	const handleDeleteChange = useCallback((model) => {
+		setNotice(null);
+		setDeleteCandidate(model);
+		setView(View.DeleteConfirm);
+	}, []);
+	const handleDeleteConfirm = useCallback(async (value) => {
+		if (isDeletingRef.current) return;
+		if (value === "back") {
+			setView(View.Delete);
+			return;
+		}
+		// v8 ignore next 3
+		if (!deleteCandidate) {
+			setView(View.Delete);
+			return;
+		}
+		try {
+			isDeletingRef.current = true;
+			setIsDeleting(true);
+			await deleteModel(deleteCandidate);
+			await loadInstalledModels();
+			setNotice({
+				tone: "success",
+				text: `✅ ${deleteCandidate} deleted successfully`
+			});
+			isDeletingRef.current = false;
+			setIsDeleting(false);
+			setDeleteCandidate(null);
+			setView(View.Delete);
+		} catch (error) {
+			isDeletingRef.current = false;
+			setIsDeleting(false);
+			setNotice({
+				tone: "error",
+				text: `❗ Error deleting model: ${error instanceof Error ? error.message : 				/* v8 ignore next */ String(error)}`
+			});
+			setView(View.Delete);
+		}
+	}, [deleteCandidate, loadInstalledModels]);
+	const renderNotice = () => notice ? /* @__PURE__ */ jsx(Text, {
+		color: notice.tone === "error" ? theme.colors.error : notice.tone === "success" ? theme.colors.status : theme.colors.secondary,
+		children: notice.text
+	}) : null;
+	if (loadError && view !== View.Menu) return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [/* @__PURE__ */ jsxs(Text, {
+			color: theme.colors.error,
+			children: ["Error loading models: ", loadError]
+		}), /* @__PURE__ */ jsx(Text, {
+			color: theme.colors.secondary,
+			dimColor: true,
+			children: "Press Esc to go back."
+		})]
 	});
-	if (!options.length) return /* @__PURE__ */ jsx(Spinner, { label: "Loading models..." });
-	return /* @__PURE__ */ jsx(SelectPrompt, {
-		options,
-		defaultValue: currentModel,
-		onChange: handleChange,
+	if (view === View.Downloading && downloadProgress) return /* @__PURE__ */ jsx(ModelDownloadingView, {
+		progress: downloadProgress,
+		theme,
+		onCancel: cancelActivePull
+	});
+	if (view === View.CustomDownload) return /* @__PURE__ */ jsx(ModelCustomDownloadView, {
+		downloadDraft,
+		notice,
+		theme,
+		onDraftChange: setDownloadDraft,
+		onHighlight: setHighlightedSuggestion,
+		onSelectSuggestion: (value) => {
+			setDownloadDraft(value);
+			setHighlightedSuggestion(value);
+		},
+		onSubmit: handleCustomDownloadSubmit
+	});
+	if (view === View.Switch) return /* @__PURE__ */ jsx(ModelSwitchView, {
+		currentModel,
+		installedModels,
+		isLoading: isLoadingModels,
+		onCancel: handleBackToMenu,
+		onSelect: handleSwitchChange
+	});
+	if (view === View.Download) return /* @__PURE__ */ jsx(ModelDownloadView, {
+		installedModels,
+		notice,
+		theme,
+		onCancel: handleBackToMenu,
+		onChange: handleDownloadChange
+	});
+	if (view === View.Delete) return /* @__PURE__ */ jsx(ModelDeleteView, {
+		currentModel,
+		installedModels,
+		isLoading: isLoadingModels,
+		notice,
+		theme,
+		onCancel: handleBackToMenu,
+		onSelect: handleDeleteChange
+	});
+	if (view === View.DeleteConfirm && deleteCandidate) return /* @__PURE__ */ jsx(ModelDeleteConfirmView, {
+		deleteCandidate,
+		isDeleting,
+		notice,
+		theme,
+		onCancel: () => {
+			setView(View.Delete);
+		},
+		onConfirm: handleDeleteConfirm
+	});
+	return /* @__PURE__ */ jsxs(SelectPrompt, {
+		options: buildMenuOptions(),
 		onCancel: onClose,
-		children: /* @__PURE__ */ jsx(SelectPromptHint, { message: "Select a model" })
+		onChange: handleMenuChange,
+		children: [
+			/* @__PURE__ */ jsxs(Text, { children: ["Current model: ", /* @__PURE__ */ jsx(Text, {
+				color: theme.colors.model,
+				children: currentModel
+			})] }),
+			renderNotice(),
+			/* @__PURE__ */ jsx(SelectPromptHint, { message: "Manage models" })
+		]
 	});
 }
 //#endregion
@@ -1751,7 +2337,6 @@ function SearchSettings({ currentUrl, onClose, onSave, theme = getTheme() }) {
 //#endregion
 //#region src/components/SessionManager.tsx
 var ACTION = {
-	BACK: "back",
 	CLOSE: "close",
 	DELETE_MENU: "delete-menu",
 	DELETE_PREFIX: "delete:",
@@ -1779,16 +2364,10 @@ function SessionManager({ currentSessionId, onClose, onDelete, onNew, onOpen, th
 	const options = view === "open" ? [...sessions.filter(({ id }) => id !== currentSessionId).map((session) => ({
 		label: formatSessionLabel(session, maxLabelWidth),
 		value: `${ACTION.OPEN_PREFIX}${session.id}`
-	})), {
-		label: "Back",
-		value: ACTION.BACK
-	}] : view === "delete" ? [...sessions.filter(({ id }) => id !== currentSessionId).map((session) => ({
+	})), BACK] : view === "delete" ? [...sessions.filter(({ id }) => id !== currentSessionId).map((session) => ({
 		label: formatSessionLabel(session, maxLabelWidth, "Delete "),
 		value: `${ACTION.DELETE_PREFIX}${session.id}`
-	})), {
-		label: "Back",
-		value: ACTION.BACK
-	}] : [
+	})), BACK] : [
 		{
 			label: "New session",
 			value: ACTION.NEW
@@ -1820,7 +2399,7 @@ function SessionManager({ currentSessionId, onClose, onDelete, onNew, onOpen, th
 			case value === ACTION.OPEN_MENU:
 				setView("open");
 				break;
-			case value === ACTION.BACK:
+			case value === BACK.value:
 				setView("main");
 				break;
 			case value.startsWith(ACTION.DELETE_PREFIX):
@@ -1972,7 +2551,7 @@ function ThemeSettings({ currentTheme, onClose, onPreview, onSave }) {
 //#region src/components/App/constants.ts
 var SCREEN = /* @__PURE__ */ function(SCREEN) {
 	SCREEN["CHAT"] = "chat";
-	SCREEN["MODEL_PICKER"] = "model-picker";
+	SCREEN["MODEL_MANAGER"] = "model-manager";
 	SCREEN["SEARCH_SETTINGS"] = "search-settings";
 	SCREEN["SESSION_MANAGER"] = "session-manager";
 	SCREEN["THEME_SETTINGS"] = "theme-settings";
@@ -1996,7 +2575,7 @@ function useScreenRouter() {
 					setScreen(SCREEN.SESSION_MANAGER);
 					break;
 				case "/model":
-					setScreen(SCREEN.MODEL_PICKER);
+					setScreen(SCREEN.MODEL_MANAGER);
 					break;
 				case "/search":
 					setScreen(SCREEN.SEARCH_SETTINGS);
@@ -2049,8 +2628,6 @@ function useSessionManager({ sessionId, model, commandColor }) {
 	}, []);
 	return {
 		activeSession,
-		sessionRef,
-		setActiveSession,
 		setSession,
 		handleCreateSession: useCallback(() => {
 			const nextSession = createSession(modelRef.current);
@@ -2089,14 +2666,12 @@ function useSessionManager({ sessionId, model, commandColor }) {
 //#region src/components/App/hooks/useThemeSettings.ts
 function useThemeSettings({ currentTheme, onUpdateConfig, setScreen }) {
 	const [previewThemeId, setPreviewThemeId] = useState(null);
-	const activeThemeId = previewThemeId ?? currentTheme;
-	const activeTheme = getTheme(activeThemeId);
+	const activeTheme = getTheme(previewThemeId ?? currentTheme);
 	const handleThemePreview = useCallback((themeId) => {
 		setPreviewThemeId(themeId);
 	}, []);
 	return {
 		activeTheme,
-		activeThemeId,
 		handleThemeClose: useCallback(() => {
 			setPreviewThemeId(null);
 			setScreen(SCREEN.CHAT);
@@ -2106,7 +2681,6 @@ function useThemeSettings({ currentTheme, onUpdateConfig, setScreen }) {
 			setPreviewThemeId(null);
 			onUpdateConfig({ theme: themeId });
 		}, [onUpdateConfig]),
-		previewThemeId,
 		setPreviewThemeId
 	};
 }
@@ -2181,8 +2755,8 @@ function App({ sessionId }) {
 	}, [handleCreateSession, setScreen]);
 	let screenContent;
 	switch (currentScreen) {
-		case SCREEN.MODEL_PICKER:
-			screenContent = /* @__PURE__ */ jsx(ModelPicker, {
+		case SCREEN.MODEL_MANAGER:
+			screenContent = /* @__PURE__ */ jsx(ModelManager, {
 				currentModel: appConfig.model,
 				onSelect: handleUpdateConfig,
 				onClose: handleClose,

package/dist/cli.js

@@ -19,7 +19,7 @@ var LIST$1 = [
 	},
 	{
 		name: "/model",
-		description: "switch the model"
+		description: "manage Ollama models"
 	},
 	{
 		name: "/theme",
@@ -37,7 +37,7 @@ var LIST$1 = [
 //#endregion
 //#region package.json
 var name = "code-ollama";
-var version = "0.17.0";
+var version = "0.18.0";
 //#endregion
 //#region src/constants/package.ts
 var NAME = name;
@@ -60,6 +60,38 @@ var LABEL = {
 	plan: "Plan"
 };
 //#endregion
+//#region src/constants/models.ts
+/**
+* @see https://ollama.com/library
+*/
+var CATALOG = [
+	{
+		label: "Gemma 4 (gemma4:latest)",
+		value: "gemma4:latest",
+		alias: "gemma4:e4b"
+	},
+	{
+		label: "Granite 4 (granite4.1:8b)",
+		value: "granite4.1:8b"
+	},
+	{
+		label: "Qwen 2.5 Coder (qwen2.5-coder:latest)",
+		value: "qwen2.5-coder:latest",
+		alias: "qwen2.5-coder:7b"
+	},
+	{
+		label: "DeepSeek Coder V2 (deepseek-coder-v2:latest)",
+		value: "deepseek-coder-v2:latest",
+		alias: "deepseek-coder-v2:16b"
+	}
+];
+//#endregion
+//#region src/constants/option.ts
+var BACK = {
+	label: "Back",
+	value: "back"
+};
+//#endregion
 //#region src/constants/prompt.ts
 var BASE_SYSTEM_PROMPT = `You are a coding assistant that helps users write, edit, and understand code. You have access to tools for reading files, writing files, running shell commands, searching code, and searching the web
 
@@ -259,6 +291,7 @@ var WEB_FETCH = "web_fetch";
 //#endregion
 //#region src/constants/ui.ts
 var HEADER_PREFIX = "🦙 ";
+var WARNING = "⚠️";
 //#endregion
 //#region src/utils/agents.ts
 var AGENTS_FILE = "AGENTS.md";
@@ -358,6 +391,15 @@ async function listModels() {
 	const { models } = await client.list();
 	return models.map(({ name }) => name);
 }
+async function pullModel(model) {
+	return client.pull({
+		model,
+		stream: true
+	});
+}
+function deleteModel(model) {
+	return client.delete({ model });
+}
 //#endregion
 //#region src/utils/screen.ts
 var clearHandler = null;
@@ -536,9 +578,6 @@ function writeError(text) {
 	process.stderr.write(text);
 }
 //#endregion
-//#region src/utils/time.ts
-var tick = (ms = 0) => new Promise((resolve) => setTimeout(resolve, ms));
-//#endregion
 //#region src/utils/tools/definitions.ts
 /**
 * Helper to define tool parameters
@@ -1085,7 +1124,7 @@ async function main(args = process.argv.slice(2)) {
 	else await launchTui();
 }
 async function launchTui(sessionId) {
-	const { renderApp } = await import("./assets/tui-CCHcdC7V.js");
+	const { renderApp } = await import("./assets/tui-4cX-bKvd.js");
 	reset();
 	renderApp(sessionId);
 }
@@ -1101,4 +1140,4 @@ function isEntrypoint(argv1 = process.argv[1]) {
 if (isEntrypoint()) main();
 // v8 ignore stop
 //#endregion
-export { PLAN_GENERATION_INSTRUCTION as A, withSystemMessage as C, ASSISTANT as D, getTheme as E, APPROVE as F, REJECT as I, VERSION as L, LABEL as M, PLAN as N, SYSTEM as O, SAFE as P, LIST$1 as R, resetSystemMessage as S, LIST as T, setClearHandler as _, tick as a, loadConfig as b, appendMessage as c, deleteSessionIfEmpty as d, listSessions as f, reset as g, clear as h, WRITE_TOOLS as i, AUTO as j, USER as k, createSession as l, updateSessionModel as m, main, READ_TOOLS as n, color as o, loadSession as p, TOOLS as r, write as s, executeTool as t, deleteSession as u, listModels as v, HEADER_PREFIX as w, saveConfig as x, streamChat as y };
+export { SYSTEM as A, REJECT as B, resetSystemMessage as C, LIST as D, WARNING as E, AUTO as F, LIST$1 as H, LABEL as I, PLAN as L, PLAN_GENERATION_INSTRUCTION as M, BACK as N, getTheme as O, CATALOG as P, SAFE as R, saveConfig as S, HEADER_PREFIX as T, VERSION as V, deleteModel as _, color as a, streamChat as b, createSession as c, listSessions as d, loadSession as f, setClearHandler as g, reset as h, WRITE_TOOLS as i, USER as j, ASSISTANT as k, deleteSession as l, clear as m, main, READ_TOOLS as n, write as o, updateSessionModel as p, TOOLS as r, appendMessage as s, executeTool as t, deleteSessionIfEmpty as u, listModels as v, withSystemMessage as w, loadConfig as x, pullModel as y, APPROVE as z };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-ollama",
-  "version": "0.17.0",
+  "version": "0.18.0",
   "description": "Ollama coding agent that runs in your terminal",
   "author": "Mark <mark@remarkablemark.org> (https://remarkablemark.org)",
   "type": "module",
@@ -65,7 +65,7 @@
     "lint-staged": "17.0.5",
     "prettier": "3.8.3",
     "publint": "0.3.21",
-    "tsx": "4.22.0",
+    "tsx": "4.22.1",
     "typescript": "6.0.3",
     "typescript-eslint": "8.59.3",
     "vite": "8.0.13",