Azure-prometheus-datasource does not work with workload identity

I am attempting to use the azure-prometheus-datasource plugin to query the Azure Managed Prometheus service. I am using workload identity federation as my grafana instance is in AKS. This works fine for the azure monitor and azure data explorer plugins but azure-prometheus-datasource continually returns a 401. Is there a special trick to making this work?

No takers?