@baryonlabs/cli 0.4.2 → 0.4.4

package/bin/baryon.js

@@ -13,6 +13,7 @@ import {
   update,
   help,
   welcome,
+  installStatusExtension,
 } from "../src/commands.js";
 import { loadConfig, piProviderConfigured, hasConfig, prunePiPackages } from "../src/config.js";
 import { DEPRECATED_EXTENSIONS } from "../src/constants.js";
@@ -120,6 +121,9 @@ async function main() {
       } catch {
         /* best-effort */
       }
+      // Keep the /status·/my slash-command extension present + up to date
+      // (cheap content-compare copy into ~/.pi/agent/extensions/).
+      installStatusExtension();
       const cfg = loadConfig();
       // Show which room (project/분반 · seat) this key is in — best-effort.
       try {

package/extensions/baryon-status.ts

@@ -0,0 +1,159 @@
+/**
+ * baryon-status/v1 — `/status` + `/my` slash commands for the baryon CLI.
+ *
+ * Claude Code's `/status`-style seat dashboard inside a pi session: today's
+ * requests vs the class daily cap, cumulative cost vs the seat budget,
+ * lifetime totals, top models and guardrail flags — all served by the
+ * baryon gateway `GET /v1/status` (key-authenticated, seat-scoped).
+ *
+ * Installed by `@baryonlabs/cli` into ~/.pi/agent/extensions/ (auto-discovered
+ * by pi). Reads the key from $BARYON_API_KEY or ~/.baryon/config.json, so it
+ * also works when pi is launched directly (not via the baryon wrapper).
+ */
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+
+interface BaryonConfig {
+	apiKey?: string;
+	baseUrl?: string;
+	defaultModel?: string;
+}
+
+function loadBaryonConfig(): BaryonConfig {
+	try {
+		const file = path.join(os.homedir(), ".baryon", "config.json");
+		return JSON.parse(fs.readFileSync(file, "utf8"));
+	} catch {
+		return {};
+	}
+}
+
+function resolveAuth(): { apiKey: string; baseUrl: string; defaultModel?: string } | null {
+	const cfg = loadBaryonConfig();
+	const apiKey = process.env.BARYON_API_KEY || cfg.apiKey || "";
+	const baseUrl = (process.env.BARYON_BASE_URL || cfg.baseUrl || "https://api.baryon.ai/v1").replace(/\/+$/, "");
+	if (!apiKey) return null;
+	return { apiKey, baseUrl, defaultModel: cfg.defaultModel };
+}
+
+async function fetchStatus(auth: { apiKey: string; baseUrl: string }): Promise<any> {
+	const ctrl = new AbortController();
+	const t = setTimeout(() => ctrl.abort(), 8000);
+	try {
+		const res = await fetch(`${auth.baseUrl}/status`, {
+			headers: { Authorization: `Bearer ${auth.apiKey}` },
+			signal: ctrl.signal,
+		});
+		if (!res.ok) throw new Error(`HTTP ${res.status}`);
+		return await res.json();
+	} finally {
+		clearTimeout(t);
+	}
+}
+
+/** `[████████░░░░] 40%` — cap 없으면 null. */
+function bar(used: number, cap: number | null | undefined, width = 12): string | null {
+	if (!cap || cap <= 0) return null;
+	const ratio = Math.min(used / cap, 1);
+	const filled = Math.round(ratio * width);
+	const icon = ratio >= 1 ? "🔴" : ratio >= 0.8 ? "🟠" : "🟢";
+	return `${icon} ${"█".repeat(filled)}${"░".repeat(width - filled)} ${Math.round(ratio * 100)}%`;
+}
+
+function usd(n: number | null | undefined): string {
+	return n == null ? "—" : `$${Number(n).toFixed(4).replace(/0+$/, "").replace(/\.$/, "")}`;
+}
+
+function fmt(n: number | null | undefined): string {
+	return n == null ? "—" : Number(n).toLocaleString("en-US");
+}
+
+function maskKey(key: string): string {
+	if (key.length <= 12) return `${key.slice(0, 4)}…`;
+	return `${key.slice(0, 12)}…${key.slice(-4)}`;
+}
+
+function renderStatus(s: any): string {
+	const lines: string[] = [];
+	const room = s.project ? `${s.project}${s.active ? "" : " (비활성)"}` : "미배정";
+	lines.push(`📊 Baryon 사용량 — 🏫 ${room} · 좌석 ${s.seat ?? "—"}`);
+	lines.push("");
+
+	const daily = bar(s.today_requests, s.daily_cap);
+	lines.push(
+		`오늘 요청   ${fmt(s.today_requests)}${s.daily_cap ? ` / ${fmt(s.daily_cap)} (남음 ${fmt(s.daily_remaining)})` : " (일일 한도 없음)"}`,
+	);
+	if (daily) lines.push(`           ${daily}`);
+
+	const cost = bar(s.cost_usd, s.cost_cap);
+	lines.push(
+		`누적 비용   ${usd(s.cost_usd)}${s.cost_cap ? ` / ${usd(s.cost_cap)} (남음 ${usd(s.cost_remaining)})` : " (좌석 예산 없음)"}`,
+	);
+	if (cost) lines.push(`           ${cost}`);
+
+	const t = s.totals || {};
+	lines.push(`전체 누계   요청 ${fmt(t.requests)} · 토큰 ${fmt(t.totalTokens)}`);
+
+	if (Array.isArray(s.top_models) && s.top_models.length) {
+		lines.push("");
+		lines.push("많이 쓴 모델");
+		for (const m of s.top_models.slice(0, 3)) {
+			lines.push(`  · ${m.model} — 요청 ${fmt(m.requests)} · ${usd(m.cost_usd)}`);
+		}
+	}
+
+	const g = s.guardrails || {};
+	const rails = [g.pii && "PII차단", g.jailbreak && "탈옥차단", g.moderation && "모더레이션"].filter(Boolean);
+	if (rails.length || s.seat_rpm) {
+		lines.push("");
+		const parts = [];
+		if (rails.length) parts.push(`가드레일 ${rails.join("·")}`);
+		if (s.seat_rpm) parts.push(`좌석 ${s.seat_rpm}rpm`);
+		lines.push(parts.join(" · "));
+	}
+	return lines.join("\n");
+}
+
+function renderMy(s: any, auth: { apiKey: string; baseUrl: string; defaultModel?: string }): string {
+	const lines: string[] = [];
+	lines.push("👤 내 Baryon 계정");
+	lines.push("");
+	lines.push(`이메일      ${s.email ?? "—"}`);
+	lines.push(`분반        ${s.project ?? "미배정"}${s.project && !s.active ? " (비활성)" : ""}`);
+	lines.push(`좌석        ${s.seat ?? "—"}`);
+	lines.push(`키          ${maskKey(auth.apiKey)}`);
+	lines.push(`엔드포인트  ${auth.baseUrl}`);
+	if (auth.defaultModel) lines.push(`기본 모델   ${auth.defaultModel}`);
+	if (Array.isArray(s.allowed_models) && s.allowed_models.length) {
+		lines.push(`허용 모델   ${s.allowed_models.slice(0, 6).join(", ")}${s.allowed_models.length > 6 ? " …" : ""}`);
+	}
+	lines.push("");
+	lines.push(`오늘 ${fmt(s.today_requests)}회 · 누적 ${usd(s.cost_usd)} — 자세히: /status`);
+	return lines.join("\n");
+}
+
+async function run(ctx: any, render: (s: any, auth: any) => string): Promise<void> {
+	const auth = resolveAuth();
+	if (!auth) {
+		ctx.ui.notify("Baryon 키가 없습니다 — `baryon setup` 으로 로그인하세요.", "warning");
+		return;
+	}
+	try {
+		const s = await fetchStatus(auth);
+		ctx.ui.notify(render(s, auth), "info");
+	} catch (e: any) {
+		ctx.ui.notify(`Baryon 상태 조회 실패: ${e?.message || e} (게이트웨이/네트워크 확인)`, "error");
+	}
+}
+
+export default function (pi: any) {
+	pi.registerCommand("status", {
+		description: "Baryon 좌석 사용량·한도 (오늘 요청/일일캡, 비용/예산, 모델별)",
+		handler: async (_args: string, ctx: any) => run(ctx, renderStatus),
+	});
+	pi.registerCommand("my", {
+		description: "내 Baryon 계정 (이메일·분반·좌석·키·엔드포인트)",
+		handler: async (_args: string, ctx: any) => run(ctx, renderMy),
+	});
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@baryonlabs/cli",
-  "version": "0.4.2",
+  "version": "0.4.4",
   "description": "Baryon CLI — AI 코딩·학습 에이전트. baryon.ai API에 기본 연결된 pi 코딩 에이전트 래퍼. 한 줄 설치, 상용·로컬 모델 전환.",
   "type": "module",
   "bin": {
@@ -10,6 +10,7 @@
     "bin",
     "src",
     "skills",
+    "extensions",
     "README.md",
     "LICENSE"
   ],

package/src/commands.js

@@ -25,6 +25,7 @@ import {
   HOMEPAGE,
   KEYS_URL,
   KEY_PREFIX,
+  PI_AGENT_DIR,
   PI_PACKAGE,
   PI_SKILLS_DIR,
   PREFERRED_DEFAULT_MODEL,
@@ -112,6 +113,10 @@ export async function setup(args) {
     installSkills();
   }
 
+  // /status·/my slash commands inside pi (seat usage dashboard). Cheap local
+  // file copy — also self-healed on every plain `baryon` launch.
+  if (installStatusExtension()) ok(t("statusExt.ok"));
+
   if (flags["no-browser"]) {
     info(t("setup.skipBrowser"));
   } else {
@@ -344,6 +349,33 @@ export function installDefaults() {
   return okc;
 }
 
+/**
+ * Install (or refresh) the bundled baryon-status pi extension, which registers
+ * the `/status` + `/my` slash commands (seat usage dashboard via the gateway's
+ * `GET /v1/status`). Copied to ~/.pi/agent/extensions/ — pi auto-discovers
+ * loose .ts files there. Content-compared so CLI upgrades refresh it in place.
+ * Returns true when a copy happened (fresh install or update), false otherwise.
+ */
+export function installStatusExtension() {
+  try {
+    const src = path.join(
+      path.dirname(fileURLToPath(import.meta.url)),
+      "..",
+      "extensions",
+      "baryon-status.ts",
+    );
+    const dstDir = path.join(PI_AGENT_DIR, "extensions");
+    const dst = path.join(dstDir, "baryon-status.ts");
+    const content = fs.readFileSync(src, "utf8");
+    if (fs.existsSync(dst) && fs.readFileSync(dst, "utf8") === content) return false;
+    fs.mkdirSync(dstDir, { recursive: true });
+    fs.writeFileSync(dst, content);
+    return true;
+  } catch {
+    return false; // best-effort — never block setup/launch
+  }
+}
+
 // Install the default Agent Skills pack (pdf/pptx/xlsx) into ~/.pi/agent/skills/,
 // which pi auto-discovers. All three are subfolders of one repo, so we shallow-
 // clone once and copy each. Idempotent (skips skills already present) and safe to

package/src/constants.js

@@ -61,7 +61,7 @@ export const PI_SKILLS_DIR = path.join(PI_AGENT_DIR, "skills");
  *
  * Baryon 3-tier lineup (기준: baryon-ai-api docs/MODELS.md — 이름은 계약,
  * 실제 업스트림 모델은 게이트웨이 라우팅으로 조절):
- *   그저(geujeo) 단순·빠름 · 두루(duru) 범용 기본 · 그루(guru) 사고·기획.
+ *   빨리(ppalli) 단순·빠름 · 두루(duru) 범용 기본 · 그루(guru) 사고·기획.
  */
 export const DEFAULT_MODELS = [
   {
@@ -73,8 +73,8 @@ export const DEFAULT_MODELS = [
     maxTokens: 8192,
   },
   {
-    id: "baryon-geujeo",
-    name: "그저 (Geujeo) — 단순·빠름",
+    id: "baryon-ppalli",
+    name: "빨리 (Ppalli) — 단순·빠름",
     reasoning: false,
     input: ["text"],
     contextWindow: 262144,

package/src/i18n.js

@@ -130,6 +130,9 @@ const MESSAGES = {
     "skills.itemOk": "skill: {name} — {note}",
     "skills.itemFail": "{name} 스킬 설치 실패 — 건너뜀 ({error})",
     "skills.cloneFail": "스킬 저장소 clone 실패(네트워크/git 확인) — 일부 스킬 건너뜀",
+
+    // ── /status·/my extension ────────────────────────────────────────────
+    "statusExt.ok": "pi 명령 /status·/my 설치됨 (좌석 사용량·계정 조회)",
     "skills.summary": "기본 스킬 {ok}/{total} 설치 → {dir}",
     "skills.bannerListTitle": "Baryon 기본 스킬 팩",
     "skills.notInstalled": "(미설치)",
@@ -272,6 +275,9 @@ const MESSAGES = {
     "skills.noSkillMd": "{name} — no SKILL.md ({dir}), skipped",
     "skills.itemOk": "skill: {name} — {note}",
     "skills.itemFail": "{name} skill install failed — skipped ({error})",
+
+    // ── /status·/my extension ────────────────────────────────────────────
+    "statusExt.ok": "Installed pi commands /status·/my (seat usage & account)",
     "skills.cloneFail": "Skill repo clone failed (check network/git) — some skills skipped",
     "skills.summary": "Default skills {ok}/{total} installed → {dir}",
     "skills.bannerListTitle": "Baryon default skills pack",