fix: prefer to use entitlement over featureVisibility

jakehwll · jakehwll · commit ff41882fd015 · 2025-11-17T02:47:33.000Z
diff --git a/site/src/pages/DeploymentSettingsPage/ObservabilitySettingsPage/ObservabilitySettingsPage.tsx b/site/src/pages/DeploymentSettingsPage/ObservabilitySettingsPage/ObservabilitySettingsPage.tsx
@@ -8,10 +8,7 @@ import { ObservabilitySettingsPageView } from "./ObservabilitySettingsPageView";
 const ObservabilitySettingsPage: FC = () => {
 	const { deploymentConfig } = useDeploymentConfig();
 	const { entitlements } = useDashboard();
-	const {
-		multiple_organizations: hasPremiumLicense,
-		aibridge: hasAIBridgeEnabled,
-	} = useFeatureVisibility();
+	const { multiple_organizations: hasPremiumLicense } = useFeatureVisibility();
 
 	return (
 		<>
@@ -20,8 +17,8 @@ const ObservabilitySettingsPage: FC = () => {
 			<ObservabilitySettingsPageView
 				options={deploymentConfig.options}
 				featureAuditLogEnabled={entitlements.features.audit_log.enabled}
+				featureAIBridgeEnabled={entitlements.features.aibridge.enabled}
 				isPremium={hasPremiumLicense}
-				isAIBridgeEnabled={hasAIBridgeEnabled}
 			/>
 		</>
 	);
diff --git a/site/src/pages/DeploymentSettingsPage/ObservabilitySettingsPage/ObservabilitySettingsPageView.tsx b/site/src/pages/DeploymentSettingsPage/ObservabilitySettingsPage/ObservabilitySettingsPageView.tsx
@@ -26,13 +26,18 @@ import OptionsTable from "../OptionsTable";
 type ObservabilitySettingsPageViewProps = {
 	options: SerpentOption[];
 	featureAuditLogEnabled: boolean;
+	featureAIBridgeEnabled: boolean;
 	isPremium: boolean;
-	isAIBridgeEnabled: boolean;
 };
 
 export const ObservabilitySettingsPageView: FC<
 	ObservabilitySettingsPageViewProps
-> = ({ options, featureAuditLogEnabled, isPremium, isAIBridgeEnabled }) => {
+> = ({
+	options,
+	featureAuditLogEnabled,
+	isPremium,
+	featureAIBridgeEnabled,
+}) => {
 	return (
 		<Stack direction="column" spacing={6}>
 			<div>
@@ -99,7 +104,7 @@ export const ObservabilitySettingsPageView: FC<
 				/>
 			</div>
 
-			{isAIBridgeEnabled && (
+			{featureAIBridgeEnabled && (
 				<div>
 					<SettingsHeader
 						actions={
