diff --git a/CHANGELOG.md b/CHANGELOG.md
index 54f957e..3d63cd2 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,6 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.22.0-alpha] - unreleased
+
+This is an alpha version! The changes listed here are not final.
+
+### Changed
+- AI Client: add types for AI assistant feature payload data branch featuresControl
+
 ## [0.21.0] - 2024-10-14
 ### Added
 - AI Client: Add image styles 'auto' and 'none' to the logo generator. Order styles so those are on top in the dropdown selector. [#39689]
@@ -435,6 +442,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Updated package dependencies. [#31659]
 - Updated package dependencies. [#31785]
 
+[0.22.0-alpha]: https://github.com/Automattic/jetpack-ai-client/compare/v0.21.0...v0.22.0-alpha
 [0.21.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.20.1...v0.21.0
 [0.20.1]: https://github.com/Automattic/jetpack-ai-client/compare/v0.20.0...v0.20.1
 [0.20.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.19.0...v0.20.0
diff --git a/build/ai-client/src/logo-generator/store/initial-state.js b/build/ai-client/src/logo-generator/store/initial-state.js
index 045105f..d34c3af 100644
--- a/build/ai-client/src/logo-generator/store/initial-state.js
+++ b/build/ai-client/src/logo-generator/store/initial-state.js
@@ -32,6 +32,12 @@ const INITIAL_STATE = {
                 asyncRequestTimerId: 0,
                 isRequestingImage: false,
             },
+            featuresControl: {
+                'logo-generator': {
+                    enabled: false,
+                    styles: [],
+                },
+            },
         },
     },
     history: [],
diff --git a/build/ai-client/src/logo-generator/store/types.d.ts b/build/ai-client/src/logo-generator/store/types.d.ts
index cd1e762..2f4c11b 100644
--- a/build/ai-client/src/logo-generator/store/types.d.ts
+++ b/build/ai-client/src/logo-generator/store/types.d.ts
@@ -59,11 +59,9 @@ export type LogoGeneratorFeatureControl = FeatureControl & {
 };
 export type FeatureControl = {
     enabled: boolean;
-    'min-jetpack-version': string;
-    [key: string]: FeatureControl | LogoGeneratorFeatureControl | boolean | string;
 };
 export type FeaturesControl = {
-    [key: string]: FeatureControl;
+    [key: string]: FeatureControl | LogoGeneratorFeatureControl;
 };
 export type AiFeatureProps = {
     hasFeature: boolean;
diff --git a/package.json b/package.json
index 27eadaa..b29dc8e 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.21.0",
+	"version": "0.22.0-alpha",
 	"description": "A JS client for consuming Jetpack AI services",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/ai-client/#readme",
 	"bugs": {
diff --git a/src/logo-generator/store/initial-state.ts b/src/logo-generator/store/initial-state.ts
index 7b130dc..d07579e 100644
--- a/src/logo-generator/store/initial-state.ts
+++ b/src/logo-generator/store/initial-state.ts
@@ -34,6 +34,12 @@ const INITIAL_STATE: LogoGeneratorStateProp = {
 				asyncRequestTimerId: 0,
 				isRequestingImage: false,
 			},
+			featuresControl: {
+				'logo-generator': {
+					enabled: false,
+					styles: [],
+				},
+			},
 		},
 	},
 	history: [],
diff --git a/src/logo-generator/store/types.ts b/src/logo-generator/store/types.ts
index c4eb52f..5f45e66 100644
--- a/src/logo-generator/store/types.ts
+++ b/src/logo-generator/store/types.ts
@@ -95,11 +95,11 @@ export type LogoGeneratorFeatureControl = FeatureControl & {
 
 export type FeatureControl = {
 	enabled: boolean;
-	'min-jetpack-version': string;
-	[ key: string ]: FeatureControl | LogoGeneratorFeatureControl | boolean | string;
 };
 
-export type FeaturesControl = { [ key: string ]: FeatureControl };
+export type FeaturesControl = {
+	[ key: string ]: FeatureControl | LogoGeneratorFeatureControl;
+};
 
 export type AiFeatureProps = {
 	hasFeature: boolean;