From 24eb5a4367a8a288f7ff1d9609c3eb93a635a2e0 Mon Sep 17 00:00:00 2001
From: ishaanxgupta <124028055+ishaanxgupta@users.noreply.github.com>
Date: Wed, 17 Jun 2026 06:22:41 +0000
Subject: [PATCH] Fix Granola Max upgrade gating (#1095)

## Summary

- Keep the Granola integration card in Upgrade state unless the active plan includes Max or above.
- Prevent the Granola API-key modal from opening unless the active plan is Max or above in both the integrations grid and add-document connections UI.
---
 apps/web/components/add-document/connections.tsx | 4 ++--
 apps/web/components/integrations-view.tsx        | 9 +++++----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/apps/web/components/add-document/connections.tsx b/apps/web/components/add-document/connections.tsx
index f37af299..f32fd5b6 100644
--- a/apps/web/components/add-document/connections.tsx
+++ b/apps/web/components/add-document/connections.tsx
@@ -948,8 +948,8 @@ export function ConnectContent({ selectedProject }: ConnectContentProps) {
 			/>
 
 			<GranolaConnectModal
-				open={granolaModalOpen}
-				onOpenChange={setGranolaModalOpen}
+				open={isMaxUser && granolaModalOpen}
+				onOpenChange={(open) => setGranolaModalOpen(open && isMaxUser)}
 				containerTags={[selectedProject]}
 			/>
 		</div>
diff --git a/apps/web/components/integrations-view.tsx b/apps/web/components/integrations-view.tsx
index 62383466..3fd395e7 100644
--- a/apps/web/components/integrations-view.tsx
+++ b/apps/web/components/integrations-view.tsx
@@ -3066,8 +3066,9 @@ export function IntegrationsView({
 			case "connector": {
 				const count = connectionsByProvider[item.provider].length
 				const isGranola = item.provider === "granola"
-				const needsPlanUpgrade =
-					!isAutumnLoading && (isGranola ? !hasMaxProduct : !hasProProduct)
+				const needsPlanUpgrade = isGranola
+					? !hasMaxProduct
+					: !isAutumnLoading && !hasProProduct
 				if (count > 0) {
 					return (
 						<div className="flex w-full items-center justify-between gap-2">
@@ -3894,8 +3895,8 @@ export function IntegrationsView({
 			</Dialog>
 
 			<GranolaConnectModal
-				open={granolaModalOpen}
-				onOpenChange={setGranolaModalOpen}
+				open={hasMaxProduct && granolaModalOpen}
+				onOpenChange={(open) => setGranolaModalOpen(open && hasMaxProduct)}
 			/>
 		</div>
 	)