Fix agent config and command toggles (#203)

Author: Josh-XT

Config/Agent.py

@@ -234,19 +234,20 @@ def get_agent_config(self):
                 self.add_agent(self.AGENT_NAME, {})
         return agent_config
 
-    def update_agent_config(self, new_config):
+    def update_agent_config(self, new_config, config_key):
         agent_name = self.AGENT_NAME
         agent_config_file = os.path.join("agents", agent_name, "config.json")
         if os.path.exists(agent_config_file):
             with open(agent_config_file, "r") as f:
                 current_config = json.load(f)
 
-            # Ensure the "settings" key is present in the current configuration
-            if "settings" not in current_config:
-                current_config["settings"] = {}
+            # Ensure the config_key is present in the current configuration
+            if config_key not in current_config:
+                current_config[config_key] = {}
 
-            # Update the "settings" key with new_config
-            current_config["settings"].update(new_config)
+            # Update the specified key with new_config while preserving other keys and values
+            for key, value in new_config.items():
+                current_config[config_key][key] = value
 
             # Save the updated configuration back to the file
             with open(agent_config_file, "w") as f:

app.py

@@ -154,7 +154,7 @@ async def rename_agent(agent_name: str, new_name: AgentNewName) -> ResponseMessa
 async def update_agent_settings(
     agent_name: str, settings: AgentSettings
 ) -> ResponseMessage:
-    update_config = Agent(agent_name).update_agent_config(settings.settings)
+    update_config = Agent(agent_name).update_agent_config(settings.settings, "settings")
     return ResponseMessage(message=update_config)
 
 
@@ -224,14 +224,14 @@ async def toggle_command(
             commands = Commands(agent_name)
             for each_command_name in commands.agent_config["commands"]:
                 commands.agent_config["commands"][each_command_name] = payload.enable
-            agent.update_agent_config(commands.agent_config)
+            agent.update_agent_config(commands.agent_config["commands"], "commands")
             return ResponseMessage(
                 message=f"All commands enabled for agent '{agent_name}'."
             )
         else:
             commands = Commands(agent_name)
             commands.agent_config["commands"][payload.command_name] = payload.enable
-            agent.update_agent_config(commands.agent_config)
+            agent.update_agent_config(commands.agent_config["commands"], "commands")
             return ResponseMessage(
                 message=f"Command '{payload.command_name}' toggled for agent '{agent_name}'."
             )