mirror of
https://github.com/openai/codex.git
synced 2026-11-14 21:40:39 +00:00
## Why Contextual fragments and extension-owned world-state sections currently return strings, limiting them to text content. ## What changed - Return `ContentItem` from fragment bodies and rendering, preserving classification and attribution. Apply text markers only to `InputText`; pass other content through unchanged. - Allow extension prompt and world-state contributions to carry typed content, and preserve media in context-window hints. - Hash non-text world-state fragments by their serialized content and role while retaining the existing text hashing behavior. - Add shared content-to-tool-output conversion that preserves image references, image detail, and audio. ## Testing Add coverage for typed world-state hashing and preservation of image content and attribution. Expand code-mode output tests to cover input text, output text, file-backed and inline images, and audio. GitOrigin-RevId: 2ed00d466a3f169b151202f3eef71e667ea64be1
35 lines
938 B
Rust
35 lines
938 B
Rust
use super::ContextualUserFragment;
|
|
use codex_protocol::models::ContentItem;
|
|
use codex_protocol::models::ContentItemKind;
|
|
use codex_protocol::models::input_text;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub(crate) struct RolloutBudgetContext {
|
|
pub(crate) remaining_tokens: i64,
|
|
}
|
|
|
|
impl ContextualUserFragment for RolloutBudgetContext {
|
|
fn content_kind(&self) -> ContentItemKind {
|
|
ContentItemKind("rollout_budget.remaining_tokens".to_string())
|
|
}
|
|
|
|
fn role(&self) -> &'static str {
|
|
"developer"
|
|
}
|
|
|
|
fn markers(&self) -> (&'static str, &'static str) {
|
|
Self::type_markers()
|
|
}
|
|
|
|
fn type_markers() -> (&'static str, &'static str) {
|
|
("<rollout_budget>\n", "\n</rollout_budget>")
|
|
}
|
|
|
|
fn body(&self) -> ContentItem {
|
|
input_text(format!(
|
|
"You have {} weighted tokens left in the shared session token budget.",
|
|
self.remaining_tokens
|
|
))
|
|
}
|
|
}
|