fix: Fix tests

Author: kennvene-aws

crates/chat-cli/src/cli/agent/mod.rs

@@ -39,7 +39,6 @@ use tracing::{
     info,
     warn,
 };
-use walkdir;
 use wrapper_types::ResourcePath;
 pub use wrapper_types::{
     OriginalToolName,

crates/chat-cli/src/cli/chat/conversation.rs

@@ -1554,6 +1554,7 @@ mod tests {
         let agents = {
             let mut agents = Agents::default();
             let mut agent = Agent::default();
+            agent.name = "TestAgent".to_string();
             agent.resources.push(AMAZONQ_FILENAME.into());
             agent.resources.push(AGENTS_FILENAME.into());
             agents.agents.insert("TestAgent".to_string(), agent);

crates/chat-cli/src/cli/chat/mod.rs

@@ -4335,7 +4335,7 @@ mod tests {
             ..Default::default()
         };
         agents.agents.insert("TestAgent".to_string(), agent);
-        agents.switch("TestAgent").expect("Failed to switch agent");
+        agents.switch("TestAgent", &os).expect("Failed to switch agent");
 
         let tool_manager = ToolManager::default();
         let tool_config = serde_json::from_str::<HashMap<String, ToolSpec>>(include_str!("tools/tool_index.json"))
@@ -4469,7 +4469,7 @@ mod tests {
             ..Default::default()
         };
         agents.agents.insert("SecurityAgent".to_string(), agent);
-        agents.switch("SecurityAgent").expect("Failed to switch agent");
+        agents.switch("SecurityAgent", &os).expect("Failed to switch agent");
 
         let tool_manager = ToolManager::default();
         let tool_config = serde_json::from_str::<HashMap<String, ToolSpec>>(include_str!("tools/tool_index.json"))