{Interactive} Fix KeyError when entering param not in sample in sce…

…nario mode (#6541) * Fix error when enter short param in scenario mode * Fix error when enter short param in scenario mode * Update HISTORY.rst * Release New Version
Azure · Aug 1, 2023 · 02742a8 · 02742a8
1 parent 8e16b69
commit 02742a8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/interactive/HISTORY.rst b/src/interactive/HISTORY.rst
@@ -3,10 +3,11 @@
 Release History
 ===============
 
-upcoming
+0.5.2
 +++++
-* Fix Scenario Idx out of bound when selecting recommended scenario
 * Add command skipped message in Scenario Execution Mode
+* Fix Scenario Idx out of bound when selecting recommended scenario
+* Fix `KeyError` when entering param not in sample in scenario mode
 
 0.5.1
 +++++

diff --git a/src/interactive/azext_interactive/azclishell/__init__.py b/src/interactive/azext_interactive/azclishell/__init__.py
@@ -3,4 +3,4 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-VERSION = '0.5.1'
+VERSION = '0.5.2'
diff --git a/src/interactive/azext_interactive/azclishell/scenario_suggest.py b/src/interactive/azext_interactive/azclishell/scenario_suggest.py
@@ -4,6 +4,7 @@
 # --------------------------------------------------------------------------------------------
 
 import re
+from collections import defaultdict
 
 from prompt_toolkit.auto_suggest import AutoSuggest, Suggestion
 
@@ -31,7 +32,7 @@ def update(self, sample: str):
         # The sample should not start with 'az '
         self.cur_sample = sample.split('az ')[-1]
         # Find parameters used in sample and its value
-        self.param_sample_value_map = {}
+        self.param_sample_value_map = defaultdict(lambda: None)
         # Find the command part of sample
         self.cur_command = self.cur_sample.split('-')[0].strip()
         cur_param = ''