openpoly 0.1.0

package/dist/tui/App.js

@@ -0,0 +1,276 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useEffect, useRef, useState } from "react";
+import { Box, Text, useApp, useInput, Static } from "ink";
+import TextInput from "ink-text-input";
+import { Banner } from "./Banner.js";
+import { Transcript } from "./Transcript.js";
+import { PermissionDialog } from "./PermissionDialog.js";
+import { StatusBar } from "./StatusBar.js";
+import { ModelMenu } from "./ModelMenu.js";
+import { KeyPrompt } from "./KeyPrompt.js";
+export function App({ session, broker, config, initialModel, switchModel, listModels, onSaveKey, promptKeyOnStart, version, intro, }) {
+    const { exit } = useApp();
+    const [items, setItems] = useState(intro ? [{ id: 0, kind: "system", text: intro }] : []);
+    const [input, setInput] = useState("");
+    const [busy, setBusy] = useState(false);
+    const [modelName, setModelName] = useState(initialModel);
+    const [reasoning, setReasoning] = useState(session.getReasoning());
+    // A queue, not a single slot: the model can request several permissions at
+    // once (concurrent tool calls), and dropping any would hang the whole turn.
+    const [permQueue, setPermQueue] = useState([]);
+    const permQueueRef = useRef([]);
+    const [menuOpen, setMenuOpen] = useState(false);
+    const [menuLoading, setMenuLoading] = useState(false);
+    const [menuItems, setMenuItems] = useState([]);
+    const [keyPromptOpen, setKeyPromptOpen] = useState(!!promptKeyOnStart);
+    const modelCache = useRef(null);
+    // A message typed while the AI is busy; sent automatically when it finishes.
+    const pendingRef = useRef(null);
+    const idRef = useRef(1);
+    const anyModalOpen = permQueue.length > 0 || keyPromptOpen || menuOpen;
+    // Esc cancels the in-flight turn — a guaranteed way to get the input back if
+    // a response ever stalls. Ignored while a modal owns the keyboard.
+    useInput((_input, key) => {
+        if (key.escape && busy && !anyModalOpen) {
+            pendingRef.current = null;
+            session.abort();
+        }
+    }, { isActive: busy && !anyModalOpen });
+    const nextId = () => idRef.current++;
+    const add = (item) => setItems((prev) => [...prev, { ...item, id: nextId() }]);
+    // Wire up agent + permission events once.
+    useEffect(() => {
+        const onDelta = (delta) => setItems((prev) => {
+            const last = prev[prev.length - 1];
+            if (last && last.kind === "assistant") {
+                const copy = [...prev];
+                copy[copy.length - 1] = { ...last, text: last.text + delta };
+                return copy;
+            }
+            return [...prev, { id: nextId(), kind: "assistant", text: delta }];
+        });
+        const onToolCall = ({ toolName, args }) => setItems((prev) => [
+            ...prev,
+            {
+                id: nextId(),
+                kind: "tool",
+                name: toolName,
+                summary: summarizeArgs(toolName, args),
+                status: "running",
+            },
+        ]);
+        const onToolResult = ({ toolName, result, }) => setItems((prev) => {
+            for (let i = prev.length - 1; i >= 0; i--) {
+                const it = prev[i];
+                if (it.kind === "tool" && it.name === toolName && it.status === "running") {
+                    const copy = [...prev];
+                    copy[i] = { ...it, status: "done", result: String(result ?? "") };
+                    return copy;
+                }
+            }
+            return prev;
+        });
+        const onError = (message) => add({ kind: "error", text: message });
+        const onDone = () => {
+            setBusy(false);
+            // Flush a message the user queued while the AI was working.
+            const queued = pendingRef.current;
+            if (queued) {
+                pendingRef.current = null;
+                add({ kind: "user", text: queued });
+                setBusy(true);
+                void session.send(queued);
+            }
+        };
+        const onRequest = (req) => {
+            permQueueRef.current = [...permQueueRef.current, req];
+            setPermQueue(permQueueRef.current);
+        };
+        session.on("text-delta", onDelta);
+        session.on("tool-call", onToolCall);
+        session.on("tool-result", onToolResult);
+        session.on("error", onError);
+        session.on("done", onDone);
+        broker.on("request", onRequest);
+        return () => {
+            session.off("text-delta", onDelta);
+            session.off("tool-call", onToolCall);
+            session.off("tool-result", onToolResult);
+            session.off("error", onError);
+            session.off("done", onDone);
+            broker.off("request", onRequest);
+        };
+    }, [session, broker]);
+    const decide = (decision) => {
+        const [first, ...rest] = permQueueRef.current;
+        if (!first)
+            return;
+        first.respond(decision);
+        permQueueRef.current = rest;
+        setPermQueue(rest);
+    };
+    // Configured models, used as a fallback if the live model list can't load.
+    const configMenuItems = () => config.models.map((m) => ({
+        label: `${m.name === modelName ? "● " : "  "}${m.name}`,
+        value: m.name,
+    }));
+    const infosToItems = (infos) => infos.map((m) => ({
+        label: `${m.id === modelName ? "● " : "  "}${m.id}${m.free ? "  ·free" : ""}`,
+        value: m.id,
+    }));
+    // Open the picker pre-populated with all tool-capable models (free first),
+    // fetched live and cached after the first open. Falls back to configured
+    // models on failure.
+    const openMenu = () => {
+        setMenuOpen(true);
+        if (modelCache.current) {
+            setMenuItems(infosToItems(modelCache.current));
+            setMenuLoading(false);
+            return;
+        }
+        setMenuItems([]);
+        setMenuLoading(true);
+        listModels()
+            .then((infos) => {
+            modelCache.current = infos;
+            setMenuItems(infosToItems(infos));
+            setMenuLoading(false);
+        })
+            .catch((err) => {
+            setMenuItems(configMenuItems());
+            setMenuLoading(false);
+            add({
+                kind: "error",
+                text: `Could not load free models (${err instanceof Error ? err.message : String(err)}). Showing configured models.`,
+            });
+        });
+    };
+    const handleMenuSelect = (value) => {
+        setMenuOpen(false);
+        const result = switchModel(value);
+        add({ kind: result.ok ? "system" : "error", text: result.message });
+        if (result.ok && result.name)
+            setModelName(result.name);
+    };
+    const handleSaveKey = (key) => {
+        setKeyPromptOpen(false);
+        const result = onSaveKey(key);
+        add({
+            kind: result.ok ? "system" : "error",
+            text: result.ok ? "API key saved." : result.message,
+        });
+        if (result.ok && result.name)
+            setModelName(result.name);
+    };
+    const handleSubmit = (value) => {
+        const text = value.trim();
+        setInput("");
+        if (!text)
+            return;
+        if (text.startsWith("/")) {
+            handleCommand(text);
+            return;
+        }
+        if (busy) {
+            // Don't drop it — queue and send when the current turn finishes.
+            pendingRef.current = text;
+            add({ kind: "system", text: "(queued — will send when the reply finishes)" });
+            return;
+        }
+        add({ kind: "user", text });
+        setBusy(true);
+        void session.send(text);
+    };
+    const handleCommand = (text) => {
+        const [cmd, ...rest] = text.slice(1).split(/\s+/);
+        const arg = rest.join(" ").trim();
+        switch (cmd) {
+            case "help":
+                add({
+                    kind: "system",
+                    text: [
+                        "Commands:",
+                        "  /think <level>  reasoning effort: off|low|medium|high (lower = faster)",
+                        "  /key            enter/update the API key for the current model",
+                        "  /model          open the model picker (free first, then all)",
+                        "  /models         same as /model",
+                        "  /model <name>   switch directly (any OpenRouter model id works)",
+                        "  /clear          clear the conversation",
+                        "  /help           show this help",
+                        "  /exit           quit polycode",
+                    ].join("\n"),
+                });
+                break;
+            case "think": {
+                const levels = ["off", "low", "medium", "high"];
+                if (!arg) {
+                    add({
+                        kind: "system",
+                        text: `Thinking level: ${reasoning}. Use /think <off|low|medium|high> (lower = faster).`,
+                    });
+                }
+                else if (levels.includes(arg)) {
+                    const level = arg;
+                    session.setReasoning(level);
+                    setReasoning(level);
+                    add({ kind: "system", text: `Thinking level set to ${level}.` });
+                }
+                else {
+                    add({
+                        kind: "error",
+                        text: `Unknown level "${arg}". Choose off, low, medium, or high.`,
+                    });
+                }
+                break;
+            }
+            case "key":
+                setKeyPromptOpen(true);
+                break;
+            case "models":
+                openMenu();
+                break;
+            case "model":
+                if (!arg) {
+                    openMenu();
+                }
+                else {
+                    const result = switchModel(arg);
+                    add({
+                        kind: result.ok ? "system" : "error",
+                        text: result.message,
+                    });
+                    if (result.ok && result.name)
+                        setModelName(result.name);
+                }
+                break;
+            case "clear":
+                session.clearHistory();
+                setItems([]);
+                break;
+            case "exit":
+            case "quit":
+                exit();
+                break;
+            default:
+                add({ kind: "error", text: `Unknown command: /${cmd}` });
+        }
+    };
+    return (_jsxs(Box, { flexDirection: "column", paddingX: 1, children: [_jsx(Static, { items: [version], children: (v) => _jsx(Banner, { version: v }, "banner") }), _jsx(Transcript, { items: items }), permQueue.length > 0 ? (_jsx(PermissionDialog, { request: permQueue[0], queued: permQueue.length - 1, onDecide: decide })) : keyPromptOpen ? (_jsx(KeyPrompt, { label: `Enter the API key for ${modelName}`, hint: "OpenRouter keys are free (no credit card): https://openrouter.ai/keys", onSubmit: handleSaveKey, onCancel: () => setKeyPromptOpen(false) })) : menuOpen ? (_jsx(ModelMenu, { items: menuItems, current: modelName, loading: menuLoading, onSelect: handleMenuSelect, onCancel: () => setMenuOpen(false) })) : (_jsxs(Box, { borderStyle: "round", borderColor: busy ? "yellow" : "cyan", borderBottom: false, borderLeft: false, borderRight: false, marginTop: 1, children: [_jsx(Text, { color: busy ? "yellow" : "cyan", bold: true, children: "❯ " }), _jsx(TextInput, { value: input, onChange: setInput, onSubmit: handleSubmit, placeholder: busy
+                            ? "Working… type to queue a message, or Esc to cancel"
+                            : "Ask polycode to build or change something…" })] })), _jsx(StatusBar, { modelName: modelName, busy: busy, reasoning: reasoning })] }));
+}
+function summarizeArgs(toolName, args) {
+    const a = (args ?? {});
+    switch (toolName) {
+        case "read_file":
+        case "write_file":
+        case "edit_file":
+            return String(a.path ?? "");
+        case "run_command":
+            return String(a.command ?? "");
+        case "search_code":
+            return String(a.pattern ?? "");
+        default:
+            return "";
+    }
+}

