Reimplement unified ai chat to use server side flag.

apps/help-center/config.js

@@ -5,7 +5,6 @@ window.configData = {
 	env_id: isProxied ? 'staging' : 'production',
 	features: {
 		'help/gpt-response': true,
-		'unified-agent': false, // Enable to use unified AI agent instead of HelpCenterGPT
 	},
 	wapuu: false,
 	i18n_default_locale_slug: 'en',

packages/agents-manager/src/global.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Global type declarations for the Agents Manager package.
+ */
+
+/**
+ * agentsManagerData is set as a global const via wp_add_inline_script
+ * in Jetpack's Agents Manager feature.
+ */
+declare const agentsManagerData:
+	| {
+			agentProviders?: string[];
+			useUnifiedExperience?: boolean;
+	  }
+	| undefined;

packages/agents-manager/src/index.ts

@@ -4,6 +4,9 @@ export type { UnifiedAIAgentProps } from './components/unified-ai-agent';
 
 export { AGENTS_MANAGER_STORE } from './stores';
 
+// Utility for checking unified experience from inline script data
+export { getUseUnifiedExperienceFromInlineData } from './utils/load-external-providers';
+
 // Extension API types for other plugins to hook into
 export type {
 	Ability,

packages/agents-manager/src/utils/load-external-providers.ts

@@ -9,8 +9,20 @@ import type { ToolProvider, ContextProvider, Suggestion } from '../types';
 import type { SubmitOptions } from '@automattic/agenttic-client';
 import type { MarkdownComponents, MarkdownExtensions } from '@automattic/agenttic-ui';
 
-// agentsManagerData is set as a global const via wp_add_inline_script
-declare const agentsManagerData: { agentProviders?: string[] } | undefined;
+/**
+ * Check if the unified experience flag is set via agentsManagerData.
+ *
+ * This is used on wp-admin environments (Atomic, Garden, Simple sites) where
+ * the flag is injected server-side by Jetpack's Agents Manager.
+ *
+ * @returns The useUnifiedExperience value, or undefined if not available.
+ */
+export function getUseUnifiedExperienceFromInlineData(): boolean | undefined {
+	if ( typeof agentsManagerData !== 'undefined' ) {
+		return agentsManagerData?.useUnifiedExperience;
+	}
+	return undefined;
+}
 
 /**
  * Navigation continuation hook type - provided by environments that support

packages/help-center/src/data/use-unified-ai-chat.ts

@@ -0,0 +1,53 @@
+import { getUseUnifiedExperienceFromInlineData } from '@automattic/agents-manager';
+import { useQuery } from '@tanstack/react-query';
+import wpcomRequest, { canAccessWpcomApis } from 'wpcom-proxy-request';
+
+interface CalypsoPreferencesResponse {
+	calypso_preferences?: {
+		unified_ai_chat?: boolean;
+	};
+}
+
+/**
+ * Determines if the user should see the unified AI chat experience.
+ *
+ * This hook uses a hybrid approach to work across all environments:
+ *
+ * 1. **wp-admin environments** (Atomic, Garden, Simple sites):
+ *    The flag is available via `agentsManagerData.useUnifiedExperience`,
+ *    injected server-side by Jetpack's Agents Manager.
+ *
+ * 2. **Calypso app** (wordpress.com):
+ *    The flag is fetched from the `/me/preferences` endpoint.
+ *
+ * The rollout logic lives in Agents Manager (Jetpack) via the
+ * `agents_manager_use_unified_experience` filter.
+ */
+export function useUnifiedAiChat( enabled = true ) {
+	return useQuery< boolean, Error >( {
+		queryKey: [ 'unified-ai-chat' ],
+		queryFn: async () => {
+			// 1. Check inline script first (available on wp-admin via Jetpack's Agents Manager)
+			const inlineValue = getUseUnifiedExperienceFromInlineData();
+			if ( inlineValue !== undefined ) {
+				return inlineValue;
+			}
+
+			// 2. Fall back to /me/preferences endpoint for Calypso app (wordpress.com)
+			if ( canAccessWpcomApis() ) {
+				const response: CalypsoPreferencesResponse = await wpcomRequest( {
+					path: '/me/preferences',
+					apiVersion: '1.1',
+				} );
+
+				return response.calypso_preferences?.unified_ai_chat ?? false;
+			}
+
+			// 3. No data available - default to false
+			return false;
+		},
+		enabled,
+		refetchOnWindowFocus: false,
+		staleTime: 300000, // 5 minutes
+	} );
+}

packages/help-center/src/hooks/use-should-use-unified-agent.ts

@@ -1,26 +1,8 @@
 /* eslint-disable no-restricted-imports */
-import config from '@automattic/calypso-config';
-import { useSupportStatus } from '../data/use-support-status';
-
-function isUnifiedAgentFlagSetInURL(): boolean {
-	const currentUrl = window.location.href;
-	const urlParams = new URLSearchParams( new URL( currentUrl ).search );
-	const flags = urlParams.get( 'flags' );
-	return flags?.split( ',' ).includes( 'unified-agent' ) ?? false;
-}
+import { useUnifiedAiChat } from '../data/use-unified-ai-chat';
 
 export const useShouldUseUnifiedAgent = () => {
-	const { data: supportStatus } = useSupportStatus();
-
-	// Check if user is eligible via support status API
-	// Note: This will need to be added to the backend support-status endpoint
-	const isEligibleViaAPI = Boolean( supportStatus?.eligibility?.unified_agent_enabled );
-
-	// Force unified agent via URL flag (for testing)
-	const isFlagSetInURL = isUnifiedAgentFlagSetInURL();
-
-	// Unified agent can be enabled via config
-	const isConfigEnabled = config.isEnabled( 'unified-agent' );
+	const { data: isEligibleViaAPI } = useUnifiedAiChat();
 
-	return isEligibleViaAPI || isFlagSetInURL || isConfigEnabled;
+	return isEligibleViaAPI;
 };