@astrofoundry/pi-astro 0.11.1 → 0.11.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -40,15 +40,19 @@ describe("detectNerdFonts", () => {
40
40
  });
41
41
 
42
42
  describe("iconsFor", () => {
43
- it("returns nerd-font icons when true", () => {
43
+ it("returns nerd-font icons when true (separator pipe, model diamond, path 'dir')", () => {
44
44
  const icons = iconsFor(true);
45
- expect(icons.separator).toBe("");
45
+ expect(icons.separator).toBe("|");
46
+ expect(icons.model).toBe("◈");
47
+ expect(icons.path).toBe("dir");
46
48
  expect(icons.git).toBe("");
47
49
  });
48
50
 
49
- it("returns ascii icons when false", () => {
51
+ it("returns ascii icons when false (pipe separator, dir prefix, ◈ model)", () => {
50
52
  const icons = iconsFor(false);
51
- expect(icons.separator).toBe("");
53
+ expect(icons.separator).toBe("|");
54
+ expect(icons.model).toBe("◈");
55
+ expect(icons.path).toBe("dir");
52
56
  expect(icons.git).toBe("git");
53
57
  });
54
58
  });
@@ -12,26 +12,26 @@ export interface IconSet {
12
12
 
13
13
  const NERD: IconSet = {
14
14
  pi: "π",
15
- model: "",
15
+ model: "",
16
16
  thinking: "",
17
- path: "",
17
+ path: "dir",
18
18
  git: "",
19
19
  tokens: "",
20
20
  cost: "",
21
21
  context: "",
22
- separator: "",
22
+ separator: "|",
23
23
  };
24
24
 
25
25
  const ASCII: IconSet = {
26
26
  pi: "π",
27
- model: "@",
27
+ model: "",
28
28
  thinking: "?",
29
- path: "/",
29
+ path: "dir",
30
30
  git: "git",
31
31
  tokens: "tok",
32
32
  cost: "$",
33
33
  context: "ctx",
34
- separator: "",
34
+ separator: "|",
35
35
  };
36
36
 
37
37
  const NERD_TERM_PROGRAMS = new Set(["iTerm.app", "WezTerm", "ghostty"]);
@@ -187,7 +187,7 @@ describe("astro-footer extension", () => {
187
187
  const lines = component.render(120);
188
188
  expect(lines).toHaveLength(1);
189
189
  const line = lines[0];
190
- expect(line).toContain("claude-sonnet-4-6");
190
+ expect(line).toContain("Anthropic: Claude Sonnet 4 6");
191
191
  expect(line).toContain("proj");
192
192
  expect(line).toContain("main");
193
193
  expect(line).toContain("1.2k");
@@ -0,0 +1,40 @@
1
+ import { describe, expect, it } from "vitest";
2
+ import { formatModelDisplay } from "./model-name.ts";
3
+
4
+ describe("formatModelDisplay", () => {
5
+ it("formats moonshotai/kimi-k2.6 as MoonshotAI: Kimi K2.6", () => {
6
+ expect(formatModelDisplay("moonshotai/kimi-k2.6")).toBe("MoonshotAI: Kimi K2.6");
7
+ });
8
+
9
+ it("formats anthropic/claude-sonnet-4-6", () => {
10
+ expect(formatModelDisplay("anthropic/claude-sonnet-4-6")).toBe("Anthropic: Claude Sonnet 4 6");
11
+ });
12
+
13
+ it("uppercases known acronyms in the leaf (gpt → GPT)", () => {
14
+ expect(formatModelDisplay("openai/gpt-5.4-mini")).toBe("OpenAI: GPT 5.4 Mini");
15
+ });
16
+
17
+ it("uses the provider mapping (xai → xAI)", () => {
18
+ expect(formatModelDisplay("xai/grok-4")).toBe("xAI: Grok 4");
19
+ });
20
+
21
+ it("falls back to capitalising unknown providers", () => {
22
+ expect(formatModelDisplay("acme/super-model")).toBe("Acme: Super Model");
23
+ });
24
+
25
+ it("handles ids without a provider prefix", () => {
26
+ expect(formatModelDisplay("kimi-k2.6")).toBe("Kimi K2.6");
27
+ });
28
+
29
+ it("returns empty string for empty input", () => {
30
+ expect(formatModelDisplay("")).toBe("");
31
+ });
32
+
33
+ it("trims whitespace", () => {
34
+ expect(formatModelDisplay(" anthropic/claude-haiku-4-5 ")).toBe("Anthropic: Claude Haiku 4 5");
35
+ });
36
+
37
+ it("preserves slashes inside the leaf when the model id has multiple slashes", () => {
38
+ expect(formatModelDisplay("openai/o4/mini")).toBe("OpenAI: O4/mini");
39
+ });
40
+ });
@@ -0,0 +1,46 @@
1
+ const PROVIDER_DISPLAY: Record<string, string> = {
2
+ anthropic: "Anthropic",
3
+ openai: "OpenAI",
4
+ "openai-codex": "OpenAI Codex",
5
+ moonshotai: "MoonshotAI",
6
+ google: "Google",
7
+ meta: "Meta",
8
+ "meta-llama": "Meta Llama",
9
+ mistralai: "MistralAI",
10
+ xai: "xAI",
11
+ deepseek: "DeepSeek",
12
+ perplexity: "Perplexity",
13
+ openrouter: "OpenRouter",
14
+ cohere: "Cohere",
15
+ together: "Together",
16
+ groq: "Groq",
17
+ qwen: "Qwen",
18
+ };
19
+
20
+ const ACRONYMS = new Set(["gpt", "ai", "llm", "api", "moe", "ssm", "vl"]);
21
+
22
+ function capitalize(s: string): string {
23
+ if (s.length === 0) return s;
24
+ return s.charAt(0).toUpperCase() + s.slice(1);
25
+ }
26
+
27
+ function formatLeafWord(word: string): string {
28
+ if (word.length === 0) return word;
29
+ if (/^\d/.test(word)) return word;
30
+ if (ACRONYMS.has(word.toLowerCase())) return word.toUpperCase();
31
+ return capitalize(word);
32
+ }
33
+
34
+ export function formatModelDisplay(id: string): string {
35
+ const trimmed = id.trim();
36
+ if (trimmed.length === 0) return "";
37
+ if (!trimmed.includes("/")) {
38
+ return trimmed.split("-").map(formatLeafWord).join(" ");
39
+ }
40
+ const slash = trimmed.indexOf("/");
41
+ const providerRaw = trimmed.slice(0, slash);
42
+ const leaf = trimmed.slice(slash + 1);
43
+ const provider = PROVIDER_DISPLAY[providerRaw.toLowerCase()] ?? capitalize(providerRaw);
44
+ const leafFormatted = leaf.split("-").map(formatLeafWord).join(" ");
45
+ return `${provider}: ${leafFormatted}`;
46
+ }
@@ -28,10 +28,10 @@ describe("segments", () => {
28
28
  expect(renderModel(fakeTheme, ICONS, undefined)).toBeNull();
29
29
  });
30
30
 
31
- it("renderModel uses the last segment of a slash-separated id", () => {
32
- const out = renderModel(fakeTheme, ICONS, "anthropic/claude-sonnet-4-6");
33
- expect(out).toContain("claude-sonnet-4-6");
34
- expect(out).not.toContain("anthropic/");
31
+ it("renderModel pretty-prints provider + model name in accent colour", () => {
32
+ const out = renderModel(fakeTheme, ICONS, "moonshotai/kimi-k2.6");
33
+ expect(out).toContain("<accent>◈</accent>");
34
+ expect(out).toContain("<accent>MoonshotAI: Kimi K2.6</accent>");
35
35
  });
36
36
 
37
37
  it("renderThinking returns null when level missing", () => {
@@ -1,6 +1,7 @@
1
1
  import type { ThemeColor } from "@mariozechner/pi-coding-agent";
2
2
  import type { IconSet } from "./icons.ts";
3
3
  import { formatCost, formatPathBasename, formatPercent, formatTokens } from "./format.ts";
4
+ import { formatModelDisplay } from "./model-name.ts";
4
5
 
5
6
  export interface ThemeFn {
6
7
  fg: (role: ThemeColor, text: string) => string;
@@ -30,8 +31,8 @@ export function renderPi(theme: ThemeFn, icons: IconSet): string {
30
31
 
31
32
  export function renderModel(theme: ThemeFn, icons: IconSet, modelId: string | undefined): string | null {
32
33
  if (!modelId) return null;
33
- const short = modelId.split("/").pop() ?? modelId;
34
- return `${theme.fg("muted", icons.model)} ${theme.fg("text", short)}`.trim();
34
+ const display = formatModelDisplay(modelId);
35
+ return `${theme.fg("accent", icons.model)} ${theme.fg("accent", display)}`.trim();
35
36
  }
36
37
 
37
38
  export function renderThinking(theme: ThemeFn, icons: IconSet, level: string | undefined): string | null {
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@astrofoundry/pi-astro",
3
- "version": "0.11.1",
3
+ "version": "0.11.2",
4
4
  "description": "Personal pi customizations (extensions, skills, prompts, themes) for the pi coding agent.",
5
5
  "keywords": [
6
6
  "pi-package"