fix: use correct function names for AI model queries
This commit is contained in:
parent
c7ffcda87b
commit
1a9f453a20
@ -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: "模型提供商配置错误" },
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user