Update code

Author: william051200

src/serial-console/azext_serialconsole/custom.py

@@ -675,6 +675,7 @@ def get_region_from_storage_account(cli_ctx, resource_group_name, vm_vmss_name,
     from azure.cli.command_modules.vm.aaz.latest.vmss.vms.instance_view import Show as VMSSVMSInstanceView
     from azure.cli.command_modules.vm.operations.vm import VMShow
     from azure.cli.command_modules.vm.operations.vmss import VMSSShow
+    from azure.cli.core.commands.client_factory import get_subscription_id
     from azext_serialconsole._client_factory import storage_client_factory
     from knack.log import get_logger
 
@@ -686,7 +687,8 @@ def get_region_from_storage_account(cli_ctx, resource_group_name, vm_vmss_name,
         command_args = {
             'instance_id': vmss_instanceid,
             'resource_group': resource_group_name,
-            'vm_scale_set_name': vm_vmss_name
+            'vm_scale_set_name': vm_vmss_name,
+            'subscription': get_subscription_id(cli_ctx)
         }
         result_data = VMSSVMSInstanceView(cli_ctx=cli_ctx)(command_args=command_args)
         result = result_data
@@ -711,15 +713,17 @@ def get_region_from_storage_account(cli_ctx, resource_group_name, vm_vmss_name,
             command_args = {
                 'resource_group': resource_group_name,
                 'vm_name': vm_vmss_name,
-                'expand': 'instanceView'
+                'expand': 'instanceView',
+                'subscription': get_subscription_id(cli_ctx)
             }
             result_data = VMShow(cli_ctx=cli_ctx)(command_args=command_args)
             result = result_data
         except HttpResponseError as e:
             try:
                 command_args = {
                     'resource_group': resource_group_name,
-                    'vm_scale_set_name': vm_vmss_name
+                    'vm_scale_set_name': vm_vmss_name,
+                    'subscription': get_subscription_id(cli_ctx)
                 }
                 VMSSShow(cli_ctx=cli_ctx)(command_args=command_args)
             except HttpResponseError: