feat(packages): Add selective pi-ai base entrypoints (#5348)

packages/ai/test/abort.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete, stream } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete, stream } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/anthropic-eager-tool-input-e2e.test.ts

@@ -1,8 +1,8 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
 import { getEnvApiKey } from "../src/env-api-keys.ts";
+import { complete } from "../src/index.ts";
 import { getModels, getProviders } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, KnownProvider, Model, ProviderStreamOptions, Tool } from "../src/types.ts";
 import { resolveApiKey } from "./oauth.ts";
 

packages/ai/test/anthropic-empty-thinking-signature-compat.test.ts

@@ -1,5 +1,5 @@
 import { describe, expect, it } from "vitest";
-import { streamSimple } from "../src/stream.ts";
+import { streamSimple } from "../src/index.ts";
 import type { AssistantMessage, Context, Model } from "../src/types.ts";
 
 interface AnthropicPayload {

packages/ai/test/anthropic-force-adaptive-thinking.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Context, Model, SimpleStreamOptions } from "../src/types.ts";
 
 interface AnthropicThinkingPayload {

packages/ai/test/anthropic-long-cache-retention-e2e.test.ts

@@ -1,7 +1,7 @@
 import { describe, expect, it } from "vitest";
 import { getEnvApiKey } from "../src/env-api-keys.ts";
+import { complete } from "../src/index.ts";
 import { getModels, getProviders } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, KnownProvider, Model, ProviderStreamOptions } from "../src/types.ts";
 import { resolveApiKey } from "./oauth.ts";
 

packages/ai/test/anthropic-opus-4-8-smoke.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Context } from "../src/types.ts";
 
 interface AnthropicThinkingPayload {

packages/ai/test/anthropic-temperature-compat.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Context, Model, SimpleStreamOptions } from "../src/types.ts";
 
 interface AnthropicTemperaturePayload {

packages/ai/test/anthropic-thinking-disable.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Context, Model, SimpleStreamOptions } from "../src/types.ts";
 
 interface AnthropicThinkingPayload {

packages/ai/test/anthropic-tool-name-normalization.test.ts

@@ -1,7 +1,7 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { stream } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { stream } from "../src/stream.ts";
 import type { Context, Tool } from "../src/types.ts";
 import { resolveApiKey } from "./oauth.ts";
 

packages/ai/test/base-entrypoint.test.ts

@@ -0,0 +1,68 @@
+import { afterEach, describe, expect, it } from "vitest";
+import { clearApiProviders, complete, getApiProvider, getApiProviders } from "../src/base.ts";
+import { register as registerAmazonBedrock } from "../src/providers/amazon-bedrock.ts";
+import { register as registerAnthropic } from "../src/providers/anthropic.ts";
+import { register as registerAzureOpenAIResponses } from "../src/providers/azure-openai-responses.ts";
+import { fauxAssistantMessage, registerFauxProvider } from "../src/providers/faux.ts";
+import { register as registerGoogle } from "../src/providers/google.ts";
+import { register as registerGoogleVertex } from "../src/providers/google-vertex.ts";
+import { register as registerMistral } from "../src/providers/mistral.ts";
+import { register as registerOpenAICodexResponses } from "../src/providers/openai-codex-responses.ts";
+import { register as registerOpenAICompletions } from "../src/providers/openai-completions.ts";
+import { register as registerOpenAIResponses } from "../src/providers/openai-responses.ts";
+
+const registrations = [
+	["bedrock-converse-stream", registerAmazonBedrock],
+	["anthropic-messages", registerAnthropic],
+	["azure-openai-responses", registerAzureOpenAIResponses],
+	["google-generative-ai", registerGoogle],
+	["google-vertex", registerGoogleVertex],
+	["mistral-conversations", registerMistral],
+	["openai-codex-responses", registerOpenAICodexResponses],
+	["openai-completions", registerOpenAICompletions],
+	["openai-responses", registerOpenAIResponses],
+] as const;
+
+afterEach(() => {
+	clearApiProviders();
+});
+
+describe("base entrypoint", () => {
+	it("starts without built-in provider registrations", () => {
+		expect(getApiProviders()).toEqual([]);
+	});
+
+	it.each(registrations)("registers the %s transport explicitly", (api, register) => {
+		register();
+		expect(getApiProvider(api)?.api).toBe(api);
+	});
+
+	it("dispatches custom providers", async () => {
+		const registration = registerFauxProvider();
+		registration.setResponses([fauxAssistantMessage("hello")]);
+		const response = await complete(registration.getModel(), {
+			messages: [{ role: "user", content: "hi", timestamp: Date.now() }],
+		});
+		expect(response.content).toEqual([{ type: "text", text: "hello" }]);
+	});
+
+	it("fails clearly when no transport is registered", async () => {
+		await expect(
+			complete(
+				{
+					id: "missing-model",
+					name: "Missing Model",
+					api: "missing-api",
+					provider: "missing-provider",
+					baseUrl: "https://example.com",
+					reasoning: false,
+					input: ["text"],
+					cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+					contextWindow: 1,
+					maxTokens: 1,
+				},
+				{ messages: [] },
+			),
+		).rejects.toThrow("No API provider registered for api: missing-api");
+	});
+});

packages/ai/test/bedrock-models.test.ts

@@ -17,8 +17,8 @@
  */
 
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModels } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Context } from "../src/types.ts";
 import { hasBedrockCredentials } from "./bedrock-utils.ts";
 

