Anemll · September 20, 2026 16:32
diff --git a/README.md b/README.md
diff --git a/pi-spark-image-history.ts b/pi-spark-image-history.ts
 // SPDX-License-Identifier: MIT
 // Install in ~/.pi/agent/extensions/, then /reload in Pi.
 // Request-only filtering: never edits transcripts, files, or model settings.
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";

 export const MAX_IMAGES = 4;
 export const OMITTED_IMAGE = "[Older image omitted from this request by the Spark four-image filter. Saved history is unchanged. Re-read or reattach the image to inspect it again.]";

 export function isSparkModel(model: any): boolean {
  return model?.id === "DeepSeek-v4.1-Flash-EXL3"
    && model?.api === "openai-completions"
    && ["gx10-dspark", "gx10-vllm"].includes(model?.provider);
 }

 // Pi serializes both user attachments and tool-result images as image_url
 // blocks in Chat Completions messages. Count across the whole request.
 export function limitImages(payload: any) {
  const messages = Array.isArray(payload?.messages) ? payload.messages : [];
  const total = messages.reduce((count: number, message: any) => count + (
    Array.isArray(message?.content)
      ? message.content.filter((part: any) => part?.type === "image_url").length
      : 0
  ), 0);
  const removed = Math.max(0, total - MAX_IMAGES);
  if (!removed) return { payload, total, removed, kept: total };

  let remaining = removed;
  const filtered = messages.map((message: any) => {
    if (!remaining || !Array.isArray(message?.content)) return message;
    let changed = false;
    const content = message.content.map((part: any) => {
      if (part?.type !== "image_url" || !remaining) return part;
      remaining--;
      changed = true;
      return { type: "text", text: OMITTED_IMAGE };
    });
    return changed ? { ...message, content } : message;
  });
  return { payload: { ...payload, messages: filtered }, total, removed, kept: MAX_IMAGES };
 }

 export default function (pi: ExtensionAPI) {
  let warned = false;
  let last: { total: number; kept: number; removed: number } | undefined;

  const status = (ctx: any, model = ctx.model) => {
    if (!ctx.hasUI) return;
    ctx.ui.setStatus("spark-image-history", isSparkModel(model)
      ? (last ? `images ${last.kept}/${last.total}` : "images: latest 4")
      : undefined);
  };

  pi.on("session_start", (_event, ctx) => {
    warned = false;
    last = undefined;
    status(ctx);
  });
  pi.on("model_select", (event, ctx) => {
    warned = false;
    last = undefined;
    status(ctx, event.model);
  });
  pi.on("agent_start", () => { warned = false; });

  pi.on("before_provider_request", (event, ctx) => {
    if (!isSparkModel(ctx.model)) return;
    const result = limitImages(event.payload);
    last = { total: result.total, kept: result.kept, removed: result.removed };
    // A footer/notification failure must not disable request filtering.
    try {
      status(ctx);
      if (result.removed && !warned && ctx.hasUI) {
        ctx.ui.notify(`Spark: sending the latest ${MAX_IMAGES} of ${result.total} images; ${result.removed} older image(s) replaced with text in this request only. Saved history is unchanged. Submit new image batches in groups of at most four.`, "warning");
        warned = true;
      }
    } catch { /* UI only; the filtered payload is still returned below. */ }
    if (result.removed) return result.payload;
  });

  pi.registerCommand("spark-images", {
    description: "Show the Spark four-image request filter status",
    handler: async (_args, ctx) => {
      ctx.ui.notify(isSparkModel(ctx.model)
        ? `Spark image filter active: newest ${MAX_IMAGES} images only; saved history unchanged.`
          + (last ? ` Last request: ${last.kept}/${last.total} images sent.` : " No request checked yet.")
        : "Spark image filter loaded but inactive for the selected model/provider.", "info");
    },
  });
 }
