fix: use correct function names for AI model queries
Some checks failed
Lint / build (20) (push) Has been cancelled
Playwright Tests / test (push) Has been cancelled

This commit is contained in:
Your Name 2026-01-24 20:23:41 +08:00
parent c7ffcda87b
commit 1a9f453a20

View File

@ -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: "模型提供商配置错误" },