free-pi-cli 0.4.7 → 0.4.9

package/dist/index.js

@@ -27,4 +27,4 @@ if (err) {
   console.error(err);
   process.exit(1);
 }
-await import("./main-ep5298x9.js");
+await import("./main-jaz36mav.js");

package/dist/{main-ep5298x9.js → main-jaz36mav.js}

@@ -3,7 +3,7 @@
 // src/main.ts
 import { spawn as spawn2 } from "node:child_process";
 import { realpathSync, writeFileSync as writeFileSync2 } from "node:fs";
-import { dirname as dirname5, join as join6 } from "node:path";
+import { dirname as dirname5, join as join7 } from "node:path";
 
 // src/browser.ts
 import { spawn } from "node:child_process";
@@ -46,7 +46,7 @@ function openBrowser(url, platform = process.platform, spawnImpl = spawn) {
 // package.json
 var package_default = {
   name: "free-pi-cli",
-  version: "0.4.7",
+  version: "0.4.9",
   description: "Free, ad-supported distribution of the pi coding agent. Inference paid for by ads.",
   license: "MIT",
   type: "module",
@@ -4906,7 +4906,8 @@ var ClientVersionResponseSchema = object({
   latest: string2(),
   model: string2().optional(),
   notice: string2().optional(),
-  models: array(object({ id: string2(), name: string2() })).optional()
+  models: array(object({ id: string2(), name: string2() })).optional(),
+  fusion: object({ planner: string2(), executor: string2() }).optional()
 });
 var SessionResetResponseSchema = object({
   released: boolean2()
@@ -4970,6 +4971,7 @@ var ConfigSchema = object({
   abuseRestrictedCapUsd: number2().default(0.5),
   abuseRestrictedHourlyCap: number2().int().default(5),
   abuseNormalHourlyCap: number2().int().default(30),
+  abuseSignedInNormalHourlyCap: number2().int().default(0),
   abuseFlipAlertThreshold: number2().int().default(10),
   abuseAdBypassMinRequests: number2().int().default(20),
   abuseAdBypassRatio: number2().int().default(10),
@@ -5042,6 +5044,7 @@ function loadConfig(env = process.env) {
     abuseRestrictedCapUsd: envNumber(env.ABUSE_RESTRICTED_CAP_USD),
     abuseRestrictedHourlyCap: envNumber(env.ABUSE_RESTRICTED_HOURLY_CAP),
     abuseNormalHourlyCap: envNumber(env.ABUSE_NORMAL_HOURLY_CAP),
+    abuseSignedInNormalHourlyCap: envNumber(env.ABUSE_SIGNED_IN_NORMAL_HOURLY_CAP),
     abuseFlipAlertThreshold: envNumber(env.ABUSE_FLIP_ALERT_THRESHOLD),
     abuseAdBypassMinRequests: envNumber(env.ABUSE_AD_BYPASS_MIN_REQUESTS),
     abuseAdBypassRatio: envNumber(env.ABUSE_AD_BYPASS_RATIO),
@@ -5086,6 +5089,42 @@ function formatCreditsLeftLine(remaining, allowance, refillAt) {
   return `${left} / ${total} credits left today · refills ${formatRefillEt(refillAt)}`;
 }
 var SLOW_LANE_METER_LINE = "Slow lane: cheapest model until refill";
+var LIMIT_NOTICE_LABELS = {
+  hourly_ceiling: "hourly limit",
+  daily_cap: "daily limit",
+  rolling_cap: "rolling limit",
+  global_cap: "service daily limit",
+  queue_wait: "queue wait",
+  concurrent: "one request at a time",
+  concurrent_session: "another session is active"
+};
+var EXTRA_CREDITS_URL = "freepi.ai/extra-credits";
+var LIMIT_NOTICE_LOCAL_TIME_AFTER_SECONDS = 90 * 60;
+function formatLimitNotice(fields, opts = {}) {
+  const slow = fields.lane === "slow" && fields.limit === "hourly_ceiling" && fields.hourly_ceiling != null;
+  const label = slow ? `You've used today's free allowance, so you're on the slow lane (${fields.hourly_ceiling.toLocaleString("en-US")} requests an hour).` : LIMIT_NOTICE_LABELS[fields.limit] ?? fields.limit.replace(/_/g, " ");
+  const parts = [label];
+  const retry = fields.retry_after_seconds;
+  if (retry != null && retry > 0) {
+    const useClock = Boolean(opts.localTime) && Boolean(fields.clears_at) && retry >= LIMIT_NOTICE_LOCAL_TIME_AFTER_SECONDS;
+    if (useClock) {
+      const formatted = new Intl.DateTimeFormat("en-US", {
+        timeZone: opts.timeZone,
+        hour: "numeric",
+        minute: "2-digit"
+      }).format(new Date(fields.clears_at));
+      parts.push(`back at ${formatted}`);
+    } else {
+      parts.push(`back in ${Math.max(1, Math.ceil(retry / 60))} min`);
+    }
+  }
+  if (slow) {
+    const wait = parts.length > 1 ? ` ${parts[1][0].toUpperCase()}${parts[1].slice(1)} — or get` : " Get";
+    return `${label}${wait} credits at ${EXTRA_CREDITS_URL} to keep going at full speed.`;
+  }
+  parts.push(`${fields.remaining_credits_today.toLocaleString("en-US")} credits left today`);
+  return parts.join(", ");
+}
 // src/gravity-pixel.ts
 var GRAVITY_PIXEL_ID = "6cd74804-0541-4d9b-899d-d7fceb465b6f";
 function userIdFromJwt(jwt) {
@@ -8129,7 +8168,7 @@ function createAdsExtension(opts) {
 }
 
 // src/pi-launch.ts
-import { dirname as dirname3, join as join4, resolve as resolve2 } from "node:path";
+import { dirname as dirname3, join as join5, resolve as resolve2 } from "node:path";
 import { fileURLToPath as fileURLToPath3 } from "node:url";
 
 // src/provider.ts
@@ -8545,6 +8584,104 @@ function createUsageToolExtension(opts) {
   };
 }
 
+// src/image-tool.ts
+import { randomBytes } from "node:crypto";
+import { mkdir as mkdir2, open } from "node:fs/promises";
+import { join as join3 } from "node:path";
+var IMAGE_TOOL_NAME = "generate_image";
+var MAX_BYTES = 8 * 1024 * 1024;
+var TIMEOUT_MS = 120000;
+var EXT = { "image/png": "png", "image/jpeg": "jpg", "image/webp": "webp" };
+var IMAGE_GUIDELINES = [
+  "Use generate_image only when the user asks for an image, mockup, icon, illustration or other visual asset. Do not call it for diagrams you can write as code, SVG or ASCII.",
+  "Make one call per asset. Put the style, any exact text to render, and the composition in the prompt, and pick aspect_ratio to suit the use (16:9 for banners, 1:1 for icons).",
+  "You cannot see the generated image. The tool saves it to a file and returns the relative path; refer to that path in code, docs or your answer, and tell the user you could not check how it looks.",
+  "Each image uses part of the user's daily allowance, so do not regenerate unless the user asks."
+];
+var ERROR_TEXT = {
+  daily_cap: "The user's free allowance for today is used up. Do not retry. Tell the user they can get credits at freepi.ai/extra-credits, and continue without this tool.",
+  rolling_cap: "The user's short-term usage limit is reached. Do not retry; continue without this tool.",
+  rate_limited: "Too many image requests right now. Do not call generate_image again in this turn unless the user asks.",
+  not_configured: "Image generation is not available on this server. Do not call generate_image again in this session.",
+  upstream_error: "The image provider failed. You may retry once.",
+  account_review: "This account is under review, so image generation is disabled. Do not retry; tell the user and continue without it.",
+  content_rejected: "The image provider rejected this prompt. Do not retry the same prompt; tell the user and offer to reword it."
+};
+async function post(opts, body, signal) {
+  try {
+    const timeout = AbortSignal.timeout(TIMEOUT_MS);
+    const res = await (opts.fetchImpl ?? fetch)(`${opts.baseUrl}/tools/image`, {
+      method: "POST",
+      headers: { Authorization: `Bearer ${await opts.getToken()}`, "Content-Type": "application/json" },
+      body: JSON.stringify(body),
+      signal: signal ? AbortSignal.any([signal, timeout]) : timeout
+    });
+    if (res.ok)
+      return { ok: true, data: await res.json() };
+    const j = await res.json().catch(() => ({}));
+    const code = j.error?.code ?? j.code ?? "";
+    const mapped = ERROR_TEXT[code];
+    if (mapped)
+      return { ok: false, text: mapped };
+    return { ok: false, text: `Request failed (HTTP ${res.status}): ${j.error?.message ?? j.message ?? "no message"}` };
+  } catch (e) {
+    if (e instanceof Error && e.name === "TimeoutError")
+      return { ok: false, text: "Image generation timed out after 120 seconds." };
+    return { ok: false, text: `Request failed: ${String(e)}` };
+  }
+}
+function slug(s) {
+  return s.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-+/, "").slice(0, 40).replace(/-+$/, "") || "image";
+}
+var done = (text, details) => ({ content: [{ type: "text", text }], details });
+var fail = (text) => done(text, { ok: false });
+function createImageToolExtension(opts) {
+  return {
+    name: "free-pi-image-tool",
+    factory(pi) {
+      pi.registerTool({
+        name: IMAGE_TOOL_NAME,
+        label: "Generate Image",
+        description: "Generate one image from a text prompt and save it as a file under images/ in the project. Returns the file path; you cannot see the image. Use it only when the user asks for an image, mockup, icon or illustration. Each call uses part of the user's daily allowance.",
+        promptSnippet: "Generate an image file from a prompt when the user asks for one (counts against the user's daily allowance)",
+        promptGuidelines: IMAGE_GUIDELINES,
+        parameters: _Object_({
+          prompt: String2({ minLength: 1, maxLength: 1000, description: "What to draw: subject, style, any exact text to render, and composition" }),
+          aspect_ratio: Optional(Union([Literal("1:1"), Literal("16:9"), Literal("9:16"), Literal("4:3"), Literal("3:2")], { description: "Default 1:1" })),
+          name: Optional(String2({ maxLength: 60, description: "Short file name hint, e.g. hero-banner; used only for the saved file's name" }))
+        }),
+        async execute(_id, params, signal) {
+          const r = await post(opts, { prompt: params.prompt, ...params.aspect_ratio ? { aspect_ratio: params.aspect_ratio } : {} }, signal);
+          if (!r.ok)
+            return fail(r.text);
+          const d = r.data;
+          const ext = EXT[d.media_type];
+          if (!ext)
+            return fail(`The server returned an unsupported image type (${d.media_type}). Nothing was saved.`);
+          const buf = Buffer.from(d.image_b64 ?? "", "base64");
+          if (buf.length === 0)
+            return fail("The server returned an empty image. Nothing was saved.");
+          if (buf.length > MAX_BYTES)
+            return fail("The generated image was too large to save (over 8 MB). Nothing was saved.");
+          const rel = `images/${slug(params.name ?? params.prompt)}-${randomBytes(3).toString("hex")}.${ext}`;
+          try {
+            await mkdir2(join3(opts.cwd ?? process.cwd(), "images"), { recursive: true });
+            const fh = await open(join3(opts.cwd ?? process.cwd(), rel), "wx");
+            try {
+              await fh.writeFile(buf);
+            } finally {
+              await fh.close();
+            }
+          } catch (e) {
+            return fail(`Could not save the image: ${String(e)}`);
+          }
+          return done(`Saved image to ${rel} (${ext}, ${buf.length} bytes). You cannot see the image; reference this path in your answer or code, and tell the user you could not check how it looks.`, { ok: true, path: rel, cost_usd: d.cost_usd, model: d.model });
+        }
+      });
+    }
+  };
+}
+
 // src/web-tools.ts
 var WEB_SEARCH_TOOL_NAME = "web_search";
 var DOCS_LOOKUP_TOOL_NAME = "docs_lookup";
@@ -8559,7 +8696,7 @@ var WEB_SEARCH_GUIDELINES = [
   "Write a specific query the way you would type it into a search engine (include the product name, the version and the exact error text). Read the returned page text before searching again, and do not repeat a query that already returned results. Cite the URLs you relied on in your answer.",
   COST_GUIDELINE
 ];
-var ERROR_TEXT = {
+var ERROR_TEXT2 = {
   daily_cap: "The user's daily free usage is used up. Do not retry; tell the user and continue without this tool.",
   rolling_cap: "The user's short-term usage limit is reached. Do not retry; continue without this tool.",
   rate_limited: "Too many calls right now. Wait for the current call to finish, and do not call again in this turn unless needed.",
@@ -8567,7 +8704,7 @@ var ERROR_TEXT = {
   not_configured: "This tool is not available on this server. Do not call it again in this session.",
   upstream_error: "The search provider failed. You may retry once."
 };
-async function post(opts, path, body, label, signal) {
+async function post2(opts, path, body, label, signal) {
   try {
     const timeout = AbortSignal.timeout(30000);
     const res = await (opts.fetchImpl ?? fetch)(`${opts.baseUrl}${path}`, {
@@ -8579,7 +8716,7 @@ async function post(opts, path, body, label, signal) {
     if (res.ok)
       return { ok: true, data: await res.json() };
     const err = await res.json().catch(() => ({}));
-    const mapped = ERROR_TEXT[err.code ?? ""];
+    const mapped = ERROR_TEXT2[err.code ?? ""];
     if (mapped)
       return { ok: false, text: typeof mapped === "function" ? mapped(label) : mapped };
     return { ok: false, text: `Request failed (HTTP ${res.status}): ${err.message ?? "no message"}` };
@@ -8589,7 +8726,7 @@ async function post(opts, path, body, label, signal) {
     return { ok: false, text: `Request failed: ${String(e)}` };
   }
 }
-var done = (o) => ({ content: [{ type: "text", text: o.text }], details: o.details });
+var done2 = (o) => ({ content: [{ type: "text", text: o.text }], details: o.details });
 function createWebToolsExtension(opts) {
   return {
     name: "free-pi-web-tools",
@@ -8605,9 +8742,9 @@ function createWebToolsExtension(opts) {
           num_results: Optional(Integer({ minimum: 1, maximum: 10, description: "1-10, default 5; use 3 for a quick fact, 8-10 for a survey" }))
         }),
         async execute(_id, params, signal) {
-          const r = await post(opts, "/tools/web-search", params, "", signal);
+          const r = await post2(opts, "/tools/web-search", params, "", signal);
           if (!r.ok)
-            return done({ text: r.text, details: { ok: false } });
+            return done2({ text: r.text, details: { ok: false } });
           const results = r.data.results ?? [];
           const text = results.length ? `Web results for "${params.query}":
 
@@ -8617,7 +8754,7 @@ Published: ${x.published_date}` : ""}
 ${x.text}`).join(`
 
 `) : `No results for "${params.query}". Try a broader or differently worded query.`;
-          return done({ text, details: { ok: true, cost_usd: r.data.cost_usd } });
+          return done2({ text, details: { ok: true, cost_usd: r.data.cost_usd } });
         }
       });
       pi.registerTool({
@@ -8632,11 +8769,11 @@ ${x.text}`).join(`
           library_id: Optional(String2({ description: "The library_id printed by an earlier docs_lookup, e.g. /vercel/next.js; skips the name search" }))
         }),
         async execute(_id, params, signal) {
-          const r = await post(opts, "/tools/docs", params, params.library, signal);
+          const r = await post2(opts, "/tools/docs", params, params.library, signal);
           if (!r.ok)
-            return done({ text: r.text, details: { ok: false } });
+            return done2({ text: r.text, details: { ok: false } });
           const d = r.data;
-          return done({ text: `${d.title} (library_id: ${d.library_id})
+          return done2({ text: `${d.title} (library_id: ${d.library_id})
 
 ${d.docs}`, details: { ok: true, cost_usd: d.cost_usd } });
         }
@@ -9051,7 +9188,7 @@ function createTodoToolExtension(options = {}) {
 
 // src/tool-aliases.ts
 import { homedir as homedir2 } from "node:os";
-import { join as join3 } from "node:path";
+import { join as join4 } from "node:path";
 import { fileURLToPath as fileURLToPath2 } from "node:url";
 import {
   createBashToolDefinition,
@@ -9077,7 +9214,7 @@ function normalizeShellPath(input, platform = process.platform, home = homedir2(
   if (p === "~")
     return home;
   if (p.startsWith("~/") || platform === "win32" && p.startsWith("~\\"))
-    return join3(home, p.slice(2));
+    return join4(home, p.slice(2));
   if (/^file:\/\//.test(p))
     return fileURLToPath2(p);
   return p;
@@ -9109,6 +9246,12 @@ function createToolAliasesExtension() {
 
 // src/error-notice.ts
 var NOTICE_DEDUPE_MS = 15000;
+function optionalNumber(value) {
+  return typeof value === "number" && Number.isFinite(value) ? value : undefined;
+}
+function optionalString(value) {
+  return typeof value === "string" && value.trim() !== "" ? value : undefined;
+}
 function parseErrorEnvelope(errorMessage) {
   if (typeof errorMessage !== "string")
     return;
@@ -9121,14 +9264,53 @@ function parseErrorEnvelope(errorMessage) {
     const code = body.code ?? body.error?.code;
     if (typeof message !== "string" || message.trim() === "")
       return;
-    return { status: Number(m[1]), code: typeof code === "string" ? code : "unknown", message };
+    const parsed = {
+      status: Number(m[1]),
+      code: typeof code === "string" ? code : "unknown",
+      message
+    };
+    const limit = optionalString(body.limit);
+    const retryAfter = optionalNumber(body.retry_after_seconds);
+    const clearsAt = optionalString(body.clears_at);
+    const remaining = optionalNumber(body.remaining_credits_today);
+    const allowance = optionalNumber(body.allowance_credits);
+    if (limit)
+      parsed.limit = limit;
+    if (retryAfter !== undefined)
+      parsed.retry_after_seconds = retryAfter;
+    if (clearsAt)
+      parsed.clears_at = clearsAt;
+    if (remaining !== undefined)
+      parsed.remaining_credits_today = remaining;
+    if (allowance !== undefined)
+      parsed.allowance_credits = allowance;
+    const ceiling = optionalNumber(body.hourly_ceiling);
+    if (body.lane === "slow")
+      parsed.lane = "slow";
+    if (ceiling !== undefined)
+      parsed.hourly_ceiling = ceiling;
+    return parsed;
   } catch {
     return;
   }
 }
+function formatLimitEnvelope(env, opts) {
+  if (env.code === "concurrent_session" || typeof env.remaining_credits_today !== "number")
+    return env.message;
+  return formatLimitNotice({
+    limit: env.limit ?? env.code,
+    retry_after_seconds: env.retry_after_seconds,
+    clears_at: env.clears_at,
+    remaining_credits_today: env.remaining_credits_today,
+    allowance_credits: env.allowance_credits,
+    lane: env.lane,
+    hourly_ceiling: env.hourly_ceiling
+  }, { localTime: true, timeZone: opts?.timeZone });
+}
 function createErrorNoticeExtension(opts = {}) {
   const lastAt = new Map;
   const now = opts.now ?? Date.now;
+  const timeZone = opts.timeZone;
   return {
     name: "free-pi-error-notice",
     factory(pi) {
@@ -9143,7 +9325,7 @@ function createErrorNoticeExtension(opts = {}) {
         if (t - (lastAt.get(env.code) ?? -Infinity) < NOTICE_DEDUPE_MS)
           return;
         lastAt.set(env.code, t);
-        ctx.ui.notify(env.message, env.status >= 500 ? "error" : "warning");
+        ctx.ui.notify(formatLimitEnvelope(env, { timeZone }), env.status >= 500 ? "error" : "warning");
       });
     }
   };
@@ -9277,6 +9459,19 @@ function installFooterRightAlign(mode) {
 
 // src/changelog.ts
 var CHANGELOG_HIGHLIGHTS = [
+  {
+    version: "0.4.9",
+    highlights: [
+      "\uD83E\uDDE8 Fusion is now much cheaper: DeepSeek V4 Pro plans and DeepSeek V4 Flash builds, so each Fusion turn uses a small fraction of the allowance it used to. Fusion is also listed first in /model."
+    ]
+  },
+  {
+    version: "0.4.8",
+    highlights: [
+      "The agent can now make images: ask for a mockup, icon or illustration and generate_image saves it into your project's images/ folder. Each image counts toward your daily allowance.",
+      "Clearer limit messages: you now see when a limit clears and how many credits you have left today."
+    ]
+  },
   {
     version: "0.4.7",
     highlights: [
@@ -9708,9 +9903,18 @@ function createHeaderExtension(opts) {
 
 // src/fusion.ts
 var FUSION_ID = "fusion";
-var FUSION_NAME = "Fusion · Argon plans, DeepSeek builds";
-var FUSION_PLANNER_ID = "gemini-4-argon-flex";
+var FUSION_PICKER_ID = "\uD83E\uDDE8 fusion";
+var FUSION_NAME = "\uD83E\uDDE8 Fusion · DeepSeek Pro plans, Flash builds";
+var FUSION_FALLBACK_NAME = "\uD83E\uDDE8 Fusion · Argon plans, DeepSeek builds";
+var FUSION_PLANNER_ID = "deepseek-v4-pro";
 var FUSION_EXECUTOR_ID = "deepseek-v4-flash";
+var FUSION_FALLBACK_PLANNER_ID = "gemini-4-argon-flex";
+function resolveFusionIds(catalogIds, server) {
+  const has = (id) => catalogIds.includes(id);
+  if (server && has(server.planner) && has(server.executor))
+    return server;
+  return { planner: has(FUSION_PLANNER_ID) ? FUSION_PLANNER_ID : FUSION_FALLBACK_PLANNER_ID, executor: FUSION_EXECUTOR_ID };
+}
 function editedThisTurn(messages) {
   for (let i = messages.length - 1;i >= 0; i--) {
     const message = messages[i];
@@ -9755,7 +9959,7 @@ function routeFusion(request, models) {
       if (restartPlan)
         return route(models.planner, "plan");
       const failed = request.failed;
-      if (!failed || failed.model.provider === PROVIDER_NAME && failed.model.id === FUSION_PLANNER_ID) {
+      if (!failed || failed.model.provider === PROVIDER_NAME && failed.model.id === models.planner?.id) {
         return route(models.executor, "build");
       }
       return { ...route(failed.model), thinkingLevel: failed.thinkingLevel ?? request.thinkingLevel };
@@ -9764,21 +9968,24 @@ function routeFusion(request, models) {
       return phase === "build" || editedThisTurn(request.messages) ? route(models.executor, "build") : route(models.planner, "plan");
   }
 }
-function createFusionExtension(defaultModelId) {
+function createFusionExtension(defaultModelId, ids = resolveFusionIds([])) {
+  const name = ids.planner === FUSION_PLANNER_ID && ids.executor === FUSION_EXECUTOR_ID ? FUSION_NAME : ids.planner === FUSION_FALLBACK_PLANNER_ID ? FUSION_FALLBACK_NAME : "\uD83E\uDDE8 Fusion";
   return {
     name: "free-pi-fusion",
     factory: (pi) => {
-      pi.registerVirtualModel({
-        provider: PROVIDER_NAME,
-        id: FUSION_ID,
-        name: FUSION_NAME,
-        input: ["text"],
-        route: (request, ctx) => routeFusion(request, {
-          planner: ctx.modelRegistry.find(PROVIDER_NAME, FUSION_PLANNER_ID),
-          executor: ctx.modelRegistry.find(PROVIDER_NAME, FUSION_EXECUTOR_ID),
-          defaultModel: ctx.modelRegistry.find(PROVIDER_NAME, defaultModelId)
-        })
-      });
+      for (const id of [FUSION_PICKER_ID, FUSION_ID]) {
+        pi.registerVirtualModel({
+          provider: PROVIDER_NAME,
+          id,
+          name,
+          input: ["text"],
+          route: (request, ctx) => routeFusion(request, {
+            planner: ctx.modelRegistry.find(PROVIDER_NAME, ids.planner),
+            executor: ctx.modelRegistry.find(PROVIDER_NAME, ids.executor),
+            defaultModel: ctx.modelRegistry.find(PROVIDER_NAME, defaultModelId)
+          })
+        });
+      }
     }
   };
 }
@@ -9788,7 +9995,7 @@ function resolveFreePiScope(modelRuntime, providerName, modelIds, env = process.
   if (env.FREEPI_KEEP_ENV_PROVIDERS)
     return [];
   const scoped = [];
-  for (const id of new Set([...modelIds, FUSION_ID])) {
+  for (const id of new Set([FUSION_PICKER_ID, ...modelIds])) {
     const model = modelRuntime.getModel(providerName, id);
     if (model)
       scoped.push({ model });
@@ -9872,6 +10079,7 @@ var ALLOWED_TOOL_NAMES = [
   DOCS_TOOL_NAME,
   WEB_SEARCH_TOOL_NAME,
   DOCS_LOOKUP_TOOL_NAME,
+  IMAGE_TOOL_NAME,
   TODO_TOOL_NAME,
   "codemode",
   "tool_search"
@@ -9934,6 +10142,10 @@ function buildRuntimeOptions(opts, sessionId) {
     baseUrl: opts.baseUrl,
     getToken: () => opts.jwt
   });
+  const imageToolExtension = createImageToolExtension({
+    baseUrl: opts.baseUrl,
+    getToken: () => opts.jwt
+  });
   const errorNoticeExtension = createErrorNoticeExtension();
   const toolGuardExtension = createToolGuardExtension({
     baseUrl: opts.baseUrl,
@@ -9956,12 +10168,13 @@ function buildRuntimeOptions(opts, sessionId) {
   const toolSearchExtension = { name: "pi-tool-search", factory: createToolSearchExtension() };
   const extensionFactories = [
     providerExtension,
-    createFusionExtension(models[0].id),
+    createFusionExtension(models[0].id, resolveFusionIds(models.map((m) => m.id), opts.fusion)),
     adsExtension,
     usageToolExtension,
     buyToolExtension,
     docsToolExtension,
     webToolsExtension,
+    imageToolExtension,
     errorNoticeExtension,
     toolGuardExtension,
     commandsExtension,
@@ -9987,7 +10200,7 @@ function buildRuntimeOptions(opts, sessionId) {
 function getSessionDir(cwd, agentDir) {
   const resolvedCwd = resolve2(cwd);
   const safePath = `--${resolvedCwd.replace(/^[/\\]/, "").replace(/[/\\:]/g, "-")}--`;
-  return join4(resolve2(agentDir), "sessions", safePath);
+  return join5(resolve2(agentDir), "sessions", safePath);
 }
 async function createSessionManager(cwd, sessionDir, sessionId, api = SessionManager) {
   if (sessionId) {
@@ -10132,7 +10345,7 @@ function createRealSpawn(io = { platform: process.platform, spawnSyncImpl: spawn
 // src/relaunch.ts
 import { mkdtempSync, existsSync as existsSync2, rmSync } from "node:fs";
 import { constants as osConstants, tmpdir } from "node:os";
-import { join as join5 } from "node:path";
+import { join as join6 } from "node:path";
 var SEMVER = /^\d+\.\d+\.\d+(?:[-+][0-9A-Za-z.-]+)?$/;
 var WIN_SAFE_ARG = /^[A-Za-z0-9._:=\/@+,-]+$/;
 async function relaunchViaNpx(latest, deps) {
@@ -10145,8 +10358,8 @@ async function relaunchViaNpx(latest, deps) {
   if (win && !deps.argv.every((a) => WIN_SAFE_ARG.test(a)))
     return;
   const target = latest && SEMVER.test(latest) ? latest : "latest";
-  const readyDir = mkdtempSync(join5(tmpdir(), "free-pi-relaunch-"));
-  const readyFile = join5(readyDir, "started");
+  const readyDir = mkdtempSync(join6(tmpdir(), "free-pi-relaunch-"));
+  const readyFile = join6(readyDir, "started");
   try {
     deps.log(`Updating free-pi to ${target}…`);
     return await new Promise((resolve) => {
@@ -10444,6 +10657,7 @@ async function run(deps, argv = process.argv) {
     session: sessionArg.session,
     model: versionCheck.model,
     models: versionCheck.models,
+    fusion: versionCheck.fusion,
     updateLatest,
     getGravityContext
   });
@@ -10500,7 +10714,7 @@ async function checkClientVersion(baseUrl, cliVersion, fetchImpl = fetch) {
     const parsed = ClientVersionResponseSchema.safeParse(await res.json());
     if (!parsed.success)
       return { action: "ok" };
-    const { min, latest, model, models, notice } = parsed.data;
+    const { min, latest, model, models, notice, fusion } = parsed.data;
     const extra = {};
     if (model !== undefined)
       extra.model = model;
@@ -10508,6 +10722,8 @@ async function checkClientVersion(baseUrl, cliVersion, fetchImpl = fetch) {
       extra.models = models;
     if (notice !== undefined)
       extra.notice = notice;
+    if (fusion !== undefined)
+      extra.fusion = fusion;
     if (compareVersions(cliVersion, min) === -1)
       return { action: "block", latest, ...extra };
     if (compareVersions(cliVersion, latest) === -1)
@@ -10576,7 +10792,7 @@ ${HELP_TEXT}`);
     launchPi,
     log: (message) => console.log(message),
     checkClientVersion,
-    startGravityPixel: (jwt) => startGravityPixel(jwt, join6(dirname5(agentDir), "gravity")),
+    startGravityPixel: (jwt) => startGravityPixel(jwt, join7(dirname5(agentDir), "gravity")),
     relaunchLatest: (latest) => relaunchViaNpx(latest, {
       env: process.env,
       argv: process.argv.slice(2),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "free-pi-cli",
-  "version": "0.4.7",
+  "version": "0.4.9",
   "description": "Free, ad-supported distribution of the pi coding agent. Inference paid for by ads.",
   "license": "MIT",
   "type": "module",