mirror of
https://github.com/openai/codex.git
synced 2026-11-12 21:20:34 +00:00
pmccrary-oai
daa48072f4
## Why Token-budget sessions need a way to recover prior conversation context and preserve working state across context-window transitions. ## What changed - Add direct-model `history` tools for listing windows and items, reading items, and searching conversation contents. - Add direct-model `notes` tools for listing, reading, searching, appending, and writing persistent notes. - Route tool calls through the configured Codex backend with trusted session and agent context, bounded request arguments, and truncation-aware output handling. - Expose the extension when `features.token_budget.use_history_notes_history` is enabled with an OpenAI provider and Codex backend authentication. ## Testing - Cover tool registration, configuration changes, provider and authentication requirements, backend request context, encrypted output preservation, and request and response limits. GitOrigin-RevId: 43b259f01014ba3f30803dd7cd5634942407a9bb
76 lines
2.4 KiB
Rust
76 lines
2.4 KiB
Rust
use codex_login::AuthHeaders;
|
|
use codex_login::AuthManager;
|
|
use codex_login::CodexAuth;
|
|
use codex_model_provider::create_model_provider;
|
|
use codex_model_provider_info::ModelProviderInfo;
|
|
use http::HeaderMap;
|
|
use http::HeaderValue;
|
|
use pretty_assertions::assert_eq;
|
|
use serde_json::json;
|
|
use wiremock::Mock;
|
|
use wiremock::MockServer;
|
|
use wiremock::ResponseTemplate;
|
|
use wiremock::matchers::header;
|
|
use wiremock::matchers::method;
|
|
use wiremock::matchers::path;
|
|
|
|
use super::HistoryNotesBackend;
|
|
|
|
#[tokio::test]
|
|
async fn routes_through_codex_backend_and_injects_trusted_session_agent_context() {
|
|
let server = MockServer::start().await;
|
|
Mock::given(method("POST"))
|
|
.and(path("/backend-api/codex/alpha/notes/v2/read_file"))
|
|
.and(header("x-openai-actor-authorization", "actor-biscuit"))
|
|
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
|
"encrypted_output": "enc_payload"
|
|
})))
|
|
.mount(&server)
|
|
.await;
|
|
let mut headers = HeaderMap::new();
|
|
headers.insert(
|
|
"x-openai-actor-authorization",
|
|
HeaderValue::from_static("actor-biscuit"),
|
|
);
|
|
let auth_manager =
|
|
AuthManager::from_auth_for_testing(CodexAuth::Headers(AuthHeaders::new(headers)));
|
|
let provider = create_model_provider(
|
|
ModelProviderInfo::create_openai_provider(Some(format!(
|
|
"{}/backend-api/codex",
|
|
server.uri()
|
|
))),
|
|
Some(auth_manager),
|
|
);
|
|
let backend = HistoryNotesBackend::new(provider);
|
|
|
|
let response = backend
|
|
.call(
|
|
"alpha/notes/v2/read_file",
|
|
"session-123",
|
|
"/root/worker",
|
|
json!({
|
|
"path": "notes.md",
|
|
"context": {
|
|
"session_id": "spoofed-session",
|
|
"current_agent_name": "/root/spoofed",
|
|
}
|
|
}),
|
|
)
|
|
.await
|
|
.expect("History request should succeed");
|
|
|
|
assert_eq!(response, json!({"encrypted_output": "enc_payload"}));
|
|
let requests = server.received_requests().await.expect("recorded requests");
|
|
assert_eq!(requests.len(), 1);
|
|
assert_eq!(
|
|
serde_json::from_slice::<serde_json::Value>(&requests[0].body).expect("JSON body"),
|
|
json!({
|
|
"path": "notes.md",
|
|
"context": {
|
|
"session_id": "session-123",
|
|
"current_agent_name": "/root/worker",
|
|
}
|
|
})
|
|
);
|
|
}
|