diff --git a/crates/agent/src/thread.rs b/crates/agent/src/thread.rs index 2e61cf3cfccb89d6793b1527a3209409d18643cc..98f505f074ac91e8be0c4defb8b8bf78b6c62bb9 100644 --- a/crates/agent/src/thread.rs +++ b/crates/agent/src/thread.rs @@ -1698,7 +1698,34 @@ impl Thread { project.set_agent_location(None, cx); }); - cx.emit (ThreadEvent::ShowError(ThreadError::Message { + // Remove the turn that was refused. + // + // https://docs.anthropic.com/en/docs/test-and-evaluate/strengthen-guardrails/handle-streaming-refusals#reset-context-after-refusal + { + let mut messages_to_remove = Vec::new(); + + for (ix, message) in thread.messages.iter().enumerate().rev() { + messages_to_remove.push(message.id); + + if message.role == Role::User { + if ix == 0 { + break; + } + + if let Some(prev_message) = thread.messages.get(ix - 1) { + if prev_message.role == Role::Assistant { + break; + } + } + } + } + + for message_id in messages_to_remove { + thread.delete_message(message_id, cx); + } + } + + cx.emit(ThreadEvent::ShowError(ThreadError::Message { header: "Language model refusal".into(), message: "Model refused to generate content for safety reasons.".into(), }));