Function useMcpToolsQuery

apps/sim/hooks/queries/mcp.ts:127–178

(workspaceId: string)

 * `useQueries`. One slow server cannot block the others.
 */
export function useMcpToolsQuery(workspaceId: string) {
  const { data: servers, isLoading: serversLoading } = useMcpServers(workspaceId)

  // Skip disabled rows (would 404 → negative-cache) and rows from a previous
  // workspace (keepPreviousData on useMcpServers).
  const serverIds = useMemo(
    () =>
      servers
        ? servers
            .filter((s) => s.enabled && s.workspaceId === workspaceId)
            .map((s) => s.id)
            .sort()
        : [],
    [servers, workspaceId]
  )

  const results = useQueries({
    queries: serverIds.map((serverId) => ({
      queryKey: mcpKeys.serverToolsList(workspaceId, serverId),
      queryFn: ({ signal }: { signal?: AbortSignal }) =>
        fetchMcpTools(workspaceId, false, signal, serverId),
      enabled: !!workspaceId,
      retry: false,
      staleTime: 30 * 1000,
      refetchOnWindowFocus: false,
    })),
  })

  return useMemo(() => {
    const tools: McpTool[] = []
    let hasData = false
    let anyServerLoading = false
    let firstError: Error | null = null
    for (const result of results) {
      // Drop stale data from servers whose latest refetch errored.
      if (result.data && !result.isError) {
        tools.push(...result.data)
        hasData = true
      }
      if (result.isLoading) anyServerLoading = true
      if (!firstError && result.error instanceof Error) firstError = result.error
    }
    return {
      data: tools,
      isLoading: (serversLoading || anyServerLoading) && !hasData,
      isFetching: serversLoading || results.some((r) => r.isFetching),
      // Suppress when any healthy server rendered; per-server errors live in `perServer`.
      error: hasData ? null : firstError,
      perServer: results,
    }
  }, [results, serversLoading])
}

export function useForceRefreshMcpTools() {
  const queryClient = useQueryClient()