From 5bdec309e69ff401cbe6aaa75189b86eb470ebce Mon Sep 17 00:00:00 2001 From: imagineer99 Date: Sun, 22 Feb 2026 01:45:51 +0000 Subject: [PATCH] feat: sort unsloth models first in HF search dropdowns --- studio/frontend/src/hooks/use-hf-model-search.ts | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/studio/frontend/src/hooks/use-hf-model-search.ts b/studio/frontend/src/hooks/use-hf-model-search.ts index 4745b4dc7b..2ec100b1bd 100644 --- a/studio/frontend/src/hooks/use-hf-model-search.ts +++ b/studio/frontend/src/hooks/use-hf-model-search.ts @@ -1,6 +1,6 @@ import type { PipelineType } from "@huggingface/hub"; import { listModels } from "@huggingface/hub"; -import { useCallback } from "react"; +import { useCallback, useMemo } from "react"; import { useHfPaginatedSearch } from "./use-hf-paginated-search"; export interface HfModelResult { @@ -84,5 +84,17 @@ export function useHfModelSearch( [query, task, accessToken], ); - return useHfPaginatedSearch(createIter, mapModel); + const search = useHfPaginatedSearch(createIter, mapModel); + + const results = useMemo( + () => + [...search.results].sort((a, b) => { + const aFirst = a.id.startsWith("unsloth/") ? 0 : 1; + const bFirst = b.id.startsWith("unsloth/") ? 0 : 1; + return aFirst - bFirst; + }), + [search.results], + ); + + return { ...search, results }; }