Fix ai status missing on CYS transitional page

plugins/woocommerce-admin/client/customize-store/index.tsx

@@ -459,14 +459,36 @@ export const customizeStoreStateMachineDefinition = createMachine( {
 					invoke: {
 						src: 'fetchSurveyCompletedOption',
 						onError: {
-							target: 'transitional', // leave it as initialised default on error
+							target: 'preCheckAiStatus', // leave it as initialised default on error
 						},
 						onDone: {
-							target: 'transitional',
+							target: 'preCheckAiStatus',
 							actions: [ 'assignHasCompleteSurvey' ],
 						},
 					},
 				},
+				preCheckAiStatus: {
+					always: [
+						{
+							cond: 'isWooExpress',
+							target: 'checkAiStatus',
+						},
+						{ cond: 'isNotWooExpress', target: 'transitional' },
+					],
+				},
+				checkAiStatus: {
+					invoke: {
+						src: 'fetchAiStatus',
+						onDone: {
+							actions: 'assignAiStatus',
+							target: 'transitional',
+						},
+						onError: {
+							actions: 'assignAiOffline',
+							target: 'transitional',
+						},
+					},
+				},
 				transitional: {
 					entry: [
 						{ type: 'updateQueryStep', step: 'transitional' },

plugins/woocommerce/changelog/44121-fix-ai-status-missing-on-cys-transitional

@@ -0,0 +1,4 @@
+Significance: minor
+Type: fix
+
+Fix the missing AI status when reloading the CYS transitional page.