@funkai/models 0.1.0

package/src/provider/resolver.ts

@@ -0,0 +1,125 @@
+import type { ModelId } from "@/catalog/index.js";
+import type { LanguageModel } from "@/provider/types.js";
+
+/**
+ * A function that resolves a provider-specific model name to a {@link LanguageModel}.
+ *
+ * @example
+ * ```typescript
+ * import { createOpenAI } from '@ai-sdk/openai'
+ *
+ * const factory: ProviderFactory = createOpenAI({ apiKey: '...' })
+ * const model = factory('gpt-4.1')
+ * ```
+ */
+export type ProviderFactory = (modelName: string) => LanguageModel;
+
+/**
+ * Mapping from provider prefix (e.g. `"openai"`, `"anthropic"`) to a factory
+ * that creates {@link LanguageModel} instances for that provider.
+ */
+export type ProviderMap = Readonly<Record<string, ProviderFactory>>;
+
+/**
+ * Configuration for {@link createModelResolver}.
+ */
+export interface ModelResolverConfig {
+  /**
+   * Direct AI SDK provider mappings by prefix.
+   *
+   * When a model ID like `"openai/gpt-4.1"` is resolved, the prefix
+   * `"openai"` is extracted and looked up in this map. If found, the
+   * factory receives the model portion (`"gpt-4.1"`).
+   */
+  readonly providers?: ProviderMap;
+
+  /**
+   * A fallback factory used when no mapped provider matches.
+   *
+   * Receives the full model ID string. Useful for routing unmapped
+   * providers through an aggregator like OpenRouter.
+   *
+   * @example
+   * ```typescript
+   * import { openrouter } from '@funkai/models'
+   *
+   * const resolve = createModelResolver({
+   *   providers: { openai: createOpenAI({ apiKey: '...' }) },
+   *   fallback: openrouter,
+   * })
+   * ```
+   */
+  readonly fallback?: (modelId: string) => LanguageModel;
+}
+
+/**
+ * A function that resolves a model ID to a {@link LanguageModel} instance.
+ */
+export type ModelResolver = (modelId: ModelId) => LanguageModel;
+
+/**
+ * Create a model resolver with configurable provider mappings.
+ *
+ * When a model ID like `"openai/gpt-4.1"` is resolved:
+ * 1. The provider prefix (`"openai"`) is extracted
+ * 2. If a direct provider is mapped for that prefix, it receives the model portion (`"gpt-4.1"`)
+ * 3. Otherwise, the fallback is called with the full ID (if configured)
+ * 4. If no fallback, an error is thrown
+ *
+ * @param config - Provider mappings and fallback configuration.
+ * @returns A resolver function that maps model IDs to {@link LanguageModel} instances.
+ *
+ * @example
+ * ```typescript
+ * import { createOpenAI } from '@ai-sdk/openai'
+ * import { createAnthropic } from '@ai-sdk/anthropic'
+ *
+ * const resolve = createModelResolver({
+ *   providers: {
+ *     openai: createOpenAI({ apiKey: '...' }),
+ *     anthropic: createAnthropic({ apiKey: '...' }),
+ *   },
+ * })
+ *
+ * const m1 = resolve('openai/gpt-4.1')           // uses @ai-sdk/openai directly
+ * const m2 = resolve('anthropic/claude-sonnet-4') // uses @ai-sdk/anthropic directly
+ * ```
+ */
+export function createModelResolver(config?: ModelResolverConfig): ModelResolver {
+  const providers = config?.providers ?? {};
+  const fallback = config?.fallback;
+
+  return (modelId: ModelId): LanguageModel => {
+    if (!modelId.trim()) {
+      throw new Error("Cannot resolve model: model ID is empty");
+    }
+
+    const slashIndex = modelId.indexOf("/");
+
+    if (slashIndex === -1) {
+      if (fallback) {
+        return fallback(modelId);
+      }
+      throw new Error(
+        `Cannot resolve model "${modelId}": no provider prefix and no fallback configured`,
+      );
+    }
+
+    const prefix = modelId.slice(0, slashIndex);
+    // eslint-disable-next-line security/detect-object-injection -- Prefix extracted from user model ID, used as record key lookup
+    const factory = providers[prefix];
+
+    if (factory) {
+      const modelName = modelId.slice(slashIndex + 1);
+      return factory(modelName);
+    }
+
+    if (fallback) {
+      return fallback(modelId);
+    }
+
+    throw new Error(
+      `Cannot resolve model "${modelId}": no provider mapped for "${prefix}" and no fallback configured`,
+    );
+  };
+}

package/src/provider/types.ts

