diff --git a/app/api/canvas/generate/image/route.ts b/app/api/canvas/generate/image/route.ts
index 7e881774b..040e5f2ee 100644
--- a/app/api/canvas/generate/image/route.ts
+++ b/app/api/canvas/generate/image/route.ts
@@ -1,6 +1,6 @@
 import { NextResponse } from "next/server";
 import { auth } from "@/app/(auth)/auth";
-import { getEnabledImageModels, getProviderById } from "@/lib/db/queries";
+import { getEnabledAiModelsWithProvider, getAiProviderById } from "@/lib/db/queries";
 import { createGeminiImage } from "@/lib/ai/providers/gemini-image";
 import { createAlibabaImage } from "@/lib/ai/providers/alibaba-image";
 
@@ -35,8 +35,9 @@ export async function POST(request: Request) {
       );
     }
 
-    // 获取可用的图片生成模型
-    const imageModels = await getEnabledImageModels();
+    // 获取可用的图片生成模型（筛选 supportsImageGen 的模型）
+    const allModels = await getEnabledAiModelsWithProvider();
+    const imageModels = allModels.filter(m => m.model.supportsImageGen);
     
     if (!imageModels || imageModels.length === 0) {
       return NextResponse.json(
@@ -54,7 +55,7 @@ export async function POST(request: Request) {
       selectedModel = imageModels[0];
     }
 
-    const provider = await getProviderById(selectedModel.model.providerId);
+    const provider = await getAiProviderById({ id: selectedModel.model.providerId });
     if (!provider) {
       return NextResponse.json(
         { success: false, error: "模型提供商配置错误" },