package/dist/tui/Banner.js

@@ -0,0 +1,9 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Box, Text } from "ink";
+/**
+ * One-time header shown at the top of the session. Rendered inside <Static> so
+ * it's printed once and scrolls naturally with the conversation.
+ */
+export function Banner({ version }) {
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, marginBottom: 1, children: [_jsxs(Text, { children: [_jsx(Text, { color: "cyan", bold: true, children: "\u25C6 polycode" }), _jsx(Text, { dimColor: true, children: `  v${version}` })] }), _jsx(Text, { dimColor: true, children: "free, open-source AI models in your terminal \u2014 type a request, or /help" })] }));
+}

package/dist/tui/KeyPrompt.js

@@ -0,0 +1,19 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useState } from "react";
+import { Box, Text, useInput } from "ink";
+import TextInput from "ink-text-input";
+/**
+ * Inline prompt to paste an API key straight into the terminal (masked). The
+ * key is saved locally and reused on later runs. Enter saves, Esc cancels.
+ */
+export function KeyPrompt({ label, hint, onSubmit, onCancel, }) {
+    const [value, setValue] = useState("");
+    useInput((_input, key) => {
+        if (key.escape)
+            onCancel();
+    });
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, marginBottom: 1, children: [_jsx(Text, { color: "cyan", bold: true, children: label }), hint ? _jsx(Text, { dimColor: true, children: hint }) : null, _jsxs(Box, { marginTop: 1, children: [_jsx(Text, { children: "key: " }), _jsx(TextInput, { value: value, onChange: setValue, onSubmit: (v) => {
+                            if (v.trim())
+                                onSubmit(v.trim());
+                        }, mask: "*", placeholder: "paste your key and press enter" })] }), _jsx(Text, { dimColor: true, children: "enter to save \u00B7 esc to cancel" })] }));
+}

