oh-pi 0.1.1 → 0.1.3

package/dist/index.js

@@ -1,3 +1,4 @@
+import * as p from "@clack/prompts";
 import { welcome } from "./tui/welcome.js";
 import { selectMode } from "./tui/mode-select.js";
 import { setupProviders } from "./tui/provider-setup.js";
@@ -49,6 +50,34 @@ async function customFlow(env) {
     const keybindings = await selectKeybindings();
     const extensions = await selectExtensions();
     const agents = await selectAgents();
+    // Advanced: auto-compaction threshold
+    const wantAdvanced = await p.confirm({
+        message: "Configure advanced settings? (compaction threshold, etc.)",
+        initialValue: false,
+    });
+    if (p.isCancel(wantAdvanced)) {
+        p.cancel("Cancelled.");
+        process.exit(0);
+    }
+    let compactThreshold = 0.75;
+    if (wantAdvanced) {
+        const threshold = await p.text({
+            message: "Auto-compact when context reaches % of window (0-100):",
+            placeholder: "75",
+            initialValue: "75",
+            validate: (v) => {
+                const n = Number(v);
+                if (isNaN(n) || n < 10 || n > 100)
+                    return "Must be a number between 10 and 100";
+                return undefined;
+            },
+        });
+        if (p.isCancel(threshold)) {
+            p.cancel("Cancelled.");
+            process.exit(0);
+        }
+        compactThreshold = Number(threshold) / 100;
+    }
     return {
         providers,
         theme,
@@ -58,5 +87,6 @@ async function customFlow(env) {
         prompts: ["review", "fix", "explain", "commit", "test", "refactor", "optimize", "security", "document", "pr"],
         agents,
         thinking: "medium",
+        compactThreshold,
     };
 }

package/dist/tui/confirm-apply.js

@@ -12,6 +12,7 @@ export async function confirmApply(config, env) {
         `Theme:        ${chalk.cyan(config.theme)}`,
         `Keybindings:  ${chalk.cyan(config.keybindings)}`,
         `Thinking:     ${chalk.cyan(config.thinking)}`,
+        `Compaction:   ${chalk.cyan(`${Math.round((config.compactThreshold ?? 0.75) * 100)}% of context`)}`,
         `Extensions:   ${chalk.cyan(config.extensions.join(", ") || "none")}`,
         `Skills:       ${chalk.cyan(config.skills.join(", ") || "none")}`,
         `Prompts:      ${chalk.cyan(`${config.prompts.length} templates`)}`,

package/dist/tui/provider-setup.js

@@ -142,7 +142,66 @@ async function setupCustomProvider() {
         defaultModel = model;
     }
     p.log.success(`${name} configured (${baseUrl})`);
-    return { name, apiKey, defaultModel, baseUrl };
+    // Model capabilities (optional)
+    const wantCaps = await p.confirm({
+        message: "Configure model capabilities? (context window, multimodal, reasoning)",
+        initialValue: false,
+    });
+    if (p.isCancel(wantCaps)) {
+        p.cancel("Cancelled.");
+        process.exit(0);
+    }
+    let contextWindow;
+    let maxTokens;
+    let reasoning;
+    let multimodal;
+    if (wantCaps) {
+        const ctxInput = await p.text({
+            message: "Context window size (tokens):",
+            placeholder: "128000",
+            initialValue: "128000",
+            validate: (v) => {
+                const n = Number(v);
+                if (isNaN(n) || n < 1024)
+                    return "Must be a number ≥ 1024";
+                return undefined;
+            },
+        });
+        if (p.isCancel(ctxInput)) {
+            p.cancel("Cancelled.");
+            process.exit(0);
+        }
+        contextWindow = Number(ctxInput);
+        const maxTokInput = await p.text({
+            message: "Max output tokens:",
+            placeholder: "8192",
+            initialValue: "8192",
+            validate: (v) => {
+                const n = Number(v);
+                if (isNaN(n) || n < 256)
+                    return "Must be a number ≥ 256";
+                return undefined;
+            },
+        });
+        if (p.isCancel(maxTokInput)) {
+            p.cancel("Cancelled.");
+            process.exit(0);
+        }
+        maxTokens = Number(maxTokInput);
+        const isMultimodal = await p.confirm({ message: "Supports image input (multimodal)?", initialValue: false });
+        if (p.isCancel(isMultimodal)) {
+            p.cancel("Cancelled.");
+            process.exit(0);
+        }
+        multimodal = isMultimodal;
+        const isReasoning = await p.confirm({ message: "Supports extended thinking (reasoning)?", initialValue: false });
+        if (p.isCancel(isReasoning)) {
+            p.cancel("Cancelled.");
+            process.exit(0);
+        }
+        reasoning = isReasoning;
+    }
+    return { name, apiKey, defaultModel, baseUrl, contextWindow, maxTokens, reasoning, multimodal };
 }
 async function selectModel(label, staticModels, baseUrl, apiKey) {
     let models = staticModels;

package/dist/types.d.ts

@@ -3,6 +3,10 @@ export interface ProviderConfig {
     apiKey: string;
     defaultModel?: string;
     baseUrl?: string;
+    contextWindow?: number;
+    maxTokens?: number;
+    reasoning?: boolean;
+    multimodal?: boolean;
 }
 export interface OhPConfig {
     providers: ProviderConfig[];
@@ -13,7 +17,16 @@ export interface OhPConfig {
     prompts: string[];
     agents: string;
     thinking: string;
+    compactThreshold?: number;
 }
+/** Official model capabilities for known providers */
+export interface ModelCapabilities {
+    contextWindow: number;
+    maxTokens: number;
+    reasoning: boolean;
+    input: ("text" | "image")[];
+}
+export declare const MODEL_CAPABILITIES: Record<string, ModelCapabilities>;
 export declare const PROVIDERS: Record<string, {
     env: string;
     label: string;

package/dist/types.js

@@ -1,3 +1,23 @@
+export const MODEL_CAPABILITIES = {
+    // Anthropic
+    "claude-sonnet-4-20250514": { contextWindow: 200000, maxTokens: 16384, reasoning: true, input: ["text", "image"] },
+    "claude-opus-4-0520": { contextWindow: 200000, maxTokens: 16384, reasoning: true, input: ["text", "image"] },
+    // OpenAI
+    "gpt-4o": { contextWindow: 128000, maxTokens: 16384, reasoning: false, input: ["text", "image"] },
+    "o3-mini": { contextWindow: 128000, maxTokens: 65536, reasoning: true, input: ["text"] },
+    // Google
+    "gemini-2.5-pro": { contextWindow: 1048576, maxTokens: 65536, reasoning: true, input: ["text", "image"] },
+    "gemini-2.5-flash": { contextWindow: 1048576, maxTokens: 65536, reasoning: true, input: ["text", "image"] },
+    // Groq
+    "llama-3.3-70b-versatile": { contextWindow: 128000, maxTokens: 32768, reasoning: false, input: ["text"] },
+    // OpenRouter
+    "anthropic/claude-sonnet-4": { contextWindow: 200000, maxTokens: 16384, reasoning: true, input: ["text", "image"] },
+    "openai/gpt-4o": { contextWindow: 128000, maxTokens: 16384, reasoning: false, input: ["text", "image"] },
+    // xAI
+    "grok-3": { contextWindow: 131072, maxTokens: 16384, reasoning: false, input: ["text", "image"] },
+    // Mistral
+    "mistral-large-latest": { contextWindow: 128000, maxTokens: 8192, reasoning: false, input: ["text"] },
+};
 export const PROVIDERS = {
     anthropic: { env: "ANTHROPIC_API_KEY", label: "Anthropic (Claude)", models: ["claude-sonnet-4-20250514", "claude-opus-4-0520"] },
     openai: { env: "OPENAI_API_KEY", label: "OpenAI (GPT)", models: ["gpt-4o", "o3-mini"] },
@@ -21,6 +41,7 @@ export const EXTENSIONS = [
     { name: "safe-guard", label: "🛡️  Safe Guard — Dangerous command confirm + path protection", default: true },
     { name: "git-guard", label: "📦 Git Guard — Auto stash checkpoint + dirty repo warning + notify", default: true },
     { name: "auto-session-name", label: "📝 Auto Session Name — Name sessions from first message", default: true },
+    { name: "custom-footer", label: "📊 Custom Footer — Enhanced status bar with tokens, cost, time, git, cwd", default: false },
     { name: "ant-colony", label: "🐜 Ant Colony — Autonomous multi-agent swarm with adaptive concurrency", default: false },
 ];
 export const KEYBINDING_SCHEMES = {

package/dist/utils/install.js

@@ -3,7 +3,7 @@ import { join, dirname } from "node:path";
 import { fileURLToPath } from "node:url";
 import { homedir } from "node:os";
 import { execSync } from "node:child_process";
-import { KEYBINDING_SCHEMES, PROVIDERS } from "../types.js";
+import { KEYBINDING_SCHEMES, MODEL_CAPABILITIES, PROVIDERS } from "../types.js";
 const PKG_ROOT = join(dirname(fileURLToPath(import.meta.url)), "..", "..");
 function ensureDir(dir) {
     mkdirSync(dir, { recursive: true });
@@ -35,13 +35,18 @@ export function applyConfig(config) {
     // 2. settings.json
     const primary = config.providers[0];
     const providerInfo = primary ? PROVIDERS[primary.name] : undefined;
+    const compactThreshold = config.compactThreshold ?? 0.75;
+    const primaryModel = primary?.defaultModel ?? providerInfo?.models[0];
+    const primaryCaps = primaryModel ? MODEL_CAPABILITIES[primaryModel] : undefined;
+    const contextWindow = primary?.contextWindow ?? primaryCaps?.contextWindow ?? 128000;
+    const reserveTokens = Math.round(contextWindow * (1 - compactThreshold));
     const settings = {
         defaultProvider: primary?.name,
-        defaultModel: primary?.defaultModel ?? providerInfo?.models[0],
+        defaultModel: primaryModel,
         defaultThinkingLevel: config.thinking,
         theme: config.theme,
         enableSkillCommands: true,
-        compaction: { enabled: true, reserveTokens: 16384, keepRecentTokens: 20000 },
+        compaction: { enabled: true, reserveTokens, keepRecentTokens: 20000 },
         retry: { enabled: true, maxRetries: 3 },
     };
     if (config.providers.length > 1) {
@@ -56,6 +61,7 @@ export function applyConfig(config) {
     if (customProviders.length > 0) {
         const models = {};
         for (const cp of customProviders) {
+            const caps = cp.defaultModel ? MODEL_CAPABILITIES[cp.defaultModel] : undefined;
             models[cp.name] = {
                 baseUrl: cp.baseUrl,
                 apiKey: cp.apiKey === "none" ? undefined : cp.apiKey,
@@ -63,10 +69,10 @@ export function applyConfig(config) {
                 models: cp.defaultModel ? [{
                         id: cp.defaultModel,
                         name: cp.defaultModel,
-                        reasoning: false,
-                        input: ["text"],
-                        contextWindow: 128000,
-                        maxTokens: 8192,
+                        reasoning: cp.reasoning ?? caps?.reasoning ?? false,
+                        input: cp.multimodal ? ["text", "image"] : (caps?.input ?? ["text"]),
+                        contextWindow: cp.contextWindow ?? caps?.contextWindow ?? 128000,
+                        maxTokens: cp.maxTokens ?? caps?.maxTokens ?? 8192,
                     }] : [],
             };
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oh-pi",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "One-click setup for pi-coding-agent. Like oh-my-zsh for pi.",
   "type": "module",
   "bin": {

package/pi-package/extensions/custom-footer.ts

@@ -0,0 +1,110 @@
+/**
+ * Custom Footer Extension — Enhanced status bar
+ *
+ * Displays: ↑input ↓output Rremaining $cost percent/contextWindow (auto) | ⏱ elapsed | 📂 cwd | 🌿 branch | model • thinking
+ * Color-coded context usage: green <50%, yellow 50-75%, red >75%
+ */
+
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+
+export default function (pi: ExtensionAPI) {
+	let sessionStart = Date.now();
+
+	function formatElapsed(ms: number): string {
+		const s = Math.floor(ms / 1000);
+		if (s < 60) return `${s}s`;
+		const m = Math.floor(s / 60);
+		const rs = s % 60;
+		if (m < 60) return `${m}m${rs > 0 ? rs + "s" : ""}`;
+		const h = Math.floor(m / 60);
+		const rm = m % 60;
+		return `${h}h${rm > 0 ? rm + "m" : ""}`;
+	}
+
+	function fmt(n: number): string {
+		if (n < 1000) return `${n}`;
+		return `${(n / 1000).toFixed(1)}k`;
+	}
+
+	pi.on("session_start", async (_event, ctx) => {
+		sessionStart = Date.now();
+
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			const unsub = footerData.onBranchChange(() => tui.requestRender());
+			const timer = setInterval(() => tui.requestRender(), 30000);
+
+			return {
+				dispose() { unsub(); clearInterval(timer); },
+				invalidate() {},
+				render(width: number): string[] {
+					// --- Tokens & Cost ---
+					let input = 0, output = 0, cost = 0;
+					for (const e of ctx.sessionManager.getBranch()) {
+						if (e.type === "message" && e.message.role === "assistant") {
+							const m = e.message as AssistantMessage;
+							input += m.usage.input;
+							output += m.usage.output;
+							cost += m.usage.cost.total;
+						}
+					}
+
+					// --- Context usage ---
+					const usage = ctx.getContextUsage();
+					const tokens = usage?.tokens ?? 0;
+					const ctxWindow = usage?.contextWindow ?? 0;
+					const pct = usage?.percent ?? 0;
+					const remaining = Math.max(0, ctxWindow - tokens);
+
+					// Color by usage level
+					const pctColor = pct > 75 ? "error" : pct > 50 ? "warning" : "success";
+					const pctStr = `${pct.toFixed(1)}%/${fmt(ctxWindow)}`;
+
+					const tokenStats = [
+						theme.fg("accent", `↑${fmt(input)}`),
+						theme.fg("dim", ` ↓${fmt(output)}`),
+						theme.fg("muted", ` R${fmt(remaining)}`),
+						theme.fg("warning", ` $${cost.toFixed(3)}`),
+						" ",
+						theme.fg(pctColor, pctStr),
+						theme.fg("dim", " (auto)"),
+					].join("");
+
+					// --- Elapsed ---
+					const elapsed = theme.fg("dim", `⏱ ${formatElapsed(Date.now() - sessionStart)}`);
+
+					// --- CWD (last 2 segments) ---
+					const cwd = process.cwd();
+					const parts = cwd.split("/");
+					const short = parts.length > 2 ? parts.slice(-2).join("/") : cwd;
+					const cwdStr = theme.fg("muted", `📂 ${short}`);
+
+					// --- Git branch ---
+					const branch = footerData.getGitBranch();
+					const branchStr = branch ? theme.fg("accent", `🌿 ${branch}`) : "";
+
+					// --- Right: model + thinking ---
+					const thinking = pi.getThinkingLevel();
+					const modelId = ctx.model?.id || "no-model";
+					const right = theme.fg("dim", `${modelId} • ${thinking}`);
+
+					// --- Layout ---
+					const sep = theme.fg("dim", " │ ");
+					const leftParts = [tokenStats, elapsed, cwdStr];
+					if (branchStr) leftParts.push(branchStr);
+					const left = leftParts.join(sep);
+
+					const pad = " ".repeat(Math.max(1, width - visibleWidth(left) - visibleWidth(sep) - visibleWidth(right)));
+					return [truncateToWidth(left + pad + right, width)];
+				},
+			};
+		});
+	});
+
+	pi.on("session_switch", async (event, _ctx) => {
+		if (event.reason === "new") {
+			sessionStart = Date.now();
+		}
+	});
+}