scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(ui): guard resumeOAuthFlow against wrong hook instance consuming OAuth result

Browse source
This commit is contained in:
Ishaan Jaffer 2026-03-11 08:24:06 -07:00
parent ef3efcff21
commit f47cce70f1
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
ui/litellm-dashboard/src/hooks/useUserMcpOAuthFlow.tsx
View file

@ -195,6 +195,19 @@ export const useUserMcpOAuthFlow = ({
const storedResult = getStorage(RESULT_KEY);
if (!storedResult) return;
// When multiple OAuth2ConnectButton components are mounted (one per server
// card), each holds its own hook instance. All run resumeOAuthFlow() on
// mount and would compete for the same RESULT_KEY. Peek at the stored
// flow state first: only the hook instance whose serverId matches the one
// that initiated the OAuth flow should consume the result.
const rawFlowState = getStorage(FLOW_STATE_KEY);
if (rawFlowState) {
try {
const peeked = JSON.parse(rawFlowState) as StoredFlowState;
if (peeked.serverId && peeked.serverId !== serverId) return;
} catch (_) {}
}
processingRef.current = true;
clearStorage(RESULT_KEY);
@ -259,7 +272,7 @@ export const useUserMcpOAuthFlow = ({
clearStorage(FLOW_STATE_KEY);
setTimeout(() => { processingRef.current = false; }, 1000);
}
}, [accessToken, onSuccess]);
}, [accessToken, serverId, onSuccess]);
useEffect(() => {
resumeOAuthFlow();