From 82c08128b6ef6d8ba5c8bb7f0c3d3d20aef92c66 Mon Sep 17 00:00:00 2001 From: icarus Date: Mon, 22 Sep 2025 12:28:03 +0800 Subject: [PATCH] refactor(useModels): merge default filter with provided filter Use lodash merge to combine provided filter with default values --- src/renderer/src/hooks/agents/useModels.ts | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/renderer/src/hooks/agents/useModels.ts b/src/renderer/src/hooks/agents/useModels.ts index f05cb777b9..e35d58dde1 100644 --- a/src/renderer/src/hooks/agents/useModels.ts +++ b/src/renderer/src/hooks/agents/useModels.ts @@ -1,4 +1,5 @@ import { ApiModelsFilter } from '@renderer/types' +import { merge } from 'lodash' import { useCallback } from 'react' import useSWR from 'swr' @@ -6,10 +7,13 @@ import { useAgentClient } from './useAgentClient' export const useApiModels = (filter?: ApiModelsFilter) => { const client = useAgentClient() - const path = client.getModelsPath(filter) + // const defaultFilter = { limit: -1 } satisfies ApiModelsFilter + const defaultFilter = {} satisfies ApiModelsFilter + const finalFilter = merge(filter, defaultFilter) + const path = client.getModelsPath(finalFilter) const fetcher = useCallback(() => { - return client.getModels(filter) - }, [client, filter]) + return client.getModels(finalFilter) + }, [client, finalFilter]) const { data, error, isLoading } = useSWR(path, fetcher) return { models: data?.data ?? [],