package/dist/tui/ModelMenu.js

@@ -0,0 +1,22 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { useState } from "react";
+import { Box, Text, useInput } from "ink";
+import SelectInput from "ink-select-input";
+import TextInput from "ink-text-input";
+/**
+ * Interactive model picker (like Claude Code's): opens already populated with
+ * the free models available. Type to filter, arrow keys to move, enter to pick,
+ * esc to cancel.
+ */
+export function ModelMenu({ items, current, loading, onSelect, onCancel, }) {
+    const [query, setQuery] = useState("");
+    useInput((_input, key) => {
+        if (key.escape)
+            onCancel();
+    });
+    const q = query.trim().toLowerCase();
+    const filtered = q
+        ? items.filter((i) => i.value.toLowerCase().includes(q))
+        : items;
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "magenta", paddingX: 1, marginBottom: 1, children: [_jsxs(Text, { color: "magenta", bold: true, children: ["Select a model", items.length ? ` (${items.length} available)` : ""] }), _jsxs(Text, { dimColor: true, children: ["type to filter \u00B7 \u2191\u2193 move \u00B7 enter select \u00B7 esc cancel \u00B7 current: ", current] }), loading ? (_jsx(Box, { marginTop: 1, children: _jsx(Text, { color: "yellow", children: "Loading free models\u2026" }) })) : (_jsxs(Box, { flexDirection: "column", marginTop: 1, children: [_jsxs(Box, { children: [_jsx(Text, { dimColor: true, children: "filter: " }), _jsx(TextInput, { value: query, onChange: setQuery, placeholder: "type to search\u2026" })] }), filtered.length === 0 ? (_jsxs(Text, { dimColor: true, children: ["no models match \"", query, "\""] })) : (_jsx(SelectInput, { items: filtered, limit: 10, onSelect: (item) => onSelect(item.value) }, q))] }))] }));
+}

