From f50b5ba7625795d35cd33979fe88f775ea6ea8a8 Mon Sep 17 00:00:00 2001 From: shrimbly Date: Wed, 18 Feb 2026 22:16:57 +1300 Subject: [PATCH] fix: type capabilitiesForMediaType return as ModelCapability[] The helper returned string[] which didn't satisfy the ModelCapability[] type in GenerationInput. Import and use the proper type. Co-Authored-By: Claude Opus 4.6 --- src/app/api/generate/route.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/app/api/generate/route.ts b/src/app/api/generate/route.ts index 8195f8f3..99659718 100644 --- a/src/app/api/generate/route.ts +++ b/src/app/api/generate/route.ts @@ -12,7 +12,7 @@ */ import { NextRequest, NextResponse } from "next/server"; import { GenerateRequest, GenerateResponse, ModelType, SelectedModel, ProviderType } from "@/types"; -import { GenerationInput } from "@/lib/providers/types"; +import { GenerationInput, ModelCapability } from "@/lib/providers/types"; import { generateWithGemini } from "./providers/gemini"; import { generateWithReplicate } from "./providers/replicate"; import { clearFalInputMappingCache as _clearFalInputMappingCache, generateWithFalQueue } from "./providers/fal"; @@ -73,8 +73,8 @@ function buildMediaResponse(output: { type: string; data: string; url?: string } }); } -function capabilitiesForMediaType(mediaType?: string): string[] { - const map: Record = { +function capabilitiesForMediaType(mediaType?: string): ModelCapability[] { + const map: Record = { audio: ["text-to-audio"], video: ["text-to-video"], "3d": ["text-to-3d"],