oh-pi 0.1.7 → 0.1.8

package/dist/index.js

@@ -1,5 +1,5 @@
 import * as p from "@clack/prompts";
-import { selectLanguage } from "./i18n.js";
+import { selectLanguage, getLocale } from "./i18n.js";
 import { t } from "./i18n.js";
 import { welcome } from "./tui/welcome.js";
 import { selectMode } from "./tui/mode-select.js";
@@ -26,6 +26,7 @@ export async function run() {
     else {
         config = await customFlow(env);
     }
+    config.locale = getLocale();
     await confirmApply(config, env);
 }
 async function quickFlow(env) {
@@ -35,7 +36,7 @@ async function quickFlow(env) {
         providers,
         theme,
         keybindings: "default",
-        extensions: ["safe-guard", "git-guard", "auto-session-name"],
+        extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "startup-banner"],
         skills: ["quick-setup", "debug-helper", "git-workflow"],
         prompts: ["review", "fix", "explain", "commit", "test"],
         agents: "general-developer",

package/dist/tui/preset-select.js

@@ -5,7 +5,7 @@ const PRESETS = {
         labelKey: "preset.starter", hintKey: "preset.starterHint",
         config: {
             theme: "oh-p-dark", keybindings: "default", thinking: "medium",
-            extensions: ["safe-guard", "git-guard", "auto-session-name"],
+            extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "startup-banner"],
             skills: ["quick-setup", "debug-helper"],
             prompts: ["review", "fix", "explain", "commit"],
             agents: "general-developer",
@@ -15,7 +15,7 @@ const PRESETS = {
         labelKey: "preset.pro", hintKey: "preset.proHint",
         config: {
             theme: "catppuccin-mocha", keybindings: "default", thinking: "high",
-            extensions: ["safe-guard", "git-guard", "auto-session-name"],
+            extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "startup-banner"],
             skills: ["quick-setup", "debug-helper", "git-workflow"],
             prompts: ["review", "fix", "explain", "commit", "test", "refactor", "optimize", "document", "pr"],
             agents: "fullstack-developer",
@@ -25,7 +25,7 @@ const PRESETS = {
         labelKey: "preset.security", hintKey: "preset.securityHint",
         config: {
             theme: "cyberpunk", keybindings: "default", thinking: "high",
-            extensions: ["safe-guard"],
+            extensions: ["safe-guard", "custom-footer", "startup-banner"],
             skills: ["debug-helper"],
             prompts: ["review", "security", "fix", "explain"],
             agents: "security-researcher",
@@ -35,7 +35,7 @@ const PRESETS = {
         labelKey: "preset.dataai", hintKey: "preset.dataaiHint",
         config: {
             theme: "tokyo-night", keybindings: "default", thinking: "medium",
-            extensions: ["safe-guard", "git-guard", "auto-session-name"],
+            extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "startup-banner"],
             skills: ["quick-setup", "debug-helper"],
             prompts: ["review", "fix", "explain", "optimize", "document", "test"],
             agents: "data-ai-engineer",
@@ -52,7 +52,7 @@ const PRESETS = {
         labelKey: "preset.full", hintKey: "preset.fullHint",
         config: {
             theme: "oh-p-dark", keybindings: "default", thinking: "high",
-            extensions: ["safe-guard", "git-guard", "auto-session-name", "ant-colony"],
+            extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "startup-banner", "ant-colony"],
             skills: ["quick-setup", "debug-helper", "git-workflow", "ant-colony"],
             prompts: ["review", "fix", "explain", "commit", "test", "refactor", "optimize", "security", "document", "pr"],
             agents: "colony-operator",

package/dist/types.d.ts

@@ -17,6 +17,7 @@ export interface OhPConfig {
     prompts: string[];
     agents: string;
     thinking: string;
+    locale?: string;
     compactThreshold?: number;
 }
 /** Official model capabilities for known providers */

package/dist/types.js

@@ -41,7 +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: "custom-footer", label: "📊 Custom Footer — Enhanced status bar with tokens, cost, time, git, cwd", default: true },
     { name: "startup-banner", label: "⚡ Startup Banner — Clean compact startup info (replaces verbose output)", default: true },
     { name: "ant-colony", label: "🐜 Ant Colony — Autonomous multi-agent swarm with adaptive concurrency", default: false },
 ];

package/dist/utils/install.js

@@ -25,10 +25,11 @@ function copyDir(src, dest) {
 export function applyConfig(config) {
     const agentDir = join(homedir(), ".pi", "agent");
     ensureDir(agentDir);
-    // 1. auth.json (skip if no providers configured)
-    if (config.providers.length > 0) {
+    // 1. auth.json (skip providers that have apiKey in models.json via baseUrl)
+    const authProviders = config.providers.filter(p => !p.baseUrl && p.apiKey !== "none");
+    if (authProviders.length > 0) {
         const auth = {};
-        for (const p of config.providers) {
+        for (const p of authProviders) {
             auth[p.name] = { type: "api_key", key: p.apiKey };
         }
         const authPath = join(agentDir, "auth.json");
@@ -61,24 +62,39 @@ export function applyConfig(config) {
     // 3. models.json (custom endpoints / providers)
     const customProviders = config.providers.filter(p => p.baseUrl);
     if (customProviders.length > 0) {
-        const models = {};
+        const providers = {};
         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,
-                api: "openai-completions",
-                models: cp.defaultModel ? [{
-                        id: cp.defaultModel,
-                        name: cp.defaultModel,
-                        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,
-                    }] : [],
-            };
+            const isBuiltin = !!PROVIDERS[cp.name];
+            if (isBuiltin) {
+                // Known provider with custom baseUrl — just override endpoint, keep built-in models
+                const entry = { baseUrl: cp.baseUrl };
+                if (cp.apiKey !== "none")
+                    entry.apiKey = cp.apiKey;
+                providers[cp.name] = entry;
+            }
+            else {
+                // Fully custom provider — need api, models, etc.
+                const caps = cp.defaultModel ? MODEL_CAPABILITIES[cp.defaultModel] : undefined;
+                const entry = {
+                    baseUrl: cp.baseUrl,
+                    api: "openai-completions",
+                };
+                if (cp.apiKey !== "none")
+                    entry.apiKey = cp.apiKey;
+                if (cp.defaultModel) {
+                    entry.models = [{
+                            id: cp.defaultModel,
+                            name: cp.defaultModel,
+                            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,
+                        }];
+                }
+                providers[cp.name] = entry;
+            }
         }
-        writeFileSync(join(agentDir, "models.json"), JSON.stringify(models, null, 2));
+        writeFileSync(join(agentDir, "models.json"), JSON.stringify({ providers }, null, 2));
     }
     // 4. keybindings.json
     const kb = KEYBINDING_SCHEMES[config.keybindings];
@@ -88,7 +104,12 @@ export function applyConfig(config) {
     // 5. AGENTS.md
     const agentsSrc = join(PKG_ROOT, "pi-package", "agents", `${config.agents}.md`);
     try {
-        const content = readFileSync(agentsSrc, "utf8");
+        let content = readFileSync(agentsSrc, "utf8");
+        if (config.locale && config.locale !== "en") {
+            const langNames = { zh: "Chinese (中文)", fr: "French (Français)" };
+            const lang = langNames[config.locale] ?? config.locale;
+            content = `## Language\nAlways respond in ${lang}. Use the user's language for all conversations and explanations. Code, commands, and technical terms can remain in English.\n\n${content}`;
+        }
         writeFileSync(join(agentDir, "AGENTS.md"), content);
     }
     catch { /* template not found, skip */ }

package/package.json

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

package/pi-package/extensions/startup-banner.ts

@@ -1,32 +1,55 @@
-import type { ExtensionContext, PiSdk } from "@anthropic/pi-sdk";
+/**
+ * oh-pi Startup Banner Extension
+ *
+ * Replaces the built-in verbose header with a compact one-line info display.
+ * Auto-restores the default header after 8 seconds.
+ */
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+export default function (pi: ExtensionAPI) {
+  pi.on("session_start", async (_event, ctx) => {
+    if (!ctx.hasUI) return;
 
-export default function activate(pi: PiSdk) {
-  pi.on("session_start", async (_event, ctx: ExtensionContext) => {
     const model = ctx.model;
     const thinking = pi.getThinkingLevel();
     const usage = ctx.getContextUsage();
-    const branch = ctx.sessionManager.getBranch();
-    const session = ctx.sessionManager.getSessionFile() ?? "ephemeral";
     const cwd = process.cwd().replace(process.env.HOME ?? "", "~");
 
+    // Sum cost from branch
+    let totalCost = 0;
+    for (const e of ctx.sessionManager.getBranch()) {
+      if (e.type === "message" && e.message.role === "assistant") {
+        totalCost += (e.message as AssistantMessage).usage.cost.total;
+      }
+    }
+
     // Git branch
     let git = "";
     try {
-      const { execSync } = await import("node:child_process");
-      git = execSync("git rev-parse --abbrev-ref HEAD", { encoding: "utf8", timeout: 2000 }).trim();
+      const { stdout } = await pi.exec("git", ["rev-parse", "--abbrev-ref", "HEAD"], { timeout: 2000 });
+      git = stdout.trim();
     } catch { /* not a git repo */ }
 
     const modelStr = model ? `${model.provider}/${model.id}` : "no model";
     const thinkStr = thinking !== "off" ? ` • 🧠 ${thinking}` : "";
     const ctxStr = usage ? ` • 📊 ${Math.round(usage.percent)}%` : "";
     const gitStr = git ? ` • 🌿 ${git}` : "";
-    const costStr = branch ? ` • $${(branch.totalCost ?? 0).toFixed(3)}` : "";
+    const costStr = totalCost > 0 ? ` • $${totalCost.toFixed(3)}` : "";
+
+    const session = ctx.sessionManager.getSessionFile?.() ?? "ephemeral";
+    const sessionName = session.split("/").pop();
 
-    const line = `⚡ ${modelStr}${thinkStr}${ctxStr}${costStr} │ 📂 ${cwd}${gitStr} │ 📋 ${session.split("/").pop()}`;
+    const line = `⚡ ${modelStr}${thinkStr}${ctxStr}${costStr} │ 📂 ${cwd}${gitStr} │ 📋 ${sessionName}`;
 
-    ctx.ui.setWidget("startup-banner", [line], { placement: "belowEditor" });
+    ctx.ui.setHeader((_tui, theme) => ({
+      render(_width: number): string[] {
+        return ["", theme.fg("accent", line), ""];
+      },
+      invalidate() {},
+    }));
 
-    // Auto-hide after 8 seconds
-    setTimeout(() => ctx.ui.setWidget("startup-banner", undefined), 8000);
+    // Restore default header after 8 seconds
+    setTimeout(() => ctx.ui.setHeader(undefined), 8000);
   });
 }