@hebo-ai/gateway 0.5.2 → 0.6.0-rc0

package/dist/models/cohere/presets.js

@@ -1,134 +0,0 @@
-import { presetFor } from "../../utils/preset";
-const COMMAND_BASE = {
-    modalities: {
-        input: ["text"],
-        output: ["text"],
-    },
-    capabilities: ["tool_call", "structured_output", "reasoning", "temperature"],
-    providers: ["cohere"],
-    knowledge: "2024-06",
-};
-const COMMAND_VISION_BASE = {
-    modalities: {
-        input: ["text", "image"],
-        output: ["text"],
-    },
-    capabilities: ["structured_output", "reasoning", "temperature"],
-    providers: ["cohere"],
-    knowledge: "2024-06",
-};
-const EMBED_V3_BASE = {
-    modalities: {
-        input: ["text", "image"],
-        output: ["embedding"],
-    },
-    providers: ["cohere"],
-};
-const EMBED_V4_BASE = {
-    modalities: {
-        input: ["text", "image", "pdf"],
-        output: ["embedding"],
-    },
-    providers: ["cohere"],
-};
-export const commandA = presetFor()("cohere/command-a", {
-    ...COMMAND_BASE,
-    name: "Cohere Command A",
-    created: "2025-03-13",
-    context: 256000,
-});
-export const commandAReasoning = presetFor()("cohere/command-a-reasoning", {
-    ...COMMAND_BASE,
-    name: "Cohere Command A Reasoning",
-    created: "2025-08-21",
-    context: 256000,
-});
-export const commandATranslate = presetFor()("cohere/command-a-translate", {
-    ...COMMAND_BASE,
-    name: "Cohere Command A Translate",
-    created: "2025-08-28",
-    context: 8000,
-});
-export const commandAVision = presetFor()("cohere/command-a-vision", {
-    ...COMMAND_VISION_BASE,
-    name: "Cohere Command A Vision",
-    created: "2025-07-31",
-    context: 128000,
-});
-export const commandR = presetFor()("cohere/command-r", {
-    ...COMMAND_BASE,
-    name: "Cohere Command R",
-    created: "2024-08-01",
-    context: 128000,
-    providers: ["cohere", "bedrock"],
-});
-export const commandRPlus = presetFor()("cohere/command-r-plus", {
-    ...COMMAND_BASE,
-    name: "Cohere Command R+",
-    created: "2024-08-01",
-    context: 128000,
-    providers: ["cohere", "bedrock"],
-});
-export const commandR7b = presetFor()("cohere/command-r7b", {
-    ...COMMAND_BASE,
-    name: "Cohere Command R7B",
-    created: "2024-12-13",
-    context: 128000,
-});
-export const embed4 = presetFor()("cohere/embed-v4.0", {
-    ...EMBED_V4_BASE,
-    name: "Cohere 4 Embeddings",
-    created: "2025-04-15",
-    context: 128000,
-    providers: ["cohere", "bedrock"],
-});
-export const embedEnglishV3 = presetFor()("cohere/embed-english-v3.0", {
-    ...EMBED_V3_BASE,
-    name: "Cohere Embed English v3",
-    created: "2024-02-07",
-    context: 512,
-    providers: ["cohere", "bedrock"],
-});
-export const embedEnglishLightV3 = presetFor()("cohere/embed-english-light-v3.0", {
-    ...EMBED_V3_BASE,
-    name: "Cohere Embed English Light v3",
-    created: "2024-02-07",
-    context: 512,
-    providers: ["cohere"],
-});
-export const embedMultilingualV3 = presetFor()("cohere/embed-multilingual-v3.0", {
-    ...EMBED_V3_BASE,
-    name: "Cohere Embed Multilingual v3",
-    created: "2024-02-07",
-    context: 512,
-    providers: ["cohere", "bedrock"],
-});
-export const embedMultilingualLightV3 = presetFor()("cohere/embed-multilingual-light-v3.0", {
-    ...EMBED_V3_BASE,
-    name: "Cohere Embed Multilingual Light v3",
-    created: "2024-02-07",
-    context: 512,
-    providers: ["cohere"],
-});
-const commandAtomic = {
-    A: [commandA, commandAReasoning, commandATranslate, commandAVision],
-    R: [commandR, commandRPlus, commandR7b],
-};
-const commandGroups = {};
-export const command = {
-    ...commandAtomic,
-    ...commandGroups,
-    latest: [commandA],
-    all: Object.values(commandAtomic).flat(),
-};
-const embedAtomic = {
-    v4: [embed4],
-    v3: [embedEnglishV3, embedEnglishLightV3, embedMultilingualV3, embedMultilingualLightV3],
-};
-const embedGroups = {};
-export const embed = {
-    ...embedAtomic,
-    ...embedGroups,
-    latest: [embed4],
-    all: Object.values(embedAtomic).flat(),
-};

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

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

