1

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;
 }