File size: 1,363 Bytes
52a9af3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | //! Typed, bounded v2 memory context.
use super::ContextualUserFragment;
use codex_protocol::models::ContentItemKind;
use codex_utils_output_truncation::TruncationPolicy;
use codex_utils_output_truncation::truncate_text;
/// Memory-owned model context, capped below 10k tokens even for byte-heavy text.
pub enum MemoryContextFragment {
ReadInstructions(String),
ExtractionEvidence(String),
}
impl ContextualUserFragment for MemoryContextFragment {
fn role(&self) -> &'static str {
match self {
Self::ReadInstructions(_) => "developer",
Self::ExtractionEvidence(_) => "user",
}
}
fn content_kind(&self) -> ContentItemKind {
ContentItemKind(
match self {
Self::ReadInstructions(_) => "memories.instructions",
Self::ExtractionEvidence(_) => "memories.extraction_evidence",
}
.to_string(),
)
}
fn requires_separate_message(&self) -> bool {
true
}
fn markers(&self) -> (&'static str, &'static str) {
Self::type_markers()
}
fn type_markers() -> (&'static str, &'static str) {
("", "")
}
fn body(&self) -> String {
let (Self::ReadInstructions(text) | Self::ExtractionEvidence(text)) = self;
truncate_text(text, TruncationPolicy::Bytes(8_900))
}
}
|