From bbdda81686ff9f0daf85897703188eb897962030 Mon Sep 17 00:00:00 2001 From: Your Name Date: Sat, 7 Mar 2026 16:25:48 +0800 Subject: [PATCH] feat: add midjourney provider and failover exclusion --- .../settings/admin-settings-client.tsx | 27 +- app/(chat)/api/chat/route.ts | 168 ++++--- app/api/admin/ai-models/route.ts | 2 + app/api/admin/ai-providers/route.ts | 4 + components/chat.tsx | 5 +- components/image-generation-loading.tsx | 1 + components/image-params-bar.tsx | 2 +- components/messages.tsx | 12 +- components/multimodal-input.tsx | 3 +- lib/ai/dynamic-provider.ts | 14 + lib/ai/provider-defaults.ts | 1 + lib/ai/providers/midjourney.ts | 459 ++++++++++++++++++ lib/canvas/generate-image.ts | 12 +- lib/db/migrate.ts | 7 +- lib/db/queries.ts | 10 +- lib/db/schema.ts | 3 +- 16 files changed, 659 insertions(+), 71 deletions(-) create mode 100644 lib/ai/providers/midjourney.ts diff --git a/app/(admin)/settings/admin-settings-client.tsx b/app/(admin)/settings/admin-settings-client.tsx index dde186d26..db207a13b 100644 --- a/app/(admin)/settings/admin-settings-client.tsx +++ b/app/(admin)/settings/admin-settings-client.tsx @@ -147,7 +147,7 @@ type ToolConfig = { type AiProvider = { id: string; name: string; - type: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "fal-video" | "zhipu" | "tripo-3d" | "volcano-multimodal"; + type: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "midjourney" | "fal-video" | "zhipu" | "tripo-3d" | "volcano-multimodal"; baseUrl: string; apiKey: string; isEnabled: boolean; @@ -184,6 +184,9 @@ type AiModel = { isHidden: boolean; isDefault: boolean; sortOrder: number; + fallbackModelId1?: string; + fallbackModelId2?: string; + excludeFromFailover?: boolean; isAgentModel?: boolean; agentCodingModelId?: string; isMultimodal?: boolean; @@ -438,6 +441,7 @@ const PROVIDER_TYPES = [ { value: "ruxa-image", label: "RUXA生图" }, { value: "fal-image", label: "FAL 生图模型" }, { value: "openai-image", label: "OpenAI 图片兼容" }, + { value: "midjourney", label: "Midjourney" }, { value: "tripo-3d", label: "Tripo3D 模型生成" }, { value: "zhipu", label: "智普官方 4.7" }, { value: "volcano-multimodal", label: "火山引擎多模态" }, @@ -461,6 +465,7 @@ const PROVIDER_DEFAULT_BASE_URL: Record = { "ruxa-image": "https://api.ruxa.ai", "fal-image": "https://fal.run", "openai-image": "https://api.openai.com/v1", + midjourney: "https://your-midjourney-api.example.com", "fal-video": "https://fal.run", "tripo-3d": "https://api.tripo3d.ai/v2/openapi", zhipu: "https://open.bigmodel.cn/api/paas/v4", @@ -1327,7 +1332,7 @@ export function AdminSettingsClient() { const [providerForm, setProviderForm] = useState({ id: "", name: "", - type: "openai" as "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "fal-video" | "zhipu" | "tripo-3d" | "volcano-multimodal", + type: "openai" as "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "midjourney" | "fal-video" | "zhipu" | "tripo-3d" | "volcano-multimodal", baseUrl: "", apiKey: "", isEnabled: true, @@ -1354,6 +1359,7 @@ export function AdminSettingsClient() { sortOrder: 0, fallbackModelId1: "", // 备用模型 ID 1 fallbackModelId2: "", // 备用模型 ID 2 + excludeFromFailover: false, // 🆕 图片生成模型特性配置 imageGenConfig: { maxImages: 2, @@ -2426,6 +2432,7 @@ export function AdminSettingsClient() { sortOrder: 0, fallbackModelId1: "", fallbackModelId2: "", + excludeFromFailover: false, imageGenConfig: { maxImages: 2, supportsSize: false, @@ -3697,6 +3704,7 @@ export function AdminSettingsClient() { "ruxa-image": "bg-emerald-500", "fal-image": "bg-sky-500", "openai-image": "bg-green-600", + "midjourney": "bg-fuchsia-600", "tripo-3d": "bg-teal-500", "openai": "bg-green-500", "zhipu": "bg-lime-500", @@ -6860,6 +6868,15 @@ export function AdminSettingsClient() { /> +
+ + setModelForm({ ...modelForm, excludeFromFailover: checked }) + } + /> + +
{/* 🆕 图片生成模型特性配置 */} {(modelForm.supportsImageGen || isVideoProviderForModelConfig) && ( @@ -7096,6 +7113,7 @@ export function AdminSettingsClient() { {model.isAgentModel && Agent} {model.isMultimodal && 多模态} {model.isHidden && 前台隐藏} + {(model as any).excludeFromFailover && 不参与故障迁移}
ID: {model.modelId} | 提供商: {provider?.name ?? "未知"} @@ -7122,6 +7140,7 @@ export function AdminSettingsClient() { sortOrder: model.sortOrder || 0, fallbackModelId1: (model as any).fallbackModelId1 || "", fallbackModelId2: (model as any).fallbackModelId2 || "", + excludeFromFailover: (model as any).excludeFromFailover || false, imageGenConfig: (model as any).imageGenConfig || { maxImages: 2, supportsSize: false, @@ -8728,7 +8747,7 @@ export function AdminSettingsClient() { const provider = (providersData?.providers || []).find((p: AiProvider) => p.id === m.providerId); if (!provider) return false; // 支持阿里生图、Gemini、既梦等图片模型 - const imageProviderTypes = ["alibaba-image", "ruxa-image", "fal-image", "openai-image", "gemini-image", "gemini", "volcano-vision"]; + const imageProviderTypes = ["alibaba-image", "ruxa-image", "fal-image", "openai-image", "midjourney", "gemini-image", "gemini", "volcano-vision"]; return imageProviderTypes.includes(provider.type); }) .map((m: AiModel) => { @@ -8743,7 +8762,7 @@ export function AdminSettingsClient() { {/* 如果没有找到生图模型，显示提示 */} {(modelsData?.models || []).filter((m: AiModel) => { const provider = (providersData?.providers || []).find((p: AiProvider) => p.id === m.providerId); - const imageProviderTypes = ["alibaba-image", "ruxa-image", "fal-image", "openai-image", "gemini-image", "gemini", "volcano-vision"]; + const imageProviderTypes = ["alibaba-image", "ruxa-image", "fal-image", "openai-image", "midjourney", "gemini-image", "gemini", "volcano-vision"]; return provider && imageProviderTypes.includes(provider.type); }).length === 0 && ( diff --git a/app/(chat)/api/chat/route.ts b/app/(chat)/api/chat/route.ts index 2161501e5..41d72d010 100644 --- a/app/(chat)/api/chat/route.ts +++ b/app/(chat)/api/chat/route.ts @@ -280,6 +280,7 @@ function normalizeImageLoadingProviderType(providerType?: string | null): string normalized === "ruxa-image" || normalized === "fal-image" || normalized === "openai-image" || + normalized === "midjourney" || normalized === "volcano-vision" || normalized === "gemini" ) { @@ -974,6 +975,56 @@ function setImageFallbackModelCooldown(modelId: string, reason: string): void { }); } +function isImageFailoverExcludedModel(model: { excludeFromFailover?: boolean } | null | undefined) { + return Boolean(model?.excludeFromFailover); +} + +function buildImageFallbackModelIds( + allModels: Array<{ model: any; provider: any }>, + options: { + currentProviderId?: string; + currentModelId?: string; + currentModelKey?: string; + configuredFallbackIds?: string[]; + } +) { + const { currentProviderId, currentModelId, currentModelKey, configuredFallbackIds = [] } = options; + + const eligibleFallbackModelIds = new Set( + allModels + .filter( + (item) => + item.model.supportsImageGen && + item.model.isEnabled && + !item.model.isSubModel && + !isImageFailoverExcludedModel(item.model) && + item.model.modelId !== currentModelKey && + item.model.id !== currentModelId + ) + .map((item) => String(item.model.modelId || "")) + .filter(Boolean) + ); + + const sameProviderFallbacks = currentProviderId + ? allModels + .filter( + (item) => + item.provider.id === currentProviderId && + item.model.supportsImageGen && + item.model.isEnabled && + !item.model.isSubModel && + !isImageFailoverExcludedModel(item.model) && + item.model.modelId !== currentModelKey && + item.model.id !== currentModelId + ) + .map((item) => item.model.modelId) + : []; + + return Array.from(new Set([...configuredFallbackIds, ...sameProviderFallbacks])).filter( + (modelId) => Boolean(modelId) && eligibleFallbackModelIds.has(String(modelId)) + ); +} + function getImageCrossProviderCandidates( allModels: any[], options: { @@ -984,7 +1035,12 @@ function getImageCrossProviderCandidates( ) { return allModels .filter((m) => { - if (!m?.model?.supportsImageGen || !m?.model?.isEnabled || m?.model?.isSubModel) { + if ( + !m?.model?.supportsImageGen || + !m?.model?.isEnabled || + m?.model?.isSubModel || + isImageFailoverExcludedModel(m?.model) + ) { return false; } if (!IMAGE_FAILOVER_PROVIDER_ALLOWLIST.has(String(m?.provider?.type || ""))) { @@ -1438,7 +1494,12 @@ function ensureImageAnalysisSummarySection(message: string, summary: string): st * 因为这些 provider 内部会自行处理图片上传到 S3 */ async function convertMessagesWithImages(messages: ChatMessage[], providerType: string) { - if (providerType === "ruxa-image" || providerType === "fal-image" || providerType === "openai-image") { + if ( + providerType === "ruxa-image" || + providerType === "fal-image" || + providerType === "openai-image" || + providerType === "midjourney" + ) { const convertedMessages = []; for (const message of messages) { @@ -2407,6 +2468,8 @@ export async function POST(request: Request) { let shouldKeepSelectedModel = false; const requestedModelId = selectedChatModel; let requestedModelProviderType: string | undefined; + let requestedModelIsFixedVisual = false; + let requestedModelIsConversationMultimodal = false; if (!shouldRunAgentConversation && requestedModelId !== "auto") { const selectedModelIsUUID = @@ -2444,6 +2507,7 @@ export async function POST(request: Request) { "ruxa-image", "fal-image", "openai-image", + "midjourney", "volcano-vision", ]); const isFixedVisualModel = @@ -2460,6 +2524,8 @@ export async function POST(request: Request) { selectedModelConfig.supportsVision && !selectedModelConfig.supportsImageGen && !isVideoProviderType(selectedModelProviderType); + requestedModelIsFixedVisual = isFixedVisualModel; + requestedModelIsConversationMultimodal = isConversationMultimodalModel; if (deepThinkingOn) { if (isAiModelAgentAuto(selectedModelConfig)) { @@ -2755,7 +2821,13 @@ export async function POST(request: Request) { console.log("[Smart Auto Mode] 🎛️ Agent recommended image params:", smartAutoRecommendedImageParams); } - if (autoGraphResult.route && !shouldKeepSelectedModel) { + const shouldForceConversationalRouteForVisualModel = + shouldKeepSelectedModel && + requestedModelIsFixedVisual && + !requestedModelIsConversationMultimodal && + Boolean(smartAutoExecution?.shouldStayConversational); + + if (autoGraphResult.route && (!shouldKeepSelectedModel || shouldForceConversationalRouteForVisualModel)) { const routeIsVideo = isVideoProviderType(autoGraphResult.route.providerType); if (routeIsVideo && !allowAgentVideoGeneration) { const fallbackIntent: "image_edit" | "image_gen" = hasImage ? "image_edit" : "image_gen"; @@ -2808,7 +2880,9 @@ export async function POST(request: Request) { } else { finalModelId = autoGraphResult.route.id; console.log( - `[Smart Auto Mode] ✅ Routed to: ${autoGraphResult.route.displayName} (${autoGraphResult.route.reason})`, + shouldForceConversationalRouteForVisualModel + ? `[Smart Auto Mode] ✅ Switched visual model to conversational route: ${autoGraphResult.route.displayName} (${autoGraphResult.route.reason})` + : `[Smart Auto Mode] ✅ Routed to: ${autoGraphResult.route.displayName} (${autoGraphResult.route.reason})`, ); } } else if (autoGraphResult.route && shouldKeepSelectedModel) { @@ -4102,6 +4176,7 @@ export async function POST(request: Request) { (languageModel as any).provider === "ruxa-image" || (languageModel as any).provider === "fal-image" || (languageModel as any).provider === "openai-image" || + (languageModel as any).provider === "midjourney" || (languageModel as any).provider === "huanke-gemini"; const isUuidModelId = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test( @@ -4218,18 +4293,12 @@ export async function POST(request: Request) { currentModelConfig?.fallbackModelId2, ].filter(Boolean) as string[]; const allModels = await getEnabledAiModelsWithProvider(); - const sameProviderFallbacks = currentModelConfig - ? allModels - .filter(m => - m.provider.id === currentModelConfig.providerId && - m.model.supportsImageGen && - m.model.isEnabled && - !m.model.isSubModel && - m.model.modelId !== actualModelId - ) - .map(m => m.model.modelId) - : []; - const fallbackModelIds = Array.from(new Set([...configuredFallbacks, ...sameProviderFallbacks])); + const fallbackModelIds = buildImageFallbackModelIds(allModels, { + currentProviderId: currentModelConfig?.providerId, + currentModelId: currentModelConfig?.id, + currentModelKey: actualModelId, + configuredFallbackIds: configuredFallbacks, + }); console.log("[Chat API] Fallback model IDs:", fallbackModelIds); const geminiStream = createUIMessageStream({ @@ -4649,7 +4718,7 @@ export async function POST(request: Request) { // ========== Gemini 处理结束 ========== // ========== 阿里云千问图片生成独立处理 ========== - const isAlibabaLikeImageProvider = ["alibaba-image", "ruxa-image", "fal-image", "openai-image"].includes((languageModel as any).provider); + const isAlibabaLikeImageProvider = ["alibaba-image", "ruxa-image", "fal-image", "openai-image", "midjourney"].includes((languageModel as any).provider); if (isAlibabaLikeImageProvider && typeof (languageModel as any).doGenerate === "function") { console.log("[Chat API] Using image generation provider:", (languageModel as any).provider); console.log("[Chat API] Image params:", effectiveImageParams); @@ -4662,18 +4731,12 @@ export async function POST(request: Request) { alibabaModelConfig?.fallbackModelId2, ].filter(Boolean) as string[]; const allModels = await getEnabledAiModelsWithProvider(); - const sameProviderFallbacks = alibabaModelConfig - ? allModels - .filter(m => - m.provider.id === alibabaModelConfig.providerId && - m.model.supportsImageGen && - m.model.isEnabled && - !m.model.isSubModel && - m.model.modelId !== actualModelId - ) - .map(m => m.model.modelId) - : []; - const fallbackModelIds = Array.from(new Set([...configuredFallbacks, ...sameProviderFallbacks])); + const fallbackModelIds = buildImageFallbackModelIds(allModels, { + currentProviderId: alibabaModelConfig?.providerId, + currentModelId: alibabaModelConfig?.id, + currentModelKey: actualModelId, + configuredFallbackIds: configuredFallbacks, + }); console.log("[Chat API] Alibaba fallback model IDs:", fallbackModelIds); const alibabaStream = createUIMessageStream({ @@ -4741,6 +4804,8 @@ export async function POST(request: Request) { ? "fal-image" : (languageModel as any).provider === "openai-image" ? "openai-image" + : (languageModel as any).provider === "midjourney" + ? "midjourney" : "alibaba-image"; const modelMessages = await convertMessagesWithImages(messagesForAlibaba, providerForConvert); @@ -4841,6 +4906,8 @@ export async function POST(request: Request) { ? "fal-image" : candidate.provider.type === "openai-image" ? "openai-image" + : candidate.provider.type === "midjourney" + ? "midjourney" : candidate.provider.type === "gemini" ? "gemini" : candidate.provider.type === "huanke-gemini" @@ -5060,18 +5127,12 @@ export async function POST(request: Request) { volcanoModelConfig?.fallbackModelId2, ].filter(Boolean) as string[]; const allModels = await getEnabledAiModelsWithProvider(); - const sameProviderFallbacks = volcanoModelConfig - ? allModels - .filter(m => - m.provider.id === volcanoModelConfig.providerId && - m.model.supportsImageGen && - m.model.isEnabled && - !m.model.isSubModel && - m.model.modelId !== actualModelId - ) - .map(m => m.model.modelId) - : []; - const fallbackModelIds = Array.from(new Set([...configuredFallbacks, ...sameProviderFallbacks])); + const fallbackModelIds = buildImageFallbackModelIds(allModels, { + currentProviderId: volcanoModelConfig?.providerId, + currentModelId: volcanoModelConfig?.id, + currentModelKey: actualModelId, + configuredFallbackIds: configuredFallbacks, + }); console.log("[Chat API] Volcano fallback model IDs:", fallbackModelIds); if (hasAnnotation) { @@ -5230,6 +5291,8 @@ export async function POST(request: Request) { ? "fal-image" : candidate.provider.type === "openai-image" ? "openai-image" + : candidate.provider.type === "midjourney" + ? "midjourney" : candidate.provider.type === "gemini" ? "gemini" : candidate.provider.type === "huanke-gemini" @@ -5754,21 +5817,12 @@ export async function POST(request: Request) { currentModelConfig?.fallbackModelId2, ].filter(Boolean) as string[]; const allModels = await getEnabledAiModelsWithProvider(); - const sameProviderFallbacks = currentModelConfig - ? allModels - .filter( - (m) => - m.provider.id === currentModelConfig.providerId && - m.model.supportsImageGen && - m.model.isEnabled && - !m.model.isSubModel && - m.model.modelId !== actualModelId - ) - .map((m) => m.model.modelId) - : []; - const fallbackModelIds = Array.from( - new Set([...configuredFallbacks, ...sameProviderFallbacks]) - ).filter((modelId) => modelId !== actualModelId); + const fallbackModelIds = buildImageFallbackModelIds(allModels, { + currentProviderId: currentModelConfig?.providerId, + currentModelId: currentModelConfig?.id, + currentModelKey: actualModelId, + configuredFallbackIds: configuredFallbacks, + }); console.log("[Chat API] OpenAI-compatible fallback model IDs:", fallbackModelIds); const lastUserMessage = uiMessagesForModel.filter((m) => m.role === "user").pop(); @@ -5901,6 +5955,8 @@ export async function POST(request: Request) { ? "fal-image" : candidate.provider.type === "openai-image" ? "openai-image" + : candidate.provider.type === "midjourney" + ? "midjourney" : candidate.provider.type === "gemini" ? "gemini" : candidate.provider.type === "huanke-gemini" diff --git a/app/api/admin/ai-models/route.ts b/app/api/admin/ai-models/route.ts index ebeb2f508..381160bf8 100644 --- a/app/api/admin/ai-models/route.ts +++ b/app/api/admin/ai-models/route.ts @@ -47,6 +47,7 @@ const createModelSchema = z.object({ sortOrder: z.number().optional(), fallbackModelId1: z.string().max(128).optional(), fallbackModelId2: z.string().max(128).optional(), + excludeFromFailover: z.boolean().optional(), isAgentModel: z.boolean().optional(), agentCodingModelId: z.string().max(128).optional(), isMultimodal: z.boolean().optional(), @@ -68,6 +69,7 @@ const updateModelSchema = z.object({ sortOrder: z.number().optional(), fallbackModelId1: z.string().max(128).optional(), fallbackModelId2: z.string().max(128).optional(), + excludeFromFailover: z.boolean().optional(), isAgentModel: z.boolean().optional(), agentCodingModelId: z.string().max(128).optional(), isMultimodal: z.boolean().optional(), diff --git a/app/api/admin/ai-providers/route.ts b/app/api/admin/ai-providers/route.ts index eb578e475..9bd7efc23 100644 --- a/app/api/admin/ai-providers/route.ts +++ b/app/api/admin/ai-providers/route.ts @@ -25,6 +25,7 @@ const providerTypes = [ "ruxa-image", "fal-image", "openai-image", + "midjourney", "fal-video", "zhipu", "tripo-3d", @@ -56,6 +57,7 @@ const DOUBAO_VIDEO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3"; const RUXA_VIDEO_BASE_URL = "https://api.ruxa.ai"; const RUXA_IMAGE_BASE_URL = "https://api.ruxa.ai"; const OPENAI_IMAGE_BASE_URL = "https://api.openai.com/v1"; +const MIDJOURNEY_BASE_URL = "https://your-midjourney-api.example.com"; const TRIPO_3D_BASE_URL = "https://api.tripo3d.ai/v2/openapi"; function normalizeProviderTypeForStorage( @@ -169,6 +171,8 @@ export async function POST(request: Request) { ? data.baseUrl || RUXA_IMAGE_BASE_URL : data.type === "openai-image" ? data.baseUrl || OPENAI_IMAGE_BASE_URL + : data.type === "midjourney" + ? data.baseUrl || MIDJOURNEY_BASE_URL : data.type === "tripo-3d" ? data.baseUrl || TRIPO_3D_BASE_URL : data.baseUrl, diff --git a/components/chat.tsx b/components/chat.tsx index e4b4084d5..ef94165b1 100644 --- a/components/chat.tsx +++ b/components/chat.tsx @@ -49,7 +49,7 @@ function normalizeImageParamsForProvider( const normalized = { ...params }; const currentCount = normalized.imageCount || 1; - if (providerType === "ruxa-image" || providerType === "openai-image") { + if (providerType === "ruxa-image" || providerType === "openai-image" || providerType === "midjourney") { normalized.imageCount = 1; return normalized; } @@ -1053,7 +1053,8 @@ export function Chat({ model?.providerType === "alibaba-image" || model?.providerType === "ruxa-image" || model?.providerType === "fal-image" || - model?.providerType === "openai-image" + model?.providerType === "openai-image" || + model?.providerType === "midjourney" ) modelType = "alibaba-image"; // 智能自动模式下，如果当前轮输入包含图片，默认按图片生成链路跟踪 diff --git a/components/image-generation-loading.tsx b/components/image-generation-loading.tsx index 7de89e293..91361df6b 100644 --- a/components/image-generation-loading.tsx +++ b/components/image-generation-loading.tsx @@ -21,6 +21,7 @@ const PROVIDER_MODEL_LABEL: Record = { "ruxa-image": "RUXA 图片", "fal-image": "FAL 图片", "openai-image": "OpenAI 图片", + midjourney: "Midjourney", "volcano-vision": "即梦 4.5", gemini: "Gemini 图片", }; diff --git a/components/image-params-bar.tsx b/components/image-params-bar.tsx index e5851c2e4..4630898fe 100644 --- a/components/image-params-bar.tsx +++ b/components/image-params-bar.tsx @@ -87,7 +87,7 @@ export function ImageParamsBar({ }: ImageParamsBarProps) { const getImageCountOptions = () => { // RUXA 当前按单图任务处理（优先级最高） - if (providerType === "ruxa-image" || providerType === "openai-image") { + if (providerType === "ruxa-image" || providerType === "openai-image" || providerType === "midjourney") { return [1]; } // 优先使用模型配置中的 maxImages diff --git a/components/messages.tsx b/components/messages.tsx index 29d9cdb8b..8efa5f9e0 100644 --- a/components/messages.tsx +++ b/components/messages.tsx @@ -124,7 +124,8 @@ function PureMessages({ (m.providerType === "alibaba-image" || m.providerType === "ruxa-image" || m.providerType === "fal-image" || - m.providerType === "openai-image") + m.providerType === "openai-image" || + m.providerType === "midjourney") ) ?? false; // 🆕 获取当前模型的显示名称 @@ -683,7 +684,12 @@ function PureMessages({ return ( ; } diff --git a/lib/ai/dynamic-provider.ts b/lib/ai/dynamic-provider.ts index 9db853cbf..947687631 100644 --- a/lib/ai/dynamic-provider.ts +++ b/lib/ai/dynamic-provider.ts @@ -21,6 +21,7 @@ import { createAlibabaImage } from "./providers/alibaba-image"; import { createRuxaImage } from "./providers/ruxa-image"; import { createFalImage } from "./providers/fal-image"; import { createOpenAIImage } from "./providers/openai-image"; +import { createMidjourney } from "./providers/midjourney"; import { createFalVideo } from "./providers/fal-video"; import { createRuxaVideo } from "./providers/ruxa-video"; import { createVolcanoMultimodal } from "./providers/volcano-multimodal"; @@ -217,6 +218,15 @@ function getOrCreateProvider(provider: Pick> = { "ruxa-image": "https://api.ruxa.ai", "fal-image": "https://fal.run", "openai-image": "https://api.openai.com/v1", + midjourney: "https://your-midjourney-api.example.com", "fal-video": "https://fal.run", "tripo-3d": "https://api.tripo3d.ai/v2/openapi", }; diff --git a/lib/ai/providers/midjourney.ts b/lib/ai/providers/midjourney.ts new file mode 100644 index 000000000..951564853 --- /dev/null +++ b/lib/ai/providers/midjourney.ts @@ -0,0 +1,459 @@ +import "server-only"; + +import sharp from "sharp"; +import { uploadFile } from "@/lib/storage/service"; + +interface MidjourneyConfig { + apiKey: string; + baseURL: string; +} + +export interface MidjourneyImageParams { + aspectRatio?: string; + imageSize?: string; + imageCount?: number; +} + +interface UploadedImage { + thumbnail: string; + original: string; +} + +interface MidjourneyTaskButton { + customId?: string; + label?: string; + emoji?: string; + style?: number; + type?: number; +} + +interface MidjourneyTaskResult { + id?: string; + imageUrl?: string; + status?: string; + progress?: string; + failReason?: string; + description?: string; + prompt?: string; + buttons?: MidjourneyTaskButton[]; +} + +const DEFAULT_TIMEOUT_MS = 300000; +const POLL_INTERVAL_MS = 3500; +const DEFAULT_IMAGE_SIZE = "2K"; + +function normalizeBaseURL(baseURL: string) { + return String(baseURL || "").trim().replace(/\/+$/, ""); +} + +function resolveMode(baseURL: string, modelId: string) { + const normalizedBaseURL = normalizeBaseURL(baseURL).toLowerCase(); + const normalizedModelId = String(modelId || "").trim().toLowerCase(); + + if ( + normalizedBaseURL.endsWith("/relax") || + normalizedBaseURL.endsWith("/mj-relax") || + normalizedModelId.includes("relax") + ) { + return "relax" as const; + } + + return "fast" as const; +} + +function buildEndpointCandidates(baseURL: string, mode: "fast" | "relax", suffix: string) { + const normalizedBaseURL = normalizeBaseURL(baseURL); + const normalizedSuffix = suffix.startsWith("/") ? suffix : `/${suffix}`; + const directPattern = /\/(fast|relax|mj-fast|mj-relax)$/i; + + if (directPattern.test(normalizedBaseURL)) { + return [`${normalizedBaseURL}${normalizedSuffix}`]; + } + + const modePrefix = mode === "relax" ? "relax" : "fast"; + const altPrefix = mode === "relax" ? "mj-relax" : "mj-fast"; + + return [ + `${normalizedBaseURL}/${modePrefix}${normalizedSuffix}`, + `${normalizedBaseURL}/${altPrefix}${normalizedSuffix}`, + ]; +} + +function normalizeImageSizeLabel(imageSize?: string): "1K" | "2K" | "4K" { + const normalized = String(imageSize || "").trim().toUpperCase(); + if (normalized === "1K" || normalized === "2K" || normalized === "4K") return normalized; + return DEFAULT_IMAGE_SIZE; +} + +function normalizeAspectRatio(value?: string) { + const normalized = String(value || "").trim(); + if (!normalized || normalized.toLowerCase() === "auto") return undefined; + if (/^\d+:\d+$/.test(normalized)) return normalized; + return undefined; +} + +function appendAspectRatio(prompt: string, aspectRatio?: string) { + const normalizedPrompt = String(prompt || "").trim(); + const normalizedAspectRatio = normalizeAspectRatio(aspectRatio); + if (!normalizedAspectRatio) return normalizedPrompt; + if (/--ar\s+\d+:\d+/i.test(normalizedPrompt)) return normalizedPrompt; + return `${normalizedPrompt} --ar ${normalizedAspectRatio}`.trim(); +} + +function detectImageExtension(imageUrl: string, contentType?: string | null) { + const type = String(contentType || "").toLowerCase(); + if (type.includes("jpeg") || type.includes("jpg")) return "jpg"; + if (type.includes("webp")) return "webp"; + if (type.includes("png")) return "png"; + if (type.includes("gif")) return "gif"; + + const cleanUrl = imageUrl.split("?")[0]?.toLowerCase() || ""; + if (cleanUrl.endsWith(".jpg") || cleanUrl.endsWith(".jpeg")) return "jpg"; + if (cleanUrl.endsWith(".webp")) return "webp"; + if (cleanUrl.endsWith(".gif")) return "gif"; + if (cleanUrl.endsWith(".png")) return "png"; + return "png"; +} + +async function createThumbnail(buffer: Buffer) { + try { + return await sharp(buffer) + .resize(400, 400, { fit: "inside", withoutEnlargement: true }) + .jpeg({ quality: 80 }) + .toBuffer(); + } catch (error) { + console.error("[Midjourney] Failed to create thumbnail:", error); + return buffer; + } +} + +function generateFilename(prefix: string, extension: string) { + const timestamp = Date.now(); + const random = Math.random().toString(36).slice(2, 8); + return `${prefix}-${timestamp}-${random}.${extension}`; +} + +async function uploadGeneratedBuffer( + buffer: Buffer, + extension: string, + mimeType: string +): Promise { + const thumbnailBuffer = await createThumbnail(buffer); + + const originalFilename = generateFilename("midjourney-orig", extension); + const originalArrayBuffer = buffer.buffer.slice( + buffer.byteOffset, + buffer.byteOffset + buffer.byteLength + ) as ArrayBuffer; + const originalUploaded = await uploadFile(originalFilename, originalArrayBuffer); + + const thumbFilename = generateFilename("midjourney-thumb", "jpg"); + const thumbArrayBuffer = thumbnailBuffer.buffer.slice( + thumbnailBuffer.byteOffset, + thumbnailBuffer.byteOffset + thumbnailBuffer.byteLength + ) as ArrayBuffer; + const thumbUploaded = await uploadFile(thumbFilename, thumbArrayBuffer); + + return { + thumbnail: thumbUploaded.url, + original: originalUploaded.url, + }; +} + +async function downloadAndUploadImage(imageUrl: string): Promise { + const response = await fetch(imageUrl, { cache: "no-store" }); + if (!response.ok) { + throw new Error(`Failed to download Midjourney image: ${response.status}`); + } + const buffer = Buffer.from(await response.arrayBuffer()); + const extension = detectImageExtension(imageUrl, response.headers.get("content-type")); + const mimeType = + response.headers.get("content-type") || + `image/${extension === "jpg" ? "jpeg" : extension}`; + return uploadGeneratedBuffer(buffer, extension, mimeType); +} + +function extractPromptAndImages(messages: Array<{ role: string; content: any }>) { + const lastUserMessage = messages.filter((message) => message.role === "user").pop(); + if (!lastUserMessage) { + throw new Error("No user message found"); + } + + let prompt = ""; + const imageUrls: string[] = []; + + const pushImageUrl = (value?: string) => { + if (!value) return; + const normalized = String(value).trim(); + if (!normalized) return; + if (!imageUrls.includes(normalized)) { + imageUrls.push(normalized); + } + }; + + if (typeof lastUserMessage.content === "string") { + prompt = lastUserMessage.content.trim(); + } else if (Array.isArray(lastUserMessage.content)) { + for (const part of lastUserMessage.content) { + if (part?.type === "text" && typeof part.text === "string") { + prompt = `${prompt} ${part.text}`.trim(); + } else if (part?.type === "image_url" && part.image_url?.url) { + pushImageUrl(part.image_url.url); + } else if (part?.type === "file") { + pushImageUrl(part.url || part.data); + } else if (part?.type === "image") { + pushImageUrl(part.image || part.url); + } + } + } + + if (!prompt) { + throw new Error("No prompt found in user message"); + } + + return { prompt, imageUrls }; +} + +async function loadInputImageAsDataUrl(source: string) { + const normalized = String(source || "").trim(); + if (!normalized) { + throw new Error("Midjourney 输入图片地址为空"); + } + + if (normalized.startsWith("data:")) { + return normalized; + } + + const response = await fetch(normalized, { cache: "no-store" }); + if (!response.ok) { + throw new Error(`Failed to fetch Midjourney input image: ${response.status}`); + } + + const mimeType = response.headers.get("content-type") || "image/png"; + const buffer = Buffer.from(await response.arrayBuffer()); + return `data:${mimeType};base64,${buffer.toString("base64")}`; +} + +async function parseJsonResponse(response: Response) { + const rawText = await response.text(); + try { + return JSON.parse(rawText); + } catch { + return rawText; + } +} + +function extractErrorMessage(payload: any) { + if (!payload) return "Midjourney 请求失败"; + if (typeof payload === "string") return payload; + if (payload.failReason) return String(payload.failReason); + if (payload.description) return String(payload.description); + if (payload.message) return String(payload.message); + if (payload.error?.message) return String(payload.error.message); + try { + return JSON.stringify(payload); + } catch { + return "Midjourney 请求失败"; + } +} + +async function requestWithEndpointFallback(params: { + baseURL: string; + mode: "fast" | "relax"; + suffix: string; + method: "GET" | "POST"; + apiKey: string; + body?: string; + contentType?: string; +}) { + const { baseURL, mode, suffix, method, apiKey, body, contentType } = params; + const endpoints = buildEndpointCandidates(baseURL, mode, suffix); + let lastPayload: any = null; + let lastStatus: number | null = null; + let lastEndpoint = endpoints[0]; + + for (const endpoint of endpoints) { + lastEndpoint = endpoint; + const response = await fetch(endpoint, { + method, + headers: { + Authorization: `Bearer ${apiKey}`, + ...(contentType ? { "Content-Type": contentType } : {}), + }, + ...(body ? { body } : {}), + signal: AbortSignal.timeout(45000), + }); + const payload = await parseJsonResponse(response); + lastPayload = payload; + lastStatus = response.status; + + if (response.ok || response.status !== 404) { + return { endpoint, response, payload }; + } + } + + throw new Error( + `Midjourney request failed: ${extractErrorMessage(lastPayload)} (${lastStatus || "no_status"}) @ ${lastEndpoint}` + ); +} + +async function submitImagineTask(params: { + baseURL: string; + apiKey: string; + mode: "fast" | "relax"; + prompt: string; + imageUrls: string[]; +}) { + const { baseURL, apiKey, mode, prompt, imageUrls } = params; + const base64Array = await Promise.all(imageUrls.map((imageUrl) => loadInputImageAsDataUrl(imageUrl))); + + const { payload } = await requestWithEndpointFallback({ + baseURL, + mode, + suffix: "/mj/submit/imagine", + method: "POST", + apiKey, + body: JSON.stringify({ + base64Array, + prompt, + }), + contentType: "application/json", + }); + + const code = Number((payload as any)?.code ?? 0); + const taskId = String((payload as any)?.result || "").trim(); + if (![1, 22].includes(code) || !taskId) { + throw new Error(extractErrorMessage(payload)); + } + + return taskId; +} + +async function fetchMidjourneyTask(params: { + baseURL: string; + apiKey: string; + mode: "fast" | "relax"; + taskId: string; +}) { + const { payload } = await requestWithEndpointFallback({ + baseURL: params.baseURL, + mode: params.mode, + suffix: `/mj/task/${params.taskId}/fetch`, + method: "GET", + apiKey: params.apiKey, + }); + + return (payload || {}) as MidjourneyTaskResult; +} + +async function waitForMidjourneyTask(params: { + baseURL: string; + apiKey: string; + mode: "fast" | "relax"; + taskId: string; +}) { + const deadlineAt = Date.now() + DEFAULT_TIMEOUT_MS; + let lastTask: MidjourneyTaskResult | null = null; + + while (Date.now() < deadlineAt) { + const task = await fetchMidjourneyTask(params); + lastTask = task; + + const status = String(task.status || "").toUpperCase(); + if (status === "SUCCESS" && task.imageUrl) { + return task; + } + if (status === "FAILURE") { + throw new Error(task.failReason || task.description || "Midjourney 任务失败"); + } + + await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS)); + } + + throw new Error( + `Midjourney task timeout: ${params.taskId} (${lastTask?.status || "unknown"})` + ); +} + +function formatMarkdown(image: UploadedImage, imageSize: string) { + return `![thumb|||${image.original}|||${imageSize}](${image.thumbnail})`; +} + +export function createMidjourney({ apiKey, baseURL }: MidjourneyConfig) { + return { + provider: "midjourney", + chat: (modelId: string) => ({ + provider: "midjourney", + doGenerate: async (options: { + messages: Array<{ role: string; content: any }>; + imageParams?: MidjourneyImageParams; + overrideModelId?: string; + }) => { + const { prompt, imageUrls } = extractPromptAndImages(options.messages); + const activeModelId = options.overrideModelId || modelId; + const mode = resolveMode(baseURL, activeModelId); + const imageSize = normalizeImageSizeLabel(options.imageParams?.imageSize); + const finalPrompt = appendAspectRatio(prompt, options.imageParams?.aspectRatio); + + console.log("[Midjourney] Submitting request", { + model: activeModelId, + mode, + imageCount: imageUrls.length, + promptPreview: finalPrompt.slice(0, 120), + }); + + const taskId = await submitImagineTask({ + baseURL, + apiKey, + mode, + prompt: finalPrompt, + imageUrls, + }); + + console.log("[Midjourney] Task submitted", { taskId, mode }); + const taskResult = await waitForMidjourneyTask({ + baseURL, + apiKey, + mode, + taskId, + }); + + if (!taskResult.imageUrl) { + throw new Error(taskResult.failReason || taskResult.description || "Midjourney 未返回图片地址"); + } + + const uploadedImage = await downloadAndUploadImage(taskResult.imageUrl); + const text = formatMarkdown(uploadedImage, imageSize); + + return { + text, + finishReason: "stop" as const, + usage: { + promptTokens: 0, + completionTokens: text.length, + }, + providerMetadata: { + midjourney: { + taskId, + mode, + progress: taskResult.progress || "", + buttons: taskResult.buttons || [], + imageUrl: taskResult.imageUrl, + }, + }, + }; + }, + doStream: async function* (options: { + messages: Array<{ role: string; content: any }>; + imageParams?: MidjourneyImageParams; + overrideModelId?: string; + }) { + const result = await this.doGenerate(options); + yield { + type: "text-delta", + textDelta: result.text, + }; + }, + }), + }; +} diff --git a/lib/canvas/generate-image.ts b/lib/canvas/generate-image.ts index 4b740f1d8..dbeda3222 100644 --- a/lib/canvas/generate-image.ts +++ b/lib/canvas/generate-image.ts @@ -6,6 +6,7 @@ import { createAlibabaImage } from "@/lib/ai/providers/alibaba-image"; import { createRuxaImage } from "@/lib/ai/providers/ruxa-image"; import { createFalImage } from "@/lib/ai/providers/fal-image"; import { createOpenAIImage } from "@/lib/ai/providers/openai-image"; +import { createMidjourney } from "@/lib/ai/providers/midjourney"; import { filterReachableRemoteUrls } from "@/lib/net/url-access"; export interface CanvasGenerateImageRequest { @@ -36,6 +37,7 @@ const SUPPORTED_CANVAS_PROVIDER_TYPES = new Set([ "ruxa-image", "fal-image", "openai-image", + "midjourney", ]); function parseImagesFromMarkdown(markdown: string): ParsedImage[] { @@ -234,9 +236,17 @@ export async function generateCanvasImage( const modelInstance = openAIImageProvider.chat(selectedModel.model.modelId); const result = await modelInstance.doGenerate({ messages, imageParams }); imageMarkdown = result.text || ""; + } else if (provider.type === "midjourney") { + const midjourneyProvider = createMidjourney({ + apiKey: provider.apiKey, + baseURL: provider.baseUrl, + }); + const modelInstance = midjourneyProvider.chat(selectedModel.model.modelId); + const result = await modelInstance.doGenerate({ messages, imageParams }); + imageMarkdown = result.text || ""; } else { throw new Error( - `不支持的图片生成模型类型: ${provider.type || "unknown"}，请改用 Gemini/阿里/RUXA/FAL/OpenAI 图片模型` + `不支持的图片生成模型类型: ${provider.type || "unknown"}，请改用 Gemini/阿里/RUXA/FAL/OpenAI/Midjourney 图片模型` ); } diff --git a/lib/db/migrate.ts b/lib/db/migrate.ts index dfbac3aa2..fc63c22f8 100644 --- a/lib/db/migrate.ts +++ b/lib/db/migrate.ts @@ -29,6 +29,7 @@ const providerBaseUrlDefaults = [ { type: "ruxa-image", baseUrl: "https://api.ruxa.ai" }, { type: "fal-image", baseUrl: "https://fal.run" }, { type: "openai-image", baseUrl: "https://api.openai.com/v1" }, + { type: "midjourney", baseUrl: "https://your-midjourney-api.example.com" }, { type: "fal-video", baseUrl: "https://fal.run" }, { type: "tripo-3d", baseUrl: "https://api.tripo3d.ai/v2/openapi" }, ] as const; @@ -90,9 +91,13 @@ const runMigrate = async () => { ADD CONSTRAINT "AiProvider_type_check" CHECK (type IN ( 'alibaba', 'openai', 'custom', 'local', 'gemini', - 'huanke-gemini', 'qianwen-video', 'doubao-video', 'ruxa-video', 'volcano-vision', 'volcano-video', 'alibaba-image', 'ruxa-image', 'fal-image', 'openai-image', 'fal-video', 'zhipu', 'tripo-3d' + 'huanke-gemini', 'qianwen-video', 'doubao-video', 'ruxa-video', 'volcano-vision', 'volcano-video', 'alibaba-image', 'ruxa-image', 'fal-image', 'openai-image', 'midjourney', 'fal-video', 'zhipu', 'tripo-3d' )); `); + await db.execute(sql` + ALTER TABLE "AiModel" + ADD COLUMN IF NOT EXISTS "excludeFromFailover" boolean NOT NULL DEFAULT false; + `); for (const item of providerBaseUrlDefaults) { await db.execute(sql` UPDATE "AiProvider" diff --git a/lib/db/queries.ts b/lib/db/queries.ts index 7d8c4ef49..fc7db5820 100644 --- a/lib/db/queries.ts +++ b/lib/db/queries.ts @@ -2005,7 +2005,7 @@ export async function createAiProvider({ updatedBy, }: { name: string; - type: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "fal-video" | "zhipu" | "tripo-3d"; + type: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "midjourney" | "fal-video" | "zhipu" | "tripo-3d"; baseUrl: string; apiKey: string; isEnabled?: boolean; @@ -2043,7 +2043,7 @@ export async function updateAiProvider({ }: { id: string; name?: string; - type?: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "fal-video" | "zhipu" | "tripo-3d"; + type?: "alibaba" | "openai" | "custom" | "local" | "gemini" | "huanke-gemini" | "qianwen-video" | "doubao-video" | "ruxa-video" | "volcano-vision" | "volcano-video" | "alibaba-image" | "ruxa-image" | "fal-image" | "openai-image" | "midjourney" | "fal-video" | "zhipu" | "tripo-3d"; baseUrl?: string; apiKey?: string; isEnabled?: boolean; @@ -2196,6 +2196,7 @@ export async function createAiModel({ sortOrder, fallbackModelId1, fallbackModelId2, + excludeFromFailover, imageGenConfig, isAgentModel, isMultimodal, @@ -2214,6 +2215,7 @@ export async function createAiModel({ sortOrder?: number; fallbackModelId1?: string; fallbackModelId2?: string; + excludeFromFailover?: boolean; imageGenConfig?: ImageGenConfig; isAgentModel?: boolean; isMultimodal?: boolean; @@ -2254,6 +2256,7 @@ export async function createAiModel({ sortOrder: sortOrder ?? 0, fallbackModelId1: fallbackModelId1 || null, fallbackModelId2: fallbackModelId2 || null, + excludeFromFailover: excludeFromFailover ?? false, imageGenConfig: imageGenConfig || null, defaultParams, capabilities: @@ -2285,6 +2288,7 @@ export async function updateAiModel({ sortOrder, fallbackModelId1, fallbackModelId2, + excludeFromFailover, imageGenConfig, isAgentModel, isMultimodal, @@ -2304,6 +2308,7 @@ export async function updateAiModel({ sortOrder?: number; fallbackModelId1?: string; fallbackModelId2?: string; + excludeFromFailover?: boolean; imageGenConfig?: ImageGenConfig; isAgentModel?: boolean; isMultimodal?: boolean; @@ -2334,6 +2339,7 @@ export async function updateAiModel({ // 备用模型 ID（空字符串转为 null） if (fallbackModelId1 !== undefined) updateData.fallbackModelId1 = fallbackModelId1 || null; if (fallbackModelId2 !== undefined) updateData.fallbackModelId2 = fallbackModelId2 || null; + if (excludeFromFailover !== undefined) updateData.excludeFromFailover = excludeFromFailover; // 🆕 图片生成配置 if (imageGenConfig !== undefined) updateData.imageGenConfig = imageGenConfig || null; if ( diff --git a/lib/db/schema.ts b/lib/db/schema.ts index dbcdcf37e..2475e0180 100644 --- a/lib/db/schema.ts +++ b/lib/db/schema.ts @@ -259,7 +259,7 @@ export type ToolConfig = InferSelectModel; export const aiProvider = pgTable("AiProvider", { id: uuid("id").primaryKey().notNull().defaultRandom(), name: varchar("name", { length: 128 }).notNull(), // 提供商名称，如 "阿里云通义千问" - type: varchar("type", { enum: ["alibaba", "openai", "custom", "local", "gemini", "huanke-gemini", "qianwen-video", "doubao-video", "ruxa-video", "volcano-vision", "volcano-video", "alibaba-image", "ruxa-image", "fal-image", "openai-image", "fal-video", "zhipu", "tripo-3d"] }) + type: varchar("type", { enum: ["alibaba", "openai", "custom", "local", "gemini", "huanke-gemini", "qianwen-video", "doubao-video", "ruxa-video", "volcano-vision", "volcano-video", "alibaba-image", "ruxa-image", "fal-image", "openai-image", "midjourney", "fal-video", "zhipu", "tripo-3d"] }) .notNull() .default("custom"), // 提供商类型 baseUrl: text("baseUrl").notNull(), // API 基础 URL @@ -296,6 +296,7 @@ export const aiModel = pgTable("AiModel", { // 🆕 备用模型 ID（当主模型请求失败时自动切换） fallbackModelId1: varchar("fallbackModelId1", { length: 128 }), // 备用模型 ID 1 fallbackModelId2: varchar("fallbackModelId2", { length: 128 }), // 备用模型 ID 2 + excludeFromFailover: boolean("excludeFromFailover").notNull().default(false), // 不参与故障迁移 // 🆕 图片生成模型特性配置（用于阿里生图、Gemini等） imageGenConfig: jsonb("imageGenConfig"), // { maxImages: 6, supportsSize: true, supportsPromptExtend: true, supportsNegativePrompt: true, enableLoadingPlaceholder: true } createdAt: timestamp("createdAt").notNull().defaultNow(),