Add variety to Az.Predictor result list (#13420)

* add variety to predictions list

* add variety to predictions list

Author: yemohyle

tools/Az.Tools.Predictor/Az.Tools.Predictor/Predictor.cs

@@ -102,6 +102,8 @@ public IDictionary<string, string> Query(Ast input, int suggestionCount, Cancell
                 var resultBuilder = new StringBuilder();
                 var usedParams = new HashSet<int>();
                 var sourceBuilder = new StringBuilder();
+                var presentCommands = new System.Collections.Generic.Dictionary<string, int>(); // Added
+                int maxAllowedCommandDupl = 1;
 
                 for (var i = 0; i < _predictions.Count && results.Count < suggestionCount; ++i)
                 {
@@ -140,7 +142,17 @@ public IDictionary<string, string> Query(Ast input, int suggestionCount, Cancell
                                     }
                                 }
 
-                                results.Add(prediction.ToString(), sourceBuilder.ToString());
+
+                                if (!presentCommands.ContainsKey(_predictions[i].Command))
+                                {
+                                    results.Add(prediction.ToString(), sourceBuilder.ToString());
+                                    presentCommands.Add(_predictions[i].Command, 1);
+                                }
+                                else if (presentCommands[_predictions[i].Command] < maxAllowedCommandDupl)
+                                {
+                                    results.Add(prediction.ToString(), sourceBuilder.ToString());
+                                    presentCommands[_predictions[i].Command] += 1;
+                                }
 
                                 if (results.Count == suggestionCount)
                                 {
@@ -152,7 +164,7 @@ public IDictionary<string, string> Query(Ast input, int suggestionCount, Cancell
                 }
             }
             catch
-            {
+            { 
             }
 
             return results;