@monotykamary/pi-better-grok 0.3.2 → 0.3.4

package/README.md

@@ -1,6 +1,6 @@
 # pi-better-grok
 
-Better Grok/xAI for [pi](https://pi.dev) — mirrors the [pi-better-openai](https://github.com/monotykamary/pi-better-openai) UX for SuperGrok subscribers: fast mode, subscription usage in the footer, footer polish, and a settings picker.
+Better Grok/xAI for [pi](https://pi.dev) — mirrors the [pi-better-openai](https://github.com/monotykamary/pi-better-openai) UX for SuperGrok subscribers: fast mode, subscription usage in the footer, footer polish, and a settings picker. Until pi-core ships it, Grok 4.7 is layered onto the existing `xai` (and sibling Grok) providers as a custom model.
 
 ## Install
 
@@ -51,7 +51,7 @@ JSON config at `~/.pi/agent/extensions/pi-better-grok.json` (global) or `<projec
 ```json
 {
   "persistState": true,
-  "supportedModels": ["xai/grok-4.6", "xai/grok-4.5"],
+  "supportedModels": ["xai/grok-4.7", "xai/grok-4.6", "xai/grok-4.5"],
   "fast": { "effort": "low" },
   "usage": {
     "enabled": true,

package/index.ts

@@ -73,6 +73,7 @@ import {
 } from "./src/grok-auth.ts";
 import { currentModelKey, FastController, modelList, supportsFast } from "./src/fast-controller.ts";
 import { isGrokSubscriptionModel, UsageController } from "./src/usage-controller.ts";
+import { registerGrok47OnProviders } from "./src/xai-models.ts";
 import { sep } from "node:path";
 
 // pi-core's getSettingsListTheme pulls the host module graph into this
@@ -731,7 +732,22 @@ export default function betterGrok(pi: ExtensionAPI): void {
     setStatusWidget(ctx, statusWidgetParts(fast, usage));
   }
 
+  const ensureGrok47 = (ctx: ExtensionContext): void => {
+    if (typeof pi.registerProvider !== "function") return;
+    const getAll = ctx.modelRegistry?.getAll;
+    if (typeof getAll !== "function") return;
+    try {
+      registerGrok47OnProviders(
+        (name, config) => pi.registerProvider(name, config),
+        getAll.call(ctx.modelRegistry),
+      );
+    } catch {
+      // Catalog registration must not break session startup.
+    }
+  };
+
   pi.on("session_start", (_event, ctx) => {
+    ensureGrok47(ctx);
     invalidateContextUsage();
     invalidateSessionName();
     multiproviderRefreshCtx = ctx;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@monotykamary/pi-better-grok",
-  "version": "0.3.2",
+  "version": "0.3.4",
   "description": "Improve Grok/xAI in pi with fast mode, subscription usage stats, banked reset redemption, multiprovider pools, footer polish, and settings — mirroring pi-better-openai.",
   "keywords": [
     "footer",
@@ -46,8 +46,8 @@
     "typecheck": "tsc --noEmit"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "0.85.1",
-    "@earendil-works/pi-tui": "0.85.1",
+    "@earendil-works/pi-coding-agent": "0.86.0",
+    "@earendil-works/pi-tui": "0.86.0",
     "@types/node": "~22.19.0",
     "oxfmt": "^0.47.0",
     "oxlint": "^1.62.0",
@@ -59,8 +59,8 @@
     "@earendil-works/pi-tui": "*"
   },
   "overrides": {
-    "@earendil-works/pi-agent-core": "0.85.1",
-    "@earendil-works/pi-ai": "0.85.1",
+    "@earendil-works/pi-agent-core": "0.86.0",
+    "@earendil-works/pi-ai": "0.86.0",
     "brace-expansion": "5.0.9",
     "nanoid": "3.3.18",
     "postcss": "8.5.25",

package/src/config.ts

@@ -8,10 +8,13 @@ export const FOOTER_MODES = ["replace", "status", "off"] as const;
 export const FAST_EFFORTS = ["low", "medium", "high"] as const;
 
 export const DEFAULT_SUPPORTED_MODELS = [
+  "xai/grok-4.7",
   "xai/grok-4.6",
   "xai/grok-4.5",
+  "xai-oauth/grok-4.7",
   "xai-oauth/grok-4.6",
   "xai-oauth/grok-4.5",
+  "grok-build/grok-4.7",
   "grok-build/grok-4.6",
   "grok-build/grok-4.5",
 ] as const;

package/src/xai-models.ts

@@ -0,0 +1,162 @@
+import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
+
+export const GROK_47_ID = "grok-4.7";
+export const GROK_47_NAME = "Grok 4.7";
+
+/** Providers that share Grok model ids and should pick up 4.7 until pi-core ships it. */
+export const GROK_CUSTOM_MODEL_PROVIDERS = ["xai", "xai-oauth", "xai-auth", "grok-build"] as const;
+
+const GROK_47_LONG_CONTEXT_TIER = {
+  inputTokensAbove: 200_000,
+  input: 4,
+  output: 12,
+  cacheRead: 1,
+  cacheWrite: 0,
+} as const;
+
+/**
+ * Native xAI Grok 4.7, cloned from pi-core's grok-4.6 catalog plus the public
+ * 4.7 card: 500k context, $2/$6 per 1M, cache read $0.50, 2x rates above 200k,
+ * reasoning efforts low/medium/high/xhigh. Used only when a sibling template exists.
+ */
+export const GROK_47_FALLBACK: ProviderModelConfig = {
+  id: GROK_47_ID,
+  name: GROK_47_NAME,
+  api: "openai-responses",
+  baseUrl: "https://api.x.ai/v1",
+  reasoning: true,
+  thinkingLevelMap: {
+    off: null,
+    minimal: null,
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: "xhigh",
+    max: null,
+  },
+  input: ["text", "image"],
+  cost: {
+    input: 2,
+    output: 6,
+    cacheRead: 0.5,
+    cacheWrite: 0,
+    tiers: [{ ...GROK_47_LONG_CONTEXT_TIER }],
+  },
+  contextWindow: 500_000,
+  maxTokens: 500_000,
+  compat: { supportsLongCacheRetention: false },
+};
+
+/** Structural catalog slice; avoids a hard pi-ai import from this package. */
+export type CatalogModel = {
+  id: string;
+  name: string;
+  api?: ProviderModelConfig["api"];
+  provider: string;
+  baseUrl?: string;
+  reasoning: boolean;
+  thinkingLevelMap?: ProviderModelConfig["thinkingLevelMap"];
+  input: ProviderModelConfig["input"];
+  cost: ProviderModelConfig["cost"];
+  promptCache?: ProviderModelConfig["promptCache"];
+  contextWindow: number;
+  maxTokens: number;
+  headers?: Record<string, string>;
+  compat?: ProviderModelConfig["compat"];
+};
+
+export function catalogModelToProviderConfig(model: CatalogModel): ProviderModelConfig {
+  return {
+    id: model.id,
+    name: model.name,
+    api: model.api,
+    baseUrl: model.baseUrl,
+    reasoning: model.reasoning,
+    thinkingLevelMap: model.thinkingLevelMap,
+    input: model.input,
+    cost: model.cost,
+    promptCache: model.promptCache,
+    contextWindow: model.contextWindow,
+    maxTokens: model.maxTokens,
+    headers: model.headers,
+    compat: model.compat,
+  };
+}
+
+export function grok47FromTemplate(template: CatalogModel | undefined): ProviderModelConfig {
+  if (!template) {
+    return {
+      ...GROK_47_FALLBACK,
+      cost: { ...GROK_47_FALLBACK.cost, tiers: [{ ...GROK_47_LONG_CONTEXT_TIER }] },
+    };
+  }
+  const thinkingLevelMap = {
+    ...template.thinkingLevelMap,
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: "xhigh",
+  };
+  return {
+    ...catalogModelToProviderConfig(template),
+    id: GROK_47_ID,
+    name: GROK_47_NAME,
+    reasoning: true,
+    input: ["text", "image"],
+    contextWindow: Math.max(template.contextWindow, 500_000),
+    maxTokens: Math.max(template.maxTokens, 500_000),
+    thinkingLevelMap,
+    cost: {
+      ...template.cost,
+      input: 2,
+      output: 6,
+      cacheRead: template.cost.cacheRead || 0.5,
+      cacheWrite: template.cost.cacheWrite ?? 0,
+      tiers: template.cost.tiers ?? [{ ...GROK_47_LONG_CONTEXT_TIER }],
+    },
+  };
+}
+
+/**
+ * Extension registerProvider models replace that provider's catalog.
+ * Return the existing models plus grok-4.7, or undefined when 4.7 is already present
+ * or the provider has nothing to layer onto.
+ */
+export function upsertGrok47(existing: readonly CatalogModel[]): ProviderModelConfig[] | undefined {
+  if (existing.length === 0) return undefined;
+  if (existing.some((model) => model.id === GROK_47_ID)) return undefined;
+  const template =
+    existing.find((model) => model.id === "grok-4.6") ??
+    existing.find((model) => model.id === "grok-4.5") ??
+    existing[0];
+  return [...existing.map(catalogModelToProviderConfig), grok47FromTemplate(template)];
+}
+
+export type Grok47Registration = { provider: string; models: ProviderModelConfig[] };
+
+export function grok47Registrations(all: readonly CatalogModel[]): Grok47Registration[] {
+  const byProvider = new Map<string, CatalogModel[]>();
+  for (const model of all) {
+    const list = byProvider.get(model.provider) ?? [];
+    list.push(model);
+    byProvider.set(model.provider, list);
+  }
+  const registrations: Grok47Registration[] = [];
+  for (const provider of GROK_CUSTOM_MODEL_PROVIDERS) {
+    const models = upsertGrok47(byProvider.get(provider) ?? []);
+    if (models) registrations.push({ provider, models });
+  }
+  return registrations;
+}
+
+export function registerGrok47OnProviders(
+  registerProvider: (name: string, config: { models: ProviderModelConfig[] }) => void,
+  all: readonly CatalogModel[],
+): string[] {
+  const registered: string[] = [];
+  for (const entry of grok47Registrations(all)) {
+    registerProvider(entry.provider, { models: entry.models });
+    registered.push(entry.provider);
+  }
+  return registered;
+}