code-ollama 0.7.0 → 0.9.0

package/dist/assets/{tui-CDaKDOEJ.js → tui-VKBxlYAz.js}

@@ -5,8 +5,10 @@ import { homedir } from "node:os";
 import { exec } from "node:child_process";
 import { Box, Text, render, useApp, useInput } from "ink";
 import { memo, useCallback, useEffect, useMemo, useRef, useState } from "react";
-import { Select, Spinner, TextInput } from "@inkjs/ui";
+import { Select, Spinner } from "@inkjs/ui";
 import { jsx, jsxs } from "react/jsx-runtime";
+import { marked } from "marked";
+import TerminalRenderer from "marked-terminal";
 //#region src/constants/command.ts
 var LIST = [
 	{
@@ -40,24 +42,173 @@ var LABEL = {
 };
 //#endregion
 //#region src/constants/ui.ts
-var HEADER_PREFIX = "🦙";
+var HEADER_PREFIX = "🦙 ";
 //#endregion
-//#region src/components/Messages.tsx
+//#region src/components/CodeBlock/CodeBlock.tsx
+async function highlightCode(code, language = "text") {
+	const { codeToANSI } = await import("@shikijs/cli");
+	try {
+		return await codeToANSI(code, language, "github-light");
+	} catch {
+		// v8 ignore next - Defensive fallback for unsupported languages
+		return code;
+	}
+}
+var CodeBlock = memo(function CodeBlock({ code, language, role }) {
+	const [highlighted, setHighlighted] = useState(code);
+	useEffect(() => {
+		let canceled = false;
+		async function loadHighlight() {
+			try {
+				const result = await highlightCode(code, language);
+				if (!canceled) setHighlighted(result);
+			} catch {}
+		}
+		loadHighlight();
+		return () => {
+			canceled = true;
+		};
+	}, [code, language]);
+	const isSystem = role === ROLE.SYSTEM;
+	return /* @__PURE__ */ jsx(Box, {
+		flexDirection: "column",
+		borderStyle: "round",
+		borderColor: isSystem ? "gray" : "dim",
+		paddingX: 1,
+		marginY: 1,
+		children: /* @__PURE__ */ jsx(Box, { children: /* @__PURE__ */ jsx(Text, {
+			dimColor: isSystem,
+			children: highlighted
+		}) })
+	});
+});
+//#endregion
+//#region src/components/Markdown/Markdown.tsx
+marked.setOptions({ renderer: new TerminalRenderer({ theme: "gitHub" }) });
+function renderMarkdown(content) {
+	const result = marked.parse(content);
+	// v8 ignore next - Defensive fallback for Promise return
+	return typeof result === "string" ? result.trim() : "";
+}
+var Markdown = memo(function Markdown({ content, color, dimColor }) {
+	const [rendered, setRendered] = useState(content);
+	useEffect(() => {
+		let canceled = false;
+		function loadMarkdown() {
+			try {
+				const result = renderMarkdown(content);
+				// v8 ignore start
+				if (!canceled) setRendered(result);
+			} catch {}
+			// v8 ignore stop
+		}
+		loadMarkdown();
+		return () => {
+			canceled = true;
+		};
+	}, [content]);
+	return /* @__PURE__ */ jsx(Text, {
+		color,
+		dimColor,
+		children: rendered
+	});
+});
+//#endregion
+//#region src/components/Messages/constants.ts
+var TURN_ABORTED_MESSAGE = [
+	"<turn_aborted>",
+	"The user interrupted the previous turn on purpose. Any running commands may still be running in the background. If any tools were aborted, they may have partially executed.",
+	"</turn_aborted>"
+].join("\n");
+//#endregion
+//#region src/components/Messages/Messages.tsx
 function getMessageColor(role) {
 	switch (role) {
 		case ROLE.USER: return "black";
-		case ROLE.ASSISTANT: return "blue";
+		case ROLE.ASSISTANT: return "cyan";
 		case ROLE.SYSTEM: return "gray";
 		default: return;
 	}
 }
-var MessageRow = memo(function MessageRow({ message }) {
+function parseContent(content) {
+	const segments = [];
+	const codeBlockRegex = /```(\w+)?\n?([\s\S]*?)```/g;
+	let lastIndex = 0;
+	let match;
+	while ((match = codeBlockRegex.exec(content)) !== null) {
+		if (match.index > lastIndex) {
+			const textContent = content.slice(lastIndex, match.index).trim();
+			// v8 ignore next 2 - Defensive check for empty trimmed content
+			if (textContent) segments.push({
+				type: "text",
+				content: textContent
+			});
+		}
+		const language = match[1];
+		const codeContent = match[2].trim();
+		// v8 ignore next 2 - Defensive check for empty code block
+		if (codeContent) segments.push({
+			type: "code",
+			content: codeContent,
+			language
+		});
+		lastIndex = match.index + match[0].length;
+	}
+	if (lastIndex < content.length) {
+		const textContent = content.slice(lastIndex).trim();
+		// v8 ignore next 2 - Defensive check for empty trimmed content
+		if (textContent) segments.push({
+			type: "text",
+			content: textContent
+		});
+	}
+	// v8 ignore next 2 - Defensive fallback for edge case
+	if (segments.length === 0 && content.trim()) segments.push({
+		type: "text",
+		content: content.trim()
+	});
+	return segments;
+}
+var Message = memo(function Message({ message }) {
+	const messageColor = getMessageColor(message.role);
+	const isSystem = message.role === ROLE.SYSTEM;
+	const isUser = message.role === ROLE.USER;
+	if (isSystem) return /* @__PURE__ */ jsx(Box, {
+		flexDirection: "column",
+		marginBottom: 1,
+		marginX: 2,
+		children: /* @__PURE__ */ jsx(Text, {
+			color: messageColor,
+			dimColor: true,
+			children: message.content
+		})
+	});
 	return /* @__PURE__ */ jsx(Box, {
+		flexDirection: "column",
 		marginBottom: 1,
-		children: /* @__PURE__ */ jsxs(Text, {
-			color: getMessageColor(message.role),
-			dimColor: message.role === ROLE.SYSTEM,
-			children: [message.role === ROLE.USER ? "> " : "", message.content]
+		children: parseContent(message.content).map((segment, index) => {
+			const prefix = isUser && index === 0 ? "> " : "";
+			if (segment.type === "code") return isUser ? /* @__PURE__ */ jsx(Text, {
+				color: messageColor,
+				children: segment.content
+			}, index) : /* @__PURE__ */ jsx(Box, {
+				marginX: 2,
+				children: /* @__PURE__ */ jsx(CodeBlock, {
+					code: segment.content,
+					language: segment.language,
+					role: message.role
+				})
+			}, index);
+			return isUser ? /* @__PURE__ */ jsx(Text, {
+				color: messageColor,
+				children: prefix + segment.content
+			}, index) : /* @__PURE__ */ jsx(Box, {
+				marginX: 2,
+				children: /* @__PURE__ */ jsx(Markdown, {
+					content: segment.content,
+					color: messageColor
+				})
+			}, index);
 		})
 	});
 });
@@ -65,11 +216,12 @@ function Messages({ messages, isLoading, streamingMessage }) {
 	return /* @__PURE__ */ jsxs(Box, {
 		flexDirection: "column",
 		children: [
-			messages.map((message, index) => /* @__PURE__ */ jsx(MessageRow, { message }, `${String(index)}-${message.role}-${message.content.slice(0, 16)}`)),
-			streamingMessage && /* @__PURE__ */ jsx(MessageRow, { message: streamingMessage }),
+			messages.filter(({ content }) => content !== TURN_ABORTED_MESSAGE).map((message, index) => /* @__PURE__ */ jsx(Message, { message }, index)),
+			streamingMessage && /* @__PURE__ */ jsx(Message, { message: streamingMessage }),
 			isLoading && !streamingMessage?.content && /* @__PURE__ */ jsx(Box, {
 				marginTop: -1,
 				marginBottom: 1,
+				marginX: 2,
 				children: /* @__PURE__ */ jsx(Spinner, { label: "Thinking..." })
 			})
 		]
@@ -226,6 +378,76 @@ function ToolApproval({ toolCall, onDecision }) {
 var ACTION_NOT_PERFORMED = "The requested action was NOT performed";
 var PLAN_CHECKLIST_REMINDER = "Then display the execution plan as an unchecked Markdown checklist only";
 var PLAN_EXECUTION_REMINDER = "Do not claim success and do not call write_file or run_shell until the user approves execution";
+var INTERRUPT_REASON = /* @__PURE__ */ function(INTERRUPT_REASON) {
+	INTERRUPT_REASON["INTERRUPTED"] = "interrupted";
+	INTERRUPT_REASON["REJECTED"] = "rejected";
+	return INTERRUPT_REASON;
+}({});
+//#endregion
+//#region src/components/TextInput/TextInput.tsx
+function TextInput({ value, isDisabled = false, placeholder, onChange, onSubmit }) {
+	const [cursorPosition, setCursorPosition] = useState(value.length);
+	const prevValueRef = useRef(value);
+	useEffect(() => {
+		const prevValue = prevValueRef.current;
+		prevValueRef.current = value;
+		if (value === "") setCursorPosition(0);
+		else if (value.length > prevValue.length && cursorPosition <= prevValue.length) setCursorPosition(value.length);
+		else if (cursorPosition > value.length) setCursorPosition(value.length);
+	}, [value, cursorPosition]);
+	useInput((input, key) => {
+		// v8 ignore next
+		if (isDisabled) return;
+		if (key.return) {
+			onSubmit(value);
+			setCursorPosition(0);
+			return;
+		}
+		if (key.backspace) {
+			if (cursorPosition > 0) {
+				onChange(value.slice(0, cursorPosition - 1) + value.slice(cursorPosition));
+				setCursorPosition(cursorPosition - 1);
+			}
+			return;
+		}
+		// v8 ignore start
+		if (key.delete) {
+			if (cursorPosition < value.length) onChange(value.slice(0, cursorPosition) + value.slice(cursorPosition + 1));
+			return;
+		}
+		// v8 ignore stop
+		if (key.leftArrow) {
+			setCursorPosition(Math.max(0, cursorPosition - 1));
+			return;
+		}
+		if (key.rightArrow) {
+			setCursorPosition(Math.min(value.length, cursorPosition + 1));
+			return;
+		}
+		if (key.home) {
+			setCursorPosition(0);
+			return;
+		}
+		if (key.end) {
+			setCursorPosition(value.length);
+			return;
+		}
+		// v8 ignore start
+		if (input) {
+			onChange(value.slice(0, cursorPosition) + input + value.slice(cursorPosition));
+			setCursorPosition(cursorPosition + input.length);
+		}
+		// v8 ignore stop
+	}, { isActive: !isDisabled });
+	const displayValue = value || (placeholder ?? "");
+	const isPlaceholder = Boolean(!value && placeholder);
+	const char = displayValue[cursorPosition] || " ";
+	const before = displayValue.slice(0, cursorPosition);
+	const after = displayValue.slice(cursorPosition + 1);
+	const dimStyle = isPlaceholder ? "\x1B[2m" : "";
+	const resetDim = isPlaceholder ? "\x1B[22m" : "";
+	return /* @__PURE__ */ jsx(Text, { children: `${dimStyle}${before}${resetDim}\x1b[7m${char}\x1b[27m${dimStyle}${after}${resetDim}` });
+}
 //#endregion
 //#region src/components/Chat/CommandMenu.tsx
 function getMatchingCommands(input) {
@@ -376,18 +598,16 @@ function FileSuggestions({ input, isDisabled = false, onChange, onSelect }) {
 function hasFileSuggestionQuery(input) {
 	return /(^|\s)@\S+$/.test(input);
 }
-function Input({ isDisabled = false, onSubmit }) {
+function Input({ isDisabled = false, onInterrupt, onSubmit }) {
 	const { exit } = useApp();
 	const [input, setInput] = useState("");
-	const [inputKey, setInputKey] = useState(0);
 	const fileSuggestionRef = useRef(null);
-	const remountTextInput = useCallback(() => {
-		setInputKey((key) => key + 1);
-	}, [setInputKey]);
+	const resetInput = useCallback(() => {
+		setInput("");
+	}, []);
 	const handleSelectFileSuggestion = useCallback((nextInput) => {
 		setInput(nextInput);
-		remountTextInput();
-	}, [remountTextInput]);
+	}, []);
 	const handleFileSuggestionChange = useCallback((nextInput) => {
 		fileSuggestionRef.current = nextInput;
 	}, []);
@@ -395,10 +615,9 @@ function Input({ isDisabled = false, onSubmit }) {
 		const trimmedInput = input.trim();
 		if (!trimmedInput) return;
 		onSubmit(trimmedInput);
-		setInput("");
+		resetInput();
 		fileSuggestionRef.current = null;
-		remountTextInput();
-	}, [onSubmit, remountTextInput]);
+	}, [onSubmit, resetInput]);
 	const showCommandMenu = input.startsWith("/");
 	const showFileSuggestions = !showCommandMenu && hasFileSuggestionQuery(input);
 	const handleSubmitText = useCallback(async (input) => {
@@ -414,21 +633,29 @@ function Input({ isDisabled = false, onSubmit }) {
 		if (LIST.find(({ name }) => name === input)) submitAndReset(input);
 	}, [submitAndReset]);
 	useInput((_input, key) => {
-		if (key.ctrl && _input === "c") if (input) {
-			setInput("");
-			remountTextInput();
-		} else exit();
+		const isCtrlC = key.ctrl && _input === "c";
+		if (isDisabled) {
+			if (key.escape || isCtrlC) onInterrupt?.();
+			return;
+		}
+		if (isCtrlC) {
+			if (input) {
+				resetInput();
+				return;
+			}
+			exit();
+		}
 	});
 	return /* @__PURE__ */ jsxs(Box, {
 		flexDirection: "column",
 		children: [
 			/* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, { children: "> " }), /* @__PURE__ */ jsx(TextInput, {
-				defaultValue: input,
+				value: input,
 				isDisabled,
 				onChange: setInput,
 				onSubmit: handleSubmitText,
 				placeholder: "Ask anything... (/ commands, @ files)"
-			}, inputKey)] }),
+			})] }),
 			showCommandMenu && /* @__PURE__ */ jsx(CommandMenu, {
 				input,
 				onSubmit: handleSubmitCommand
@@ -458,12 +685,15 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 	const [isLoading, setIsLoading] = useState(false);
 	const [pendingToolCall, setPendingToolCall] = useState(null);
 	const [pendingPlan, setPendingPlan] = useState(null);
+	const [interruptReason, setInterruptReason] = useState(null);
+	const abortControllerRef = useRef(null);
 	useEffect(() => {
 		setMessages([]);
 		setStreamingMessage(null);
 		setIsLoading(false);
 		setPendingToolCall(null);
 		setPendingPlan(null);
+		setInterruptReason(null);
 	}, [sessionId]);
 	const buildToolResultMessage = useCallback((toolName, result) => {
 		if (result.error?.startsWith("Tool not allowed:")) return {
@@ -490,7 +720,20 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 			PLAN_EXECUTION_REMINDER
 		].join("\n")
 	}), []);
+	const handleInterrupt = useCallback(() => {
+		abortControllerRef.current?.abort();
+		abortControllerRef.current = null;
+		setIsLoading(false);
+		setStreamingMessage(null);
+		setInterruptReason(INTERRUPT_REASON.INTERRUPTED);
+		setMessages((prev) => [...prev, {
+			role: ROLE.USER,
+			content: TURN_ABORTED_MESSAGE
+		}]);
+	}, []);
 	const processStream = useCallback(async (currentMessages, executionMode = mode) => {
+		const controller = new AbortController();
+		abortControllerRef.current = controller;
 		const assistantMessage = {
 			role: ROLE.ASSISTANT,
 			content: ""
@@ -518,33 +761,40 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 		};
 		setStreamingMessage(assistantMessage);
 		try {
-			for await (const chunk of streamChat(withSystemMessage(currentMessages), model, TOOLS)) if (chunk.type === "content") {
-				assistantMessage.content += chunk.content;
-				setStreamingMessage({ ...assistantMessage });
-			} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
-				const requiresApproval = WRITE_TOOLS.has(toolCall.function.name);
-				// v8 ignore start
-				const allowedTools = executionMode === NAME.PLAN ? READ_TOOLS : void 0;
-				// v8 ignore stop
-				const updatedMessages = commitAssistantMessage();
-				if (executionMode === NAME.SAFE && requiresApproval) {
-					setPendingToolCall(toolCall);
-					setIsLoading(false);
+			for await (const chunk of streamChat(withSystemMessage(currentMessages), model, TOOLS, controller.signal)) {
+				// v8 ignore next 3
+				if (controller.signal.aborted) return;
+				if (chunk.type === "content") {
+					assistantMessage.content += chunk.content;
+					setStreamingMessage({ ...assistantMessage });
+				} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
+					const requiresApproval = WRITE_TOOLS.has(toolCall.function.name);
+					// v8 ignore start
+					const allowedTools = executionMode === NAME.PLAN ? READ_TOOLS : void 0;
+					// v8 ignore stop
+					const updatedMessages = commitAssistantMessage();
+					if (executionMode === NAME.SAFE && requiresApproval) {
+						setPendingToolCall(toolCall);
+						setIsLoading(false);
+						return;
+					}
+					const result = await executeTool(toolCall.function.name, toolCall.function.arguments, { allowedTools });
+					const toolResultMessage = buildToolResultMessage(toolCall.function.name, result);
+					const newMessages = [...updatedMessages, toolResultMessage];
+					setMessages(newMessages);
+					await processStream(newMessages, executionMode);
 					return;
 				}
-				const result = await executeTool(toolCall.function.name, toolCall.function.arguments, { allowedTools });
-				const toolResultMessage = buildToolResultMessage(toolCall.function.name, result);
-				const newMessages = [...updatedMessages, toolResultMessage];
-				setMessages(newMessages);
-				await processStream(newMessages, executionMode);
-				return;
 			}
 			commitAssistantMessage();
 		} catch (error) {
 			// v8 ignore next
-			assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
-			commitAssistantMessage();
+			if (!controller.signal.aborted) {
+				assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
+				commitAssistantMessage();
+			}
 		} finally {
+			if (abortControllerRef.current === controller) abortControllerRef.current = null;
 			setIsLoading(false);
 		}
 	}, [
@@ -553,6 +803,8 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 		mode
 	]);
 	const processStreamReadOnly = useCallback(async (currentMessages) => {
+		const controller = new AbortController();
+		abortControllerRef.current = controller;
 		const assistantMessage = {
 			role: ROLE.ASSISTANT,
 			content: ""
@@ -581,24 +833,28 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 		setStreamingMessage(assistantMessage);
 		try {
 			const readOnlyTools = TOOLS.filter((tool) => READ_TOOLS.has(tool.function.name));
-			for await (const chunk of streamChat(withSystemMessage(currentMessages), model, readOnlyTools)) if (chunk.type === "content") {
-				assistantMessage.content += chunk.content;
-				setStreamingMessage({ ...assistantMessage });
-			} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
-				const updatedMessages = commitAssistantMessage();
-				if (!READ_TOOLS.has(toolCall.function.name)) {
-					const correctionMessage = buildPlanModeCorrectionMessage(toolCall.function.name);
-					const newMessages = [...updatedMessages, correctionMessage];
+			for await (const chunk of streamChat(withSystemMessage(currentMessages), model, readOnlyTools, controller.signal)) {
+				// v8 ignore next 3
+				if (controller.signal.aborted) return;
+				if (chunk.type === "content") {
+					assistantMessage.content += chunk.content;
+					setStreamingMessage({ ...assistantMessage });
+				} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
+					const updatedMessages = commitAssistantMessage();
+					if (!READ_TOOLS.has(toolCall.function.name)) {
+						const correctionMessage = buildPlanModeCorrectionMessage(toolCall.function.name);
+						const newMessages = [...updatedMessages, correctionMessage];
+						setMessages(newMessages);
+						await processStreamReadOnly(newMessages);
+						return;
+					}
+					const result = await executeTool(toolCall.function.name, toolCall.function.arguments, { allowedTools: READ_TOOLS });
+					const toolResultMessage = buildToolResultMessage(toolCall.function.name, result);
+					const newMessages = [...updatedMessages, toolResultMessage];
 					setMessages(newMessages);
 					await processStreamReadOnly(newMessages);
 					return;
 				}
-				const result = await executeTool(toolCall.function.name, toolCall.function.arguments, { allowedTools: READ_TOOLS });
-				const toolResultMessage = buildToolResultMessage(toolCall.function.name, result);
-				const newMessages = [...updatedMessages, toolResultMessage];
-				setMessages(newMessages);
-				await processStreamReadOnly(newMessages);
-				return;
 			}
 			const researchMessages = commitAssistantMessage();
 			const planInstruction = {
@@ -612,9 +868,13 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 			};
 			setStreamingMessage(planAssistantMessage);
 			try {
-				for await (const chunk of streamChat(withSystemMessage(planMessages), model, [])) if (chunk.type === "content") {
-					planAssistantMessage.content += chunk.content;
-					setStreamingMessage({ ...planAssistantMessage });
+				for await (const chunk of streamChat(withSystemMessage(planMessages), model, [], controller.signal)) {
+					// v8 ignore next 3
+					if (controller.signal.aborted) return;
+					if (chunk.type === "content") {
+						planAssistantMessage.content += chunk.content;
+						setStreamingMessage({ ...planAssistantMessage });
+					}
 				}
 			} catch (error) {
 				// v8 ignore next
@@ -634,9 +894,12 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 			setIsLoading(false);
 		} catch (error) {
 			// v8 ignore next
-			assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
-			commitAssistantMessage();
+			if (!controller.signal.aborted) {
+				assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
+				commitAssistantMessage();
+			}
 		} finally {
+			if (abortControllerRef.current === controller) abortControllerRef.current = null;
 			setIsLoading(false);
 		}
 	}, [
@@ -689,16 +952,14 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 				await processStream(newMessages);
 				break;
 			}
-			case REJECT: {
-				const rejectionMessage = {
-					role: ROLE.SYSTEM,
-					content: `User declined to execute tool ${toolCall.function.name}`
-				};
-				const newMessages = [...messages, rejectionMessage];
-				setMessages((previousMessages) => [...previousMessages, rejectionMessage]);
-				await processStream(newMessages);
+			case REJECT:
+				setMessages((previousMessages) => [...previousMessages, {
+					role: ROLE.USER,
+					content: TURN_ABORTED_MESSAGE
+				}]);
+				setIsLoading(false);
+				setInterruptReason(INTERRUPT_REASON.REJECTED);
 				break;
-			}
 		}
 	}, [
 		pendingToolCall,
@@ -706,6 +967,7 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 		processStream
 	]);
 	const handleSubmit = useCallback(async (value) => {
+		setInterruptReason(null);
 		const userContent = value.trim();
 		if (!userContent) return;
 		if (userContent.startsWith("/")) {
@@ -744,8 +1006,16 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 				toolCall: pendingToolCall,
 				onDecision: handleToolApproval
 			}),
+			interruptReason && !isLoading && /* @__PURE__ */ jsx(Box, {
+				marginBottom: 1,
+				children: /* @__PURE__ */ jsx(Text, {
+					color: "red",
+					children: interruptReason === INTERRUPT_REASON.REJECTED ? "❗ Tool call rejected." : "❗ Execution interrupted."
+				})
+			}),
 			!pendingPlan && !pendingToolCall && /* @__PURE__ */ jsx(Input, {
 				isDisabled: isLoading,
+				onInterrupt: handleInterrupt,
 				onSubmit: handleSubmit
 			})
 		]
@@ -949,7 +1219,6 @@ function renderApp() {
 	const tree = /* @__PURE__ */ jsx(App, {});
 	const app = render(tree, {
 		exitOnCtrlC: false,
-		incrementalRendering: true,
 		maxFps: 60
 	});
 	setClearHandler(() => {

package/dist/cli.js

@@ -8,7 +8,7 @@ import { exec } from "node:child_process";
 import { promisify } from "node:util";
 //#endregion
 //#region src/constants/package.ts
-var VERSION = "0.7.0";
+var VERSION = "0.9.0";
 //#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, and searching code
@@ -134,22 +134,32 @@ function saveConfig(patch) {
 //#region src/utils/ollama.ts
 var { host, model: DEFAULT_MODEL } = loadConfig();
 var client = new Ollama({ host });
-async function* streamChat(messages, model = DEFAULT_MODEL, tools) {
+async function* streamChat(messages, model = DEFAULT_MODEL, tools, signal) {
 	const response = await client.chat({
 		model,
 		messages,
 		stream: true,
-		tools
+		tools,
+		// v8 ignore next
+		...signal ? { signal } : {}
 	});
-	for await (const chunk of response) {
-		if (chunk.message.content) yield {
-			type: "content",
-			content: chunk.message.content
-		};
-		if (chunk.message.tool_calls) yield {
-			type: "tool_calls",
-			tool_calls: chunk.message.tool_calls
-		};
+	try {
+		for await (const chunk of response) {
+			// v8 ignore next 3
+			if (signal?.aborted) return;
+			if (chunk.message.content) yield {
+				type: "content",
+				content: chunk.message.content
+			};
+			if (chunk.message.tool_calls) yield {
+				type: "tool_calls",
+				tool_calls: chunk.message.tool_calls
+			};
+		}
+	} catch (error) {
+		// v8 ignore start
+		if (error instanceof Error && (error.name === "AbortError" || signal?.aborted)) return;
+		throw error;
 	}
 }
 async function listModels() {
@@ -502,7 +512,7 @@ async function processRunStream(messages, model) {
 }
 async function main(args = process.argv.slice(2)) {
 	if (!args.length) {
-		const { renderApp } = await import("./assets/tui-CDaKDOEJ.js");
+		const { renderApp } = await import("./assets/tui-VKBxlYAz.js");
 		process.stdout.write("\x1Bc");
 		renderApp();
 		return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-ollama",
-  "version": "0.7.0",
+  "version": "0.9.0",
   "description": "Ollama coding agent that runs in your terminal",
   "author": "Mark <mark@remarkablemark.org> (https://remarkablemark.org)",
   "type": "module",
@@ -40,8 +40,11 @@
   ],
   "dependencies": {
     "@inkjs/ui": "2.0.0",
+    "@shikijs/cli": "4.0.2",
     "cac": "7.0.0",
     "ink": "7.0.2",
+    "marked": "15.0.12",
+    "marked-terminal": "7.3.0",
     "ollama": "0.6.3",
     "react": "19.2.6"
   },