From 51001544e456412b96697d99237f71a2b1a75e9d Mon Sep 17 00:00:00 2001
From: NFish <douxc512@gmail.com>
Date: Thu, 8 May 2025 10:42:48 +0800
Subject: [PATCH] fix: refresh license limits when invite member success

---
 .../account-setting/members-page/invite-modal/index.tsx      | 2 ++
 web/context/provider-context.tsx                             | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/app/components/header/account-setting/members-page/invite-modal/index.tsx b/web/app/components/header/account-setting/members-page/invite-modal/index.tsx
index a1d3807155..e999253ea7 100644
--- a/web/app/components/header/account-setting/members-page/invite-modal/index.tsx
+++ b/web/app/components/header/account-setting/members-page/invite-modal/index.tsx
@@ -32,6 +32,7 @@ const InviteModal = ({
 }: IInviteModalProps) => {
   const { t } = useTranslation()
   const licenseLimit = useProviderContextSelector(s => s.licenseLimit)
+  const refreshLicenseLimit = useProviderContextSelector(s => s.refreshLicenseLimit)
   const [emails, setEmails] = useState<string[]>([])
   const { notify } = useContext(ToastContext)
   const [isLimited, setIsLimited] = useState(false)
@@ -59,6 +60,7 @@ const InviteModal = ({
         })
 
         if (result === 'success') {
+          refreshLicenseLimit()
           onCancel()
           onSend(invitation_results)
         }
diff --git a/web/context/provider-context.tsx b/web/context/provider-context.tsx
index 8ad09ad751..70c9019aca 100644
--- a/web/context/provider-context.tsx
+++ b/web/context/provider-context.tsx
@@ -54,7 +54,8 @@ type ProviderContextState = {
       size: number
       limit: number
     }
-  }
+  },
+  refreshLicenseLimit: () => void
 }
 const ProviderContext = createContext<ProviderContextState>({
   modelProviders: [],
@@ -95,6 +96,7 @@ const ProviderContext = createContext<ProviderContextState>({
       limit: 0,
     },
   },
+  refreshLicenseLimit: noop,
 })
 
 export const useProviderContext = () => useContext(ProviderContext)
@@ -223,6 +225,7 @@ export const ProviderContextProvider = ({
       isEducationAccount: isEducationAccount?.result || false,
       webappCopyrightEnabled,
       licenseLimit,
+      refreshLicenseLimit: fetchPlan,
     }}>
       {children}
     </ProviderContext.Provider>