State company tested how artificial intelligence could minimise electricity disruptions and now looks to expand the technology.