package/dist/tui/PermissionDialog.js

@@ -0,0 +1,17 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { Box, Text, useInput } from "ink";
+/**
+ * Modal-style prompt shown when a tool requests permission. Captures keystrokes
+ * while visible: (y) allow once, (a) always allow this tool, (n/esc) deny.
+ */
+export function PermissionDialog({ request, queued = 0, onDecide, }) {
+    useInput((input, key) => {
+        if (key.escape || input === "n" || input === "d")
+            onDecide("deny");
+        else if (input === "y" || key.return)
+            onDecide("allow");
+        else if (input === "a")
+            onDecide("always");
+    });
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "yellow", paddingX: 1, marginBottom: 1, children: [_jsxs(Text, { color: "yellow", bold: true, children: ["Permission required", queued > 0 ? `  (${queued} more pending)` : ""] }), _jsx(Text, { children: request.title }), request.detail ? (_jsx(Box, { marginTop: 1, flexDirection: "column", children: _jsx(Text, { dimColor: true, children: request.detail }) })) : null, _jsx(Box, { marginTop: 1, children: _jsxs(Text, { children: [_jsx(Text, { color: "green", children: "(y)" }), " allow", "  ", _jsx(Text, { color: "cyan", children: "(a)" }), " always allow", "  ", _jsx(Text, { color: "red", children: "(n)" }), " deny"] }) })] }));
+}

package/dist/tui/StatusBar.js

@@ -0,0 +1,6 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Box, Text } from "ink";
+import Spinner from "ink-spinner";
+export function StatusBar({ modelName, busy, reasoning, }) {
+    return (_jsxs(Box, { marginTop: 1, children: [_jsx(Text, { dimColor: true, children: "model: " }), _jsx(Text, { color: "magenta", children: modelName }), reasoning ? _jsx(Text, { dimColor: true, children: `  ·  think:${reasoning}` }) : null, _jsx(Text, { dimColor: true, children: "  ·  " }), busy ? (_jsxs(Text, { color: "yellow", children: [_jsx(Spinner, { type: "dots" }), " thinking\u2026", " ", _jsx(Text, { dimColor: true, children: "(Esc to cancel)" })] })) : (_jsx(Text, { dimColor: true, children: "/help for commands \u00B7 ctrl+c to exit" }))] }));
+}

