check for disk_list before removing

natankeddem · Oct 13, 2024 · b790be7 · b790be7
1 parent 6809b34
commit b790be7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -102,7 +102,7 @@ def response(answer: str, system_info: Dict[str, Any], data: Dict[str, Any]):
             if "network" in default_data and "network" in answer_data:
                 default_data["network"].update(answer_data["network"])
             if "disk-setup" in default_data and "disk-setup" in answer_data:
-                if any("filter" in k for k in answer_data["disk-setup"]):
+                if any("filter" in k for k in answer_data["disk-setup"]) and "disk_list" in default_data["disk-setup"]:
                     del default_data["disk-setup"]["disk_list"]
                 if "disk_list" in answer_data["disk-setup"]:
                     for key in list(default_data["disk-setup"].keys()):