pi-spark 0.22.0 → 0.22.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-spark",
-  "version": "0.22.0",
+  "version": "0.22.1",
   "description": "Pi package that polishes your daily experience.",
   "keywords": [
     "pi-coding-agent",

package/src/features/credits/index.ts

@@ -18,7 +18,7 @@ export function registerCredits(pi: ExtensionAPI): void {
 
   pi.on("session_start", (_event, ctx) => {
     const config = loadConfig(ctx).credits;
-    if (ctx.mode !== "tui" || !config) return;
+    if (!["tui", "rpc"].includes(ctx.mode) || !config) return;
 
     const providers = getEnabledProviders(config);
     if (providers.length === 0) return;

package/src/features/footer/index.ts

@@ -5,7 +5,7 @@ import { hyperlink, Text } from "@earendil-works/pi-tui";
 
 import { SplitLine } from "../../components/split-line";
 import { loadConfig } from "../../config";
-import { formatContextUsage, formatCost, formatCwd, sanitizeText } from "../../utils/format";
+import { formatContextUsage, formatCwd, sanitizeText } from "../../utils/format";
 import { getEntryUsage } from "../../utils/usage";
 
 import type { ExtensionContext, ExtensionAPI, ReadonlyFooterDataProvider, Theme } from "@earendil-works/pi-coding-agent";
@@ -76,11 +76,9 @@ class FooterComponent implements Component {
 
   private getStyledCostText(): string {
     const totalCost = this.ctx.sessionManager.getEntries().reduce((acc, entry) => acc + (getEntryUsage(entry)?.cost.total ?? 0), 0);
+    if (totalCost < 0.0005) return ""; // Hide cost below half a millicent, since it would render as $0.000.
 
-    // Hide cost below half a cent, since it would render as $0.00.
-    if (totalCost < 0.005) return "";
-
-    const costText = formatCost(totalCost);
+    const costText = `$${totalCost.toFixed(3)}`;
 
     if (totalCost > 20) return this.theme.fg("warning", costText);
     return this.theme.fg("dim", costText);

package/src/features/recap/manager.ts

@@ -87,6 +87,7 @@ export class RecapManager {
     const options: SimpleStreamOptions = { maxTokens: MAX_TOKENS, signal };
     if (auth.apiKey) options.apiKey = auth.apiKey;
     if (auth.headers) options.headers = auth.headers;
+    if (auth.env) options.env = auth.env;
     if (thinkingLevel !== "off") options.reasoning = thinkingLevel;
 
     const response = await completeBackground(model, {

package/src/features/title/manager.ts

@@ -70,6 +70,7 @@ export class TitleManager {
     const options: SimpleStreamOptions = { maxTokens: MAX_TOKENS, signal };
     if (auth.apiKey) options.apiKey = auth.apiKey;
     if (auth.headers) options.headers = auth.headers;
+    if (auth.env) options.env = auth.env;
     if (thinkingLevel !== "off") options.reasoning = thinkingLevel;
 
     const response = await completeBackground(model, {

package/src/utils/format.ts

@@ -26,10 +26,6 @@ export function formatContextUsage(contextUsage: ContextUsage | undefined): stri
   return `${tokens === null ? "?" : formatTokens(tokens)}/${contextWindow === null ? "?" : formatTokens(contextWindow)} (${percentText})`;
 }
 
-export function formatCost(cost: number): string {
-  return `$${cost.toFixed(2)}`;
-}
-
 export function formatCwd(cwd: string, home: string): string {
   const resolvedCwd = resolve(cwd);
   const resolvedHome = resolve(home);

package/src/utils/model.ts

@@ -59,7 +59,7 @@ export async function resolveModelSettings(ctx: ExtensionContext, config: Option
     return;
   }
 
-  const { model, warning: modelWarning } = await resolveModelSelection(ctx, config, feature, fallbackModel);
+  const { model, warning: modelWarning } = await resolveModel(ctx, config, feature, fallbackModel);
   const { thinkingLevel, warning: thinkingLevelWarning } = resolveThinkingLevel(model, config.thinkingLevel ?? DEFAULT_THINKING_LEVEL);
 
   return {
@@ -69,41 +69,28 @@ export async function resolveModelSettings(ctx: ExtensionContext, config: Option
   };
 }
 
-async function resolveModelSelection(ctx: ExtensionContext, config: OptionalModelConfig, feature: string, fallbackModel: Model<Api>): Promise<ModelSelection> {
-  if (!config.provider && !config.model) {
-    return {
-      model: fallbackModel,
-      warning: undefined,
-    };
-  }
-
-  if (!config.provider || !config.model) {
-    return {
-      model: fallbackModel,
-      warning: `Both ${feature}.provider and ${feature}.model are required; using the current model.`,
-    };
-  }
-
-  const model = ctx.modelRegistry.find(config.provider, config.model);
-  if (!model) {
-    return {
-      model: fallbackModel,
-      warning: `Model ${formatModel(config.provider, config.model)} not found; using the current model.`,
-    };
+async function resolveModel(ctx: ExtensionContext, config: OptionalModelConfig, feature: string, fallbackModel: Model<Api>): Promise<ModelSelection> {
+  let model = fallbackModel;
+  let warning: string | undefined = undefined;
+
+  if (config.provider && config.model) {
+    const reference = formatModel(config.provider, config.model);
+    const configuredModel = ctx.modelRegistry.find(config.provider, config.model);
+    if (!configuredModel) {
+      warning = `Model ${reference} not found; using the current model.`;
+    } else {
+      const auth = await ctx.modelRegistry.getApiKeyAndHeaders(configuredModel);
+      if (auth.ok) {
+        model = configuredModel;
+      } else {
+        warning = `Model ${reference} unavailable: ${auth.error}; using the current model.`;
+      }
+    }
+  } else if (config.provider || config.model) {
+    warning = `Both ${feature}.provider and ${feature}.model are required; using the current model.`;
   }
 
-  const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
-  if (!auth.ok) {
-    return {
-      model: fallbackModel,
-      warning: `Model ${formatModel(config.provider, config.model)} unavailable: ${auth.error}; using the current model.`,
-    };
-  }
-
-  return {
-    model,
-    warning: undefined,
-  };
+  return { model, warning };
 }
 
 function resolveThinkingLevel(model: Model<Api>, requested: ModelThinkingLevel): ThinkingLevelSelection {