Function useMcpTools

apps/sim/hooks/mcp/use-mcp-tools.ts:39–84

(workspaceId: string)

}

export function useMcpTools(workspaceId: string): UseMcpToolsResult {
  const queryClient = useQueryClient()

  const { data: mcpToolsData, isLoading, error: queryError } = useMcpToolsQuery(workspaceId)

  const mcpTools = useMemo<McpToolForUI[]>(() => {
    return mcpToolsData.map((tool) => ({
      id: createMcpToolId(tool.serverId, tool.name),
      name: tool.name,
      description: tool.description,
      serverId: tool.serverId,
      serverName: tool.serverName,
      type: 'mcp' as const,
      inputSchema: tool.inputSchema,
      bgColor: '#6366F1',
      icon: McpIcon,
    }))
  }, [mcpToolsData])

  // Soft refresh — invalidate per-server entries. For cache-bypass, use `useForceRefreshMcpTools`.
  const refreshTools = useCallback(async () => {
    if (!workspaceId) {
      logger.warn('Cannot refresh tools: no workspaceId provided')
      return
    }

    await queryClient.invalidateQueries({
      queryKey: mcpKeys.serverToolsWorkspace(workspaceId),
    })
  }, [workspaceId, queryClient])

  const getToolsByServer = useCallback(
    (serverId: string): McpToolForUI[] => {
      return mcpTools.filter((tool) => tool.serverId === serverId)
    },
    [mcpTools]
  )

  return {
    mcpTools,
    isLoading,
    error: queryError instanceof Error ? queryError.message : null,
    refreshTools,
    getToolsByServer,
  }
}