@gajae-code/ai 0.1.1

package/src/providers/register-builtins.ts

@@ -0,0 +1,412 @@
+/**
+ * Lazy provider module loading.
+ *
+ * Each provider module is loaded only when its stream function is first called.
+ * This avoids eagerly importing heavy SDK dependencies (e.g., @anthropic-ai/sdk,
+ * openai) at startup. The loaded module promise is cached so subsequent calls
+ * reuse the same import.
+ *
+ * NOTE: stream.ts currently imports providers directly, so this file is not yet
+ * wired into the main streaming path. It provides the infrastructure for lazy
+ * loading that can be integrated when stream.ts is refactored.
+ */
+import type {
+	Api,
+	AssistantMessage,
+	AssistantMessageEvent,
+	AssistantMessageEventStream,
+	Context,
+	Model,
+	OptionsForApi,
+} from "../types";
+import { type AbortSourceTracker, createAbortSourceTracker } from "../utils/abort";
+import { AssistantMessageEventStream as EventStreamImpl } from "../utils/event-stream";
+import { getStreamFirstEventTimeoutMs, getStreamIdleTimeoutMs, iterateWithIdleTimeout } from "../utils/idle-iterator";
+import type { BedrockOptions } from "./amazon-bedrock";
+import type { AnthropicOptions } from "./anthropic";
+import type { AzureOpenAIResponsesOptions } from "./azure-openai-responses";
+import type { CursorOptions } from "./cursor";
+import type { GoogleOptions } from "./google";
+import type { GoogleGeminiCliOptions } from "./google-gemini-cli";
+import type { GoogleVertexOptions } from "./google-vertex";
+import type { OllamaChatOptions } from "./ollama";
+import type { OpenAICodexResponsesOptions } from "./openai-codex-responses";
+import type { OpenAICompletionsOptions } from "./openai-completions";
+import type { OpenAIResponsesOptions } from "./openai-responses";
+
+// ---------------------------------------------------------------------------
+// Lazy provider module shape
+// ---------------------------------------------------------------------------
+
+interface LazyProviderModule<TApi extends Api> {
+	stream: (model: Model<TApi>, context: Context, options: OptionsForApi<TApi>) => AsyncIterable<AssistantMessageEvent>;
+}
+
+interface AnthropicProviderModule {
+	streamAnthropic: (
+		model: Model<"anthropic-messages">,
+		context: Context,
+		options: AnthropicOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface AzureOpenAIResponsesProviderModule {
+	streamAzureOpenAIResponses: (
+		model: Model<"azure-openai-responses">,
+		context: Context,
+		options: AzureOpenAIResponsesOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface GoogleProviderModule {
+	streamGoogle: (
+		model: Model<"google-generative-ai">,
+		context: Context,
+		options: GoogleOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface GoogleGeminiCliProviderModule {
+	streamGoogleGeminiCli: (
+		model: Model<"google-gemini-cli">,
+		context: Context,
+		options: GoogleGeminiCliOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface GoogleVertexProviderModule {
+	streamGoogleVertex: (
+		model: Model<"google-vertex">,
+		context: Context,
+		options: GoogleVertexOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface OpenAICodexResponsesProviderModule {
+	streamOpenAICodexResponses: (
+		model: Model<"openai-codex-responses">,
+		context: Context,
+		options: OpenAICodexResponsesOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface OpenAICompletionsProviderModule {
+	streamOpenAICompletions: (
+		model: Model<"openai-completions">,
+		context: Context,
+		options: OpenAICompletionsOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface OpenAIResponsesProviderModule {
+	streamOpenAIResponses: (
+		model: Model<"openai-responses">,
+		context: Context,
+		options: OpenAIResponsesOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface OllamaProviderModule {
+	streamOllama: (
+		model: Model<"ollama-chat">,
+		context: Context,
+		options: OllamaChatOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface CursorProviderModule {
+	streamCursor: (
+		model: Model<"cursor-agent">,
+		context: Context,
+		options: CursorOptions,
+	) => AssistantMessageEventStream;
+}
+
+interface BedrockProviderModule {
+	streamBedrock: (
+		model: Model<"bedrock-converse-stream">,
+		context: Context,
+		options: BedrockOptions,
+	) => AssistantMessageEventStream;
+}
+
+// ---------------------------------------------------------------------------
+// Module-level lazy promise caches
+// ---------------------------------------------------------------------------
+
+let anthropicProviderModulePromise: Promise<LazyProviderModule<"anthropic-messages">> | undefined;
+let azureOpenAIResponsesProviderModulePromise: Promise<LazyProviderModule<"azure-openai-responses">> | undefined;
+let googleProviderModulePromise: Promise<LazyProviderModule<"google-generative-ai">> | undefined;
+let googleGeminiCliProviderModulePromise: Promise<LazyProviderModule<"google-gemini-cli">> | undefined;
+let googleVertexProviderModulePromise: Promise<LazyProviderModule<"google-vertex">> | undefined;
+let openAICodexResponsesProviderModulePromise: Promise<LazyProviderModule<"openai-codex-responses">> | undefined;
+let openAICompletionsProviderModulePromise: Promise<LazyProviderModule<"openai-completions">> | undefined;
+let openAIResponsesProviderModulePromise: Promise<LazyProviderModule<"openai-responses">> | undefined;
+let ollamaProviderModulePromise: Promise<LazyProviderModule<"ollama-chat">> | undefined;
+let cursorProviderModulePromise: Promise<LazyProviderModule<"cursor-agent">> | undefined;
+let bedrockProviderModuleOverride: LazyProviderModule<"bedrock-converse-stream"> | undefined;
+let bedrockProviderModulePromise: Promise<LazyProviderModule<"bedrock-converse-stream">> | undefined;
+
+export function setBedrockProviderModule(module: BedrockProviderModule): void {
+	bedrockProviderModuleOverride = {
+		stream: module.streamBedrock,
+	};
+}
+
+// ---------------------------------------------------------------------------
+// Stream forwarding / error helpers
+// ---------------------------------------------------------------------------
+
+const LAZY_STREAM_IDLE_TIMEOUT_ERROR = "Provider stream stalled while waiting for the next event";
+const LAZY_STREAM_FIRST_EVENT_TIMEOUT_ERROR = "Provider stream timed out while waiting for the first event";
+
+function hasFinalResult(
+	source: AsyncIterable<AssistantMessageEvent>,
+): source is AsyncIterable<AssistantMessageEvent> & { result(): Promise<AssistantMessage> } {
+	return typeof (source as { result?: unknown }).result === "function";
+}
+
+/**
+ * Per-provider default overrides for the lazy stream watchdogs. These widen the
+ * floor used when neither caller option nor env var pins a value. The env vars
+ * (`PI_STREAM_FIRST_EVENT_TIMEOUT_MS`, `PI_STREAM_IDLE_TIMEOUT_MS`) still take
+ * precedence; `StreamOptions.streamFirstEventTimeoutMs` / `streamIdleTimeoutMs`
+ * still trump everything.
+ */
+interface LazyStreamLimits {
+	defaultFirstEventTimeoutMs?: number;
+	defaultIdleTimeoutMs?: number;
+}
+
+/**
+ * Cloud Code Assist (google-gemini-cli / google-antigravity) routinely takes
+ * longer than the global 100s default to emit its first SSE event when serving
+ * the heavier Gemini 3.x Pro tiers at high thinking levels. Bump the first-event
+ * floor to five minutes so duke et al. stop seeing spurious "stream timed out
+ * while waiting for the first event" aborts on legitimate cold reasoning starts.
+ * The steady-state idle watchdog stays on the global default since the upstream
+ * emits thinking tokens frequently once it gets going.
+ */
+const GOOGLE_GEMINI_CLI_LAZY_STREAM_LIMITS: LazyStreamLimits = {
+	defaultFirstEventTimeoutMs: 300_000,
+};
+
+function forwardStream<TApi extends Api>(
+	target: EventStreamImpl,
+	source: AsyncIterable<AssistantMessageEvent>,
+	model: Model<TApi>,
+	options: OptionsForApi<TApi>,
+	abortTracker: AbortSourceTracker,
+	limits?: LazyStreamLimits,
+): void {
+	(async () => {
+		try {
+			const idleTimeoutMs = options.streamIdleTimeoutMs ?? getStreamIdleTimeoutMs(limits?.defaultIdleTimeoutMs);
+			const watchedSource = iterateWithIdleTimeout(source, {
+				idleTimeoutMs,
+				firstItemTimeoutMs:
+					options.streamFirstEventTimeoutMs ??
+					getStreamFirstEventTimeoutMs(idleTimeoutMs, limits?.defaultFirstEventTimeoutMs),
+				errorMessage: LAZY_STREAM_IDLE_TIMEOUT_ERROR,
+				firstItemErrorMessage: LAZY_STREAM_FIRST_EVENT_TIMEOUT_ERROR,
+				onIdle: () => abortTracker.abortLocally(new Error(LAZY_STREAM_IDLE_TIMEOUT_ERROR)),
+				onFirstItemTimeout: () => abortTracker.abortLocally(new Error(LAZY_STREAM_FIRST_EVENT_TIMEOUT_ERROR)),
+				abortSignal: options.signal,
+				// The synthetic `start` event is yielded immediately by every provider before
+				// the upstream model has emitted any tokens. Treating it as the first "real"
+				// item would flip the watchdog from `firstItemTimeoutMs` to the much shorter
+				// `idleTimeoutMs` while we're still legitimately waiting on the model's
+				// first response (slow first-token from reasoning models, cold proxies, etc.).
+				isProgressItem: event => (event as AssistantMessageEvent).type !== "start",
+			});
+
+			for await (const event of watchedSource) {
+				target.push(event);
+			}
+			if (hasFinalResult(source)) {
+				target.end(await source.result());
+			} else {
+				target.end();
+			}
+		} catch (error) {
+			const stopReason = abortTracker.wasCallerAbort() ? "aborted" : "error";
+			const message = createLazyLoadErrorMessage(model, error, stopReason);
+			target.push({ type: "error", reason: stopReason, error: message });
+			target.end(message);
+		}
+	})();
+}
+
+function createLazyLoadErrorMessage<TApi extends Api>(
+	model: Model<TApi>,
+	error: unknown,
+	stopReason: Extract<AssistantMessage["stopReason"], "aborted" | "error"> = "error",
+): AssistantMessage {
+	return {
+		role: "assistant",
+		content: [],
+		api: model.api,
+		provider: model.provider,
+		model: model.id,
+		usage: {
+			input: 0,
+			output: 0,
+			cacheRead: 0,
+			cacheWrite: 0,
+			totalTokens: 0,
+			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+		},
+		stopReason,
+		errorMessage:
+			stopReason === "aborted" ? "Request was aborted" : error instanceof Error ? error.message : String(error),
+		timestamp: Date.now(),
+	};
+}
+
+// ---------------------------------------------------------------------------
+// Generic lazy stream factory
+// ---------------------------------------------------------------------------
+
+function createLazyStream<TApi extends Api>(
+	loadModule: () => Promise<LazyProviderModule<TApi>>,
+	limits?: LazyStreamLimits,
+): (model: Model<TApi>, context: Context, options: OptionsForApi<TApi>) => EventStreamImpl {
+	return (model, context, options) => {
+		const outer = new EventStreamImpl();
+		const streamOptions = (options ?? {}) as OptionsForApi<TApi>;
+
+		loadModule()
+			.then(module => {
+				const abortTracker = createAbortSourceTracker(streamOptions.signal);
+				const providerOptions = { ...streamOptions, signal: abortTracker.requestSignal } as OptionsForApi<TApi>;
+				const inner = module.stream(model, context, providerOptions);
+				forwardStream(outer, inner, model, streamOptions, abortTracker, limits);
+			})
+			.catch(error => {
+				const message = createLazyLoadErrorMessage(model, error);
+				outer.push({ type: "error", reason: "error", error: message });
+				outer.end(message);
+			});
+
+		return outer;
+	};
+}
+
+// ---------------------------------------------------------------------------
+// Module loaders (one per provider, cached via ||=)
+// ---------------------------------------------------------------------------
+
+function loadAnthropicProviderModule(): Promise<LazyProviderModule<"anthropic-messages">> {
+	anthropicProviderModulePromise ||= import("./anthropic").then(module => {
+		const provider = module as AnthropicProviderModule;
+		return { stream: provider.streamAnthropic };
+	});
+	return anthropicProviderModulePromise;
+}
+
+function loadAzureOpenAIResponsesProviderModule(): Promise<LazyProviderModule<"azure-openai-responses">> {
+	azureOpenAIResponsesProviderModulePromise ||= import("./azure-openai-responses").then(module => {
+		const provider = module as AzureOpenAIResponsesProviderModule;
+		return { stream: provider.streamAzureOpenAIResponses };
+	});
+	return azureOpenAIResponsesProviderModulePromise;
+}
+
+function loadGoogleProviderModule(): Promise<LazyProviderModule<"google-generative-ai">> {
+	googleProviderModulePromise ||= import("./google").then(module => {
+		const provider = module as GoogleProviderModule;
+		return { stream: provider.streamGoogle };
+	});
+	return googleProviderModulePromise;
+}
+
+function loadGoogleGeminiCliProviderModule(): Promise<LazyProviderModule<"google-gemini-cli">> {
+	googleGeminiCliProviderModulePromise ||= import("./google-gemini-cli").then(module => {
+		const provider = module as GoogleGeminiCliProviderModule;
+		return { stream: provider.streamGoogleGeminiCli };
+	});
+	return googleGeminiCliProviderModulePromise;
+}
+
+function loadGoogleVertexProviderModule(): Promise<LazyProviderModule<"google-vertex">> {
+	googleVertexProviderModulePromise ||= import("./google-vertex").then(module => {
+		const provider = module as GoogleVertexProviderModule;
+		return { stream: provider.streamGoogleVertex };
+	});
+	return googleVertexProviderModulePromise;
+}
+
+function loadOpenAICodexResponsesProviderModule(): Promise<LazyProviderModule<"openai-codex-responses">> {
+	openAICodexResponsesProviderModulePromise ||= import("./openai-codex-responses").then(module => {
+		const provider = module as OpenAICodexResponsesProviderModule;
+		return { stream: provider.streamOpenAICodexResponses };
+	});
+	return openAICodexResponsesProviderModulePromise;
+}
+
+function loadOpenAICompletionsProviderModule(): Promise<LazyProviderModule<"openai-completions">> {
+	openAICompletionsProviderModulePromise ||= import("./openai-completions").then(module => {
+		const provider = module as OpenAICompletionsProviderModule;
+		return { stream: provider.streamOpenAICompletions };
+	});
+	return openAICompletionsProviderModulePromise;
+}
+
+function loadOpenAIResponsesProviderModule(): Promise<LazyProviderModule<"openai-responses">> {
+	openAIResponsesProviderModulePromise ||= import("./openai-responses").then(module => {
+		const provider = module as OpenAIResponsesProviderModule;
+		return { stream: provider.streamOpenAIResponses };
+	});
+	return openAIResponsesProviderModulePromise;
+}
+
+function loadOllamaProviderModule(): Promise<LazyProviderModule<"ollama-chat">> {
+	ollamaProviderModulePromise ||= import("./ollama").then(module => {
+		const provider = module as OllamaProviderModule;
+		return { stream: provider.streamOllama };
+	});
+	return ollamaProviderModulePromise;
+}
+
+function loadCursorProviderModule(): Promise<LazyProviderModule<"cursor-agent">> {
+	cursorProviderModulePromise ||= import("./cursor").then(module => {
+		const provider = module as CursorProviderModule;
+		return { stream: provider.streamCursor };
+	});
+	return cursorProviderModulePromise;
+}
+
+function loadBedrockProviderModule(): Promise<LazyProviderModule<"bedrock-converse-stream">> {
+	if (bedrockProviderModuleOverride) {
+		return Promise.resolve(bedrockProviderModuleOverride);
+	}
+	bedrockProviderModulePromise ||= import("./amazon-bedrock").then(module => {
+		const provider = module as BedrockProviderModule;
+		return { stream: provider.streamBedrock };
+	});
+	return bedrockProviderModulePromise;
+}
+
+// ---------------------------------------------------------------------------
+// Lazy stream function exports
+//
+// These use the same names as the direct provider stream functions. When
+// stream.ts is updated to import from this module instead of individual
+// providers, the lazy loading will take effect on the main code path.
+// ---------------------------------------------------------------------------
+
+export const streamAnthropic = createLazyStream(loadAnthropicProviderModule);
+export const streamAzureOpenAIResponses = createLazyStream(loadAzureOpenAIResponsesProviderModule);
+export const streamGoogle = createLazyStream(loadGoogleProviderModule);
+export const streamGoogleGeminiCli = createLazyStream(
+	loadGoogleGeminiCliProviderModule,
+	GOOGLE_GEMINI_CLI_LAZY_STREAM_LIMITS,
+);
+export const streamGoogleVertex = createLazyStream(loadGoogleVertexProviderModule);
+export const streamOpenAICodexResponses = createLazyStream(loadOpenAICodexResponsesProviderModule);
+export const streamOpenAICompletions = createLazyStream(loadOpenAICompletionsProviderModule);
+export const streamOpenAIResponses = createLazyStream(loadOpenAIResponsesProviderModule);
+export const streamCursor = createLazyStream(loadCursorProviderModule);
+export const streamOllama = createLazyStream(loadOllamaProviderModule);
+
+export const streamBedrock = createLazyStream(loadBedrockProviderModule);

package/src/providers/synthetic.ts

@@ -0,0 +1,50 @@
+/**
+ * Synthetic provider - wraps OpenAI or Anthropic API based on format setting.
+ *
+ * Synthetic offers both OpenAI-compatible and Anthropic-compatible APIs:
+ * - OpenAI: https://api.synthetic.new/openai/v1/chat/completions
+ * - Anthropic: https://api.synthetic.new/anthropic/v1/messages
+ *
+ * @see https://dev.synthetic.new/docs/api/overview
+ */
+
+import type { Api, Context, Model } from "../types";
+import type { AssistantMessageEventStream } from "../utils/event-stream";
+import {
+	type OpenAIAnthropicApiFormat,
+	type OpenAIAnthropicShimOptions,
+	streamOpenAIAnthropicShim,
+} from "./openai-anthropic-shim";
+
+export type SyntheticApiFormat = OpenAIAnthropicApiFormat;
+
+const SYNTHETIC_NEW_BASE_URL = "https://api.synthetic.new/openai/v1";
+const SYNTHETIC_NEW_ANTHROPIC_BASE_URL = "https://api.synthetic.new/anthropic";
+
+export interface SyntheticOptions extends OpenAIAnthropicShimOptions {
+	/** API format: "openai" or "anthropic". Default: "openai" */
+	format?: SyntheticApiFormat;
+}
+
+/**
+ * Stream from Synthetic, routing to either OpenAI or Anthropic API based on format.
+ * Returns synchronously like other providers - async processing happens internally.
+ */
+export function streamSynthetic(
+	model: Model<"openai-completions">,
+	context: Context,
+	options?: SyntheticOptions,
+): AssistantMessageEventStream {
+	return streamOpenAIAnthropicShim(model, context, options, {
+		anthropicBaseUrl: SYNTHETIC_NEW_ANTHROPIC_BASE_URL,
+		openaiBaseUrl: SYNTHETIC_NEW_BASE_URL,
+		defaultFormat: "openai",
+	});
+}
+
+/**
+ * Check if a model is a Synthetic model.
+ */
+export function isSyntheticModel(model: Model<Api>): boolean {
+	return model.provider === "synthetic";
+}