packages/ai/test/cache-retention.test.ts

@@ -1,10 +1,10 @@
 import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { stream } from "../src/index.ts";
 import { MODELS } from "../src/models.generated.ts";
 import { getModel } from "../src/models.ts";
 import { streamAnthropic } from "../src/providers/anthropic.ts";
 import { streamOpenAICompletions } from "../src/providers/openai-completions.ts";
 import { streamOpenAIResponses } from "../src/providers/openai-responses.ts";
-import { stream } from "../src/stream.ts";
 import type { Context, Model } from "../src/types.ts";
 
 class PayloadCaptured extends Error {

packages/ai/test/context-overflow.test.ts

@@ -14,8 +14,8 @@
 import type { ChildProcess } from "child_process";
 import { execSync, spawn } from "child_process";
 import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel, getModels } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { AssistantMessage, Context, Model, Usage } from "../src/types.ts";
 import { isContextOverflow } from "../src/utils/overflow.ts";
 import { hasAzureOpenAICredentials } from "./azure-utils.ts";

packages/ai/test/cross-provider-handoff.test.ts

@@ -25,8 +25,8 @@
 import { writeFileSync } from "fs";
 import { Type } from "typebox";
 import { beforeAll, describe, expect, it } from "vitest";
+import { completeSimple, getEnvApiKey } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { completeSimple, getEnvApiKey } from "../src/stream.ts";
 import type { Api, AssistantMessage, Message, Model, Tool, ToolResultMessage } from "../src/types.ts";
 import { hasAzureOpenAICredentials } from "./azure-utils.ts";
 import { hasCloudflareAiGatewayCredentials, hasCloudflareWorkersAICredentials } from "./cloudflare-utils.ts";

packages/ai/test/empty.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, AssistantMessage, Context, Model, StreamOptions, UserMessage } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/google-thinking-disable.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Api, Context, Model, SimpleStreamOptions } from "../src/types.ts";
 
 type SimpleOptionsWithExtras = SimpleStreamOptions & Record<string, unknown>;

packages/ai/test/interleaved-thinking.test.ts

@@ -1,8 +1,8 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
 import { getEnvApiKey } from "../src/env-api-keys.ts";
