feat: integrate Pi backend and Pi Web

2026-10-03 02:59:35 +00:00 · 2026-07-30 19:37:53 +08:00
commit 7392ab9dd7
1390 changed files with 337197 additions and 0 deletions
@@ -0,0 +1,178 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { streamSimple as streamAnthropic } from "../src/api/anthropic-messages.ts";
+import { streamSimple as streamAzureOpenAIResponses } from "../src/api/azure-openai-responses.ts";
+import { streamSimple as streamGoogleGenerativeAI } from "../src/api/google-generative-ai.ts";
+import { streamSimple as streamGoogleVertex } from "../src/api/google-vertex.ts";
+import { streamSimple as streamMistral } from "../src/api/mistral-conversations.ts";
+import { streamSimple as streamOpenAICodexResponses } from "../src/api/openai-codex-responses.ts";
+import { streamSimple as streamOpenAICompletions } from "../src/api/openai-completions.ts";
+import { streamSimple as streamOpenAIResponses } from "../src/api/openai-responses.ts";
+import { generateImages } from "../src/api/openrouter-images.ts";
+import { streamSimple as streamPiMessages } from "../src/api/pi-messages.ts";
+import type { Api, Context, FetchFunction, ImagesModel, Model } from "../src/types.ts";
+
+const context: Context = {
+	messages: [{ role: "user", content: "hello", timestamp: 1 }],
+};
+
+function createModel<TApi extends Api>(api: TApi): Model<TApi> {
+	return {
+		id: "test-model",
+		name: "Test Model",
+		api,
+		provider: "test-provider",
+		baseUrl: "https://upstream.test/v1",
+		reasoning: false,
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 10_000,
+		maxTokens: 1_000,
+	};
+}
+
+function mockFetches() {
+	const fallback = vi.fn<FetchFunction>(async () => {
+		throw new Error("ambient fetch must not be called");
+	});
+	const custom = vi.fn<FetchFunction>(
+		async () =>
+			new Response(JSON.stringify({ error: { message: "upstream rejected request" } }), {
+				status: 401,
+				headers: { "content-type": "application/json" },
+			}),
+	);
+	vi.stubGlobal("fetch", fallback);
+	return { custom, fallback };
+}
+
+function expectOnlyCustomFetch(
+	custom: ReturnType<typeof vi.fn<FetchFunction>>,
+	fallback: ReturnType<typeof vi.fn<FetchFunction>>,
+) {
+	expect(custom).toHaveBeenCalled();
+	expect(fallback).not.toHaveBeenCalled();
+	expect(globalThis.fetch).toBe(fallback);
+}
+
+afterEach(() => {
+	vi.unstubAllGlobals();
+});
+
+describe("fetch stream option", () => {
+	it("passes fetch through streamSimple to the Anthropic SDK", async () => {
+		const { custom, fallback } = mockFetches();
+		await streamAnthropic(createModel("anthropic-messages"), context, {
+			apiKey: "test-key",
+			fetch: custom,
+			maxRetries: 0,
+		}).result();
+		expectOnlyCustomFetch(custom, fallback);
+	});
+
+	it("passes fetch through streamSimple to OpenAI SDK adapters", async () => {
+		const adapters = [
+			() =>
+				streamOpenAICompletions(createModel("openai-completions"), context, {
+					apiKey: "test-key",
+					fetch: custom,
+					maxRetries: 0,
+				}).result(),
+			() =>
+				streamOpenAIResponses(createModel("openai-responses"), context, {
+					apiKey: "test-key",
+					fetch: custom,
+					maxRetries: 0,
+				}).result(),
+			() =>
+				streamAzureOpenAIResponses(createModel("azure-openai-responses"), context, {
+					apiKey: "test-key",
+					fetch: custom,
+					maxRetries: 0,
+				}).result(),
+		];
+		const { custom, fallback } = mockFetches();
+		for (const run of adapters) {
+			await run();
+		}
+		expect(custom).toHaveBeenCalledTimes(adapters.length);
+		expect(fallback).not.toHaveBeenCalled();
+		expect(globalThis.fetch).toBe(fallback);
+	});
+
+	it("uses fetch for Mistral, Codex SSE, and pi-messages HTTP requests", async () => {
+		const { custom, fallback } = mockFetches();
+		await streamMistral(createModel("mistral-conversations"), context, {
+			apiKey: "test-key",
+			fetch: custom,
+		}).result();
+		await streamOpenAICodexResponses(createModel("openai-codex-responses"), context, {
+			apiKey: `header.${btoa(JSON.stringify({ "https://api.openai.com/auth": { chatgpt_account_id: "account" } }))}.signature`,
+			fetch: custom,
+			transport: "sse",
+			maxRetries: 0,
+		}).result();
+		await streamPiMessages(createModel("pi-messages"), context, {
+			apiKey: "test-key",
+			fetch: custom,
+		}).result();
+		expect(custom).toHaveBeenCalledTimes(3);
+		expect(fallback).not.toHaveBeenCalled();
+		expect(globalThis.fetch).toBe(fallback);
+	});
+
+	it("rejects custom fetch for Google adapters instead of silently bypassing it", async () => {
+		const { custom, fallback } = mockFetches();
+		const google = await streamGoogleGenerativeAI(createModel("google-generative-ai"), context, {
+			apiKey: "test-key",
+			fetch: custom,
+		}).result();
+		const vertex = await streamGoogleVertex(createModel("google-vertex"), context, {
+			apiKey: "test-key",
+			fetch: custom,
+		}).result();
+
+		expect(google.errorMessage).toContain("Custom fetch is not supported by the Google Generative AI adapter");
+		expect(vertex.errorMessage).toContain("Custom fetch is not supported by the Google Vertex adapter");
+		expect(custom).not.toHaveBeenCalled();
+		expect(fallback).not.toHaveBeenCalled();
+		expect(globalThis.fetch).toBe(fallback);
+	});
+
+	it("allows Google adapters to receive globalThis.fetch explicitly", async () => {
+		const ambient = vi.fn<FetchFunction>(
+			async () =>
+				new Response(JSON.stringify({ error: { message: "upstream rejected request" } }), {
+					status: 401,
+					headers: { "content-type": "application/json" },
+				}),
+		);
+		vi.stubGlobal("fetch", ambient);
+		const result = await streamGoogleGenerativeAI(createModel("google-generative-ai"), context, {
+			apiKey: "test-key",
+			fetch: ambient,
+		}).result();
+
+		expect(ambient).toHaveBeenCalledOnce();
+		expect(result.errorMessage).not.toContain("Custom fetch is not supported");
+		expect(globalThis.fetch).toBe(ambient);
+	});
+
+	it("uses fetch for image generation", async () => {
+		const { custom, fallback } = mockFetches();
+		const model: ImagesModel<"openrouter-images"> = {
+			...createModel("openrouter-images"),
+			provider: "openrouter",
+			output: ["image"],
+		};
+		await generateImages(
+			model,
+			{ input: [{ type: "text", text: "draw" }] },
+			{
+				apiKey: "test-key",
+				fetch: custom,
+				maxRetries: 0,
+			},
+		);
+		expectOnlyCustomFetch(custom, fallback);
+	});
+});