fix agent-mode session resumption bug

This commit is contained in:
Dhanji R. Prasanna
2026-01-03 16:44:58 +11:00
parent 76bfb77f84
commit f4a1bf5e93
3 changed files with 479 additions and 13 deletions

View File

@@ -1012,25 +1012,39 @@ impl<W: UiWriter> Agent<W> {
fn generate_session_id(&self, description: &str) -> String {
use std::collections::hash_map::DefaultHasher;
use std::hash::{Hash, Hasher};
use std::time::{SystemTime, UNIX_EPOCH};
// Clean and truncate the description for a readable filename
let clean_description = description
.chars()
.filter(|c| c.is_alphanumeric() || c.is_whitespace() || *c == '-' || *c == '_')
.collect::<String>()
.split_whitespace()
.take(5) // Take first 5 words
.collect::<Vec<_>>()
.join("_")
.to_lowercase();
// For agent mode, use agent name as prefix for clarity
// For regular mode, use first 5 words of description
let prefix = if let Some(ref agent_name) = self.agent_name {
agent_name.clone()
} else {
description
.chars()
.filter(|c| c.is_alphanumeric() || c.is_whitespace() || *c == '-' || *c == '_')
.collect::<String>()
.split_whitespace()
.take(5)
.collect::<Vec<_>>()
.join("_")
.to_lowercase()
};
// Create a hash for uniqueness
// Create a hash for uniqueness (description + agent name + timestamp)
let mut hasher = DefaultHasher::new();
description.hash(&mut hasher);
if let Some(ref agent_name) = self.agent_name {
agent_name.hash(&mut hasher);
}
let timestamp = SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|d| d.as_nanos())
.unwrap_or(0);
timestamp.hash(&mut hasher);
let hash = hasher.finish();
// Format: clean_description_hash
format!("{}_{:x}", clean_description, hash)
// Format: prefix_hash (agent_name_hash for agents, description_hash for regular)
format!("{}_{:x}", prefix, hash)
}
/// Save the entire context window to a per-session file
@@ -1765,6 +1779,15 @@ impl<W: UiWriter> Agent<W> {
debug!("Agent mode enabled for agent: {}", agent_name);
}
/// Initialize session ID manually (primarily for testing).
/// This allows tests to verify session ID generation without calling execute_task,
/// which would require an LLM provider.
pub fn init_session_id_for_test(&mut self, description: &str) {
if self.session_id.is_none() {
self.session_id = Some(self.generate_session_id(description));
}
}
/// Clear session state and continuation artifacts (for /clear command)
pub fn clear_session(&mut self) {
use crate::session_continuation::clear_continuation;

View File

@@ -0,0 +1,145 @@
//! Tests for agent session ID generation
//!
//! This test verifies that:
//! 1. Agent mode sessions use the agent name as prefix (e.g., "fowler_<hash>")
//! 2. Different agents get different session IDs even with the same task
//! 3. Regular (non-agent) sessions use the task description as prefix
use g3_config::Config;
use g3_core::ui_writer::NullUiWriter;
use g3_core::Agent;
use serial_test::serial;
use tempfile::TempDir;
/// Helper to create a test agent
async fn create_test_agent(temp_dir: &TempDir) -> Agent<NullUiWriter> {
std::env::set_current_dir(temp_dir.path()).unwrap();
let config = Config::default();
Agent::new_with_readme_and_quiet(config, NullUiWriter, None, true)
.await
.unwrap()
}
/// Helper to create a test agent in agent mode
async fn create_agent_mode_agent(temp_dir: &TempDir, agent_name: &str) -> Agent<NullUiWriter> {
std::env::set_current_dir(temp_dir.path()).unwrap();
let config = Config::default();
let mut agent = Agent::new_with_readme_and_quiet(config, NullUiWriter, None, true)
.await
.unwrap();
agent.set_agent_mode(agent_name);
agent
}
// =============================================================================
// AGENT MODE SESSION ID TESTS
// =============================================================================
#[tokio::test]
#[serial]
async fn test_agent_session_id_uses_agent_name_prefix() {
let temp_dir = TempDir::new().unwrap();
let mut agent = create_agent_mode_agent(&temp_dir, "fowler").await;
// Trigger session ID generation
agent.init_session_id_for_test("Test task");
let session_id = agent.get_session_id();
assert!(session_id.is_some(), "Session ID should be set after adding a message");
let session_id = session_id.unwrap();
assert!(
session_id.starts_with("fowler_"),
"Agent session ID should start with agent name. Got: {}",
session_id
);
}
#[tokio::test]
#[serial]
async fn test_different_agents_get_different_session_ids() {
let temp_dir1 = TempDir::new().unwrap();
let temp_dir2 = TempDir::new().unwrap();
let mut agent1 = create_agent_mode_agent(&temp_dir1, "fowler").await;
let mut agent2 = create_agent_mode_agent(&temp_dir2, "lamport").await;
// Use the exact same task description for both
let task = "Begin your analysis and work on the current project.";
agent1.init_session_id_for_test(task);
agent2.init_session_id_for_test(task);
let session_id1 = agent1.get_session_id().unwrap();
let session_id2 = agent2.get_session_id().unwrap();
// Session IDs should be different
assert_ne!(
session_id1, session_id2,
"Different agents should get different session IDs even with same task"
);
// Each should have the correct prefix
assert!(
session_id1.starts_with("fowler_"),
"Fowler session should start with 'fowler_'. Got: {}",
session_id1
);
assert!(
session_id2.starts_with("lamport_"),
"Lamport session should start with 'lamport_'. Got: {}",
session_id2
);
}
#[tokio::test]
#[serial]
async fn test_regular_session_uses_description_prefix() {
let temp_dir = TempDir::new().unwrap();
let mut agent = create_test_agent(&temp_dir).await;
// Add a message with a specific description
agent.init_session_id_for_test("implement fibonacci function");
let session_id = agent.get_session_id();
assert!(session_id.is_some(), "Session ID should be set");
let session_id = session_id.unwrap();
// Regular sessions should use the description (first 5 words, lowercased)
assert!(
session_id.starts_with("implement_fibonacci_function_"),
"Regular session ID should start with description. Got: {}",
session_id
);
}
#[tokio::test]
#[serial]
async fn test_same_agent_different_runs_get_different_session_ids() {
let temp_dir1 = TempDir::new().unwrap();
let temp_dir2 = TempDir::new().unwrap();
let mut agent1 = create_agent_mode_agent(&temp_dir1, "fowler").await;
let mut agent2 = create_agent_mode_agent(&temp_dir2, "fowler").await;
// Same agent, same task
let task = "Begin your analysis and work on the current project.";
agent1.init_session_id_for_test(task);
// Small delay to ensure different timestamps
std::thread::sleep(std::time::Duration::from_millis(1));
agent2.init_session_id_for_test(task);
let session_id1 = agent1.get_session_id().unwrap();
let session_id2 = agent2.get_session_id().unwrap();
// Session IDs should be different due to timestamp
assert_ne!(
session_id1, session_id2,
"Same agent running twice should get different session IDs"
);
// Both should have the same prefix
assert!(session_id1.starts_with("fowler_"), "Got: {}", session_id1);
assert!(session_id2.starts_with("fowler_"), "Got: {}", session_id2);
}