diff --git a/projects/plugins/jetpack/changelog/update-create-general-purpose-image-beta-flag b/projects/plugins/jetpack/changelog/update-create-general-purpose-image-beta-flag
new file mode 100644
index 0000000000000..db5a935d5ce4c
--- /dev/null
+++ b/projects/plugins/jetpack/changelog/update-create-general-purpose-image-beta-flag
@@ -0,0 +1,4 @@
+Significance: minor
+Type: other
+
+Jetpack AI: register ai-general-purpose-image-generator beta flag to control the changes we are doing to the image generation tool. 
diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-assistant.php b/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-assistant.php
index 972c1b7645daf..3553ee4e0bd87 100644
--- a/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-assistant.php
+++ b/projects/plugins/jetpack/extensions/blocks/ai-assistant/ai-assistant.php
@@ -173,3 +173,15 @@ function () {
 		}
 	}
 );
+
+/**
+ * Register the `ai-general-purpose-image-generator` extension.
+ */
+add_action(
+	'jetpack_register_gutenberg_extensions',
+	function () {
+		if ( apply_filters( 'jetpack_ai_enabled', true ) ) {
+			\Jetpack_Gutenberg::set_extension_available( 'ai-general-purpose-image-generator' );
+		}
+	}
+);
diff --git a/projects/plugins/jetpack/extensions/index.json b/projects/plugins/jetpack/extensions/index.json
index 2c11d20a724b1..614401ba1146f 100644
--- a/projects/plugins/jetpack/extensions/index.json
+++ b/projects/plugins/jetpack/extensions/index.json
@@ -78,7 +78,8 @@
 		"videopress/video-chapters",
 		"ai-assistant-backend-prompts",
 		"ai-logo-generator",
-		"ai-assistant-extensions-support"
+		"ai-assistant-extensions-support",
+		"ai-general-purpose-image-generator"
 	],
 	"experimental": [ "ai-image", "ai-paragraph" ],
 	"no-post-editor": [