diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/mcpServers/useMCPToolsets.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/mcpServers/useMCPToolsets.ts new file mode 100644 index 00000000000..aacb446d4bc --- /dev/null +++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/mcpServers/useMCPToolsets.ts @@ -0,0 +1,16 @@ +import { useQuery } from "@tanstack/react-query"; +import { createQueryKeys } from "../common/queryKeysFactory"; +import { fetchMCPToolsets } from "@/components/networking"; +import { MCPToolset } from "@/components/mcp_tools/types"; +import useAuthorized from "../useAuthorized"; + +const mcpToolsetKeys = createQueryKeys("mcpToolsets"); + +export const useMCPToolsets = () => { + const { accessToken } = useAuthorized(); + return useQuery({ + queryKey: mcpToolsetKeys.list(), + queryFn: async () => await fetchMCPToolsets(accessToken!), + enabled: !!accessToken, + }); +};