Field	Value
Model ID	`DeepSeek-v4.1-Flash-EXL3`
API	`openai-completions`
Provider ID	`gx10-dspark` or `gx10-vllm`
Images retained in a request	Approximate image-token budget	Preprocessed image tensors, one resident copy
1	1,025	~10 MiB
4	4,100	~40 MiB
16	16,400	~160 MiB
100	102,500	~1 GiB
	// SPDX-License-Identifier: MIT
	// Install in ~/.pi/agent/extensions/, then /reload in Pi.
	// Request-only filtering: never edits transcripts, files, or model settings.
	import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";

	export const MAX_IMAGES = 4;
	export const OMITTED_IMAGE = "[Older image omitted from this request by the Spark four-image filter. Saved history is unchanged. Re-read or reattach the image to inspect it again.]";

	export function isSparkModel(model: any): boolean {
	return model?.id === "DeepSeek-v4.1-Flash-EXL3"
	&& model?.api === "openai-completions"
	&& ["gx10-dspark", "gx10-vllm"].includes(model?.provider);
	}

	// Pi serializes both user attachments and tool-result images as image_url
	// blocks in Chat Completions messages. Count across the whole request.
	export function limitImages(payload: any) {
	const messages = Array.isArray(payload?.messages) ? payload.messages : [];
	const total = messages.reduce((count: number, message: any) => count + (
	Array.isArray(message?.content)
	? message.content.filter((part: any) => part?.type === "image_url").length
	: 0
	), 0);
	const removed = Math.max(0, total - MAX_IMAGES);
	if (!removed) return { payload, total, removed, kept: total };

	let remaining = removed;
	const filtered = messages.map((message: any) => {
	if (!remaining \|\| !Array.isArray(message?.content)) return message;
	let changed = false;
	const content = message.content.map((part: any) => {
	if (part?.type !== "image_url" \|\| !remaining) return part;
	remaining--;
	changed = true;
	return { type: "text", text: OMITTED_IMAGE };
	});
	return changed ? { ...message, content } : message;
	});
	return { payload: { ...payload, messages: filtered }, total, removed, kept: MAX_IMAGES };
	}

	export default function (pi: ExtensionAPI) {
	let warned = false;
	let last: { total: number; kept: number; removed: number } \| undefined;

	const status = (ctx: any, model = ctx.model) => {
	if (!ctx.hasUI) return;
	ctx.ui.setStatus("spark-image-history", isSparkModel(model)
	? (last ? `images ${last.kept}/${last.total}` : "images: latest 4")
	: undefined);
	};

	pi.on("session_start", (_event, ctx) => {
	warned = false;
	last = undefined;
	status(ctx);
	});
	pi.on("model_select", (event, ctx) => {
	warned = false;
	last = undefined;
	status(ctx, event.model);
	});
	pi.on("agent_start", () => { warned = false; });

	pi.on("before_provider_request", (event, ctx) => {
	if (!isSparkModel(ctx.model)) return;
	const result = limitImages(event.payload);
	last = { total: result.total, kept: result.kept, removed: result.removed };
	// A footer/notification failure must not disable request filtering.
	try {
	status(ctx);
	if (result.removed && !warned && ctx.hasUI) {
	ctx.ui.notify(`Spark: sending the latest ${MAX_IMAGES} of ${result.total} images; ${result.removed} older image(s) replaced with text in this request only. Saved history is unchanged. Submit new image batches in groups of at most four.`, "warning");
	warned = true;
	}
	} catch { /* UI only; the filtered payload is still returned below. */ }
	if (result.removed) return result.payload;
	});

	pi.registerCommand("spark-images", {
	description: "Show the Spark four-image request filter status",
	handler: async (_args, ctx) => {
	ctx.ui.notify(isSparkModel(ctx.model)
	? `Spark image filter active: newest ${MAX_IMAGES} images only; saved history unchanged.`
	+ (last ? ` Last request: ${last.kept}/${last.total} images sent.` : " No request checked yet.")
	: "Spark image filter loaded but inactive for the selected model/provider.", "info");
	},
	});
	}