@Senthilnath TM Thanks for your response. I have modified the above queries and please use the below query which will give you the complete information like vm name, vm size, data disk names and disk sizes as well. I have uploaded the query in this GitHub repo as well.
Here is the sample output screenshot for your reference.
Hope this helps and let me know if you still have any further questions on this.