code-ollama 0.4.0 → 0.6.0

package/README.md

@@ -1,6 +1,10 @@
 > [!NOTE]
 > TUI is under active development. APIs may change.
 
+<p align="center">
+  <img alt="Ollama" height="200" src="https://github.com/ai-action/assets/blob/master/logos/ollama.svg?raw=true">
+</p>
+
 # Code Ollama
 
 [![NPM](https://nodei.co/npm/code-ollama.svg)](https://www.npmjs.com/package/code-ollama)

package/dist/assets/{tui-BLJeczya.js → tui-UfyDivSA.js}

@@ -1,9 +1,40 @@
-import { _ as NAME, a as setClearHandler, b as NAMES, c as loadConfig, d as withSystemMessage, f as HEADER_PREFIX, g as LABEL, h as VERSION, i as executeTool, l as saveConfig, m as PLAN_GENERATION_INSTRUCTION, n as READ_ONLY_TOOLS, o as listModels, p as ROLE, r as TOOLS, s as streamChat, t as DANGEROUS_TOOLS, u as resetSystemMessage, v as APPROVE, y as REJECT } from "../cli.js";
+import { a as tick, c as streamChat, d as resetSystemMessage, f as withSystemMessage, h as VERSION, i as executeTool, l as loadConfig, m as PLAN_GENERATION_INSTRUCTION, n as TOOLS, o as setClearHandler, p as ROLE, r as WRITE_TOOLS, s as listModels, t as READ_TOOLS, u as saveConfig } from "../cli.js";
+import { readdirSync } from "node:fs";
+import { join, relative } from "node:path";
 import { homedir } from "node:os";
-import { Box, Text, render, useInput } from "ink";
-import { memo, useCallback, useEffect, useState } from "react";
+import { exec } from "node:child_process";
+import { Box, Text, render, useApp, useInput } from "ink";
+import { memo, useCallback, useEffect, useMemo, useState } from "react";
 import { Select, Spinner, TextInput } from "@inkjs/ui";
 import { jsx, jsxs } from "react/jsx-runtime";
+//#region src/constants/command.ts
+var LIST = [{
+	name: "/clear",
+	description: "clear the current session"
+}, {
+	name: "/model",
+	description: "switch the model"
+}];
+//#endregion
+//#region src/constants/decision.ts
+var APPROVE = "approve";
+var REJECT = "reject";
+//#endregion
+//#region src/constants/mode.ts
+var NAME = {
+	SAFE: "safe",
+	AUTO: "auto",
+	PLAN: "plan"
+};
+var LABEL = {
+	safe: "Safe",
+	auto: "Auto",
+	plan: "Plan"
+};
+//#endregion
+//#region src/constants/ui.ts
+var HEADER_PREFIX = "🦙";
+//#endregion
 //#region src/components/Messages.tsx
 function getMessageColor(role) {
 	switch (role) {
@@ -48,6 +79,45 @@ function SelectPrompt({ children, onEscape, ...selectProps }) {
 		children: [children, /* @__PURE__ */ jsx(Select, { ...selectProps })]
 	});
 }
+function SelectPromptHint({ message = "Select option", escapeLabel = "cancel" }) {
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "row",
+		children: [
+			/* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [message, " ("]
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				bold: true,
+				children: "↑↓"
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				dimColor: true,
+				children: " + "
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				bold: true,
+				children: "Enter"
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				dimColor: true,
+				children: " to confirm, "
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				bold: true,
+				children: "Esc"
+			}),
+			/* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [
+					" to ",
+					escapeLabel,
+					")"
+				]
+			})
+		]
+	});
+}
 //#endregion
 //#region src/components/PlanApproval.tsx
 var options$1 = [
@@ -86,10 +156,7 @@ function PlanApproval({ planContent, onModeChange }) {
 					marginY: 1,
 					children: /* @__PURE__ */ jsx(Text, { children: planContent })
 				}),
-				/* @__PURE__ */ jsx(Text, {
-					dimColor: true,
-					children: "Select execution mode (↑↓ + Enter to confirm, Esc to cancel)"
-				})
+				/* @__PURE__ */ jsx(SelectPromptHint, { message: "Select execution mode" })
 			]
 		})
 	});
@@ -140,9 +207,9 @@ function ToolApproval({ toolCall, onDecision }) {
 					args
 				] })]
 			}),
-			/* @__PURE__ */ jsx(Text, {
-				dimColor: true,
-				children: "Select approval action (↑↓ + Enter to confirm, Esc to reject)"
+			/* @__PURE__ */ jsx(SelectPromptHint, {
+				message: "Select approval action",
+				escapeLabel: "reject"
 			})
 		]
 	});
