// Vercel Ai Gateway provider module implements model/runtime integration.
import type { ModelProviderConfig } from "openclaw/plugin-sdk/provider-model-shared";
import {
  discoverVercelAiGatewayModels,
  getStaticVercelAiGatewayModelCatalog,
  resolveVercelAiGatewayDynamicModel,
  VERCEL_AI_GATEWAY_BASE_URL,
  VERCEL_AI_GATEWAY_PROVIDER_ID,
} from "./models.js";
import { resolveVercelAiGatewayThinkingProfile } from "./thinking.js";

const VERCEL_AI_GATEWAY_IMAGE_MODEL_IDS = new Set([
  "openai/gpt-5.5",
  "openai/gpt-5.5-pro",
  "openai/gpt-5.4",
  "openai/gpt-5.4-pro",
  "openai/gpt-5.4-mini",
  "openai/gpt-5.4-nano",
  "openai/gpt-5.3-codex",
  "openai/gpt-5.3-codex-spark",
  "openai/gpt-5.2",
  "openai/gpt-5.2-codex",
  "openai/gpt-5.1-codex",
]);

export function resolveVercelAiGatewayModel(modelId: string) {
  const model = resolveVercelAiGatewayDynamicModel(modelId);
  const input: Array<"text" | "image"> = model.input.includes("image")
    ? ["text", "image"]
    : VERCEL_AI_GATEWAY_IMAGE_MODEL_IDS.has(modelId) ||
        /^anthropic\/claude-(?:opus|sonnet|haiku)-/.test(modelId)
      ? ["text", "image"]
      : ["text"];
  return {
    ...model,
    reasoning: model.reasoning || Boolean(resolveVercelAiGatewayThinkingProfile(modelId)),
    input,
    api: "anthropic-messages" as const,
    provider: VERCEL_AI_GATEWAY_PROVIDER_ID,
    baseUrl: VERCEL_AI_GATEWAY_BASE_URL,
  };
}

export function buildStaticVercelAiGatewayProvider(): ModelProviderConfig {
  return {
    baseUrl: VERCEL_AI_GATEWAY_BASE_URL,
    api: "anthropic-messages",
    models: getStaticVercelAiGatewayModelCatalog(),
  };
}

export async function buildVercelAiGatewayProvider(): Promise<ModelProviderConfig> {
  return {
    baseUrl: VERCEL_AI_GATEWAY_BASE_URL,
    api: "anthropic-messages",
    models: await discoverVercelAiGatewayModels(),
  };
}