@@ -0,0 +1,39 @@
+import { type LanguageModel as BaseLanguageModel } from "ai";
+
+/**
+ * AI SDK language model instance (v3 specification).
+ *
+ * Narrowed from the base `LanguageModel` union (which includes `string`) to
+ * only concrete v3 model objects. This is required because AI SDK functions
+ * like `wrapLanguageModel` expect `LanguageModelV3` specifically.
+ *
+ * When the AI SDK introduces a new specification version, update the
+ * `specificationVersion` literal here and verify compatibility with
+ * downstream consumers (e.g. `@funkai/agents` middleware).
+ */
+export type LanguageModel = Extract<BaseLanguageModel, { specificationVersion: "v3" }>;
+
+/**
+ * Base token counts shared by raw tracking records and final output.
+ *
+ * All fields are resolved `number` (0 when absent).
+ */
+export interface TokenUsage {
+  /** Number of input (prompt) tokens. */
+  readonly inputTokens: number;
+
+  /** Number of output (completion) tokens. */
+  readonly outputTokens: number;
+
+  /** Total tokens (input + output). */
+  readonly totalTokens: number;
+
+  /** Tokens served from the provider's prompt cache. */
+  readonly cacheReadTokens: number;
+
+  /** Tokens written into the provider's prompt cache. */
+  readonly cacheWriteTokens: number;
+
+  /** Tokens consumed by the model's internal reasoning (e.g. o3/o4). */
+  readonly reasoningTokens: number;
+}

