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,223 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+
+const googleGenAiMock = vi.hoisted(() => ({
+	constructorCalls: [] as Array<Record<string, unknown>>,
+}));
+
+vi.mock("@google/genai", () => {
+	class GoogleGenAI {
+		models = {
+			generateContentStream: async function* () {
+				yield {
+					responseId: "vertex-response-id",
+					candidates: [
+						{
+							content: { parts: [{ text: "ok" }] },
+							finishReason: "STOP",
+						},
+					],
+					usageMetadata: {
+						promptTokenCount: 1,
+						candidatesTokenCount: 1,
+						totalTokenCount: 2,
+					},
+				};
+			},
+		};
+
+		constructor(config: Record<string, unknown>) {
+			googleGenAiMock.constructorCalls.push(config);
+		}
+	}
+
+	return {
+		GoogleGenAI,
+		ResourceScope: {
+			COLLECTION: "COLLECTION",
+		},
+		ThinkingLevel: {
+			THINKING_LEVEL_UNSPECIFIED: "THINKING_LEVEL_UNSPECIFIED",
+			MINIMAL: "MINIMAL",
+			LOW: "LOW",
+			MEDIUM: "MEDIUM",
+			HIGH: "HIGH",
+		},
+	};
+});
+
+import { stream as streamGoogleVertex } from "../src/api/google-vertex.ts";
+import { getModel } from "../src/compat.ts";
+import type { Context, Model } from "../src/types.ts";
+
+const model = getModel("google-vertex", "gemini-3-flash-preview");
+const context: Context = {
+	messages: [{ role: "user", content: "hello", timestamp: Date.now() }],
+};
+
+const originalGoogleCloudApiKey = process.env.GOOGLE_CLOUD_API_KEY;
+
+beforeEach(() => {
+	googleGenAiMock.constructorCalls.length = 0;
+	delete process.env.GOOGLE_CLOUD_API_KEY;
+});
+
+afterEach(() => {
+	if (originalGoogleCloudApiKey === undefined) {
+		delete process.env.GOOGLE_CLOUD_API_KEY;
+	} else {
+		process.env.GOOGLE_CLOUD_API_KEY = originalGoogleCloudApiKey;
+	}
+});
+
+describe("google-vertex api key resolution", () => {
+	it("falls back to ADC when options.apiKey is a placeholder marker", async () => {
+		const stream = streamGoogleVertex(model, context, {
+			apiKey: "<authenticated>",
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			project: "test-project",
+			location: "us-central1",
+			apiVersion: "v1",
+		});
+		expect(googleGenAiMock.constructorCalls[0]).not.toHaveProperty("apiKey");
+	});
+
+	it("falls back to ADC when options.apiKey is the gcp-vertex-credentials marker", async () => {
+		const stream = streamGoogleVertex(model, context, {
+			apiKey: "gcp-vertex-credentials",
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			project: "test-project",
+			location: "us-central1",
+			apiVersion: "v1",
+		});
+		expect(googleGenAiMock.constructorCalls[0]).not.toHaveProperty("apiKey");
+	});
+
+	it("falls back to ADC when GOOGLE_CLOUD_API_KEY is a placeholder marker", async () => {
+		process.env.GOOGLE_CLOUD_API_KEY = "<authenticated>";
+
+		const stream = streamGoogleVertex(model, context, {
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			project: "test-project",
+			location: "us-central1",
+			apiVersion: "v1",
+		});
+		expect(googleGenAiMock.constructorCalls[0]).not.toHaveProperty("apiKey");
+	});
+
+	it("still uses the API key client for real API keys", async () => {
+		const stream = streamGoogleVertex(model, context, {
+			apiKey: "AIzaSyExampleRealisticLookingApiKey123456",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			apiKey: "AIzaSyExampleRealisticLookingApiKey123456",
+			apiVersion: "v1",
+		});
+		expect(googleGenAiMock.constructorCalls[0]).not.toHaveProperty("project");
+		expect(googleGenAiMock.constructorCalls[0]).not.toHaveProperty("location");
+	});
+
+	it("does not forward generated Vertex base URL placeholders", async () => {
+		const stream = streamGoogleVertex(model, context, {
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]?.httpOptions).toBeUndefined();
+	});
+
+	it("forwards custom baseUrl to the ADC client", async () => {
+		const customModel: Model<"google-vertex"> = { ...model, baseUrl: "https://proxy.example.com" };
+		const stream = streamGoogleVertex(customModel, context, {
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			project: "test-project",
+			location: "us-central1",
+			apiVersion: "v1",
+			httpOptions: {
+				baseUrl: "https://proxy.example.com",
+				baseUrlResourceScope: "COLLECTION",
+			},
+		});
+	});
+
+	it("forwards custom baseUrl to the API key client", async () => {
+		const customModel: Model<"google-vertex"> = { ...model, baseUrl: "https://proxy.example.com" };
+		const stream = streamGoogleVertex(customModel, context, {
+			apiKey: "AIzaSyExampleRealisticLookingApiKey123456",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			vertexai: true,
+			apiKey: "AIzaSyExampleRealisticLookingApiKey123456",
+			apiVersion: "v1",
+			httpOptions: {
+				baseUrl: "https://proxy.example.com",
+				baseUrlResourceScope: "COLLECTION",
+			},
+		});
+	});
+
+	it("does not append apiVersion when custom baseUrl already includes one", async () => {
+		const customModel: Model<"google-vertex"> = {
+			...model,
+			baseUrl: "https://proxy.example.com/v1/projects/test-project/locations/global",
+		};
+		const stream = streamGoogleVertex(customModel, context, {
+			project: "test-project",
+			location: "us-central1",
+		});
+
+		await stream.result();
+
+		expect(googleGenAiMock.constructorCalls).toHaveLength(1);
+		expect(googleGenAiMock.constructorCalls[0]).toMatchObject({
+			httpOptions: {
+				baseUrl: "https://proxy.example.com/v1/projects/test-project/locations/global",
+				baseUrlResourceScope: "COLLECTION",
+				apiVersion: "",
+			},
+		});
+	});
+});