feat: add Flux (Black Forest Lab) provider support

Author: monkeyWie

src/app/i18n/locales/en.json

@@ -225,6 +225,36 @@
 					"description": "FLUX.1 Kontext [pro] handles both text and reference images as inputs, seamlessly enabling targeted, local edits and complex transformations of entire scenes."
 				}
 			}
+		},
+		"flux": {
+			"description": "High-quality AI image generation by Black Forest Labs",
+			"settings": {
+				"apiKey": {
+					"title": "API Key",
+					"description": "Your BFL API key for authentication",
+					"placeholder": "Enter your BFL API key"
+				}
+			},
+			"models": {
+				"flux-kontext-pro": {
+					"description": "FLUX.1 Kontext Pro - Advanced image-to-image generation with superior prompt adherence and editing capabilities."
+				},
+				"flux-kontext-max": {
+					"description": "FLUX.1 Kontext Max - Premium image-to-image generation with maximum quality and consistency for professional editing."
+				},
+				"flux-dev": {
+					"description": "FLUX.1 Dev - High-quality text-to-image generation model optimized for development and experimentation."
+				},
+				"flux-pro": {
+					"description": "FLUX.1 Pro - Professional-grade text-to-image generation with excellent prompt adherence and image quality."
+				},
+				"flux-pro-1.1": {
+					"description": "FLUX.1 Pro 1.1 - Enhanced version of FLUX Pro with improved performance and quality."
+				},
+				"flux-pro-1.1-ultra": {
+					"description": "FLUX.1 Pro 1.1 Ultra - The most advanced FLUX model with ultra-high quality generation and precision."
+				}
+			}
 		}
 	}
 }

src/app/i18n/locales/zh.json

@@ -225,6 +225,36 @@
 					"description": "FLUX.1 Kontext [pro] 将文本和参考图像作为输入进行处理，无缝地实现了对整个场景的有针对性的局部编辑和复杂转换。"
 				}
 			}
+		},
+		"flux": {
+			"description": "由Black Forest Labs提供的高质量AI图像生成",
+			"settings": {
+				"apiKey": {
+					"title": "API密钥",
+					"description": "用于身份验证的BFL API密钥",
+					"placeholder": "输入您的BFL API密钥"
+				}
+			},
+			"models": {
+				"flux-kontext-pro": {
+					"description": "FLUX.1 Kontext Pro - 高级图像到图像生成，具有卓越的提示遵循和编辑能力。"
+				},
+				"flux-kontext-max": {
+					"description": "FLUX.1 Kontext Max - 顶级图像到图像生成，为专业编辑提供最高质量和一致性。"
+				},
+				"flux-dev": {
+					"description": "FLUX.1 Dev - 为开发和实验优化的高质量文本到图像生成模型。"
+				},
+				"flux-pro": {
+					"description": "FLUX.1 Pro - 专业级文本到图像生成，具有出色的提示遵循和图像质量。"
+				},
+				"flux-pro-1.1": {
+					"description": "FLUX.1 Pro 1.1 - FLUX Pro的增强版本，提升了性能和质量。"
+				},
+				"flux-pro-1.1-ultra": {
+					"description": "FLUX.1 Pro 1.1 Ultra - 最先进的FLUX模型，具有超高质量生成和精度。"
+				}
+			}
 		}
 	}
 }

src/server/ai/provider/flux.ts

