code-ollama 0.1.0 → 0.2.0

package/dist/assets/tui-DSR1MJGd.js

@@ -0,0 +1,438 @@
+import { a as streamChat, c as createSystemMessage, i as listModels, l as ROLE, n as TOOLS_REQUIRING_APPROVAL, o as loadConfig, r as executeTool, s as saveConfig, t as TOOLS, u as VERSION } from "../cli.js";
+import { homedir } from "node:os";
+import { Box, Text, render, useInput } from "ink";
+import { useCallback, useEffect, useState } from "react";
+import { Select, Spinner, TextInput } from "@inkjs/ui";
+import { jsx, jsxs } from "react/jsx-runtime";
+//#region src/constants/commands.ts
+var COMMANDS = [{
+	name: "/model",
+	description: "switch the model"
+}];
+//#endregion
+//#region src/constants/ui.ts
+var HEADER_PREFIX = "🦙";
+//#endregion
+//#region src/components/Autocomplete.tsx
+function getMatches(input) {
+	if (!input.startsWith("/")) return [];
+	return COMMANDS.filter((command) => command.name.startsWith(input));
+}
+function Autocomplete({ isDisabled = false, onSubmit }) {
+	const [value, setValue] = useState("");
+	const [selectedIndex, setSelectedIndex] = useState(0);
+	const [inputKey, setInputKey] = useState(0);
+	const matches = getMatches(value);
+	const isCommandMode = value.startsWith("/");
+	useInput((_char, key) => {
+		// v8 ignore next
+		if (!isCommandMode) return;
+		if (key.upArrow) {
+			setSelectedIndex((i) => Math.max(0, i - 1));
+			return;
+		}
+		if (key.downArrow) {
+			setSelectedIndex((i) => Math.min(matches.length - 1, i + 1));
+			return;
+		}
+		if (key.tab && matches.length > 0) {
+			setValue((matches[selectedIndex] ?? matches[0]).name);
+			setSelectedIndex(0);
+			setInputKey((key) => key + 1);
+			return;
+		}
+	}, { isActive: !isDisabled && isCommandMode });
+	const handleSubmit = useCallback((input) => {
+		const trimmed = (isCommandMode && matches.length > 0 && matches[selectedIndex] ? matches[selectedIndex].name : input).trim();
+		if (trimmed) {
+			onSubmit(trimmed);
+			setValue("");
+			setSelectedIndex(0);
+			setInputKey((key) => key + 1);
+		}
+	}, [
+		isCommandMode,
+		matches,
+		onSubmit,
+		selectedIndex
+	]);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [/* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, { children: "> " }), /* @__PURE__ */ jsx(TextInput, {
+			isDisabled,
+			defaultValue: value,
+			onChange: setValue,
+			onSubmit: handleSubmit
+		}, inputKey)] }), isCommandMode && matches.length > 0 && /* @__PURE__ */ jsx(Box, {
+			flexDirection: "column",
+			marginLeft: 2,
+			children: matches.map((command, index) => {
+				const isHighlighted = index === selectedIndex;
+				return /* @__PURE__ */ jsxs(Box, {
+					gap: 3,
+					children: [/* @__PURE__ */ jsx(Text, {
+						color: isHighlighted ? "cyan" : void 0,
+						bold: isHighlighted,
+						children: command.name
+					}), /* @__PURE__ */ jsx(Text, {
+						dimColor: true,
+						children: command.description
+					})]
+				}, command.name);
+			})
+		})]
+	});
+}
+//#endregion
+//#region src/components/Messages.tsx
+function getMessageColor(role) {
+	switch (role) {
+		case ROLE.USER: return "black";
+		case ROLE.ASSISTANT: return "blue";
+		case ROLE.SYSTEM: return "gray";
+		default: return;
+	}
+}
+function Messages({ messages, isLoading }) {
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [messages.map((message, index) => /* @__PURE__ */ jsx(Box, {
+			marginBottom: 1,
+			children: /* @__PURE__ */ jsxs(Text, {
+				color: getMessageColor(message.role),
+				dimColor: message.role === ROLE.SYSTEM,
+				children: [message.role === ROLE.USER ? "> " : "", message.content]
+			})
+		}, index)), isLoading && messages[messages.length - 1]?.content === "" && /* @__PURE__ */ jsx(Box, {
+			marginTop: -1,
+			marginBottom: 1,
+			children: /* @__PURE__ */ jsx(Spinner, { label: "Thinking..." })
+		})]
+	});
+}
+//#endregion
+//#region src/components/ToolApproval.tsx
+function ToolApproval({ toolCall, onApprove, onReject }) {
+	const [selected, setSelected] = useState("yes");
+	useInput((_, key) => {
+		if (key.return) if (selected === "yes") onApprove();
+		else onReject();
+		else if (key.leftArrow || key.rightArrow) setSelected((prev) => prev === "yes" ? "no" : "yes");
+		// v8 ignore stop
+	});
+	const args = JSON.stringify(toolCall.function.arguments, null, 2);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		marginY: 1,
+		children: [
+			/* @__PURE__ */ jsx(Text, {
+				color: "yellow",
+				bold: true,
+				children: "⚠️ Tool requires approval:"
+			}),
+			/* @__PURE__ */ jsxs(Box, {
+				marginX: 2,
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsxs(Text, { children: [
+					/* @__PURE__ */ jsx(Text, {
+						bold: true,
+						children: "Tool:"
+					}),
+					" ",
+					toolCall.function.name
+				] }), /* @__PURE__ */ jsxs(Text, { children: [
+					/* @__PURE__ */ jsx(Text, {
+						bold: true,
+						children: "Arguments:"
+					}),
+					" ",
+					args
+				] })]
+			}),
+			/* @__PURE__ */ jsxs(Box, {
+				marginTop: 1,
+				gap: 2,
+				children: [/* @__PURE__ */ jsx(Text, { children: /* @__PURE__ */ jsxs(Text, {
+					color: selected === "yes" ? "green" : void 0,
+					children: [selected === "yes" ? "▶ " : "  ", "✓ Yes (Enter)"]
+				}) }), /* @__PURE__ */ jsx(Text, { children: /* @__PURE__ */ jsxs(Text, {
+					color: selected === "no" ? "red" : void 0,
+					children: [selected === "no" ? "▶ " : "  ", "✗ No (Esc)"]
+				}) })]
+			})
+		]
+	});
+}
+//#endregion
+//#region src/components/Chat.tsx
+function Chat({ model, onCommand, autoExecute }) {
+	const [messages, setMessages] = useState([createSystemMessage()]);
+	const [submitKey, setSubmitKey] = useState(0);
+	const [isLoading, setIsLoading] = useState(false);
+	const [pendingToolCall, setPendingToolCall] = useState(null);
+	const processStream = useCallback(async (currentMessages) => {
+		const assistantMessage = {
+			role: ROLE.ASSISTANT,
+			content: ""
+		};
+		setMessages((previousMessages) => [...previousMessages, assistantMessage]);
+		try {
+			for await (const chunk of streamChat(currentMessages, model, TOOLS)) if (chunk.type === "content") {
+				assistantMessage.content += chunk.content;
+				setMessages((previousMessages) => {
+					const newMessages = [...previousMessages];
+					newMessages[newMessages.length - 1] = { ...assistantMessage };
+					return newMessages;
+				});
+			} else if (chunk.type === "tool_calls") for (const toolCall of chunk.tool_calls) {
+				const requiresApproval = TOOLS_REQUIRING_APPROVAL.has(toolCall.function.name);
+				if (!autoExecute && requiresApproval) {
+					setPendingToolCall(toolCall);
+					setIsLoading(false);
+					return;
+				}
+				const result = await executeTool(toolCall.function.name, toolCall.function.arguments);
+				const toolResultMessage = {
+					role: ROLE.SYSTEM,
+					content: `Tool ${toolCall.function.name} result:\n${result.content}${result.error ? `\nError: ${result.error}` : ""}`
+				};
+				const newMessages = [
+					...currentMessages,
+					assistantMessage,
+					toolResultMessage
+				];
+				setMessages((previousMessages) => [...previousMessages, toolResultMessage]);
+				await processStream(newMessages);
+				return;
+			}
+		} catch (error) {
+			assistantMessage.content = `Error: ${error instanceof Error ? error.message : String(error)}`;
+			setMessages((previousMessages) => {
+				const newMessages = [...previousMessages];
+				newMessages[newMessages.length - 1] = { ...assistantMessage };
+				return newMessages;
+			});
+		} finally {
+			setIsLoading(false);
+		}
+	}, [model, autoExecute]);
+	const handleToolApproval = useCallback(async (approved) => {
+		// v8 ignore next
+		if (!pendingToolCall) return;
+		const toolCall = pendingToolCall;
+		setPendingToolCall(null);
+		setIsLoading(true);
+		if (approved) {
+			const result = await executeTool(toolCall.function.name, toolCall.function.arguments);
+			const toolResultMessage = {
+				role: ROLE.SYSTEM,
+				content: `Tool ${toolCall.function.name} result:\n${result.content}${result.error ? `\nError: ${result.error}` : ""}`
+			};
+			const newMessages = [...messages, toolResultMessage];
+			setMessages((previousMessages) => [...previousMessages, toolResultMessage]);
+			await processStream(newMessages);
+		} else {
+			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);
+		}
+	}, [
+		pendingToolCall,
+		messages,
+		processStream
+	]);
+	const handleSubmit = useCallback(async (value) => {
+		const userContent = value.trim();
+		if (!userContent) return;
+		setSubmitKey((key) => key + 1);
+		if (userContent.startsWith("/")) {
+			onCommand(userContent);
+			return;
+		}
+		setIsLoading(true);
+		const userMessage = {
+			role: ROLE.USER,
+			content: userContent
+		};
+		setMessages((previousMessages) => [...previousMessages, userMessage]);
+		await processStream([...messages, userMessage]);
+	}, [
+		messages,
+		onCommand,
+		processStream
+	]);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [
+			/* @__PURE__ */ jsx(Messages, {
+				messages: messages.slice(1),
+				isLoading
+			}),
+			pendingToolCall && /* @__PURE__ */ jsx(ToolApproval, {
+				toolCall: pendingToolCall,
+				onApprove: () => void handleToolApproval(true),
+				onReject: () => void handleToolApproval(false)
+			}),
+			!pendingToolCall && /* @__PURE__ */ jsx(Autocomplete, {
+				isDisabled: isLoading,
+				onSubmit: (val) => {
+					handleSubmit(val);
+				}
+			}, submitKey)
+		]
+	});
+}
+//#endregion
+//#region src/components/Footer.tsx
+function Footer({ autoExecute, onToggleMode }) {
+	useInput((_, key) => {
+		if (key.tab && key.shift) onToggleMode();
+	});
+	return /* @__PURE__ */ jsx(Box, {
+		justifyContent: "space-between",
+		marginTop: 1,
+		children: /* @__PURE__ */ jsxs(Text, {
+			dimColor: true,
+			children: [
+				"Mode: ",
+				autoExecute ? "Auto" : "Safe",
+				" (Shift+Tab to toggle)"
+			]
+		})
+	});
+}
+//#endregion
+//#region src/components/Header.tsx
+function abbreviatePath(dir) {
+	const home = homedir();
+	return dir.startsWith(home) ? `~${dir.slice(home.length)}` : dir;
+}
+function Header({ model }) {
+	const directory = abbreviatePath(process.cwd());
+	return /* @__PURE__ */ jsxs(Box, {
+		borderStyle: "round",
+		flexDirection: "column",
+		paddingX: 1,
+		children: [
+			/* @__PURE__ */ jsxs(Text, { children: [/* @__PURE__ */ jsxs(Text, {
+				bold: true,
+				children: [HEADER_PREFIX, "Code Ollama"]
+			}), /* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [
+					" (v",
+					VERSION,
+					")"
+				]
+			})] }),
+			/* @__PURE__ */ jsx(Text, { children: " " }),
+			/* @__PURE__ */ jsxs(Box, { children: [
+				/* @__PURE__ */ jsx(Text, {
+					dimColor: true,
+					children: "model:".padEnd(11)
+				}),
+				/* @__PURE__ */ jsxs(Text, { children: [model, "   "] }),
+				/* @__PURE__ */ jsx(Text, {
+					color: "cyan",
+					children: "/model"
+				}),
+				/* @__PURE__ */ jsx(Text, {
+					dimColor: true,
+					children: " to switch"
+				})
+			] }),
+			/* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, {
+				dimColor: true,
+				children: "directory:".padEnd(11)
+			}), /* @__PURE__ */ jsx(Text, { children: directory })] })
+		]
+	});
+}
+//#endregion
+//#region src/components/ModelPicker.tsx
+function ModelPicker({ currentModel, onSelect, onCancel }) {
+	const [options, setOptions] = useState([]);
+	const [error, setError] = useState(null);
+	useEffect(() => {
+		async function load() {
+			try {
+				setOptions((await listModels()).map((name) => ({
+					label: name,
+					value: name
+				})));
+			} catch (err) {
+				setError(err instanceof Error ? err.message : String(err));
+			}
+		}
+		load();
+	}, []);
+	useInput((_, key) => {
+		if (key.escape) onCancel();
+	});
+	if (error) return /* @__PURE__ */ jsxs(Text, {
+		color: "red",
+		children: ["Error loading models: ", error]
+	});
+	if (!options.length) return /* @__PURE__ */ jsx(Spinner, { label: "Loading models..." });
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [/* @__PURE__ */ jsx(Text, {
+			dimColor: true,
+			children: "Select a model (↑↓ + Enter to confirm, Esc to cancel)"
+		}), /* @__PURE__ */ jsx(Select, {
+			options,
+			defaultValue: currentModel,
+			onChange: onSelect
+		})]
+	});
+}
+//#endregion
+//#region src/components/App.tsx
+function App() {
+	const [model, setModel] = useState(() => loadConfig().model);
+	const [picking, setPicking] = useState(false);
+	const [autoExecute, setAutoExecute] = useState(false);
+	const handleCommand = useCallback((command) => {
+		if (command === "/model") setPicking(true);
+	}, []);
+	const handleSelect = useCallback((selected) => {
+		setModel(selected);
+		saveConfig({ model: selected });
+		setPicking(false);
+	}, []);
+	const handleCancel = useCallback(() => {
+		setPicking(false);
+	}, []);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [
+			/* @__PURE__ */ jsx(Header, { model }),
+			picking ? /* @__PURE__ */ jsx(ModelPicker, {
+				currentModel: model,
+				onSelect: handleSelect,
+				onCancel: handleCancel
+			}) : /* @__PURE__ */ jsx(Chat, {
+				model,
+				onCommand: handleCommand,
+				autoExecute
+			}),
+			/* @__PURE__ */ jsx(Footer, {
+				autoExecute,
+				onToggleMode: () => {
+					setAutoExecute((isAutoExecute) => !isAutoExecute);
+				}
+			})
+		]
+	});
+}
+//#endregion
+//#region src/tui.tsx
+function renderApp() {
+	render(/* @__PURE__ */ jsx(App, {}));
+}
+//#endregion
+export { renderApp };