pi-fireworks-provider 1.4.2 → 1.5.1

package/index.ts

@@ -26,7 +26,7 @@
  */
 
 import { getAgentDir, type ExtensionAPI, type ModelRegistry } from "@earendil-works/pi-coding-agent";
-import type { Input, matchesKey, Key, truncateToWidth, visibleWidth, wrapTextWithAnsi, fuzzyFilter, SettingsListTheme } from "@earendil-works/pi-tui";
+import type { Input, matchesKey, Key, KeyId, truncateToWidth, visibleWidth, wrapTextWithAnsi, fuzzyFilter, SettingsListTheme } from "@earendil-works/pi-tui";
 import modelsData from "./models.json" with { type: "json" };
 import customModelsData from "./custom-models.json" with { type: "json" };
 import patchData from "./patch.json" with { type: "json" };
@@ -44,7 +44,7 @@ interface JsonModel {
   baseUrl?: string;
   reasoning: boolean;
   thinkingLevelMap?: Record<string, string | null>;
-  input: string[];
+  input: ("text" | "image")[];
   cost: {
     input: number;
     output: number;
@@ -64,7 +64,7 @@ interface PatchEntry {
   baseUrl?: string;
   reasoning?: boolean;
   thinkingLevelMap?: Record<string, string | null>;
-  input?: string[];
+  input?: ("text" | "image")[];
   cost?: {
     input?: number;
     output?: number;
@@ -371,7 +371,7 @@ type PreserveMode = boolean; // true = inject reasoning_history:"preserved"; fal
 
 interface ServiceTierConfig {
   default: ServiceTier;
-  keybinding: string;
+  keybinding: KeyId;
   display: "statusbar" | "off";
 }
 
@@ -425,7 +425,7 @@ function isValidTier(v: unknown): v is ServiceTier {
   return v === "standard" || v === "priority";
 }
 
-function isValidKeybinding(v: unknown): v is string {
+function isValidKeybinding(v: unknown): v is KeyId {
   return typeof v === "string" && v.length > 0;
 }
 
@@ -1318,7 +1318,58 @@ export default function (pi: ExtensionAPI) {
     description: "Configure Fireworks: preserved thinking + service tier + logit bias + display",
     async handler(_args, ctx) {
       if (ctx.mode !== "tui") {
-        ctx.ui.notify("/fireworks-settings requires TUI mode.", "error");
+        // GUI/RPC fallback: a select -> select flow. The custom SettingsList and
+        // the nested LogitBiasEditor are TUI-only, so logit bias isn't offered
+        // here (configure it in a TUI session). One setting per run.
+        if (!ctx.hasUI) {
+          ctx.ui.notify("/fireworks-settings requires a UI (TUI or GUI).", "error");
+          return;
+        }
+        const guiItems = [
+          { id: "preserveThinking", label: "Preserved thinking", current: preserveOn ? "on" : "off", values: ["on", "off"] },
+          { id: "serviceTier", label: "Service tier", current: currentTier, values: ["standard", "priority"] },
+          { id: "serviceTier.display", label: "Service tier display", current: fireworksConfig.serviceTier.display, values: ["statusbar", "off"] },
+        ];
+        const pick = await ctx.ui.select(
+          "Fireworks settings — pick a setting (logit-bias editor is TUI-only)",
+          guiItems.map((i) => `${i.label}: ${i.current}`),
+        );
+        if (pick === undefined) return;
+        const item = guiItems.find((i) => pick.startsWith(`${i.label}:`));
+        if (!item) return;
+        const newValue = await ctx.ui.select(item.label, item.values);
+        if (newValue === undefined) return;
+        if (item.id === "preserveThinking") {
+          const on = newValue === "on";
+          setPreserve(on);
+          const raw = readRawFireworksConfig();
+          raw.preserveThinking = { ...(raw.preserveThinking ?? {}), default: on };
+          writeRawFireworksConfig(raw);
+          fireworksConfig = loadFireworksConfig();
+          ctx.ui.notify(`Preserved thinking ${on ? "on" : "off"} — takes effect now.`, "info");
+        } else if (item.id === "serviceTier") {
+          let model: any;
+          try { model = ctx.model; } catch { model = undefined; }
+          if (!model || model.provider !== "fireworks" || !isPriorityApplicable(model.id)) {
+            ctx.ui.notify("Service tier only applies to supported Fireworks models.", "info");
+            return;
+          }
+          const tier = newValue as ServiceTier;
+          setTier(ctx, tier);
+          const raw = readRawFireworksConfig();
+          raw.serviceTier = { ...(raw.serviceTier ?? {}), default: tier };
+          writeRawFireworksConfig(raw);
+          fireworksConfig = loadFireworksConfig();
+          ctx.ui.notify(`Fireworks service tier: ${tier}`, "info");
+        } else if (item.id === "serviceTier.display") {
+          const raw = readRawFireworksConfig();
+          raw.serviceTier = { ...(raw.serviceTier ?? {}), display: newValue };
+          writeRawFireworksConfig(raw);
+          fireworksConfig = loadFireworksConfig();
+          updateTierStatus(ctx);
+          ctx.ui.notify("Service tier display updated.", "info");
+        }
+        ctx.ui.notify("Run /fireworks-settings again for more.", "info");
         return;
       }
       const { SettingsList, Container, Input, matchesKey, Key, truncateToWidth, visibleWidth, wrapTextWithAnsi, fuzzyFilter } = await import("@earendil-works/pi-tui");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-fireworks-provider",
-  "version": "1.4.2",
+  "version": "1.5.1",
   "description": "Fireworks AI provider extension for pi - Access Kimi, MiniMax, GLM, DeepSeek, and GPT-OSS models through the Fireworks AI API",
   "type": "module",
   "main": "index.ts",