package/dist/tui/Transcript.js

@@ -0,0 +1,44 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { Box, Text } from "ink";
+const TOOL_LABEL = {
+    read_file: "read",
+    write_file: "write",
+    edit_file: "edit",
+    run_command: "run",
+    search_code: "search",
+};
+/** A tool whose result indicates a denial or error gets a red marker. */
+function toolTone(item) {
+    if (item.status === "running")
+        return { icon: "◐", color: "yellow" };
+    const r = item.result ?? "";
+    if (/^Denied|^Error/i.test(r))
+        return { icon: "⏺", color: "red" };
+    return { icon: "⏺", color: "green" };
+}
+function ToolView({ item }) {
+    const label = TOOL_LABEL[item.name] ?? item.name;
+    const { icon, color } = toolTone(item);
+    return (_jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [_jsxs(Text, { children: [_jsxs(Text, { color: color, children: [icon, " "] }), _jsx(Text, { color: color, bold: true, children: label }), item.summary ? _jsx(Text, { dimColor: true, children: `  ${item.summary}` }) : null] }), item.status === "done" && item.result ? (_jsx(Box, { marginLeft: 2, flexDirection: "column", children: _jsx(Text, { dimColor: true, children: truncate(item.result, 600) }) })) : null] }));
+}
+export function Transcript({ items }) {
+    return (_jsx(Box, { flexDirection: "column", children: items.map((item) => {
+            switch (item.kind) {
+                case "user":
+                    return (_jsxs(Box, { marginTop: 1, marginBottom: 1, children: [_jsx(Text, { color: "cyan", bold: true, children: "❯ " }), _jsx(Text, { bold: true, children: item.text })] }, item.id));
+                case "assistant":
+                    return (_jsx(Box, { marginBottom: 1, children: _jsx(Text, { children: item.text }) }, item.id));
+                case "tool":
+                    return _jsx(ToolView, { item: item }, item.id);
+                case "error":
+                    return (_jsxs(Box, { marginBottom: 1, children: [_jsx(Text, { color: "red", children: "✗ " }), _jsx(Text, { color: "red", children: item.text })] }, item.id));
+                case "system":
+                    return (_jsx(Box, { marginBottom: 1, children: _jsx(Text, { dimColor: true, children: item.text }) }, item.id));
+            }
+        }) }));
+}
+function truncate(s, max) {
+    if (s.length <= max)
+        return s;
+    return s.slice(0, max) + `\n… (${s.length - max} more chars)`;
+}

package/dist/tui/types.js

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "openpoly",
+  "version": "0.1.0",
+  "description": "An open-source, model-agnostic terminal coding agent. Free open-source models (DeepSeek, GLM, Qwen, Llama) via OpenRouter or local Ollama — or bring your own GPT/Claude/Gemini keys.",
+  "type": "module",
+  "bin": {
+    "polycode": "dist/cli.js",
+    "openpoly": "dist/cli.js"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "dev": "tsx src/cli.tsx",
+    "build": "tsc",
+    "start": "node dist/cli.js",
+    "typecheck": "tsc --noEmit",
+    "prepare": "npm run build",
+    "install:global": "npm run build && npm install -g ."
+  },
+  "keywords": [
+    "ai",
+    "agent",
+    "cli",
+    "tui",
+    "coding-assistant",
+    "llm",
+    "openai",
+    "anthropic",
+    "ollama"
+  ],
+  "license": "MIT",
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@ai-sdk/anthropic": "^1.2.12",
+    "@ai-sdk/google": "^1.2.22",
+    "@ai-sdk/openai": "^1.3.24",
+    "@ai-sdk/openai-compatible": "^0.2.16",
+    "ai": "^4.3.19",
+    "ink": "^5.1.0",
+    "ink-select-input": "^6.0.0",
+    "ink-spinner": "^5.0.0",
+    "ink-text-input": "^6.0.0",
+    "react": "^18.3.1",
+    "zod": "^3.25.76"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.0",
+    "@types/react": "^18.3.12",
+    "tsx": "^4.19.2",
+    "typescript": "^5.7.2"
+  }
+}