diff --git a/web/hooks/useEngineManagement.ts b/web/hooks/useEngineManagement.ts index 96fa281e9..02044ec65 100644 --- a/web/hooks/useEngineManagement.ts +++ b/web/hooks/useEngineManagement.ts @@ -477,20 +477,23 @@ export const useRefreshModelList = (engine: string) => { const [refreshingModels, setRefreshingModels] = useState(false) const { mutate: fetchRemoteModels } = useGetRemoteModels(engine) - const refreshModels = useCallback(() => { - setRefreshingModels(true) - fetchRemoteModels() - .then((remoteModelList) => - Promise.all( - remoteModelList?.data?.map((model: { id?: string }) => - model?.id - ? addRemoteEngineModel(model.id, engine).catch(() => {}) - : {} - ) ?? [] + const refreshModels = useCallback( + (engine: string) => { + setRefreshingModels(true) + fetchRemoteModels() + .then((remoteModelList) => + Promise.all( + remoteModelList?.data?.map((model: { id?: string }) => + model?.id + ? addRemoteEngineModel(model.id, engine).catch(() => {}) + : {} + ) ?? [] + ) ) - ) - .finally(() => setRefreshingModels(false)) - }, [fetchRemoteModels]) + .finally(() => setRefreshingModels(false)) + }, + [fetchRemoteModels] + ) return { refreshingModels, refreshModels } } diff --git a/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx b/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx index 20c3a2edc..d091b97b5 100644 --- a/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx +++ b/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx @@ -6,15 +6,15 @@ import Spinner from '@/containers/Loader/Spinner' import { useRefreshModelList } from '@/hooks/useEngineManagement' -function RemoteModelRefresh({ id }: { id: string }) { - const { refreshingModels, refreshModels } = useRefreshModelList(id) +function RemoteModelRefresh({ engine }: { engine: string }) { + const { refreshingModels, refreshModels } = useRefreshModelList(engine) return ( - +