Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-10 07:19:02 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(useModels): merge default filter with provided filter

Browse Source 
Use lodash merge to combine provided filter with default values
This commit is contained in:
icarus 2025-09-22 12:28:03 +08:00
parent 8cd40a471e
commit 82c08128b6
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/renderer/src/hooks/agents/useModels.ts
View File

@ -1,4 +1,5 @@
import { ApiModelsFilter } from '@renderer/types' import { ApiModelsFilter } from '@renderer/types'
import { merge } from 'lodash'
import { useCallback } from 'react' import { useCallback } from 'react'
import useSWR from 'swr' import useSWR from 'swr'
@ -6,10 +7,13 @@ import { useAgentClient } from './useAgentClient'
export const useApiModels = (filter?: ApiModelsFilter) => { export const useApiModels = (filter?: ApiModelsFilter) => {
const client = useAgentClient() 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(() => { const fetcher = useCallback(() => {
return client.getModels(filter) return client.getModels(finalFilter)
}, [client, filter]) }, [client, finalFilter])
const { data, error, isLoading } = useSWR(path, fetcher) const { data, error, isLoading } = useSWR(path, fetcher)
return { return {
models: data?.data ?? [], models: data?.data ?? [],