@asterxsk/kiln 0.1.0 → 0.3.0

package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs

@@ -1,170 +1,170 @@
-import assert from "node:assert/strict";
-import { spawnSync } from "node:child_process";
-import { mkdtemp, writeFile } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import { test } from "node:test";
-
-const videoModuleUrl = new URL("../video-extract.ts", import.meta.url).href;
-
-function runChild(script, env) {
-	const childEnv = { ...process.env };
-	for (const key of ["PI_CODING_AGENT_DIR", "XDG_CONFIG_HOME", "GEMINI_API_KEY", "GOOGLE_GEMINI_BASE_URL", "CLOUDFLARE_API_KEY"]) {
-		delete childEnv[key];
-	}
-	Object.assign(childEnv, env);
-	return spawnSync(process.execPath, ["--input-type=module"], {
-		input: script,
-		encoding: "utf8",
-		env: childEnv,
-		maxBuffer: 2 * 1024 * 1024,
-	});
-}
-
-test("Gemini generate, upload, status, and delete use only shared header auth", async () => {
-	const root = await mkdtemp(join(tmpdir(), "pi-web-access-gemini-transport-"));
-	const mediaPath = join(root, "synthetic.webm");
-	await writeFile(mediaPath, "synthetic media", "utf8");
-	await writeFile(join(root, "web-search.json"), JSON.stringify({ geminiApiKey: "synthetic-gemini-key" }) + "\n", "utf8");
-
-	const child = runChild(`
-		const requests = [];
-		globalThis.fetch = async (url, init = {}) => {
-			const request = {
-				url: String(url),
-				method: init.method ?? "GET",
-				headers: Object.fromEntries(new Headers(init.headers)),
-			};
-			requests.push(request);
-			if (request.url.endsWith("/upload/v1beta/files")) {
-				return new Response("", {
-					status: 200,
-					headers: { "x-goog-upload-url": "https://generativelanguage.googleapis.com/upload/v1beta/files?upload_id=synthetic" },
-				});
-			}
-			if (request.url.includes("upload_id=synthetic")) {
-				return new Response(JSON.stringify({ file: { name: "files/synthetic", uri: "files/synthetic" } }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "GET") {
-				return new Response(JSON.stringify({ state: "ACTIVE" }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.includes(":generateContent")) {
-				return new Response(JSON.stringify({ candidates: [{ content: { parts: [{ text: "# Synthetic video" }] } }] }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "DELETE") {
-				return new Response("", { status: 204 });
-			}
-			throw new Error("Unexpected fetch: " + request.url);
-		};
-
-		const { extractVideo } = await import(${JSON.stringify(videoModuleUrl)});
-		const result = await extractVideo({
-			absolutePath: ${JSON.stringify(mediaPath)},
-			mimeType: "video/webm",
-			sizeBytes: 15,
-			maxSizeBytes: 50 * 1024 * 1024,
-			withinUploadLimit: true,
-		});
-		await new Promise(resolve => setImmediate(resolve));
-		console.log(JSON.stringify({ result, requests }));
-	`, {
-		HOME: root,
-		USERPROFILE: root,
-		PI_CODING_AGENT_DIR: root,
-	});
-
-	assert.equal(child.status, 0, child.stderr);
-	const output = JSON.parse(child.stdout.trim());
-	assert.equal(output.result.content, "# Synthetic video");
-	assert.deepEqual(output.requests.map(request => request.method), ["POST", "PUT", "GET", "POST", "DELETE"]);
-	const generateRequest = output.requests.find(request => request.url.includes(":generateContent"));
-	assert.equal(generateRequest.url, "https://generativelanguage.googleapis.com/v1beta/models/gemini-3.6-flash:generateContent");
-	for (const request of output.requests) {
-		assert.equal(request.headers["x-goog-api-key"], "synthetic-gemini-key");
-		const url = new URL(request.url);
-		assert.equal(url.searchParams.has("key"), false);
-		assert.equal(url.searchParams.has("api_key"), false);
-		assert.equal(request.url.includes("synthetic-gemini-key"), false);
-	}
-});
-
-test("Gemini video upload, status, and delete respect a configured geminiBaseUrl relay", async () => {
-	const root = await mkdtemp(join(tmpdir(), "pi-web-access-gemini-relay-"));
-	const mediaPath = join(root, "synthetic.webm");
-	await writeFile(mediaPath, "synthetic media", "utf8");
-	await writeFile(join(root, "web-search.json"), JSON.stringify({ geminiApiKey: "synthetic-gemini-key", geminiBaseUrl: "https://relay.example.com" }) + "\n", "utf8");
-
-	const child = runChild(`
-		const requests = [];
-		globalThis.fetch = async (url, init = {}) => {
-			const request = {
-				url: String(url),
-				method: init.method ?? "GET",
-				headers: Object.fromEntries(new Headers(init.headers)),
-			};
-			requests.push(request);
-			if (request.url.endsWith("/upload/v1beta/files")) {
-				return new Response("", {
-					status: 200,
-					headers: { "x-goog-upload-url": "https://relay.example.com/upload/v1beta/files?upload_id=synthetic" },
-				});
-			}
-			if (request.url.includes("upload_id=synthetic")) {
-				return new Response(JSON.stringify({ file: { name: "files/synthetic", uri: "files/synthetic" } }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "GET") {
-				return new Response(JSON.stringify({ state: "ACTIVE" }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.includes(":generateContent")) {
-				return new Response(JSON.stringify({ candidates: [{ content: { parts: [{ text: "# Relay video" }] } }] }), {
-					status: 200,
-					headers: { "content-type": "application/json" },
-				});
-			}
-			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "DELETE") {
-				return new Response("", { status: 204 });
-			}
-			throw new Error("Unexpected fetch: " + request.url);
-		};
-
-		const { extractVideo } = await import(${JSON.stringify(videoModuleUrl)});
-		const result = await extractVideo({
-			absolutePath: ${JSON.stringify(mediaPath)},
-			mimeType: "video/webm",
-			sizeBytes: 15,
-			maxSizeBytes: 50 * 1024 * 1024,
-			withinUploadLimit: true,
-		});
-		await new Promise(resolve => setImmediate(resolve));
-		console.log(JSON.stringify({ result, requests }));
-	`, {
-		HOME: root,
-		USERPROFILE: root,
-		PI_CODING_AGENT_DIR: root,
-	});
-
-	assert.equal(child.status, 0, child.stderr);
-	const output = JSON.parse(child.stdout.trim());
-	assert.equal(output.result.content, "# Relay video");
-	assert.deepEqual(output.requests.map(request => request.method), ["POST", "PUT", "GET", "POST", "DELETE"]);
-	for (const request of output.requests) {
-		assert.equal(new URL(request.url).origin, "https://relay.example.com", request.url);
-		assert.equal(request.headers["x-goog-api-key"], "synthetic-gemini-key");
-		assert.equal(request.url.includes("generativelanguage.googleapis.com"), false);
-	}
-});
+import assert from "node:assert/strict";
+import { spawnSync } from "node:child_process";
+import { mkdtemp, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { test } from "node:test";
+
+const videoModuleUrl = new URL("../video-extract.ts", import.meta.url).href;
+
+function runChild(script, env) {
+	const childEnv = { ...process.env };
+	for (const key of ["PI_CODING_AGENT_DIR", "XDG_CONFIG_HOME", "GEMINI_API_KEY", "GOOGLE_GEMINI_BASE_URL", "CLOUDFLARE_API_KEY"]) {
+		delete childEnv[key];
+	}
+	Object.assign(childEnv, env);
+	return spawnSync(process.execPath, ["--input-type=module"], {
+		input: script,
+		encoding: "utf8",
+		env: childEnv,
+		maxBuffer: 2 * 1024 * 1024,
+	});
+}
+
+test("Gemini generate, upload, status, and delete use only shared header auth", async () => {
+	const root = await mkdtemp(join(tmpdir(), "pi-web-access-gemini-transport-"));
+	const mediaPath = join(root, "synthetic.webm");
+	await writeFile(mediaPath, "synthetic media", "utf8");
+	await writeFile(join(root, "web-search.json"), JSON.stringify({ geminiApiKey: "synthetic-gemini-key" }) + "\n", "utf8");
+
+	const child = runChild(`
+		const requests = [];
+		globalThis.fetch = async (url, init = {}) => {
+			const request = {
+				url: String(url),
+				method: init.method ?? "GET",
+				headers: Object.fromEntries(new Headers(init.headers)),
+			};
+			requests.push(request);
+			if (request.url.endsWith("/upload/v1beta/files")) {
+				return new Response("", {
+					status: 200,
+					headers: { "x-goog-upload-url": "https://generativelanguage.googleapis.com/upload/v1beta/files?upload_id=synthetic" },
+				});
+			}
+			if (request.url.includes("upload_id=synthetic")) {
+				return new Response(JSON.stringify({ file: { name: "files/synthetic", uri: "files/synthetic" } }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "GET") {
+				return new Response(JSON.stringify({ state: "ACTIVE" }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.includes(":generateContent")) {
+				return new Response(JSON.stringify({ candidates: [{ content: { parts: [{ text: "# Synthetic video" }] } }] }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "DELETE") {
+				return new Response("", { status: 204 });
+			}
+			throw new Error("Unexpected fetch: " + request.url);
+		};
+
+		const { extractVideo } = await import(${JSON.stringify(videoModuleUrl)});
+		const result = await extractVideo({
+			absolutePath: ${JSON.stringify(mediaPath)},
+			mimeType: "video/webm",
+			sizeBytes: 15,
+			maxSizeBytes: 50 * 1024 * 1024,
+			withinUploadLimit: true,
+		});
+		await new Promise(resolve => setImmediate(resolve));
+		console.log(JSON.stringify({ result, requests }));
+	`, {
+		HOME: root,
+		USERPROFILE: root,
+		PI_CODING_AGENT_DIR: root,
+	});
+
+	assert.equal(child.status, 0, child.stderr);
+	const output = JSON.parse(child.stdout.trim());
+	assert.equal(output.result.content, "# Synthetic video");
+	assert.deepEqual(output.requests.map(request => request.method), ["POST", "PUT", "GET", "POST", "DELETE"]);
+	const generateRequest = output.requests.find(request => request.url.includes(":generateContent"));
+	assert.equal(generateRequest.url, "https://generativelanguage.googleapis.com/v1beta/models/gemini-3.6-flash:generateContent");
+	for (const request of output.requests) {
+		assert.equal(request.headers["x-goog-api-key"], "synthetic-gemini-key");
+		const url = new URL(request.url);
+		assert.equal(url.searchParams.has("key"), false);
+		assert.equal(url.searchParams.has("api_key"), false);
+		assert.equal(request.url.includes("synthetic-gemini-key"), false);
+	}
+});
+
+test("Gemini video upload, status, and delete respect a configured geminiBaseUrl relay", async () => {
+	const root = await mkdtemp(join(tmpdir(), "pi-web-access-gemini-relay-"));
+	const mediaPath = join(root, "synthetic.webm");
+	await writeFile(mediaPath, "synthetic media", "utf8");
+	await writeFile(join(root, "web-search.json"), JSON.stringify({ geminiApiKey: "synthetic-gemini-key", geminiBaseUrl: "https://relay.example.com" }) + "\n", "utf8");
+
+	const child = runChild(`
+		const requests = [];
+		globalThis.fetch = async (url, init = {}) => {
+			const request = {
+				url: String(url),
+				method: init.method ?? "GET",
+				headers: Object.fromEntries(new Headers(init.headers)),
+			};
+			requests.push(request);
+			if (request.url.endsWith("/upload/v1beta/files")) {
+				return new Response("", {
+					status: 200,
+					headers: { "x-goog-upload-url": "https://relay.example.com/upload/v1beta/files?upload_id=synthetic" },
+				});
+			}
+			if (request.url.includes("upload_id=synthetic")) {
+				return new Response(JSON.stringify({ file: { name: "files/synthetic", uri: "files/synthetic" } }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "GET") {
+				return new Response(JSON.stringify({ state: "ACTIVE" }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.includes(":generateContent")) {
+				return new Response(JSON.stringify({ candidates: [{ content: { parts: [{ text: "# Relay video" }] } }] }), {
+					status: 200,
+					headers: { "content-type": "application/json" },
+				});
+			}
+			if (request.url.endsWith("/v1beta/files/synthetic") && request.method === "DELETE") {
+				return new Response("", { status: 204 });
+			}
+			throw new Error("Unexpected fetch: " + request.url);
+		};
+
+		const { extractVideo } = await import(${JSON.stringify(videoModuleUrl)});
+		const result = await extractVideo({
+			absolutePath: ${JSON.stringify(mediaPath)},
+			mimeType: "video/webm",
+			sizeBytes: 15,
+			maxSizeBytes: 50 * 1024 * 1024,
+			withinUploadLimit: true,
+		});
+		await new Promise(resolve => setImmediate(resolve));
+		console.log(JSON.stringify({ result, requests }));
+	`, {
+		HOME: root,
+		USERPROFILE: root,
+		PI_CODING_AGENT_DIR: root,
+	});
+
+	assert.equal(child.status, 0, child.stderr);
+	const output = JSON.parse(child.stdout.trim());
+	assert.equal(output.result.content, "# Relay video");
+	assert.deepEqual(output.requests.map(request => request.method), ["POST", "PUT", "GET", "POST", "DELETE"]);
+	for (const request of output.requests) {
+		assert.equal(new URL(request.url).origin, "https://relay.example.com", request.url);
+		assert.equal(request.headers["x-goog-api-key"], "synthetic-gemini-key");
+		assert.equal(request.url.includes("generativelanguage.googleapis.com"), false);
+	}
+});

package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs

@@ -1,133 +1,133 @@
-import assert from "node:assert/strict";
-import { after, test } from "node:test";
-
-const originalFetch = globalThis.fetch;
-const originalApiKey = process.env.GEMINI_API_KEY;
-process.env.GEMINI_API_KEY = "synthetic-gemini-key";
-
-const { extractPDFViaGemini } = await import("../gemini-pdf-extract.ts");
-
-after(() => {
-	globalThis.fetch = originalFetch;
-	if (originalApiKey === undefined) delete process.env.GEMINI_API_KEY;
-	else process.env.GEMINI_API_KEY = originalApiKey;
-});
-
-test("Gemini PDF conversion sends inline PDF data and normalizes valid Markdown", async () => {
-	let capturedUrl = "";
-	let capturedInit;
-	globalThis.fetch = async (url, init) => {
-		capturedUrl = String(url);
-		capturedInit = init;
-		return jsonResponse({
-			candidates: [{
-				finishReason: "STOP",
-				content: { parts: [{ text: "```markdown\n<!-- Page 1 -->\n# Sample Document\n\nFirst page.\n\n<!-- Page 2 -->\n\nSecond page.\n```" }] },
-			}],
-		});
-	};
-
-	const input = Uint8Array.from([1, 2, 3, 4]).buffer;
-	const markdown = await extractPDFViaGemini(input, {
-		pages: 5,
-		maxPages: 2,
-		title: "Sample Document",
-	});
-
-	assert.match(capturedUrl, /gemini-3\.6-flash:generateContent$/);
-	assert.equal(new Headers(capturedInit.headers).get("x-goog-api-key"), "synthetic-gemini-key");
-	const body = JSON.parse(capturedInit.body);
-	assert.equal(body.contents[0].role, "user");
-	assert.equal(body.contents[0].parts[0].inlineData.mimeType, "application/pdf");
-	assert.equal(body.contents[0].parts[0].inlineData.data, Buffer.from(input).toString("base64"));
-	assert.match(body.contents[0].parts[1].text, /pages 1 through 2/);
-	assert.match(markdown, /^<!-- Page 1 -->/);
-	assert.doesNotMatch(markdown, /^<!-- Page 1 -->\s*\n# Sample Document/m);
-	assert.match(markdown, /<!-- Page 2 -->/);
-});
-
-test("Gemini PDF conversion supports Cloudflare AI Gateway without a direct Gemini key", async () => {
-	const savedApiKey = process.env.GEMINI_API_KEY;
-	const savedBaseUrl = process.env.GOOGLE_GEMINI_BASE_URL;
-	const savedCloudflareKey = process.env.CLOUDFLARE_API_KEY;
-	delete process.env.GEMINI_API_KEY;
-	process.env.GOOGLE_GEMINI_BASE_URL = "https://gateway.ai.cloudflare.com/v1/example/gemini";
-	process.env.CLOUDFLARE_API_KEY = "synthetic-cloudflare-key";
-
-	try {
-		globalThis.fetch = async (url, init) => {
-			assert.match(String(url), /^https:\/\/gateway\.ai\.cloudflare\.com\//);
-			assert.equal(new Headers(init.headers).get("cf-aig-authorization"), "Bearer synthetic-cloudflare-key");
-			assert.equal(new Headers(init.headers).get("x-goog-api-key"), null);
-			return jsonResponse({
-				candidates: [{
-					finishReason: "STOP",
-					content: { parts: [{ text: "<!-- Page 1 -->\nGateway result" }] },
-				}],
-			});
-		};
-
-		const markdown = await extractPDFViaGemini(new ArrayBuffer(1), {
-			pages: 1,
-			maxPages: 1,
-			title: "Document",
-		});
-		assert.match(markdown, /Gateway result/);
-	} finally {
-		if (savedApiKey === undefined) delete process.env.GEMINI_API_KEY;
-		else process.env.GEMINI_API_KEY = savedApiKey;
-		if (savedBaseUrl === undefined) delete process.env.GOOGLE_GEMINI_BASE_URL;
-		else process.env.GOOGLE_GEMINI_BASE_URL = savedBaseUrl;
-		if (savedCloudflareKey === undefined) delete process.env.CLOUDFLARE_API_KEY;
-		else process.env.CLOUDFLARE_API_KEY = savedCloudflareKey;
-	}
-});
-
-test("Gemini PDF conversion rejects truncated output", async () => {
-	globalThis.fetch = async () => jsonResponse({
-		candidates: [{
-			finishReason: "MAX_TOKENS",
-			content: { parts: [{ text: "<!-- Page 1 -->\nPartial" }] },
-		}],
-	});
-
-	await assert.rejects(
-		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
-		/MAX_TOKENS/,
-	);
-});
-
-test("Gemini PDF conversion rejects missing or out-of-sequence page markers", async () => {
-	globalThis.fetch = async () => jsonResponse({
-		candidates: [{
-			finishReason: "STOP",
-			content: { parts: [{ text: "<!-- Page 1 -->\nFirst\n\n<!-- Page 3 -->\nThird" }] },
-		}],
-	});
-
-	await assert.rejects(
-		extractPDFViaGemini(new ArrayBuffer(1), { pages: 2, maxPages: 2, title: "Document" }),
-		/out of sequence/,
-	);
-});
-
-test("Gemini PDF conversion surfaces safety blocks and empty responses", async () => {
-	globalThis.fetch = async () => jsonResponse({ promptFeedback: { blockReason: "SAFETY" } });
-	await assert.rejects(
-		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
-		/blocked PDF extraction: SAFETY/,
-	);
-
-	globalThis.fetch = async () => jsonResponse({ candidates: [{ finishReason: "STOP", content: { parts: [] } }] });
-	await assert.rejects(
-		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
-		/empty PDF extraction/,
-	);
-});
-
-function jsonResponse(value) {
-	return new Response(JSON.stringify(value), {
-		status: 200,
-		headers: { "content-type": "application/json" },
-	});
-}
+import assert from "node:assert/strict";
+import { after, test } from "node:test";
+
+const originalFetch = globalThis.fetch;
+const originalApiKey = process.env.GEMINI_API_KEY;
+process.env.GEMINI_API_KEY = "synthetic-gemini-key";
+
+const { extractPDFViaGemini } = await import("../gemini-pdf-extract.ts");
+
+after(() => {
+	globalThis.fetch = originalFetch;
+	if (originalApiKey === undefined) delete process.env.GEMINI_API_KEY;
+	else process.env.GEMINI_API_KEY = originalApiKey;
+});
+
+test("Gemini PDF conversion sends inline PDF data and normalizes valid Markdown", async () => {
+	let capturedUrl = "";
+	let capturedInit;
+	globalThis.fetch = async (url, init) => {
+		capturedUrl = String(url);
+		capturedInit = init;
+		return jsonResponse({
+			candidates: [{
+				finishReason: "STOP",
+				content: { parts: [{ text: "```markdown\n<!-- Page 1 -->\n# Sample Document\n\nFirst page.\n\n<!-- Page 2 -->\n\nSecond page.\n```" }] },
+			}],
+		});
+	};
+
+	const input = Uint8Array.from([1, 2, 3, 4]).buffer;
+	const markdown = await extractPDFViaGemini(input, {
+		pages: 5,
+		maxPages: 2,
+		title: "Sample Document",
+	});
+
+	assert.match(capturedUrl, /gemini-3\.6-flash:generateContent$/);
+	assert.equal(new Headers(capturedInit.headers).get("x-goog-api-key"), "synthetic-gemini-key");
+	const body = JSON.parse(capturedInit.body);
+	assert.equal(body.contents[0].role, "user");
+	assert.equal(body.contents[0].parts[0].inlineData.mimeType, "application/pdf");
+	assert.equal(body.contents[0].parts[0].inlineData.data, Buffer.from(input).toString("base64"));
+	assert.match(body.contents[0].parts[1].text, /pages 1 through 2/);
+	assert.match(markdown, /^<!-- Page 1 -->/);
+	assert.doesNotMatch(markdown, /^<!-- Page 1 -->\s*\n# Sample Document/m);
+	assert.match(markdown, /<!-- Page 2 -->/);
+});
+
+test("Gemini PDF conversion supports Cloudflare AI Gateway without a direct Gemini key", async () => {
+	const savedApiKey = process.env.GEMINI_API_KEY;
+	const savedBaseUrl = process.env.GOOGLE_GEMINI_BASE_URL;
+	const savedCloudflareKey = process.env.CLOUDFLARE_API_KEY;
+	delete process.env.GEMINI_API_KEY;
+	process.env.GOOGLE_GEMINI_BASE_URL = "https://gateway.ai.cloudflare.com/v1/example/gemini";
+	process.env.CLOUDFLARE_API_KEY = "synthetic-cloudflare-key";
+
+	try {
+		globalThis.fetch = async (url, init) => {
+			assert.match(String(url), /^https:\/\/gateway\.ai\.cloudflare\.com\//);
+			assert.equal(new Headers(init.headers).get("cf-aig-authorization"), "Bearer synthetic-cloudflare-key");
+			assert.equal(new Headers(init.headers).get("x-goog-api-key"), null);
+			return jsonResponse({
+				candidates: [{
+					finishReason: "STOP",
+					content: { parts: [{ text: "<!-- Page 1 -->\nGateway result" }] },
+				}],
+			});
+		};
+
+		const markdown = await extractPDFViaGemini(new ArrayBuffer(1), {
+			pages: 1,
+			maxPages: 1,
+			title: "Document",
+		});
+		assert.match(markdown, /Gateway result/);
+	} finally {
+		if (savedApiKey === undefined) delete process.env.GEMINI_API_KEY;
+		else process.env.GEMINI_API_KEY = savedApiKey;
+		if (savedBaseUrl === undefined) delete process.env.GOOGLE_GEMINI_BASE_URL;
+		else process.env.GOOGLE_GEMINI_BASE_URL = savedBaseUrl;
+		if (savedCloudflareKey === undefined) delete process.env.CLOUDFLARE_API_KEY;
+		else process.env.CLOUDFLARE_API_KEY = savedCloudflareKey;
+	}
+});
+
+test("Gemini PDF conversion rejects truncated output", async () => {
+	globalThis.fetch = async () => jsonResponse({
+		candidates: [{
+			finishReason: "MAX_TOKENS",
+			content: { parts: [{ text: "<!-- Page 1 -->\nPartial" }] },
+		}],
+	});
+
+	await assert.rejects(
+		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
+		/MAX_TOKENS/,
+	);
+});
+
+test("Gemini PDF conversion rejects missing or out-of-sequence page markers", async () => {
+	globalThis.fetch = async () => jsonResponse({
+		candidates: [{
+			finishReason: "STOP",
+			content: { parts: [{ text: "<!-- Page 1 -->\nFirst\n\n<!-- Page 3 -->\nThird" }] },
+		}],
+	});
+
+	await assert.rejects(
+		extractPDFViaGemini(new ArrayBuffer(1), { pages: 2, maxPages: 2, title: "Document" }),
+		/out of sequence/,
+	);
+});
+
+test("Gemini PDF conversion surfaces safety blocks and empty responses", async () => {
+	globalThis.fetch = async () => jsonResponse({ promptFeedback: { blockReason: "SAFETY" } });
+	await assert.rejects(
+		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
+		/blocked PDF extraction: SAFETY/,
+	);
+
+	globalThis.fetch = async () => jsonResponse({ candidates: [{ finishReason: "STOP", content: { parts: [] } }] });
+	await assert.rejects(
+		extractPDFViaGemini(new ArrayBuffer(1), { pages: 1, maxPages: 1, title: "Document" }),
+		/empty PDF extraction/,
+	);
+});
+
+function jsonResponse(value) {
+	return new Response(JSON.stringify(value), {
+		status: 200,
+		headers: { "content-type": "application/json" },
+	});
+}