tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

feat(goose-acp): enable parallel sessions with isolated agent state (#6392)

Browse Source 
Signed-off-by: Adrian Cole <adrian@tetrate.io>
This commit is contained in:
Adrian Cole
2026-01-15 06:19:17 +08:00
committed by GitHub
parent fb0eca2c36
commit 7d4a6bd8ff
86 changed files with 2594 additions and 1938 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/agents/execute_commands.rs
+20 -7
View File
@@ -5,7 +5,6 @@ use anyhow::{anyhow, Result};
use crate::context_mgmt::compact_messages;
use crate::conversation::message::{Message, SystemNotificationType};
use crate::recipe::build_recipe::build_recipe_from_template_with_positional_params;
use crate::session::SessionManager;
use super::Agent;
@@ -81,7 +80,8 @@ impl Agent {
}
async fn handle_compact_command(&self, session_id: &str) -> Result<Option<Message>> {
let session = SessionManager::get_session(session_id, true).await?;
let manager = self.config.session_manager.clone();
let session = manager.get_session(session_id, true).await?;
let conversation = session
.conversation
.ok_or_else(|| anyhow!("Session has no conversation"))?;
@@ -93,7 +93,9 @@ impl Agent {
)
.await?;
SessionManager::replace_conversation(session_id, &compacted_conversation).await?;
manager
.replace_conversation(session_id, &compacted_conversation)
.await?;
Ok(Some(Message::assistant().with_system_notification(
SystemNotificationType::InlineMessage,
@@ -104,9 +106,13 @@ impl Agent {
async fn handle_clear_command(&self, session_id: &str) -> Result<Option<Message>> {
use crate::conversation::Conversation;
SessionManager::replace_conversation(session_id, &Conversation::default()).await?;
let manager = self.config.session_manager.clone();
manager
.replace_conversation(session_id, &Conversation::default())
.await?;
SessionManager::update_session(session_id)
manager
.update(session_id)
.total_tokens(Some(0))
.input_tokens(Some(0))
.output_tokens(Some(0))
@@ -238,10 +244,17 @@ impl Agent {
return Ok(Some(Message::assistant().with_text(error_msg)));
}
SessionManager::add_message(session_id, &msg).await?;
self.config
.session_manager
.clone()
.add_message(session_id, &msg)
.await?;
}
let last_message = SessionManager::get_session(session_id, true)
let last_message = self
.config
.session_manager
.get_session(session_id, true)
.await?
.conversation
.ok_or_else(|| anyhow!("No conversation found"))?