@kernl-sdk/ai 0.4.3 → 0.4.4

package/src/providers/anthropic.ts

@@ -1,29 +1,67 @@
-import { anthropic as _anthropic } from "@ai-sdk/anthropic";
+import {
+  anthropic as _anthropic,
+  createAnthropic as _createAnthropic,
+} from "@ai-sdk/anthropic";
 import { AISDKLanguageModel } from "../language-model";
+import { createOAuthFetch } from "../oauth/anthropic";
+import type { OAuthCredentials } from "../oauth/types";
 
 /**
- * Anthropic model IDs.
+ * Anthropic model IDs (derived from @ai-sdk/anthropic).
  */
-export type AnthropicModelId =
-  | "claude-haiku-4-5"
-  | "claude-haiku-4-5-20251001"
-  | "claude-sonnet-4-5"
-  | "claude-sonnet-4-5-20250929"
-  | "claude-opus-4-1"
-  | "claude-opus-4-0"
-  | "claude-sonnet-4-0"
-  | "claude-opus-4-1-20250805"
-  | "claude-opus-4-20250514"
-  | "claude-sonnet-4-20250514"
-  | "claude-3-7-sonnet-latest"
-  | "claude-3-7-sonnet-20250219"
-  | "claude-3-5-haiku-latest"
-  | "claude-3-5-haiku-20241022"
-  | "claude-3-haiku-20240307"
-  | (string & {});
+export type AnthropicModelId = Parameters<typeof _anthropic>[0];
+
+/**
+ * Options for creating a custom Anthropic provider.
+ */
+export interface AnthropicProviderOptions {
+  /** API key for standard authentication */
+  apiKey?: string;
+  /** OAuth credentials for Claude Pro/Max authentication */
+  oauth?: OAuthCredentials;
+  /** Custom base URL */
+  baseURL?: string;
+  /** Custom headers */
+  headers?: Record<string, string>;
+}
+
+/**
+ * Create a custom Anthropic provider with explicit credentials.
+ *
+ * @example API key auth
+ * ```ts
+ * const anthropic = createAnthropic({ apiKey: "sk-..." });
+ * const model = anthropic("claude-sonnet-4-5");
+ * ```
+ *
+ * @example OAuth auth (Claude Pro/Max)
+ * ```ts
+ * const anthropic = createAnthropic({
+ *   oauth: {
+ *     accessToken: "...",
+ *     refreshToken: "...",
+ *     expiresAt: Date.now() + 3600000,
+ *     onRefresh: (tokens) => saveTokens(tokens),
+ *   }
+ * });
+ * const model = anthropic("claude-sonnet-4-5");
+ * ```
+ */
+export function createAnthropic(options: AnthropicProviderOptions = {}) {
+  const provider = _createAnthropic({
+    apiKey: options.oauth ? undefined : options.apiKey,
+    baseURL: options.baseURL,
+    headers: options.headers,
+    fetch: options.oauth ? createOAuthFetch(options.oauth) : undefined,
+  });
+
+  return (modelId: AnthropicModelId) =>
+    new AISDKLanguageModel(provider(modelId));
+}
 
 /**
  * Create a kernl-compatible Anthropic language model.
+ * Uses ANTHROPIC_API_KEY environment variable.
  *
  * @example
  * ```ts
@@ -38,4 +76,7 @@ export function anthropic(modelId: AnthropicModelId) {
   return new AISDKLanguageModel(model);
 }
 
+// Re-export types
+export type { OAuthCredentials } from "../oauth/types";
+
 // Note: Anthropic does not currently support embeddings

package/src/providers/google.ts

@@ -1,45 +1,12 @@
 import { google as _google } from "@ai-sdk/google";
+import { registerEmbeddingProvider } from "@kernl-sdk/retrieval";
 import { AISDKLanguageModel } from "../language-model";
 import { AISDKEmbeddingModel } from "../embedding-model";
-import { registerEmbeddingProvider } from "@kernl-sdk/retrieval";
 
 /**
- * Google Generative AI model IDs.
+ * Google model IDs (derived from @ai-sdk/google).
  */
-type GoogleGenerativeAIModelId =
-  | "gemini-1.5-flash"
-  | "gemini-1.5-flash-latest"
-  | "gemini-1.5-flash-001"
-  | "gemini-1.5-flash-002"
-  | "gemini-1.5-flash-8b"
-  | "gemini-1.5-flash-8b-latest"
-  | "gemini-1.5-flash-8b-001"
-  | "gemini-1.5-pro"
-  | "gemini-1.5-pro-latest"
-  | "gemini-1.5-pro-001"
-  | "gemini-1.5-pro-002"
-  | "gemini-2.0-flash"
-  | "gemini-2.0-flash-001"
-  | "gemini-2.0-flash-live-001"
-  | "gemini-2.0-flash-lite"
-  | "gemini-2.0-pro-exp-02-05"
-  | "gemini-2.0-flash-thinking-exp-01-21"
-  | "gemini-2.0-flash-exp"
-  | "gemini-2.5-pro"
-  | "gemini-2.5-flash"
-  | "gemini-2.5-flash-image-preview"
-  | "gemini-2.5-flash-lite"
-  | "gemini-2.5-flash-lite-preview-09-2025"
-  | "gemini-2.5-flash-preview-04-17"
-  | "gemini-2.5-flash-preview-09-2025"
-  | "gemini-pro-latest"
-  | "gemini-flash-latest"
-  | "gemini-flash-lite-latest"
-  | "gemini-2.5-pro-exp-03-25"
-  | "gemini-exp-1206"
-  | "gemma-3-12b-it"
-  | "gemma-3-27b-it"
-  | (string & {});
+export type GoogleModelId = Parameters<typeof _google>[0];
 
 /**
  * Create a kernl-compatible Google Generative AI language model.
@@ -52,7 +19,7 @@ type GoogleGenerativeAIModelId =
  * const response = await gemini.generate([...], {});
  * ```
  */
