Update the cmdlet parameters according to feedback. (#14217)

Author: kceiw

tools/Az.Tools.Predictor/Az.Tools.Predictor/Commands/DisableAzPredictor.cs

@@ -12,7 +12,6 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
-using System;
 using System.Text;
 using System.Management.Automation;
 
@@ -29,11 +28,10 @@ public sealed class DisableAzPredictor : PSCmdlet
         };
 
         /// <summary>
-        /// Gets and sets the session that this cmdlet applies to.
+        /// Indicates whether it's applied to all sessions.
         /// </summary>
-        [Parameter(Position = 0)]
-        [ValidateSet(nameof(SessionParameterValue.All), nameof(SessionParameterValue.Current))]
-        public SessionParameterValue Session { get; set; }
+        [Parameter(Mandatory = false)]
+        public SwitchParameter AllSession { get; set; }
 
         /// <summary>
         /// Indicates whether the user would like to receive output.
@@ -47,7 +45,7 @@ protected override void ProcessRecord()
             var scriptToRun = new StringBuilder();
             var _ = scriptToRun.Append(DisableAzPredictor._DisableStatements[0]);
 
-            if (Session == SessionParameterValue.All)
+            if (AllSession.IsPresent)
             {
                 _ = scriptToRun.Append(";Write-Host \"To disable Az Predictor, please edit your profile ($PROFILE) and remove the following lines:`nImport-Module Az.Tools.Predictor`nSet-PSReadLineOption -PredictionSource HistoryAndPlugin`n\"");
             }

tools/Az.Tools.Predictor/Az.Tools.Predictor/Commands/EnableAzPredictor.cs

@@ -12,7 +12,6 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
-using System;
 using System.Text;
 using System.Management.Automation;
 
@@ -30,11 +29,10 @@ public sealed class EnableAzPredictor : PSCmdlet
         };
 
         /// <summary>
-        /// Gets and sets the session that this cmdlet applies to.
+        /// Indicates whether it's applied to all sessions.
         /// </summary>
-        [Parameter(Position = 0)]
-        [ValidateSet(nameof(SessionParameterValue.All), nameof(SessionParameterValue.Current))]
-        public SessionParameterValue Session { get; set; }
+        [Parameter(Mandatory = false)]
+        public SwitchParameter AllSession { get; set; }
 
         /// <summary>
         /// Indicates whether the user would like to receive output.
@@ -48,7 +46,7 @@ protected override void ProcessRecord()
             var scriptToRun = new StringBuilder();
             var _ = scriptToRun.Append(EnableAzPredictor._EnableStatements[1]);
 
-            if (Session == SessionParameterValue.All)
+            if (AllSession.IsPresent)
             {
                 _ = scriptToRun.Append($";Add-Content -Path $PROFILE -Value \"`n{string.Join("`n", EnableAzPredictor._EnableStatements)}\" -NoNewline -Encoding UTF8 -Force")
                                 .Append($";Write-Host \"User profile ($PROFILE) has been updated.`n\"");