From 2e94944d2bedb99435373fa4b3a2e769d28d4199 Mon Sep 17 00:00:00 2001
From: Cats Juice <wp1298554944@hotmail.com>
Date: Wed, 20 Nov 2024 14:23:30 +0800
Subject: [PATCH] fix(core): ai subscribe button is not show in ai usage page
 (#8869)

---
 .../dialogs/setting/account-setting/ai-usage-panel.tsx     | 1 +
 .../setting/general-setting/plans/ai/actions/subscribe.tsx | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx b/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
index af5b07a6f5..357adf3595 100644
--- a/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
+++ b/packages/frontend/core/src/desktop/dialogs/setting/account-setting/ai-usage-panel.tsx
@@ -31,6 +31,7 @@ export const AIUsagePanel = ({
   useEffect(() => {
     // revalidate latest subscription status
     subscriptionService.subscription.revalidate();
+    subscriptionService.prices.revalidate();
   }, [subscriptionService]);
   const copilotQuotaService = useService(UserCopilotQuotaService);
   useEffect(() => {
diff --git a/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx b/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
index 59da7eca76..0d501a30fa 100644
--- a/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
+++ b/packages/frontend/core/src/desktop/dialogs/setting/general-setting/plans/ai/actions/subscribe.tsx
@@ -74,7 +74,12 @@ export const AISubscribe = ({
       onBeforeCheckout={onBeforeCheckout}
       checkoutOptions={checkoutOptions}
       renderer={props => (
-        <Button variant="primary" {...props} {...btnProps}>
+        <Button
+          variant="primary"
+          {...props}
+          {...btnProps}
+          data-testid="ai-subscribe-button"
+        >
           {btnProps.children ?? `${priceReadable} / ${priceFrequency}`}
           {displayedFrequency === 'monthly' ? (
             <span