1

server-rs/crates/shared-contracts/src/puzzle_agent.rs

@@ -6,6 +6,10 @@ pub struct CreatePuzzleAgentSessionRequest {
     #[serde(default)]
     pub seed_text: Option<String>,
     #[serde(default)]
+    pub work_title: Option<String>,
+    #[serde(default)]
+    pub work_description: Option<String>,
+    #[serde(default)]
     pub picture_description: Option<String>,
     #[serde(default)]
     pub reference_image_src: Option<String>,
@@ -33,11 +37,32 @@ pub struct ExecutePuzzleAgentActionRequest {
     #[serde(default)]
     pub candidate_id: Option<String>,
     #[serde(default)]
+    pub level_id: Option<String>,
+    #[serde(default)]
+    pub work_title: Option<String>,
+    #[serde(default)]
+    pub work_description: Option<String>,
+    #[serde(default)]
+    pub picture_description: Option<String>,
+    #[serde(default)]
     pub level_name: Option<String>,
     #[serde(default)]
     pub summary: Option<String>,
     #[serde(default)]
     pub theme_tags: Option<Vec<String>>,
+    #[serde(default)]
+    pub levels_json: Option<String>,
+}
+
+#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
+#[serde(rename_all = "camelCase")]
+pub struct PuzzleFormDraftResponse {
+    #[serde(default)]
+    pub work_title: Option<String>,
+    #[serde(default)]
+    pub work_description: Option<String>,
+    #[serde(default)]
+    pub picture_description: Option<String>,
 }
 
 #[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
@@ -88,6 +113,8 @@ pub struct PuzzleCreatorIntentResponse {
 #[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
 #[serde(rename_all = "camelCase")]
 pub struct PuzzleResultDraftResponse {
+    pub work_title: String,
+    pub work_description: String,
     pub level_name: String,
     pub summary: String,
     pub theme_tags: Vec<String>,
@@ -103,6 +130,25 @@ pub struct PuzzleResultDraftResponse {
     #[serde(default)]
     pub cover_asset_id: Option<String>,
     pub generation_status: String,
+    pub levels: Vec<PuzzleDraftLevelResponse>,
+    #[serde(default)]
+    pub form_draft: Option<PuzzleFormDraftResponse>,
+}
+
+#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
+#[serde(rename_all = "camelCase")]
+pub struct PuzzleDraftLevelResponse {
+    pub level_id: String,
+    pub level_name: String,
+    pub picture_description: String,
+    pub candidates: Vec<PuzzleGeneratedImageCandidateResponse>,
+    #[serde(default)]
+    pub selected_candidate_id: Option<String>,
+    #[serde(default)]
+    pub cover_image_src: Option<String>,
+    #[serde(default)]
+    pub cover_asset_id: Option<String>,
+    pub generation_status: String,
 }
 
 #[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
@@ -153,6 +199,7 @@ pub struct PuzzleResultPreviewEnvelopeResponse {
 #[serde(rename_all = "camelCase")]
 pub struct PuzzleAgentSessionSnapshotResponse {
     pub session_id: String,
+    pub seed_text: String,
     pub current_turn: u32,
     pub progress_percent: u32,
     pub stage: String,