+import { completeSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { completeSimple } from "../src/stream.ts";
 import type { Api, Context, Model, StopReason, Tool, ToolCall, ToolResultMessage } from "../src/types.ts";
 import { StringEnum } from "../src/utils/typebox-helpers.ts";
 import { hasBedrockCredentials } from "./bedrock-utils.ts";

packages/ai/test/lazy-module-load.test.ts

@@ -5,6 +5,7 @@ import { describe, expect, it } from "vitest";
 
 const packageRoot = resolve(dirname(fileURLToPath(import.meta.url)), "..");
 const aiEntryUrl = new URL("../src/index.ts", import.meta.url).href;
+const baseEntryUrl = new URL("../src/base.ts", import.meta.url).href;
 
 const SDK_SPECIFIERS = [
 	"@anthropic-ai/sdk",
@@ -16,9 +17,10 @@ const SDK_SPECIFIERS = [
 
 type ProbeResult = {
 	loadedSpecifiers: string[];
+	value?: unknown;
 };
 
-function runProbe(action: string): ProbeResult {
+function runProbe(action: string, entryUrl = aiEntryUrl): ProbeResult {
 	const script = `
 		import { registerHooks } from "node:module";
 
@@ -34,9 +36,11 @@ function runProbe(action: string): ProbeResult {
 			},
 		});
 
-		const mod = await import(${JSON.stringify(aiEntryUrl)});
-		${action}
-		console.log(JSON.stringify({ loadedSpecifiers: [...new Set(loaded)] }));
+		const mod = await import(${JSON.stringify(entryUrl)});
+		const value = await (async () => {
+			${action}
+		})();
+		console.log(JSON.stringify({ loadedSpecifiers: [...new Set(loaded)], value }));
 	`;
 
 	const result = spawnSync(process.execPath, ["--input-type=module", "--eval", script], {
@@ -66,6 +70,33 @@ describe("lazy provider module loading", () => {
 		expect(result.loadedSpecifiers).toEqual([]);
 	});
 
+	it("registers built-in transports when importing the root barrel", () => {
+		const result = runProbe(`return mod.getApiProviders().map((provider) => provider.api).sort();`);
+		expect(result.value).toEqual([
+			"anthropic-messages",
+			"azure-openai-responses",
+			"bedrock-converse-stream",
+			"google-generative-ai",
+			"google-vertex",
+			"mistral-conversations",
+			"openai-codex-responses",
+			"openai-completions",
+			"openai-responses",
+		]);
+	});
+
+	it("registers built-in image transports when importing the root barrel", () => {
+		const result = runProbe(`return mod.getImagesApiProvider("openrouter-images")?.api;`);
+		expect(result.loadedSpecifiers).toEqual([]);
+		expect(result.value).toBe("openrouter-images");
+	});
+
+	it("does not load provider SDKs or register transports when importing the base barrel", () => {
+		const result = runProbe(`return mod.getApiProviders().map((provider) => provider.api);`, baseEntryUrl);
+		expect(result.loadedSpecifiers).toEqual([]);
+		expect(result.value).toEqual([]);
+	});
+
 	it("loads only the Anthropic SDK when calling the root lazy wrapper", () => {
 		const result = runProbe(`
 			const model = {
@@ -96,4 +127,17 @@ describe("lazy provider module loading", () => {
 
 		expect(result.loadedSpecifiers).toEqual(["@anthropic-ai/sdk"]);
 	});
+
+	it("dispatches through a lazy wrapper again after resetting providers", () => {
+		const result = runProbe(`
+			const model = mod.getModel("anthropic", "claude-sonnet-4-6");
+			const context = { messages: [{ role: "user", content: "hi" }] };
+			await mod.streamSimple(model, context).result();
+			mod.resetApiProviders();
+			return (await mod.streamSimple(model, context).result()).stopReason;
+		`);
+
+		expect(result.loadedSpecifiers).toEqual(["@anthropic-ai/sdk"]);
+		expect(result.value).toBe("error");
+	});
 });

packages/ai/test/mistral-reasoning-mode.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 import type { Context, Model, SimpleStreamOptions } from "../src/types.ts";
 
 interface MistralPayload {

packages/ai/test/mistral-tool-schema.test.ts

@@ -1,7 +1,7 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Context, Model } from "../src/types.ts";
 
 interface MistralToolPayload {

packages/ai/test/openai-codex-cache-affinity-e2e.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Context } from "../src/types.ts";
 import { resolveApiKey } from "./oauth.ts";
 

packages/ai/test/openai-completions-empty-tools.test.ts

@@ -1,6 +1,6 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
+import { streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { streamSimple } from "../src/stream.ts";
 
 // Empty tools arrays must NOT be serialized as `tools: []` — some OpenAI-compatible
 // backends (e.g. DashScope / Aliyun Qwen via compatible-mode) reject the request with

packages/ai/test/openai-completions-response-model.test.ts

@@ -1,5 +1,5 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
-import { complete } from "../src/stream.ts";
+import { complete } from "../src/index.ts";
 import type { Model } from "../src/types.ts";
 
 // Router/virtual ids (e.g. OpenRouter `auto`) keep `model` pinned to the

packages/ai/test/openai-completions-tool-choice.test.ts

@@ -1,8 +1,8 @@
 import { Type } from "typebox";
 import { beforeEach, describe, expect, it, vi } from "vitest";
+import { stream, streamSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
 import { convertMessages } from "../src/providers/openai-completions.ts";
-import { stream, streamSimple } from "../src/stream.ts";
 import type { AssistantMessage, Model, Tool, ToolResultMessage } from "../src/types.ts";
 
 const mockState = vi.hoisted(() => ({

packages/ai/test/openai-responses-cache-affinity-e2e.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Context } from "../src/types.ts";
 
 describe.skipIf(!process.env.OPENAI_API_KEY)("openai responses cache affinity e2e", () => {

packages/ai/test/openai-responses-reasoning-replay-e2e.test.ts

@@ -1,7 +1,7 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { complete, getEnvApiKey } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete, getEnvApiKey } from "../src/stream.ts";
 import type { AssistantMessage, Context, Message, Tool, ToolCall } from "../src/types.ts";
 
 const testToolSchema = Type.Object({

packages/ai/test/openrouter-cache-write-repro.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { completeSimple } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { completeSimple } from "../src/stream.ts";
 
 function createLongSystemPrompt(): string {
 	const nonce = `${Date.now()}-${Math.random()}`;

packages/ai/test/openrouter-images.test.ts

@@ -1,5 +1,8 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { generateImages } from "../src/images.ts";
+import { clearImagesApiProviders, getImagesApiProvider } from "../src/images-api-registry.ts";
+import { register as registerOpenRouterImages } from "../src/providers/images/openrouter.ts";
+import { registerBuiltInImagesApiProviders } from "../src/providers/register-builtins.ts";
 import type { ImagesContext, ImagesModel } from "../src/types.ts";
 
 const mockState = vi.hoisted(() => ({
@@ -62,6 +65,15 @@ describe("openrouter images", () => {
 	beforeEach(() => {
 		mockState.lastParams = undefined;
 		mockState.lastRequestOptions = undefined;
+		clearImagesApiProviders();
+		registerBuiltInImagesApiProviders();
+	});
+
+	it("registers the direct OpenRouter images transport explicitly", () => {
+		clearImagesApiProviders();
+		expect(getImagesApiProvider("openrouter-images")).toBeUndefined();
+		registerOpenRouterImages();
+		expect(getImagesApiProvider("openrouter-images")?.api).toBe("openrouter-images");
 	});
 
 	it("returns text plus images in final output", async () => {

packages/ai/test/responseid.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions } from "../src/types.ts";
 import { hasAzureOpenAICredentials, resolveAzureDeploymentName } from "./azure-utils.ts";
 import { resolveApiKey } from "./oauth.ts";

packages/ai/test/stream.test.ts

@@ -4,8 +4,8 @@ import { dirname, join } from "path";
 import { Type } from "typebox";
 import { fileURLToPath } from "url";
 import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { complete, stream } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete, stream } from "../src/stream.ts";
 import type { Api, Context, ImageContent, Model, StreamOptions, Tool, ToolResultMessage } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/tokens.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { stream } from "../src/index.ts";
 import { getModel, getModels } from "../src/models.ts";
-import { stream } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/tool-call-id-normalization.test.ts

@@ -12,8 +12,8 @@
 
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { completeSimple, getEnvApiKey } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { completeSimple, getEnvApiKey } from "../src/stream.ts";
 import type { AssistantMessage, Message, Tool, ToolResultMessage } from "../src/types.ts";
 import { resolveApiKey } from "./oauth.ts";
 

packages/ai/test/tool-call-without-result.test.ts

@@ -1,7 +1,7 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions, Tool } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/total-tokens.test.ts

@@ -13,8 +13,8 @@
  */
 
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions, Usage } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/unicode-surrogate.test.ts

@@ -1,7 +1,7 @@
 import { Type } from "typebox";
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { complete } from "../src/stream.ts";
 import type { Api, Context, Model, StreamOptions, ToolResultMessage } from "../src/types.ts";
 
 type StreamOptionsWithExtras = StreamOptions & Record<string, unknown>;

packages/ai/test/xhigh.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { stream } from "../src/index.ts";
 import { getModel } from "../src/models.ts";
-import { stream } from "../src/stream.ts";
 import type { Context, Model } from "../src/types.ts";
 
 function makeContext(): Context {

packages/ai/test/xiaomi-token-plan-ams-anthropic-empty-signature-smoke.test.ts

@@ -1,5 +1,5 @@
 import { describe, expect, it } from "vitest";
-import { completeSimple, getEnvApiKey, streamSimple } from "../src/stream.ts";
+import { completeSimple, getEnvApiKey, streamSimple } from "../src/index.ts";
 import type { AssistantMessage, Context, Model } from "../src/types.ts";
 
 const provider = "xiaomi-token-plan-ams";

packages/ai/test/zen.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from "vitest";
+import { complete } from "../src/index.ts";
 import { MODELS } from "../src/models.generated.ts";
-import { complete } from "../src/stream.ts";
 import type { Model } from "../src/types.ts";
 
 describe.skipIf(!process.env.OPENCODE_API_KEY)("OpenCode Models Smoke Test", () => {