package/dist/models/google/index.js

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

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

@@ -1,7 +0,0 @@
-import type { EmbeddingModelMiddleware, LanguageModelMiddleware } from "ai";
-import type { ChatCompletionsReasoningEffort } from "../../endpoints/chat-completions/schema";
-export declare const geminiDimensionsMiddleware: EmbeddingModelMiddleware;
-export declare function mapGeminiReasoningEffort(effort: ChatCompletionsReasoningEffort, modelId: string): ChatCompletionsReasoningEffort | undefined;
-export declare const GEMINI_DEFAULT_MAX_OUTPUT_TOKENS = 65536;
-export declare const GEMINI_2_5_PRO_MIN_THINKING_BUDGET = 128;
-export declare const geminiReasoningMiddleware: LanguageModelMiddleware;

package/dist/models/google/middleware.js

@@ -1,103 +0,0 @@
-import { modelMiddlewareMatcher } from "../../middleware/matcher";
-import { calculateReasoningBudgetFromEffort } from "../../middleware/utils";
-// Convert `dimensions` (OpenAI) to `outputDimensionality` (Google)
-export const geminiDimensionsMiddleware = {
-    specificationVersion: "v3",
-    // eslint-disable-next-line require-await
-    transformParams: async ({ params }) => {
-        const unknown = params.providerOptions?.["unknown"];
-        if (!unknown)
-            return params;
-        const dimensions = unknown["dimensions"];
-        if (!dimensions)
-            return params;
-        (params.providerOptions["google"] ??= {})["outputDimensionality"] = dimensions;
-        delete unknown["dimensions"];
-        return params;
-    },
-};
-// https://ai.google.dev/gemini-api/docs/thinking#thinking-levels
-export function mapGeminiReasoningEffort(effort, modelId) {
-    if (modelId.includes("gemini-3.1-pro")) {
-        switch (effort) {
-            case "none":
-            case "minimal":
-            case "low":
-                return "low";
-            case "medium":
-                return "medium";
-            case "high":
-            case "xhigh":
-            case "max":
-                return "high";
-        }
-    }
-    if (modelId.includes("gemini-3-pro")) {
-        switch (effort) {
-            case "none":
-            case "minimal":
-            case "low":
-                return "low";
-            case "medium":
-            case "high":
-            case "xhigh":
-            case "max":
-                return "high";
-        }
-    }
-    if (modelId.includes("gemini-3-flash")) {
-        switch (effort) {
-            case "none":
-            case "minimal":
-                return "minimal";
-            case "low":
-                return "low";
-            case "medium":
-                return "medium";
-            case "high":
-            case "xhigh":
-            case "max":
-                return "high";
-        }
-    }
-    return effort;
-}
-export const GEMINI_DEFAULT_MAX_OUTPUT_TOKENS = 65536;
-export const GEMINI_2_5_PRO_MIN_THINKING_BUDGET = 128;
-export const geminiReasoningMiddleware = {
-    specificationVersion: "v3",
-    // eslint-disable-next-line require-await
-    transformParams: async ({ params, model }) => {
-        const unknown = params.providerOptions?.["unknown"];
-        if (!unknown)
-            return params;
-        const reasoning = unknown["reasoning"];
-        if (!reasoning)
-            return params;
-        const target = (params.providerOptions["google"] ??= {});
-        const modelId = model.modelId;
-        if (modelId.includes("gemini-2")) {
-            const is25Pro = modelId.includes("gemini-2.5-pro");
-            target["thinkingConfig"] = {
-                thinkingBudget: reasoning.max_tokens ??
-                    calculateReasoningBudgetFromEffort(reasoning.effort ?? "none", params.maxOutputTokens ?? GEMINI_DEFAULT_MAX_OUTPUT_TOKENS, is25Pro ? GEMINI_2_5_PRO_MIN_THINKING_BUDGET : 0),
-            };
-        }
-        else if (modelId.includes("gemini-3") && reasoning.effort) {
-            target["thinkingConfig"] = {
-                thinkingLevel: mapGeminiReasoningEffort(reasoning.effort, modelId),
-            };
-            // FUTURE: warn if model is gemini-3 and max_tokens (unsupported) was ignored
-        }
-        (target["thinkingConfig"] ??= {})["includeThoughts"] =
-            reasoning.enabled ? !reasoning.exclude : false;
-        delete unknown["reasoning"];
-        return params;
-    },
-};
-modelMiddlewareMatcher.useForModel("google/gemini-*embedding-*", {
-    embedding: [geminiDimensionsMiddleware],
-});
-modelMiddlewareMatcher.useForModel(["google/gemini-2*", "google/gemini-3*"], {
-    language: [geminiReasoningMiddleware],
-});