diff --git a/src/renderer/src/hooks/agents/useModels.ts b/src/renderer/src/hooks/agents/useModels.ts new file mode 100644 index 000000000..53d681f50 --- /dev/null +++ b/src/renderer/src/hooks/agents/useModels.ts @@ -0,0 +1,19 @@ +import { ApiModelsFilter } from '@renderer/types' +import { useCallback } from 'react' +import useSWR from 'swr' + +import { useAgentClient } from './useAgentClient' + +export const useModels = (filter: ApiModelsFilter) => { + const client = useAgentClient() + const path = client.getModelsPath(filter) + const fetcher = useCallback(() => { + return client.getModels(filter) + }, [client, filter]) + const { data, error, isLoading } = useSWR(path, fetcher) + return { + models: data?.data, + error, + isLoading + } +}