@monotykamary/pi-better-grok 0.3.6 → 0.3.7

package/index.ts

@@ -771,6 +771,9 @@ export default function betterGrok(pi: ExtensionAPI): void {
           registerGrok47OnProviders(
             (name, config) => pi.registerProvider(name, config),
             getAll.call(registry),
+            typeof registry.getModelsOfType === "function"
+              ? [...registry.getModelsOfType("image"), ...registry.getModelsOfType("classifier")]
+              : [],
           );
         } catch {
           // Catalog registration must not break session startup.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@monotykamary/pi-better-grok",
-  "version": "0.3.6",
+  "version": "0.3.7",
   "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",
@@ -43,11 +43,12 @@
     "lint": "oxlint --deny-warnings index.ts src/*.ts tests/*.ts",
     "lint:fix": "oxlint --fix index.ts src/*.ts tests/*.ts",
     "test": "vitest run",
-    "typecheck": "tsc --noEmit"
+    "typecheck": "tsc --noEmit",
+    "test:pi": "node scripts/test-pi-compat.mjs"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "0.87.0",
-    "@earendil-works/pi-tui": "0.87.0",
+    "@earendil-works/pi-coding-agent": "0.99.0",
+    "@earendil-works/pi-tui": "0.99.0",
     "@types/node": "~22.19.0",
     "oxfmt": "^0.47.0",
     "oxlint": "^1.62.0",
@@ -59,8 +60,8 @@
     "@earendil-works/pi-tui": "*"
   },
   "overrides": {
-    "@earendil-works/pi-agent-core": "0.87.0",
-    "@earendil-works/pi-ai": "0.87.0",
+    "@earendil-works/pi-agent-core": "0.99.0",
+    "@earendil-works/pi-ai": "0.99.0",
     "brace-expansion": "5.0.9",
     "nanoid": "3.3.18",
     "postcss": "8.5.25",

package/src/xai-models.ts

@@ -2,6 +2,8 @@ import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { dirname, join } from "node:path";
 import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 
+type ProviderChatModelConfig = Extract<ProviderModelConfig, { type?: "chat" }>;
+
 export const GROK_47_ID = "grok-4.7";
 export const GROK_47_NAME = "Grok 4.7";
 
@@ -38,7 +40,7 @@ const GROK_47_THINKING_LEVEL_MAP = {
  * Native xAI Grok 4.7, cloned from pi-core's grok-4.6 catalog plus the public
  * 4.7 card. Used as an in-memory fallback when models.json cannot be updated.
  */
-export const GROK_47_FALLBACK: ProviderModelConfig = {
+export const GROK_47_FALLBACK: ProviderChatModelConfig = {
   id: GROK_47_ID,
   name: GROK_47_NAME,
   api: "openai-responses",
@@ -72,21 +74,21 @@ export const GROK_47_MODELS_JSON_DEFINITION = {
 export type CatalogModel = {
   id: string;
   name: string;
-  api?: ProviderModelConfig["api"];
+  api?: ProviderChatModelConfig["api"];
   provider: string;
   baseUrl?: string;
   reasoning: boolean;
-  thinkingLevelMap?: ProviderModelConfig["thinkingLevelMap"];
-  input: ProviderModelConfig["input"];
-  cost: ProviderModelConfig["cost"];
-  promptCache?: ProviderModelConfig["promptCache"];
+  thinkingLevelMap?: ProviderChatModelConfig["thinkingLevelMap"];
+  input: ProviderChatModelConfig["input"];
+  cost: ProviderChatModelConfig["cost"];
+  promptCache?: ProviderChatModelConfig["promptCache"];
   contextWindow: number;
   maxTokens: number;
   headers?: Record<string, string>;
-  compat?: ProviderModelConfig["compat"];
+  compat?: ProviderChatModelConfig["compat"];
 };
 
-export function catalogModelToProviderConfig(model: CatalogModel): ProviderModelConfig {
+export function catalogModelToProviderConfig(model: CatalogModel): ProviderChatModelConfig {
   return {
     id: model.id,
     name: model.name,
@@ -104,7 +106,7 @@ export function catalogModelToProviderConfig(model: CatalogModel): ProviderModel
   };
 }
 
-export function grok47FromTemplate(template: CatalogModel | undefined): ProviderModelConfig {
+export function grok47FromTemplate(template: CatalogModel | undefined): ProviderChatModelConfig {
   if (!template) {
     return {
       ...GROK_47_FALLBACK,
@@ -143,7 +145,9 @@ export function grok47FromTemplate(template: CatalogModel | undefined): Provider
  * 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 {
+export function upsertGrok47(
+  existing: readonly CatalogModel[],
+): ProviderChatModelConfig[] | undefined {
   if (existing.length === 0) return undefined;
   if (existing.some((model) => model.id === GROK_47_ID)) return undefined;
   const template =
@@ -153,7 +157,7 @@ export function upsertGrok47(existing: readonly CatalogModel[]): ProviderModelCo
   return [...existing.map(catalogModelToProviderConfig), grok47FromTemplate(template)];
 }
 
-export type Grok47Registration = { provider: string; models: ProviderModelConfig[] };
+export type Grok47Registration = { provider: string; models: ProviderChatModelConfig[] };
 
 export function grok47Registrations(all: readonly CatalogModel[]): Grok47Registration[] {
   const byProvider = new Map<string, CatalogModel[]>();
@@ -173,10 +177,16 @@ export function grok47Registrations(all: readonly CatalogModel[]): Grok47Registr
 export function registerGrok47OnProviders(
   registerProvider: (name: string, config: { models: ProviderModelConfig[] }) => void,
   all: readonly CatalogModel[],
+  nonChat: readonly (Exclude<ProviderModelConfig, ProviderChatModelConfig> & {
+    provider: string;
+  })[] = [],
 ): string[] {
   const registered: string[] = [];
   for (const entry of grok47Registrations(all)) {
-    registerProvider(entry.provider, { models: entry.models });
+    // Pi 0.99 replaces every operation when a legacy catalog is registered.
+    registerProvider(entry.provider, {
+      models: [...entry.models, ...nonChat.filter((model) => model.provider === entry.provider)],
+    });
     registered.push(entry.provider);
   }
   return registered;