Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: fix model hub sorting (#4722)

Browse Source 
* chore: fix model hub sorting

* chore: linter fix
This commit is contained in:
Louis 2025-02-24 15:23:06 +07:00 committed by GitHub
parent 2e9cca376c
commit 2d7dd8e8ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
web/screens/Hub/index.tsx
View File

@ -121,22 +121,8 @@ const HubScreen = () => {
[sources, searchValue]
)
const sortedModels = useMemo(() => {
if (!sources) return []
return sources.sort((a, b) => {
if (sortSelected === 'most-downloaded') {
return b.metadata.downloads - a.metadata.downloads
} else {
return (
new Date(b.metadata.createdAt).getTime() -
new Date(a.metadata.createdAt).getTime()
)
}
})
}, [sortSelected, sources])
const filteredModels = useMemo(() => {
return sortedModels.filter((model) => {
return (sources ?? []).filter((model) => {
const isCompatible =
!compatible ||
model.models?.some((e) => e.size * 1.5 < totalRam * (1 << 20))
@ -153,7 +139,7 @@ const HubScreen = () => {
return isCompatible && matchesCtxLen && matchesMinSize && matchesMaxSize
})
}, [
sortedModels,
sources,
compatible,
ctxLenFilter,
minModelSizeFilter,
@ -161,6 +147,19 @@ const HubScreen = () => {
totalRam,
])
const sortedModels = useMemo(() => {
return filteredModels.sort((a, b) => {
if (sortSelected === 'most-downloaded') {
return b.metadata.downloads - a.metadata.downloads
} else {
return (
new Date(b.metadata.createdAt).getTime() -
new Date(a.metadata.createdAt).getTime()
)
}
})
}, [sortSelected, filteredModels])
useEffect(() => {
if (modelDetail) {
setSelectedModel(sources?.find((e) => e.id === modelDetail))
@ -516,7 +515,7 @@ const HubScreen = () => {
{(filterOption === 'on-device' ||
filterOption === 'all') && (
<ModelList
models={filteredModels}
models={sortedModels}
onSelectedModel={(model) => setSelectedModel(model)}
/>
)}