Enable image generation continuation

codex-rs/features/src/lib.rs

@@ -192,6 +192,9 @@ pub enum Feature {
     ExternalMigration,
     /// Allow the model to invoke the built-in image generation tool.
     ImageGeneration,
+    /// Continue sampling after image generation so workflows can return text
+    /// and artifact metadata alongside the generated image.
+    ImageGenerationContinuation,
     /// Allow prompting and installing missing MCP dependencies.
     SkillMcpDependencyInstall,
     /// Prompt for missing skill env var dependencies.
@@ -1022,6 +1025,12 @@ pub const FEATURES: &[FeatureSpec] = &[
         stage: Stage::Stable,
         default_enabled: true,
     },
+    FeatureSpec {
+        id: Feature::ImageGenerationContinuation,
+        key: "image_generation_continuation",
+        stage: Stage::Stable,
+        default_enabled: true,
+    },
     FeatureSpec {
         id: Feature::SkillMcpDependencyInstall,
         key: "skill_mcp_dependency_install",

codex-rs/features/src/tests.rs

@@ -226,6 +226,12 @@ fn image_generation_is_stable_and_enabled_by_default() {
     assert_eq!(Feature::ImageGeneration.default_enabled(), true);
 }
 
+#[test]
+fn image_generation_continuation_is_stable_and_enabled_by_default() {
+    assert_eq!(Feature::ImageGenerationContinuation.stage(), Stage::Stable);
+    assert_eq!(Feature::ImageGenerationContinuation.default_enabled(), true);
+}
+
 #[test]
 fn use_legacy_landlock_config_records_deprecation_notice() {
     let mut entries = BTreeMap::new();