@@ -153,20 +220,196 @@ 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";
 //#endregion
+//#region src/components/Chat/CommandMenu.tsx
+function getMatchingCommands(input) {
+	const normalizedInput = input.trim().toLowerCase();
+	if (!normalizedInput.startsWith("/")) return [];
+	return LIST.filter(({ name }) => name.toLowerCase().startsWith(normalizedInput)).map(({ name, description }) => ({
+		label: `${name} - ${description}`,
+		value: name
+	}));
+}
+function CommandMenu({ input, onSubmit }) {
+	const commandOptions = useMemo(() => getMatchingCommands(input), [input]);
+	if (!commandOptions.length) return null;
+	return /* @__PURE__ */ jsx(SelectPrompt, {
+		highlightText: input,
+		onChange: onSubmit,
+		options: commandOptions
+	});
+}
+//#endregion
+//#region src/components/Chat/FileSuggestions.tsx
+var MAX_VISIBLE_OPTIONS = 5;
+var MENTION_PATTERN = /(^|\s)@(\S+)$/;
+var RIPGREP_MAX_BUFFER = 10 * 1024 * 1024;
+function normalizePath(filePath) {
+	return filePath.replaceAll("\\", "/");
+}
+function getMentionMatch(input) {
+	const match = MENTION_PATTERN.exec(input);
+	if (!match) return null;
+	return {
+		prefix: input.slice(0, match.index + match[1].length),
+		query: match[2]
+	};
+}
+function buildNextInput(input, filePath) {
+	const mentionMatch = getMentionMatch(input);
+	// v8 ignore next 3
+	if (!mentionMatch) return input;
+	return `${mentionMatch.prefix}${filePath} `;
+}
+function listProjectFilesFallback(rootDir) {
+	const filePaths = [];
+	function walk(currentPath) {
+		const entries = readdirSync(currentPath, { withFileTypes: true });
+		for (const entry of entries) {
+			if (entry.name === ".git") continue;
+			const fullPath = join(currentPath, entry.name);
+			if (entry.isDirectory()) {
+				walk(fullPath);
+				continue;
+			}
+			if (entry.isFile()) filePaths.push(normalizePath(relative(rootDir, fullPath)));
+		}
+	}
+	walk(rootDir);
+	return filePaths.sort((left, right) => left.localeCompare(right));
+}
+function listProjectFilesWithRipgrep(rootDir) {
+	return new Promise((resolve, reject) => {
+		exec("rg --files --hidden -g \"!**/.git/**\"", {
+			cwd: rootDir,
+			maxBuffer: RIPGREP_MAX_BUFFER
+		}, (error, stdout) => {
+			if (error) {
+				reject(error);
+				return;
+			}
+			resolve(stdout.split("\n").map((line) => line.trim()).filter(Boolean).map(normalizePath).sort((left, right) => left.localeCompare(right)));
+		});
+	});
+}
+async function listProjectFiles(rootDir) {
+	try {
+		return await listProjectFilesWithRipgrep(rootDir);
+	} catch {
+		return listProjectFilesFallback(rootDir);
+	}
+}
+function FileSuggestions({ input, isDisabled = false, onSelect }) {
+	const [filePaths, setFilePaths] = useState([]);
+	const [focusedIndex, setFocusedIndex] = useState(0);
+	useEffect(() => {
+		async function loadProjectFiles() {
+			setFilePaths(await listProjectFiles(process.cwd()));
+		}
+		loadProjectFiles();
+	}, []);
+	const mentionMatch = getMentionMatch(input);
+	const options = useMemo(() => {
+		if (!mentionMatch) return [];
+		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]);
+	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) 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);
+		})
+	});
+}
+//#endregion
 //#region src/components/Chat/Input.tsx
