1

src/components/puzzle-agent/puzzleImageModelOptions.ts

@@ -0,0 +1,33 @@
+export const PUZZLE_IMAGE_MODEL_ORIGINAL = 'original';
+export const PUZZLE_IMAGE_MODEL_GPT_IMAGE_2 = 'gpt-image-2';
+export const PUZZLE_IMAGE_MODEL_NANOBANANA2 = 'gemini-3.1-flash-image-preview';
+
+export type PuzzleImageModelId =
+  | typeof PUZZLE_IMAGE_MODEL_ORIGINAL
+  | typeof PUZZLE_IMAGE_MODEL_GPT_IMAGE_2
+  | typeof PUZZLE_IMAGE_MODEL_NANOBANANA2;
+
+export const PUZZLE_IMAGE_MODEL_OPTIONS: Array<{
+  id: PuzzleImageModelId;
+  label: string;
+}> = [
+  { id: PUZZLE_IMAGE_MODEL_ORIGINAL, label: '原模型' },
+  { id: PUZZLE_IMAGE_MODEL_GPT_IMAGE_2, label: 'gpt-image-2' },
+  { id: PUZZLE_IMAGE_MODEL_NANOBANANA2, label: 'nanobanana2' },
+];
+
+export function normalizePuzzleImageModel(
+  value: string | null | undefined,
+): PuzzleImageModelId {
+  return (
+    PUZZLE_IMAGE_MODEL_OPTIONS.find((option) => option.id === value)?.id ??
+    PUZZLE_IMAGE_MODEL_ORIGINAL
+  );
+}
+
+export function getPuzzleImageModelLabel(model: PuzzleImageModelId) {
+  return (
+    PUZZLE_IMAGE_MODEL_OPTIONS.find((option) => option.id === model)?.label ??
+    '原模型'
+  );
+}