Does the APIDataSet fit your usecase? Or you could have a custom api dataset that wrap around your logic