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 ?? [],