+function hasActiveMentionQuery(input) {
+	return /(^|\s)@\S+$/.test(input);
+}
 function Input({ isDisabled = false, onSubmit }) {
-	const [resetKey, setResetKey] = useState(0);
-	const handleSubmit = useCallback((input) => {
-		const trimmed = input.trim();
-		if (!trimmed) return;
-		onSubmit(trimmed);
-		setResetKey((key) => key + 1);
-	}, [onSubmit]);
-	return /* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, { children: "> " }), /* @__PURE__ */ jsx(TextInput, {
-		isDisabled,
-		suggestions: NAMES,
-		onSubmit: handleSubmit
-	}, resetKey)] });
+	const { exit } = useApp();
+	const [input, setInput] = useState("");
+	const [inputKey, setInputKey] = useState(0);
+	const remountTextInput = useCallback(() => {
+		setInputKey((key) => key + 1);
+	}, [setInputKey]);
+	const handleSubmitText = useCallback(async (input) => {
+		await tick();
+		if (input.startsWith("/")) return;
+		const trimmedInput = input.trim();
+		if (!trimmedInput) return;
+		onSubmit(trimmedInput);
+		setInput("");
+		remountTextInput();
+	}, [onSubmit, remountTextInput]);
+	const handleSubmitCommand = useCallback((input) => {
+		if (!LIST.find(({ name }) => name === input)) return;
+		onSubmit(input);
+		setInput("");
+		remountTextInput();
+	}, [onSubmit, remountTextInput]);
+	const handleSelectFileSuggestion = useCallback((nextInput) => {
+		setInput(nextInput);
+		remountTextInput();
+	}, [remountTextInput]);
+	useInput((_input, key) => {
+		if (key.ctrl && _input === "c") if (input) {
+			setInput("");
+			remountTextInput();
+		} else exit();
+	});
+	const showCommandMenu = input.startsWith("/");
+	const showFileSuggestions = !showCommandMenu && hasActiveMentionQuery(input);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [
+			/* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, { children: "> " }), /* @__PURE__ */ jsx(TextInput, {
+				defaultValue: input,
+				isDisabled,
+				onChange: setInput,
+				onSubmit: handleSubmitText,
+				placeholder: "Ask anything... (/ commands, @ files)"
+			}, inputKey)] }),
+			showCommandMenu && /* @__PURE__ */ jsx(CommandMenu, {
+				input,
+				onSubmit: handleSubmitCommand
+			}),
+			showFileSuggestions && /* @__PURE__ */ jsx(FileSuggestions, {
+				input,
+				isDisabled,
+				onSelect: handleSelectFileSuggestion
+			})
+		]
+	});
 }
 //#endregion
 //#region src/components/Chat/plan.ts
@@ -248,9 +491,9 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 				assistantMessage.content += chunk.content;
 				setStreamingMessage({ ...assistantMessage });
 			} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
-				const requiresApproval = DANGEROUS_TOOLS.has(toolCall.function.name);
+				const requiresApproval = WRITE_TOOLS.has(toolCall.function.name);
 				// v8 ignore start
-				const allowedTools = executionMode === NAME.PLAN ? READ_ONLY_TOOLS : void 0;
+				const allowedTools = executionMode === NAME.PLAN ? READ_TOOLS : void 0;
 				// v8 ignore stop
 				const updatedMessages = commitAssistantMessage();
 				if (executionMode === NAME.SAFE && requiresApproval) {
@@ -267,6 +510,7 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 			}
 			commitAssistantMessage();
 		} catch (error) {
+			// v8 ignore next
 			assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
 			commitAssistantMessage();
 		} finally {
@@ -305,20 +549,20 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 		};
 		setStreamingMessage(assistantMessage);
 		try {
-			const readOnlyTools = TOOLS.filter((tool) => READ_ONLY_TOOLS.has(tool.function.name));
+			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_ONLY_TOOLS.has(toolCall.function.name)) {
+				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_ONLY_TOOLS });
+				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);
@@ -342,6 +586,7 @@ function Chat({ model, onCommand, mode, onModeChange, sessionId }) {
 					setStreamingMessage({ ...planAssistantMessage });
 				}
 			} catch (error) {
+				// v8 ignore next
 				planAssistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
 				setMessages([...planMessages, { ...planAssistantMessage }]);
 				setStreamingMessage(null);
@@ -357,6 +602,7 @@ 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();
 		} finally {
@@ -591,10 +837,7 @@ function ModelPicker({ currentModel, onSelect, onClose }) {
 		defaultValue: currentModel,
 		onChange: onSelect,
 		onEscape: onClose,
-		children: /* @__PURE__ */ jsx(Text, {
-			dimColor: true,
-			children: "Select a model (↑↓ + Enter to confirm, Esc to cancel)"
-		})
+		children: /* @__PURE__ */ jsx(SelectPromptHint, { message: "Select a model" })
 	});
 }
 //#endregion
