From 8efafc6ba9f526d57f142f77e810a19eef8300c1 Mon Sep 17 00:00:00 2001 From: icarus Date: Fri, 19 Sep 2025 23:51:46 +0800 Subject: [PATCH] feat(hooks): add useModels hook for fetching agent models --- src/renderer/src/hooks/agents/useModels.ts | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 src/renderer/src/hooks/agents/useModels.ts diff --git a/src/renderer/src/hooks/agents/useModels.ts b/src/renderer/src/hooks/agents/useModels.ts new file mode 100644 index 0000000000..53d681f50d --- /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 + } +}