1

packages/shared/src/contracts/puzzleAgentActions.ts

@@ -13,6 +13,7 @@ export interface PuzzleAgentSuggestedAction {
 }
 
 export type PuzzleAgentActionType =
+  | 'save_puzzle_form_draft'
   | 'compile_puzzle_draft'
   | 'generate_puzzle_images'
   | 'select_puzzle_image'
@@ -39,27 +40,42 @@ export interface PuzzleAgentOperationRecord {
 }
 
 export type PuzzleAgentActionRequest =
+  | {
+      action: 'save_puzzle_form_draft';
+      promptText?: string | null;
+      workTitle?: string;
+      workDescription?: string;
+      pictureDescription?: string;
+    }
   | {
       action: 'compile_puzzle_draft';
       promptText?: string | null;
+      workTitle?: string;
+      workDescription?: string;
+      pictureDescription?: string;
       referenceImageSrc?: string | null;
       candidateCount?: number;
     }
   | {
       action: 'generate_puzzle_images';
+      levelId?: string | null;
       promptText?: string | null;
       referenceImageSrc?: string | null;
       candidateCount?: number;
     }
   | {
       action: 'select_puzzle_image';
+      levelId?: string | null;
       candidateId: string;
     }
   | {
       action: 'publish_puzzle_work';
+      workTitle?: string;
+      workDescription?: string;
       levelName?: string;
       summary?: string;
       themeTags?: string[];
+      levelsJson?: string;
     };
 
 /**