package/src/providers/alibaba.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { ALIBABA_MODELS } from "../catalog/providers/alibaba.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for Alibaba (Qwen).
+ *
+ * @example
+ * ```typescript
+ * import type { AlibabaModelId } from '@funkai/models/alibaba'
+ *
+ * const id: AlibabaModelId = 'qwen-vl-plus'
+ * ```
+ */
+export type AlibabaModelId = (typeof ALIBABA_MODELS)[number]["id"];
+
+/**
+ * All Alibaba (Qwen) models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { alibabaModels } from '@funkai/models/alibaba'
+ *
+ * for (const m of alibabaModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const alibabaModels = ALIBABA_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(ALIBABA_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up an Alibaba (Qwen) model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { alibabaModel } from '@funkai/models/alibaba'
+ *
+ * const m = alibabaModel('qwen-vl-plus')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function alibabaModel(id: LiteralUnion<AlibabaModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/amazon-bedrock.ts

@@ -0,0 +1,67 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { AMAZON_BEDROCK_MODELS } from "../catalog/providers/amazon-bedrock.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for Amazon Bedrock.
+ *
+ * @example
+ * ```typescript
+ * import type { AmazonBedrockModelId } from '@funkai/models/amazon-bedrock'
+ *
+ * const id: AmazonBedrockModelId = 'deepseek.r1-v1:0'
+ * ```
+ */
+export type AmazonBedrockModelId = (typeof AMAZON_BEDROCK_MODELS)[number]["id"];
+
+/**
+ * All Amazon Bedrock models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { amazonBedrockModels } from '@funkai/models/amazon-bedrock'
+ *
+ * for (const m of amazonBedrockModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const amazonBedrockModels = AMAZON_BEDROCK_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(AMAZON_BEDROCK_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up an Amazon Bedrock model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { amazonBedrockModel } from '@funkai/models/amazon-bedrock'
+ *
+ * const m = amazonBedrockModel('deepseek.r1-v1:0')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function amazonBedrockModel(
+  id: LiteralUnion<AmazonBedrockModelId, string>,
+): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/anthropic.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { ANTHROPIC_MODELS } from "../catalog/providers/anthropic.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for Anthropic.
+ *
+ * @example
+ * ```typescript
+ * import type { AnthropicModelId } from '@funkai/models/anthropic'
+ *
+ * const id: AnthropicModelId = 'claude-opus-4-5-20251101'
+ * ```
+ */
+export type AnthropicModelId = (typeof ANTHROPIC_MODELS)[number]["id"];
+
+/**
+ * All Anthropic models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { anthropicModels } from '@funkai/models/anthropic'
+ *
+ * for (const m of anthropicModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const anthropicModels = ANTHROPIC_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(ANTHROPIC_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up an Anthropic model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { anthropicModel } from '@funkai/models/anthropic'
+ *
+ * const m = anthropicModel('claude-opus-4-5-20251101')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function anthropicModel(id: LiteralUnion<AnthropicModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/cerebras.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { CEREBRAS_MODELS } from "../catalog/providers/cerebras.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for Cerebras.
+ *
+ * @example
+ * ```typescript
+ * import type { CerebrasModelId } from '@funkai/models/cerebras'
+ *
+ * const id: CerebrasModelId = 'qwen-3-235b-a22b-instruct-2507'
+ * ```
+ */
+export type CerebrasModelId = (typeof CEREBRAS_MODELS)[number]["id"];
+
+/**
+ * All Cerebras models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { cerebrasModels } from '@funkai/models/cerebras'
+ *
+ * for (const m of cerebrasModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const cerebrasModels = CEREBRAS_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(CEREBRAS_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up a Cerebras model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { cerebrasModel } from '@funkai/models/cerebras'
+ *
+ * const m = cerebrasModel('qwen-3-235b-a22b-instruct-2507')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function cerebrasModel(id: LiteralUnion<CerebrasModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/cohere.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { COHERE_MODELS } from "../catalog/providers/cohere.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for Cohere.
+ *
+ * @example
+ * ```typescript
+ * import type { CohereModelId } from '@funkai/models/cohere'
+ *
+ * const id: CohereModelId = 'c4ai-aya-expanse-32b'
+ * ```
+ */
+export type CohereModelId = (typeof COHERE_MODELS)[number]["id"];
+
+/**
+ * All Cohere models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { cohereModels } from '@funkai/models/cohere'
+ *
+ * for (const m of cohereModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const cohereModels = COHERE_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(COHERE_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up a Cohere model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { cohereModel } from '@funkai/models/cohere'
+ *
+ * const m = cohereModel('c4ai-aya-expanse-32b')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function cohereModel(id: LiteralUnion<CohereModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/deepinfra.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { DEEPINFRA_MODELS } from "../catalog/providers/deepinfra.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for DeepInfra.
+ *
+ * @example
+ * ```typescript
+ * import type { DeepInfraModelId } from '@funkai/models/deepinfra'
+ *
+ * const id: DeepInfraModelId = 'zai-org/GLM-4.7-Flash'
+ * ```
+ */
+export type DeepInfraModelId = (typeof DEEPINFRA_MODELS)[number]["id"];
+
+/**
+ * All DeepInfra models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { deepInfraModels } from '@funkai/models/deepinfra'
+ *
+ * for (const m of deepInfraModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const deepInfraModels = DEEPINFRA_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(DEEPINFRA_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up a DeepInfra model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { deepInfraModel } from '@funkai/models/deepinfra'
+ *
+ * const m = deepInfraModel('zai-org/GLM-4.7-Flash')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function deepInfraModel(id: LiteralUnion<DeepInfraModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}

package/src/providers/deepseek.ts

@@ -0,0 +1,65 @@
+// ──────────────────────────────────────────────────────────────
+// ███████╗██╗   ██╗███╗   ██╗██╗  ██╗ █████╗ ██╗
+// ██╔════╝██║   ██║████╗  ██║██║ ██╔╝██╔══██╗██║
+// █████╗  ██║   ██║██╔██╗ ██║█████╔╝ ███████║██║
+// ██╔══╝  ██║   ██║██║╚██╗██║██╔═██╗ ██╔══██║██║
+// ██║     ╚██████╔╝██║ ╚████║██║  ██╗██║  ██║██║
+// ╚═╝      ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝  ╚═╝╚═╝
+//
+// AUTO-GENERATED — DO NOT EDIT
+// Source: https://models.dev
+// Update: pnpm --filter=@funkai/models generate:models
+// ──────────────────────────────────────────────────────────────
+
+import type { LiteralUnion } from "type-fest";
+
+import { DEEPSEEK_MODELS } from "../catalog/providers/deepseek.js";
+import type { ModelDefinition } from "../catalog/types.js";
+
+/**
+ * Known model identifiers for DeepSeek.
+ *
+ * @example
+ * ```typescript
+ * import type { DeepSeekModelId } from '@funkai/models/deepseek'
+ *
+ * const id: DeepSeekModelId = 'deepseek-reasoner'
+ * ```
+ */
+export type DeepSeekModelId = (typeof DEEPSEEK_MODELS)[number]["id"];
+
+/**
+ * All DeepSeek models in the catalog.
+ *
+ * @example
+ * ```typescript
+ * import { deepSeekModels } from '@funkai/models/deepseek'
+ *
+ * for (const m of deepSeekModels) {
+ *   console.log(m.id, m.pricing.input)
+ * }
+ * ```
+ */
+export const deepSeekModels = DEEPSEEK_MODELS;
+
+const MODEL_INDEX = new Map<string, ModelDefinition>(DEEPSEEK_MODELS.map((m) => [m.id, m]));
+
+/**
+ * Look up a DeepSeek model by ID.
+ *
+ * @param id - The provider-native model identifier.
+ * @returns The matching model definition, or `null`.
+ *
+ * @example
+ * ```typescript
+ * import { deepSeekModel } from '@funkai/models/deepseek'
+ *
+ * const m = deepSeekModel('deepseek-reasoner')
+ * if (m) {
+ *   console.log(m.pricing.input)
+ * }
+ * ```
+ */
+export function deepSeekModel(id: LiteralUnion<DeepSeekModelId, string>): ModelDefinition | null {
+  return MODEL_INDEX.get(id) ?? null;
+}