1

src/components/puzzle-agent/PuzzleAgentWorkspace.tsx

@@ -8,6 +8,12 @@ import type {
   SendPuzzleAgentMessageRequest,
 } from '../../../packages/shared/src/contracts/puzzleAgentSession';
 import { readPuzzleReferenceImageAsDataUrl } from '../../services/puzzleReferenceImage';
+import {
+  normalizePuzzleImageModel,
+  PUZZLE_IMAGE_MODEL_ORIGINAL,
+  type PuzzleImageModelId,
+} from './puzzleImageModelOptions';
+import { PuzzleImageModelPicker } from './PuzzleImageModelPicker';
 
 type PuzzleAgentWorkspaceProps = {
   session: PuzzleAgentSessionSnapshot | null;
@@ -27,6 +33,7 @@ type PuzzleFormState = {
   pictureDescription: string;
   referenceImageSrc: string;
   referenceImageLabel: string;
+  imageModel: PuzzleImageModelId;
 };
 
 const EMPTY_FORM_STATE: PuzzleFormState = {
@@ -35,6 +42,7 @@ const EMPTY_FORM_STATE: PuzzleFormState = {
   pictureDescription: '',
   referenceImageSrc: '',
   referenceImageLabel: '',
+  imageModel: PUZZLE_IMAGE_MODEL_ORIGINAL,
 };
 
 function resolveInitialFormState(
@@ -51,6 +59,7 @@ function resolveInitialFormState(
       referenceImageLabel: initialFormPayload?.referenceImageSrc
         ? '已选择参考图'
         : '',
+      imageModel: normalizePuzzleImageModel(initialFormPayload?.imageModel),
     };
   }
 
@@ -64,6 +73,7 @@ function resolveInitialFormState(
       referenceImageLabel: initialFormPayload.referenceImageSrc
         ? '已选择参考图'
         : '',
+      imageModel: normalizePuzzleImageModel(initialFormPayload.imageModel),
     };
   }
 
@@ -87,6 +97,7 @@ function resolveInitialFormState(
       session.draft?.summary || session.anchorPack.visualSubject.value || '',
     referenceImageSrc: '',
     referenceImageLabel: '',
+    imageModel: PUZZLE_IMAGE_MODEL_ORIGINAL,
   };
 }
 
@@ -151,9 +162,11 @@ export function PuzzleAgentWorkspace({
       workDescription,
       pictureDescription,
       referenceImageSrc: formState.referenceImageSrc || null,
+      imageModel: formState.imageModel,
     }),
     [
       formState.referenceImageSrc,
+      formState.imageModel,
       pictureDescription,
       workDescription,
       workTitle,
@@ -163,6 +176,7 @@ export function PuzzleAgentWorkspace({
     autosavePayload.workTitle,
     autosavePayload.workDescription,
     autosavePayload.pictureDescription,
+    autosavePayload.imageModel,
   ]);
   const lastAutosaveSignatureRef = useRef(autosaveSignature);
   const autosaveSessionIdRef = useRef(session?.sessionId ?? null);
@@ -240,6 +254,7 @@ export function PuzzleAgentWorkspace({
       workDescription,
       pictureDescription,
       referenceImageSrc: formState.referenceImageSrc || null,
+      imageModel: formState.imageModel,
     };
 
     if (!session && onCreateFromForm) {
@@ -254,6 +269,7 @@ export function PuzzleAgentWorkspace({
       workDescription,
       pictureDescription,
       referenceImageSrc: formState.referenceImageSrc || null,
+      imageModel: formState.imageModel,
       candidateCount: 1,
     });
   };
@@ -332,6 +348,16 @@ export function PuzzleAgentWorkspace({
                   className="w-full resize-none rounded-[1rem] border border-[var(--platform-subpanel-border)] bg-white/90 px-4 py-3 pb-16 text-sm leading-6 text-[var(--platform-text-strong)] outline-none"
                   aria-label="画面描述"
                 />
+                <PuzzleImageModelPicker
+                  value={formState.imageModel}
+                  disabled={isBusy}
+                  onChange={(imageModel) =>
+                    setFormState((current) => ({
+                      ...current,
+                      imageModel,
+                    }))
+                  }
+                />
                 <label
                   className={`absolute bottom-3 right-3 inline-flex h-10 w-10 cursor-pointer items-center justify-center rounded-full border border-amber-300/70 bg-white/96 text-amber-700 shadow-sm transition hover:bg-amber-50 ${isBusy ? 'cursor-not-allowed opacity-55' : ''}`}
                   title={