From cf24d66101b93347e9f1d415c01b5a5295ea24ef Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Sun, 22 Jun 2025 08:40:30 -0400 Subject: [PATCH] refactor: streamline useToolToggle hook by integrating setEphemeralAgent directly into toggle logic and removing redundant setValue function --- client/src/hooks/Plugins/useToolToggle.ts | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/client/src/hooks/Plugins/useToolToggle.ts b/client/src/hooks/Plugins/useToolToggle.ts index 51034f38fa..739e8caef2 100644 --- a/client/src/hooks/Plugins/useToolToggle.ts +++ b/client/src/hooks/Plugins/useToolToggle.ts @@ -64,20 +64,10 @@ export function useToolToggle({ /** Track previous value to prevent infinite loops */ const prevIsToolEnabled = useRef(isToolEnabled); - const setValue = useCallback( - (isChecked: boolean) => { - setEphemeralAgent((prev) => ({ - ...prev, - [toolKey]: isChecked, - })); - }, - [setEphemeralAgent, toolKey], - ); - const [toggleState, setToggleState] = useLocalStorage( `${localStorageKey}${key}`, isToolEnabled, - setValue, + undefined, storageCondition, ); @@ -89,8 +79,12 @@ export function useToolToggle({ return; } setToggleState(isChecked); + setEphemeralAgent((prev) => ({ + ...prev, + [toolKey]: isChecked, + })); }, - [setToggleState, setIsDialogOpen, isAuthenticated], + [setToggleState, setIsDialogOpen, isAuthenticated, setEphemeralAgent, toolKey], ); const debouncedChange = useMemo(