-export function google(modelId: GoogleGenerativeAIModelId) {
+export function google(modelId: GoogleModelId) {
   const model = _google(modelId);
   return new AISDKLanguageModel(model);
 }

package/src/providers/openai.ts

@@ -1,52 +1,70 @@
-import { openai as _openai } from "@ai-sdk/openai";
+import {
+  openai as _openai,
+  createOpenAI as _createOpenAI,
+} from "@ai-sdk/openai";
+import { registerEmbeddingProvider } from "@kernl-sdk/retrieval";
+
 import { AISDKLanguageModel } from "../language-model";
 import { AISDKEmbeddingModel } from "../embedding-model";
-import { registerEmbeddingProvider } from "@kernl-sdk/retrieval";
+import { createOAuthFetch } from "../oauth/openai";
+import type { OpenAIOAuthCredentials } from "../oauth/types";
+
+/**
+ * OpenAI model IDs (derived from @ai-sdk/openai).
+ */
+export type OpenAIModelId = Parameters<typeof _openai>[0];
 
 /**
- * OpenAI model IDs.
+ * Options for creating a custom OpenAI provider.
  */
-type OpenAIModelId =
-  | "chatgpt-4o-latest"
-  | "gpt-3.5-turbo-0125"
-  | "gpt-3.5-turbo-1106"
-  | "gpt-3.5-turbo"
-  | "gpt-4-0613"
-  | "gpt-4-turbo-2024-04-09"
-  | "gpt-4-turbo"
-  | "gpt-4.1-2025-04-14"
-  | "gpt-4.1-mini-2025-04-14"
-  | "gpt-4.1-mini"
-  | "gpt-4.1-nano-2025-04-14"
-  | "gpt-4.1-nano"
-  | "gpt-4.1"
-  | "gpt-4"
-  | "gpt-4o-2024-05-13"
-  | "gpt-4o-2024-08-06"
-  | "gpt-4o-2024-11-20"
-  | "gpt-4o-mini-2024-07-18"
-  | "gpt-4o-mini"
-  | "gpt-4o"
-  | "gpt-5-2025-08-07"
-  | "gpt-5-chat-latest"
-  | "gpt-5-codex"
-  | "gpt-5-mini-2025-08-07"
-  | "gpt-5-mini"
-  | "gpt-5-nano-2025-08-07"
-  | "gpt-5-nano"
-  | "gpt-5-pro-2025-10-06"
-  | "gpt-5-pro"
-  | "gpt-5"
-  | "o1-2024-12-17"
-  | "o1"
-  | "o3-2025-04-16"
-  | "o3-mini-2025-01-31"
-  | "o3-mini"
-  | "o3"
-  | (string & {});
+export interface OpenAIProviderOptions {
+  /** API key for standard authentication */
+  apiKey?: string;
+  /** OAuth credentials for ChatGPT Plus/Pro (Codex) authentication */
+  oauth?: OpenAIOAuthCredentials;
+  /** Custom base URL (ignored for OAuth - uses Codex endpoint) */
+  baseURL?: string;
+  /** Custom headers */
+  headers?: Record<string, string>;
+}
+
+/**
+ * Create a custom OpenAI provider with explicit credentials.
+ *
+ * @example API key auth
+ * ```ts
+ * const openai = createOpenAI({ apiKey: "sk-..." });
+ * const model = openai("gpt-4o");
+ * ```
+ *
+ * @example OAuth auth (ChatGPT Plus/Pro via Codex)
+ * ```ts
+ * const openai = createOpenAI({
+ *   oauth: {
+ *     accessToken: "...",
+ *     refreshToken: "...",
+ *     expiresAt: Date.now() + 3600000,
+ *     accountId: "...", // for org subscriptions
+ *     onRefresh: (tokens) => saveTokens(tokens),
+ *   }
+ * });
+ * const model = openai("gpt-4o");
+ * ```
+ */
+export function createOpenAI(options: OpenAIProviderOptions = {}) {
+  const provider = _createOpenAI({
+    apiKey: options.oauth ? undefined : options.apiKey,
+    baseURL: options.oauth ? undefined : options.baseURL,
+    headers: options.headers,
+    fetch: options.oauth ? createOAuthFetch(options.oauth) : undefined,
+  });
+
+  return (modelId: OpenAIModelId) => new AISDKLanguageModel(provider(modelId));
+}
 
 /**
  * Create a kernl-compatible OpenAI language model.
+ * Uses OPENAI_API_KEY environment variable.
  *
  * @example
  * ```ts
@@ -61,6 +79,9 @@ export function openai(modelId: OpenAIModelId) {
   return new AISDKLanguageModel(model);
 }
 
+// Re-export types
+export type { OpenAIOAuthCredentials } from "../oauth/types";
+
 // Auto-register OpenAI embedding provider
 registerEmbeddingProvider(
   "openai",