rei-forks/fred
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

agent: Remove selections as context once message is sent (#29030)

Browse Source 
Release Notes:

- N/A
This commit is contained in:
Bennet Bo Fenner 2025-04-18 10:50:46 +02:00 committed by GitHub
parent 6254efe39d
commit 87512d0814
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
crates/agent/src/message_editor.rs
View File

@ -2,7 +2,7 @@ use std::collections::BTreeMap;
use std::sync::Arc; use std::sync::Arc;
use crate::assistant_model_selector::ModelType; use crate::assistant_model_selector::ModelType;
use crate::context::format_context_as_string; use crate::context::{AssistantContext, format_context_as_string};
use crate::tool_compatibility::{IncompatibleToolsState, IncompatibleToolsTooltip}; use crate::tool_compatibility::{IncompatibleToolsState, IncompatibleToolsTooltip};
use buffer_diff::BufferDiff; use buffer_diff::BufferDiff;
use collections::HashSet; use collections::HashSet;
@ -293,6 +293,21 @@ impl MessageEditor {
}) })
.log_err(); .log_err();
context_store
.update(cx, |context_store, cx| {
let excerpt_ids = context_store
.context()
.iter()
.filter(|ctx| matches!(ctx, AssistantContext::Excerpt(_)))
.map(|ctx| ctx.id())
.collect::<Vec<_>>();
for id in excerpt_ids {
context_store.remove_context(id, cx);
}
})
.log_err();
if let Some(wait_for_summaries) = context_store if let Some(wait_for_summaries) = context_store
.update(cx, |context_store, cx| context_store.wait_for_summaries(cx)) .update(cx, |context_store, cx| context_store.wait_for_summaries(cx))
.log_err() .log_err()