@hebo-ai/gateway 0.1.2 → 0.2.0

package/dist/models/openai/presets.js

@@ -0,0 +1,206 @@
+import { presetFor } from "../../utils/preset";
+const GPT_OSS_BASE = {
+    modalities: {
+        input: ["text", "file"],
+        output: ["text"],
+    },
+    capabilities: [
+        "attachments",
+        "reasoning",
+        "tool_call",
+        "structured_output",
+        "temperature",
+    ],
+    context: 131072,
+    providers: ["groq", "bedrock", "vertex"],
+};
+const GPT_BASE = {
+    modalities: {
+        input: ["text", "image"],
+        output: ["text"],
+    },
+    capabilities: [
+        "attachments",
+        "reasoning",
+        "tool_call",
+        "structured_output",
+        "temperature",
+    ],
+    providers: ["openai", "azure"],
+};
+const GPT_PRO_BASE = {
+    modalities: {
+        input: ["text", "image"],
+        output: ["text"],
+    },
+    capabilities: [
+        "attachments",
+        "reasoning",
+        "tool_call",
+        "structured_output",
+        "temperature",
+    ],
+    providers: ["openai", "azure"],
+};
+const EMBEDDINGS_BASE = {
+    modalities: {
+        input: ["text"],
+        output: ["embeddings"],
+    },
+    providers: ["openai", "azure"],
+};
+export const gpt5 = presetFor()("openai/gpt-5", {
+    ...GPT_BASE,
+    name: "GPT-5",
+    created: "2025-08-07",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt5Pro = presetFor()("openai/gpt-5-pro", {
+    ...GPT_PRO_BASE,
+    name: "GPT-5 Pro",
+    created: "2025-10-06",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt5Mini = presetFor()("openai/gpt-5-mini", {
+    ...GPT_BASE,
+    name: "GPT-5 Mini",
+    created: "2025-08-07",
+    knowledge: "2024-05",
+    context: 400000,
+});
+export const gpt5Nano = presetFor()("openai/gpt-5-nano", {
+    ...GPT_BASE,
+    name: "GPT-5 Nano",
+    created: "2025-08-07",
+    knowledge: "2024-05",
+    context: 400000,
+});
+export const gpt51 = presetFor()("openai/gpt-5.1", {
+    ...GPT_BASE,
+    name: "GPT-5.1",
+    created: "2025-11-13",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt51Chat = presetFor()("openai/gpt-5.1-chat", {
+    ...GPT_BASE,
+    name: "GPT-5.1 Chat",
+    created: "2025-11-13",
+    knowledge: "2024-09",
+    context: 128000,
+});
+export const gpt51Codex = presetFor()("openai/gpt-5.1-codex", {
+    ...GPT_BASE,
+    name: "GPT-5.1 Codex",
+    created: "2025-11-13",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt51CodexMax = presetFor()("openai/gpt-5.1-codex-max", {
+    ...GPT_BASE,
+    name: "GPT-5.1 Codex Max",
+    created: "2025-11-19",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt5Codex = presetFor()("openai/gpt-5-codex", {
+    ...GPT_BASE,
+    name: "GPT-5 Codex",
+    created: "2025-09-15",
+    knowledge: "2024-09",
+    context: 400000,
+});
+export const gpt52 = presetFor()("openai/gpt-5.2", {
+    ...GPT_BASE,
+    name: "GPT-5.2",
+    created: "2025-12-11",
+    knowledge: "2025-08",
+    context: 400000,
+});
+export const gpt52Chat = presetFor()("openai/gpt-5.2-chat", {
+    ...GPT_BASE,
+    name: "GPT-5.2 Chat",
+    created: "2025-12-11",
+    knowledge: "2025-08",
+    context: 128000,
+});
+export const gpt52Pro = presetFor()("openai/gpt-5.2-pro", {
+    ...GPT_PRO_BASE,
+    name: "GPT-5.2 Pro",
+    created: "2025-12-11",
+    knowledge: "2025-08",
+    context: 400000,
+});
+export const gpt52Codex = presetFor()("openai/gpt-5.2-codex", {
+    ...GPT_BASE,
+    name: "GPT-5.2 Codex",
+    created: "2025-12-18",
+    knowledge: "2025-08",
+});
+export const textEmbedding3Small = presetFor()("openai/text-embedding-3-small", {
+    ...EMBEDDINGS_BASE,
+    name: "Text Embedding 3 Small",
+    created: "2024-01-25",
+    context: 8192,
+});
+export const textEmbedding3Large = presetFor()("openai/text-embedding-3-large", {
+    ...EMBEDDINGS_BASE,
+    name: "Text Embedding 3 Large",
+    created: "2024-01-25",
+    context: 8192,
+});
+export const gptOss20b = presetFor()("openai/gpt-oss-20b", {
+    ...GPT_OSS_BASE,
+    name: "GPT-OSS 20B",
+    created: "2025-08-05",
+    knowledge: "2024-06",
+});
+export const gptOss120b = presetFor()("openai/gpt-oss-120b", {
+    ...GPT_OSS_BASE,
+    name: "GPT-OSS 120B",
+    created: "2025-08-05",
+    knowledge: "2024-06",
+});
+const gptOssAtomic = {
+    v1: [gptOss20b, gptOss120b],
+};
+const gptOssGroups = {
+    "v1.x": [...gptOssAtomic["v1"]],
+};
+const gptAtomic = {
+    v5: [gpt5, gpt5Mini, gpt5Nano, gpt5Pro],
+    "v5.1": [gpt51, gpt51Chat, gpt51Codex, gpt51CodexMax],
+    "v5.2": [gpt52, gpt52Chat, gpt52Pro, gpt52Codex],
+    codex: [gpt5Codex, gpt51Codex, gpt51CodexMax, gpt52Codex],
+    chat: [gpt51Chat, gpt52Chat],
+    pro: [gpt5Pro, gpt52Pro],
+};
+const gptGroups = {
+    "v5.x": [...gptAtomic["v5"], ...gptAtomic["v5.1"], ...gptAtomic["v5.2"]],
+};
+const textEmbeddingsAtomic = {
+    v3: [textEmbedding3Small, textEmbedding3Large],
+};
+const textEmbeddingsGroups = {
+    "v3.x": [...textEmbeddingsAtomic["v3"]],
+};
+export const gptOss = {
+    ...gptOssAtomic,
+    ...gptOssGroups,
+    latest: [...gptOssAtomic["v1"]],
+    all: Object.values(gptOssAtomic).flat(),
+};
+export const gpt = {
+    ...gptAtomic,
+    ...gptGroups,
+    latest: [gpt52, gpt5Mini, gpt5Nano],
+    all: Object.values(gptAtomic).flat(),
+};
+export const textEmbeddings = {
+    ...textEmbeddingsAtomic,
+    ...textEmbeddingsGroups,
+    latest: [...textEmbeddingsAtomic["v3"]],
+    all: Object.values(textEmbeddingsAtomic).flat(),
+};

package/dist/models/types.d.ts

@@ -1,12 +1,12 @@
 import type { ProviderId } from "../providers/types";
-export declare const CANONICAL_MODEL_IDS: readonly ["anthropic/claude-haiku-4.5", "anthropic/claude-sonnet-4.5", "anthropic/claude-opus-4.5", "anthropic/claude-opus-4.1", "anthropic/claude-opus-4", "anthropic/claude-sonnet-4", "anthropic/claude-sonnet-3.7", "anthropic/claude-sonnet-3.5", "anthropic/claude-haiku-3.5", "anthropic/claude-haiku-3", "openai/gpt-oss-20b", "openai/gpt-oss-120b", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash", "google/gemini-2.5-pro", "google/gemini-3-flash-preview", "google/gemini-3-pro-preview", "meta/llama-3.1-8b", "meta/llama-3.1-70b", "meta/llama-3.1-405b", "meta/llama-3.2-1b", "meta/llama-3.2-3b", "meta/llama-3.2-11b", "meta/llama-3.2-90b", "meta/llama-3.3-70b", "meta/llama-4-scout", "meta/llama-4-maverick", "cohere/embed-v4.0", "cohere/command-a", "cohere/command-r7b", "cohere/command-a-translate", "cohere/command-a-reasoning", "cohere/command-a-vision", "cohere/command-r", "cohere/command-r-plus", "voyage/voyage-2-code", "voyage/voyage-2-law", "voyage/voyage-2-finance", "voyage/voyage-3-code", "voyage/voyage-3-large", "voyage/voyage-3.5-lite", "voyage/voyage-3.5", "voyage/voyage-4-lite", "voyage/voyage-4", "voyage/voyage-4-large"];
+export declare const CANONICAL_MODEL_IDS: readonly ["anthropic/claude-haiku-4.5", "anthropic/claude-sonnet-4.5", "anthropic/claude-opus-4.5", "anthropic/claude-opus-4.1", "anthropic/claude-opus-4", "anthropic/claude-sonnet-4", "anthropic/claude-sonnet-3.7", "anthropic/claude-sonnet-3.5", "anthropic/claude-haiku-3.5", "anthropic/claude-haiku-3", "openai/gpt-oss-20b", "openai/gpt-oss-120b", "openai/gpt-5", "openai/gpt-5-pro", "openai/gpt-5.2", "openai/gpt-5.2-chat", "openai/gpt-5.2-pro", "openai/gpt-5.2-codex", "openai/gpt-5-mini", "openai/gpt-5-nano", "openai/gpt-5-codex", "openai/gpt-5.1-codex", "openai/gpt-5.1-codex-max", "openai/gpt-5.1-chat", "openai/gpt-5.1", "openai/text-embedding-3-small", "openai/text-embedding-3-large", "amazon/nova-micro", "amazon/nova-lite", "amazon/nova-pro", "amazon/nova-premier", "amazon/nova-2-lite", "amazon/nova-2-multimodal-embeddings", "google/gemini-2.5-flash-lite", "google/gemini-2.5-flash", "google/gemini-2.5-pro", "google/gemini-3-flash-preview", "google/gemini-3-pro-preview", "google/embedding-001", "meta/llama-3.1-8b", "meta/llama-3.1-70b", "meta/llama-3.1-405b", "meta/llama-3.2-1b", "meta/llama-3.2-3b", "meta/llama-3.2-11b", "meta/llama-3.2-90b", "meta/llama-3.3-70b", "meta/llama-4-scout", "meta/llama-4-maverick", "cohere/embed-v4.0", "cohere/embed-english-v3.0", "cohere/embed-english-light-v3.0", "cohere/embed-multilingual-v3.0", "cohere/embed-multilingual-light-v3.0", "cohere/command-a", "cohere/command-r7b", "cohere/command-a-translate", "cohere/command-a-reasoning", "cohere/command-a-vision", "cohere/command-r", "cohere/command-r-plus", "voyage/voyage-2-code", "voyage/voyage-2-law", "voyage/voyage-2-finance", "voyage/voyage-3-code", "voyage/voyage-3-large", "voyage/voyage-3.5-lite", "voyage/voyage-3.5", "voyage/voyage-4-lite", "voyage/voyage-4", "voyage/voyage-4-large"];
 export type CanonicalModelId = (typeof CANONICAL_MODEL_IDS)[number];
 export type ModelId = CanonicalModelId | (string & {});
 export type CatalogModel = {
-    name: string;
+    name?: string;
     created?: string;
     knowledge?: string;
-    modalities: {
+    modalities?: {
         input: readonly ("text" | "image" | "file" | "audio" | "video" | "pdf")[];
         output: readonly ("text" | "image" | "audio" | "video" | "embeddings")[];
     };

package/dist/models/types.js

@@ -13,12 +13,35 @@ export const CANONICAL_MODEL_IDS = [
     // OpenAI
     "openai/gpt-oss-20b",
     "openai/gpt-oss-120b",
+    "openai/gpt-5",
+    "openai/gpt-5-pro",
+    "openai/gpt-5.2",
+    "openai/gpt-5.2-chat",
+    "openai/gpt-5.2-pro",
+    "openai/gpt-5.2-codex",
+    "openai/gpt-5-mini",
+    "openai/gpt-5-nano",
+    "openai/gpt-5-codex",
+    "openai/gpt-5.1-codex",
+    "openai/gpt-5.1-codex-max",
+    "openai/gpt-5.1-chat",
+    "openai/gpt-5.1",
+    "openai/text-embedding-3-small",
+    "openai/text-embedding-3-large",
+    // Amazon
+    "amazon/nova-micro",
+    "amazon/nova-lite",
+    "amazon/nova-pro",
+    "amazon/nova-premier",
+    "amazon/nova-2-lite",
+    "amazon/nova-2-multimodal-embeddings",
     // Google
     "google/gemini-2.5-flash-lite",
     "google/gemini-2.5-flash",
     "google/gemini-2.5-pro",
     "google/gemini-3-flash-preview",
     "google/gemini-3-pro-preview",
+    "google/embedding-001",
     // Meta
     "meta/llama-3.1-8b",
     "meta/llama-3.1-70b",
@@ -32,6 +55,10 @@ export const CANONICAL_MODEL_IDS = [
     "meta/llama-4-maverick",
     // Cohere
     "cohere/embed-v4.0",
+    "cohere/embed-english-v3.0",
+    "cohere/embed-english-light-v3.0",
+    "cohere/embed-multilingual-v3.0",
+    "cohere/embed-multilingual-light-v3.0",
     "cohere/command-a",
     "cohere/command-r7b",
     "cohere/command-a-translate",

package/dist/models/voyage/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./middleware";
+export * from "./presets";

package/dist/models/voyage/index.js

@@ -0,0 +1,2 @@
+export * from "./middleware";
+export * from "./presets";

package/dist/models/voyage/middleware.d.ts

@@ -0,0 +1,2 @@
+import type { EmbeddingModelMiddleware } from "ai";
+export declare const voyageEmbeddingModelMiddleware: EmbeddingModelMiddleware;

package/dist/models/voyage/middleware.js

@@ -0,0 +1,18 @@
+import { modelMiddlewareMatcher } from "../../middleware/matcher";
+// Convert `dimensions` (OpenAI) to `outputDimension` (Voyage)
+export const voyageEmbeddingModelMiddleware = {
+    specificationVersion: "v3",
+    // eslint-disable-next-line require-await
+    transformParams: async ({ params }) => {
+        const unknown = params.providerOptions?.["unknown"];
+        if (!unknown)
+            return params;
+        let dimensions = unknown["dimensions"];
+        if (!dimensions)
+            dimensions = 1024;
+        (params.providerOptions["voyage"] ??= {})["outputDimension"] = dimensions;
+        delete unknown["dimensions"];
+        return params;
+    },
+};
+modelMiddlewareMatcher.useForModel("voyage/*", { embedding: voyageEmbeddingModelMiddleware });