1

packages/shared/src/contracts/puzzleAgentActions.ts

@@ -46,6 +46,7 @@ export type PuzzleAgentActionRequest =
       workTitle?: string;
       workDescription?: string;
       pictureDescription?: string;
+      imageModel?: string | null;
     }
   | {
       action: 'compile_puzzle_draft';
@@ -54,6 +55,7 @@ export type PuzzleAgentActionRequest =
       workDescription?: string;
       pictureDescription?: string;
       referenceImageSrc?: string | null;
+      imageModel?: string | null;
       candidateCount?: number;
     }
   | {
@@ -61,6 +63,7 @@ export type PuzzleAgentActionRequest =
       levelId?: string | null;
       promptText?: string | null;
       referenceImageSrc?: string | null;
+      imageModel?: string | null;
       candidateCount?: number;
       levelsJson?: string;
     }

packages/shared/src/contracts/puzzleAgentSession.ts

@@ -1,4 +1,7 @@
-import type { PuzzleAgentActionResponse, PuzzleAgentSuggestedAction } from './puzzleAgentActions';
+import type {
+  PuzzleAgentActionResponse,
+  PuzzleAgentSuggestedAction,
+} from './puzzleAgentActions';
 import type { PuzzleAnchorPack, PuzzleResultDraft } from './puzzleAgentDraft';
 import type { PuzzleResultPreviewEnvelope } from './puzzleResultPreview';
 
@@ -47,6 +50,7 @@ export interface CreatePuzzleAgentSessionRequest {
   workDescription?: string;
   pictureDescription?: string;
   referenceImageSrc?: string | null;
+  imageModel?: string | null;
 }
 
 export interface CreatePuzzleAgentSessionResponse {
@@ -59,6 +63,7 @@ export interface SendPuzzleAgentMessageRequest {
   quickFillRequested?: boolean;
 }
 
-export interface SendPuzzleAgentMessageResponse extends PuzzleAgentActionResponse {
+export interface SendPuzzleAgentMessageResponse
+  extends PuzzleAgentActionResponse {
   session: PuzzleAgentSessionSnapshot;
 }