fix: Fixes comments

Author: kennvene-aws

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

@@ -150,7 +150,7 @@ pub struct ConversationState {
     /// Tangent mode checkpoint - stores main conversation when in tangent mode
     #[serde(default, skip_serializing_if = "Option::is_none")]
     tangent_state: Option<ConversationCheckpoint>,
-    /// Current continuation ID for billing tracking - generated per conversation turn
+    /// Current continuation ID - generated per conversation turn
     #[serde(default, skip_serializing_if = "Option::is_none")]
     current_continuation_id: Option<String>,
 }
@@ -383,12 +383,12 @@ impl ConversationState {
         self.next_message = None;
     }
 
-    /// Returns the current continuation ID for billing tracking
+    /// Returns the current continuation ID
     pub fn continuation_id(&self) -> Option<&str> {
         self.current_continuation_id.as_deref()
     }
 
-    /// Generate a new continuation ID for billing tracking
+    /// Generate a new continuation ID
     fn generate_new_continuation_id(&mut self) {
         let new_continuation_id = uuid::Uuid::new_v4().to_string();
         self.current_continuation_id = Some(new_continuation_id);

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

@@ -1,6 +1,4 @@
-// ABOUTME: Tests for continuation ID billing functionality
-// ABOUTME: Ensures continuation IDs are generated and propagated correctly for billing consistency
-
+// ABOUTME: Ensures continuation IDs are generated and propagated correctly
 #[cfg(test)]
 mod tests {
     use std::collections::HashMap;

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

@@ -2129,7 +2129,7 @@ impl ChatSession {
                     user_input
                 };
 
-                // For tool approval responses (y/n/t), preserve continuation ID for billing consistency
+                // For tool approval responses (y/n/t), preserve continuation ID
                 let preserve_continuation_id = is_simple_tool_interaction(&input_trimmed);
                 self.conversation.abandon_tool_use_with_continuation_policy(
                     &self.tool_uses,
@@ -3700,7 +3700,7 @@ fn is_reject_response(input: &str) -> bool {
 }
 
 /// Check if input is any simple tool interaction response (y/Y/n/N/t/T)
-/// These responses should preserve continuation ID for billing consistency
+/// These responses should preserve continuation ID
 fn is_simple_tool_interaction(input: &str) -> bool {
     is_trust_response(input) || is_accept_response(input) || is_reject_response(input)
 }
@@ -4455,28 +4455,3 @@ mod tests {
         }
     }
 }
-
-// Helper method to save the agent config to file
-async fn save_agent_config(os: &mut Os, config: &Agent, agent_name: &str, is_global: bool) -> Result<(), ChatError> {
-    let config_dir = if is_global {
-        directories::chat_global_agent_path(os)
-            .map_err(|e| ChatError::Custom(format!("Could not find global agent directory: {}", e).into()))?
-    } else {
-        directories::chat_local_agent_dir(os)
-            .map_err(|e| ChatError::Custom(format!("Could not find local agent directory: {}", e).into()))?
-    };
-
-    tokio::fs::create_dir_all(&config_dir)
-        .await
-        .map_err(|e| ChatError::Custom(format!("Failed to create config directory: {}", e).into()))?;
-
-    let config_file = config_dir.join(format!("{}.json", agent_name));
-    let config_json = serde_json::to_string_pretty(config)
-        .map_err(|e| ChatError::Custom(format!("Failed to serialize agent config: {}", e).into()))?;
-
-    tokio::fs::write(&config_file, config_json)
-        .await
-        .map_err(|e| ChatError::Custom(format!("Failed to write agent config file: {}", e).into()))?;
-
-    Ok(())
-}