@@ -0,0 +1,158 @@
+import { fetchUrlToDataURI } from "@/server/lib/util";
+import type { AiProvider, ApiProviderSettings, ApiProviderSettingsItem } from "../types/provider";
+import { type ProviderSettingsType, chooseAblility, doParseSettings, findModel } from "../types/provider";
+
+const fluxSettingsSchema = [
+	{
+		key: "apiKey",
+		type: "password",
+		required: true,
+	},
+] as const satisfies ApiProviderSettingsItem[];
+
+// Automatically generate type from schema
+export type FluxSettings = ProviderSettingsType<typeof fluxSettingsSchema>;
+
+interface FluxSubmitResponse {
+	id: string;
+	polling_url: string;
+}
+
+interface FluxPollResponse {
+	id: string;
+	status: "Pending" | "Running" | "Ready" | "Error" | "Failed";
+	result?: {
+		sample: string;
+	};
+	error?: string;
+}
+
+const Flux: AiProvider = {
+	id: "flux",
+	name: "Flux",
+	supportCors: false,
+	enabledByDefault: true,
+	settings: fluxSettingsSchema,
+	models: [
+		{
+			id: "flux-kontext-max",
+			name: "FLUX.1 Kontext [max]",
+			ability: "i2i",
+			enabledByDefault: true,
+		},
+		{
+			id: "flux-kontext-pro",
+			name: "FLUX.1 Kontext [pro]",
+			ability: "i2i",
+			enabledByDefault: true,
+		},
+		{
+			id: "flux-pro-1.1-ultra",
+			name: "FLUX1.1 [pro] Ultra",
+			ability: "t2i",
+			enabledByDefault: true,
+		},
+		{
+			id: "flux-pro-1.1",
+			name: "FLUX1.1 [pro]",
+			ability: "t2i",
+			enabledByDefault: true,
+		},
+		{
+			id: "flux-pro",
+			name: "FLUX.1 [pro]",
+			ability: "t2i",
+			enabledByDefault: true,
+		},
+		{
+			id: "flux-dev",
+			name: "FLUX.1 [dev]",
+			ability: "t2i",
+			enabledByDefault: true,
+		},
+	],
+	parseSettings: <FluxSettings>(settings: ApiProviderSettings) => {
+		return doParseSettings(settings, fluxSettingsSchema) as FluxSettings;
+	},
+	generate: async (request, settings) => {
+		const { apiKey } = Flux.parseSettings<FluxSettings>(settings);
+
+		const model = findModel(Flux, request.modelId);
+		const genType = chooseAblility(request, model.ability);
+
+		const requestBody: any = {
+			prompt: request.prompt,
+		};
+		if (genType === "i2i" && request.images?.[0]) {
+			requestBody.image_url = request.images[0];
+		}
+
+		const submitResponse = await fetch(`https://api.bfl.ai/v1/${request.modelId}`, {
+			method: "POST",
+			headers: {
+				accept: "application/json",
+				"x-key": apiKey,
+				"Content-Type": "application/json",
+			},
+			body: JSON.stringify(requestBody),
+		});
+
+		if (!submitResponse.ok) {
+			if (submitResponse.status === 403) {
+				return {
+					errorReason: "CONFIG_ERROR",
+					images: [],
+				};
+			}
+			throw new Error(`Flux API error: ${submitResponse.status} ${submitResponse.statusText}`);
+		}
+
+		const submitData: FluxSubmitResponse = await submitResponse.json();
+		const { id: requestId, polling_url: pollingUrl } = submitData;
+
+		let attempts = 0;
+		const maxAttempts = 120;
+
+		while (attempts < maxAttempts) {
+			await new Promise((resolve) => setTimeout(resolve, 500));
+			attempts++;
+
+			const pollUrl = new URL(pollingUrl);
+			pollUrl.searchParams.set("id", requestId);
+
+			const pollResponse = await fetch(pollUrl.toString(), {
+				method: "GET",
+				headers: {
+					accept: "application/json",
+					"x-key": apiKey,
+				},
+			});
+
+			if (!pollResponse.ok) {
+				throw new Error(`Flux polling error: ${pollResponse.status} ${pollResponse.statusText}`);
+			}
+
+			const pollData: FluxPollResponse = await pollResponse.json();
+
+			if (pollData.status === "Ready" && pollData.result?.sample) {
+				try {
+					const imageDataUri = await fetchUrlToDataURI(pollData.result.sample);
+					return {
+						images: [imageDataUri],
+					};
+				} catch (error) {
+					console.error("Flux image fetch error:", error);
+					return {
+						images: [],
+					};
+				}
+			} else if (pollData.status === "Error" || pollData.status === "Failed") {
+				throw new Error(`Flux generation failed: ${pollData.error || "Unknown error"}`);
+			}
+		}
+
+		throw new Error("Flux generation timeout - exceeded maximum polling attempts");
+	},
+};
+
+export default Flux;

src/server/ai/provider/index.ts

@@ -4,9 +4,10 @@ import { ServiceException } from "@/server/lib/exception";
 import type { AiProvider } from "../types/provider";
 import { default as cloudflare } from "./cloudflare";
 import { default as fal } from "./fal";
+import { default as flux } from "./flux";
 import { default as openAI } from "./openai";
 
-export const AI_PROVIDERS = [cloudflare, openAI, fal].map(enhancedProvider);
+export const AI_PROVIDERS = [cloudflare, openAI, fal, flux].map(enhancedProvider);
 
 export function getDefaultProvider() {
 	return AI_PROVIDERS[0]!;