Preserve partial creation replies on stream failure

server-rs/crates/api-server/src/prompt/mod.rs

@@ -4,6 +4,7 @@ pub(crate) mod character_visual;
 pub(crate) mod puzzle;
 pub(crate) mod rpg;
 pub(crate) mod scene_background;
+pub(crate) mod square_hole;
 
 pub(crate) use rpg::agent_chat;
 pub(crate) use rpg::foundation_draft;

server-rs/crates/api-server/src/prompt/square_hole.rs

@@ -0,0 +1,164 @@
+use serde_json::{Value as JsonValue, json};
+use spacetime_client::{SquareHoleAgentMessageRecord, SquareHoleAgentSessionRecord};
+
+use crate::creation_agent_chat::render_quick_fill_extra_rules;
+
+/// 方洞挑战共创 Agent 的系统提示词。
+///
+/// 这里只定义模型职责与输出约束，具体的模型调用、解析和写库由方洞 Agent turn 负责。
+pub(crate) const SQUARE_HOLE_AGENT_SYSTEM_PROMPT: &str = r#"你是一个负责和百梦主共创“方洞挑战”竖屏玩法的中文创意策划。
+
+你要把用户灵感收束成一个反直觉形状分拣小游戏：玩家会本能把形状投入对应洞口，但真实规则可能让所有形状都优先进入方洞，形成类似参考视频“所有东西都进方洞”的喜剧反差。
+
+你必须同时输出：
+1. 一段直接发给用户的中文回复 replyText
+2. 当前进度 progressPercent
+3. 下一轮完整可用的 nextConfig
+
+硬约束：
+1. 只能输出 JSON，不能输出代码块或解释
+2. nextConfig 必须是完整对象，不能只输出 patch
+3. replyText 必须是自然中文，不能提“字段”“结构”“JSON”“后端”等内部词
+4. replyText 一次最多推进一个最关键问题
+5. 如果用户要求自动配置，就直接补齐可发布草稿需要的题材、反差规则、形状数量和难度，不要继续提问
+6. 默认核心反差优先使用“方洞万能”或“方洞优先”，但可以根据用户题材包装成更有记忆点的规则
+7. progressPercent 范围只能是 0 到 100
+8. shapeCount 只能是 6 到 24 的整数，difficulty 只能是 1 到 10 的整数
+"#;
+
+const SQUARE_HOLE_AGENT_OUTPUT_CONTRACT: &str = r#"请严格按以下 JSON 输出，不要输出其他文字：
+{
+  "replyText": "",
+  "progressPercent": 0,
+  "nextConfig": {
+    "themeText": "",
+    "twistRule": "",
+    "shapeCount": 12,
+    "difficulty": 4
+  }
+}"#;
+
+pub(crate) const SQUARE_HOLE_AGENT_JSON_TURN_USER_PROMPT: &str = "请按约定输出这一轮的 JSON。";
+
+/// 方洞挑战草稿生成对话提示词脚本。
+///
+/// 方洞首版只需要四个可写回 SpacetimeDB 的配置项，因此提示词直接围绕配置收束，
+/// 不在模型输出层引入额外锚点，避免和当前持久化 schema 产生漂移。
+pub(crate) fn build_square_hole_agent_prompt(
+    session: &SquareHoleAgentSessionRecord,
+    quick_fill_requested: bool,
+) -> String {
+    let quick_fill_rules = if quick_fill_requested {
+        format!(
+            "\n\n{}",
+            render_quick_fill_extra_rules(
+                "当前方洞挑战方向里的题材、反差规则、形状数量和难度",
+                "不要要求用户再提供洞口、形状、演出或难度信息",
+                "输出完整 nextConfig，直接补齐空缺或仍过于泛化的项",
+                "生成结果页",
+            )
+        )
+    } else {
+        String::new()
+    };
+    format!(
+        "模板目标：收束成可试玩、可发布的方洞挑战玩法草稿。{quick_fill_rules}\n\n当前是第 {turn} 轮，当前进度 {progress}% 。\n\n是否要求自动配置：{quick_fill_requested_text}\n\n当前配置：\n{current_config}\n\n最近聊天记录：\n{chat_history}\n\n收束要求：\n1. themeText 描述本局的玩具、道具或场景题材，保持短句。\n2. twistRule 描述真实判定规则，优先体现方洞优先或类似反直觉逻辑。\n3. shapeCount 决定单局形状数量，移动端短局建议 8 到 16。\n4. difficulty 决定误导强度和节奏，建议 3 到 7。\n5. 用户给出明确方向时优先吸收并推进，不要机械问完四个问题。\n\n{contract}",
+        quick_fill_rules = quick_fill_rules,
+        turn = session.current_turn.saturating_add(1),
+        progress = session.progress_percent,
+        quick_fill_requested_text = if quick_fill_requested { "是" } else { "否" },
+        current_config = serialize_square_hole_session_config(session),
+        chat_history =
+            serde_json::to_string_pretty(&build_chat_history(session.messages.as_slice()))
+                .unwrap_or_else(|_| "[]".to_string()),
+        contract = SQUARE_HOLE_AGENT_OUTPUT_CONTRACT,
+    )
+}
+
+fn serialize_square_hole_session_config(session: &SquareHoleAgentSessionRecord) -> String {
+    serde_json::to_string_pretty(&json!({
+        "themeText": session.config.theme_text,
+        "twistRule": session.config.twist_rule,
+        "shapeCount": session.config.shape_count,
+        "difficulty": session.config.difficulty,
+    }))
+    .unwrap_or_else(|_| "{}".to_string())
+}
+
+fn build_chat_history(messages: &[SquareHoleAgentMessageRecord]) -> Vec<JsonValue> {
+    messages
+        .iter()
+        .map(|message| {
+            json!({
+                "role": message.role,
+                "kind": message.kind,
+                "content": message.text,
+            })
+        })
+        .collect()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::build_square_hole_agent_prompt;
+
+    fn message(role: &str, text: &str) -> spacetime_client::SquareHoleAgentMessageRecord {
+        spacetime_client::SquareHoleAgentMessageRecord {
+            id: format!("message-{role}"),
+            role: role.to_string(),
+            kind: "chat".to_string(),
+            text: text.to_string(),
+            created_at: "2026-05-04T10:00:00.000Z".to_string(),
+        }
+    }
+
+    fn session_record() -> spacetime_client::SquareHoleAgentSessionRecord {
+        spacetime_client::SquareHoleAgentSessionRecord {
+            session_id: "square-hole-session-test".to_string(),
+            current_turn: 1,
+            progress_percent: 25,
+            stage: "collecting_config".to_string(),
+            anchor_pack: spacetime_client::SquareHoleAnchorPackRecord {
+                theme: anchor("theme", "题材主题", "积木纸箱"),
+                twist_rule: anchor("twistRule", "反差规则", ""),
+                shape_count: anchor("shapeCount", "形状数量", "12"),
+                difficulty: anchor("difficulty", "难度", "4"),
+            },
+            config: spacetime_client::SquareHoleCreatorConfigRecord {
+                theme_text: "积木纸箱".to_string(),
+                twist_rule: "方洞万能".to_string(),
+                shape_count: 12,
+                difficulty: 4,
+            },
+            draft: None,
+            messages: vec![message("user", "做成办公室文具版")],
+            last_assistant_reply: Some("这次可以从办公室文具题材开始。".to_string()),
+            published_profile_id: None,
+            updated_at: "2026-05-04T10:00:00.000Z".to_string(),
+        }
+    }
+
+    fn anchor(key: &str, label: &str, value: &str) -> spacetime_client::SquareHoleAnchorItemRecord {
+        spacetime_client::SquareHoleAnchorItemRecord {
+            key: key.to_string(),
+            label: label.to_string(),
+            value: value.to_string(),
+            status: if value.is_empty() {
+                "missing"
+            } else {
+                "confirmed"
+            }
+            .to_string(),
+        }
+    }
+
+    #[test]
+    fn quick_fill_prompt_requires_complete_config() {
+        let prompt = build_square_hole_agent_prompt(&session_record(), true);
+
+        assert!(prompt.contains("用户刚刚主动要求你自动补充剩余关键字"));
+        assert!(prompt.contains("不要再继续提问"));
+        assert!(prompt.contains("nextConfig"));
+        assert!(prompt.contains("progressPercent 直接输出为 100"));
+    }
+}