diff --git a/extensions/inference-cortex-extension/bin/version.txt b/extensions/inference-cortex-extension/bin/version.txt index 5bf81f993..de9a33e5d 100644 --- a/extensions/inference-cortex-extension/bin/version.txt +++ b/extensions/inference-cortex-extension/bin/version.txt @@ -1 +1 @@ -1.0.11-rc4 +1.0.11-rc5 diff --git a/web/containers/Providers/DeepLinkListener.tsx b/web/containers/Providers/DeepLinkListener.tsx index f8eebe618..41a8d9e24 100644 --- a/web/containers/Providers/DeepLinkListener.tsx +++ b/web/containers/Providers/DeepLinkListener.tsx @@ -6,7 +6,10 @@ import { useDebouncedCallback } from 'use-debounce' import { MainViewState } from '@/constants/screens' -import { useModelSourcesMutation } from '@/hooks/useModelSource' +import { + useGetModelSources, + useModelSourcesMutation, +} from '@/hooks/useModelSource' import { loadingModalInfoAtom } from '../LoadingModal' import { toaster } from '../Toast' @@ -19,6 +22,7 @@ const DeepLinkListener: React.FC = () => { const setLoadingInfo = useSetAtom(loadingModalInfoAtom) const setMainView = useSetAtom(mainViewStateAtom) const setModelDetail = useSetAtom(modelDetailAtom) + const { mutate } = useGetModelSources() const handleDeepLinkAction = useDebouncedCallback( async (deepLinkAction: DeepLinkAction) => { @@ -37,7 +41,7 @@ const DeepLinkListener: React.FC = () => { title: 'Getting Hugging Face model details', message: 'Please wait..', }) - await addModelSource(deepLinkAction.resource) + await addModelSource(deepLinkAction.resource).then(() => mutate()) setLoadingInfo(undefined) setMainView(MainViewState.Hub) setModelDetail(deepLinkAction.resource) diff --git a/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx b/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx new file mode 100644 index 000000000..20c3a2edc --- /dev/null +++ b/web/screens/Hub/ModelPage/RemoteModelRefresh.tsx @@ -0,0 +1,29 @@ +import { Button } from '@janhq/joi' + +import { RefreshCwIcon } from 'lucide-react' + +import Spinner from '@/containers/Loader/Spinner' + +import { useRefreshModelList } from '@/hooks/useEngineManagement' + +function RemoteModelRefresh({ id }: { id: string }) { + const { refreshingModels, refreshModels } = useRefreshModelList(id) + + return ( + + ) +} + +export default RemoteModelRefresh diff --git a/web/screens/Hub/ModelPage/index.tsx b/web/screens/Hub/ModelPage/index.tsx index b52ae72a9..26f5d4e0d 100644 --- a/web/screens/Hub/ModelPage/index.tsx +++ b/web/screens/Hub/ModelPage/index.tsx @@ -25,6 +25,8 @@ import { MarkdownTextMessage } from '@/screens/Thread/ThreadCenterPanel/TextMess import { toGigabytes } from '@/utils/converter' import { extractModelName, removeYamlFrontMatter } from '@/utils/modelSource' +import RemoteModelRefresh from './RemoteModelRefresh' + import { mainViewStateAtom } from '@/helpers/atoms/App.atom' import { selectedSettingAtom, @@ -39,7 +41,6 @@ type Props = { const ModelPage = ({ model, onGoBack }: Props) => { const setSelectedSetting = useSetAtom(selectedSettingAtom) const setMainViewState = useSetAtom(mainViewStateAtom) - const { refreshingModels, refreshModels } = useRefreshModelList(model.id) const showScrollBar = useAtomValue(showScrollBarAtom) return ( @@ -157,23 +158,7 @@ const ModelPage = ({ model, onGoBack }: Props) => { )} {model.type === 'cloud' && ( - + )}