diff --git a/crates/agent/src/agent.rs b/crates/agent/src/agent.rs index d4062fec85cb458ade372085d23fa42a47e631ed..140c872ecacc9d572881d2745bbbfc2810b1d309 100644 --- a/crates/agent/src/agent.rs +++ b/crates/agent/src/agent.rs @@ -1429,15 +1429,16 @@ impl acp_thread::AgentConnection for NativeAgentConnection { session_id: &acp::SessionId, cx: &mut App, ) -> Task> { - self.0.update(cx, |agent, _cx| { - let project_id = agent.sessions.get(session_id).map(|s| s.project_id); - agent.sessions.remove(session_id); - - if let Some(project_id) = project_id { - let has_remaining = agent.sessions.values().any(|s| s.project_id == project_id); - if !has_remaining { - agent.projects.remove(&project_id); - } + self.0.update(cx, |agent, cx| { + let Some(session) = agent.sessions.remove(session_id) else { + return; + }; + let project_id = session.project_id; + agent.save_thread(session.thread, cx); + + let has_remaining = agent.sessions.values().any(|s| s.project_id == project_id); + if !has_remaining { + agent.projects.remove(&project_id); } }); Task::ready(Ok(()))