@@ -670,6 +913,7 @@ function App() {
 function renderApp() {
 	const tree = /* @__PURE__ */ jsx(App, {});
 	const app = render(tree, {
+		exitOnCtrlC: false,
 		incrementalRendering: true,
 		maxFps: 60
 	});

package/dist/cli.js

@@ -6,32 +6,9 @@ import { homedir } from "node:os";
 import { Ollama } from "ollama";
 import { exec } from "node:child_process";
 import { promisify } from "node:util";
-var NAMES = [{
-	name: "/clear",
-	description: "clear the current session"
-}, {
-	name: "/model",
-	description: "switch the model"
-}].map(({ name }) => name);
-//#endregion
-//#region src/constants/decision.ts
-var APPROVE = "approve";
-var REJECT = "reject";
-//#endregion
-//#region src/constants/mode.ts
-var NAME$1 = {
-	SAFE: "safe",
-	AUTO: "auto",
-	PLAN: "plan"
-};
-var LABEL = {
-	safe: "Safe",
-	auto: "Auto",
-	plan: "Plan"
-};
 //#endregion
 //#region src/constants/package.ts
-var VERSION = "0.4.0";
+var VERSION = "0.6.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
@@ -90,9 +67,6 @@ var NAME = {
 	VIEW_RANGE: "view_range"
 };
 //#endregion
-//#region src/constants/ui.ts
-var HEADER_PREFIX = "🦙";
-//#endregion
 //#region src/utils/agents.ts
 var AGENTS_FILE = "AGENTS.md";
 function loadAgentsContent() {
@@ -184,6 +158,9 @@ async function listModels() {
 }
 function setClearHandler(handler) {}
 //#endregion
+//#region src/utils/time.ts
+var tick = (ms = 0) => new Promise((resolve) => setTimeout(resolve, ms));
+//#endregion
 //#region src/utils/tools.ts
 var execAsync = promisify(exec);
 /**
@@ -276,13 +253,13 @@ var TOOLS = [
 		"end"
 	])
 ];
-var READ_ONLY_TOOLS = new Set([
+var READ_TOOLS = new Set([
 	NAME.READ_FILE,
 	NAME.LIST_DIR,
 	NAME.GREP_SEARCH,
 	NAME.VIEW_RANGE
 ]);
-var DANGEROUS_TOOLS = new Set([
+var WRITE_TOOLS = new Set([
 	NAME.WRITE_FILE,
 	NAME.EDIT_FILE,
 	NAME.RUN_SHELL
@@ -442,7 +419,7 @@ async function grepSearch(pattern, dirPath) {
 			}
 		}
 		searchDirectory(dirPath);
-		if (results.length === 0) return { content: "No matches found" };
+		if (!results.length) return { content: "No matches found" };
 		return { content: results.join("\n") };
 	} catch (error) {
 		return {
@@ -525,7 +502,7 @@ async function processRunStream(messages, model) {
 }
 async function main(args = process.argv.slice(2)) {
 	if (!args.length) {
-		const { renderApp } = await import("./assets/tui-BLJeczya.js");
+		const { renderApp } = await import("./assets/tui-UfyDivSA.js");
 		process.stdout.write("\x1Bc");
 		renderApp();
 		return;
@@ -548,4 +525,4 @@ function isEntrypoint(argv1 = process.argv[1]) {
 if (isEntrypoint()) main();
 // v8 ignore stop
 //#endregion
-export { NAME$1 as _, setClearHandler as a, NAMES as b, loadConfig as c, withSystemMessage as d, HEADER_PREFIX as f, LABEL as g, VERSION as h, executeTool as i, saveConfig as l, PLAN_GENERATION_INSTRUCTION as m, main, READ_ONLY_TOOLS as n, listModels as o, ROLE as p, TOOLS as r, streamChat as s, DANGEROUS_TOOLS as t, resetSystemMessage as u, APPROVE as v, REJECT as y };
+export { tick as a, streamChat as c, resetSystemMessage as d, withSystemMessage as f, VERSION as h, executeTool as i, loadConfig as l, PLAN_GENERATION_INSTRUCTION as m, main, TOOLS as n, setClearHandler as o, ROLE as p, WRITE_TOOLS as r, listModels as s, READ_TOOLS as t, saveConfig as u };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-ollama",
-  "version": "0.4.0",
+  "version": "0.6.0",
   "description": "Ollama coding agent that runs in your terminal",
   "author": "Mark <mark@remarkablemark.org> (https://remarkablemark.org)",
   "type": "module",
@@ -50,7 +50,7 @@
     "@commitlint/config-conventional": "20.5.3",
     "@eslint/compat": "2.0.5",
     "@eslint/js": "10.0.1",
-    "@types/node": "25.6.0",
+    "@types/node": "25.6.2",
     "@types/react": "19.2.14",
     "@vitest/coverage-v8": "4.1.5",
     "eslint": "10.3.0",
@@ -65,7 +65,7 @@
     "tsx": "4.21.0",
     "typescript": "6.0.3",
     "typescript-eslint": "8.59.2",
-    "vite": "8.0.10",
+    "vite": "8.0.11",
     "vitest": "4.1.5"
   },
   "files": [