import type { AiProvider } from "@/lib/db/schema";

const PROVIDER_DEFAULT_BASE_URL: Partial<Record<AiProvider["type"], string>> = {
  alibaba: "https://dashscope.aliyuncs.com/compatible-mode/v1",
  openai: "https://api.openai.com/v1",
  "grsai-gpt-image": "https://grsai.dakka.com.cn",
  "grsai-video": "https://grsai.dakka.com.cn",
  "huanke-gemini": "https://api.huandutech.com/v1",
  "qianwen-video": "https://dashscope.aliyuncs.com/api/v1/services/aigc/video-generation",
  "doubao-video": "https://ark.cn-beijing.volces.com/api/v3",
  "ruxa-video": "https://api.ruxa.ai",
  "volcano-vision": "https://ark.cn-beijing.volces.com",
  "volcano-video": "https://ark.cn-beijing.volces.com/api/v3",
  "alibaba-image":
    "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation",
  "ruxa-image": "https://api.ruxa.ai",
  "fal-image": "https://fal.run",
  "snapgen-image": "https://api.snapgen.ai",
  "openai-image": "https://api.openai.com/v1",
  bananaimage: "https://bananaimage.aigenmedia.art",
  visionary: "https://api.visionary.beer",
  atlascloud: "https://api.atlascloud.ai",
  "stepfun-image": "https://api.stepfun.com",
  "bfl-image": "https://api.bfl.ai",
  midjourney: "https://your-midjourney-api.example.com",
  "fal-video": "https://fal.run",
  "tripo-3d": "https://api.tripo3d.ai/v2/openapi",
  "worldlabs-3d": "https://api.worldlabs.ai",
  "kittycad-cad": "https://api.zoo.dev",
  runway: "https://api.dev.runwayml.com",
  krea: "https://api.krea.ai",
};

export function getProviderDefaultBaseUrl(type: AiProvider["type"]): string | undefined {
  return PROVIDER_DEFAULT_BASE_URL[type];
}