mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
sayan-oai
a22706dfae
## Summary Add the extension-backed standalone `web.run` tool so Codex can call the standalone search endpoint through the `codex-api` search client and return its encrypted output to Responses. - gate the new tool behind `standalone_web_search` - install the extension in the app-server thread registry and hide hosted `web_search` when standalone search is enabled for OpenAI providers so the two paths stay mutually exclusive - build search context from persisted history using a small tail heuristic: previous user message, assistant text between the last two user turns capped at about 1k tokens, and current user message ## Test Plan - `cargo test -p codex-web-search-extension` - `cargo test -p codex-api` - `cargo test -p codex-core hosted_tools_follow_provider_auth_model_and_config_gates`
73 lines
2.3 KiB
Rust
73 lines
2.3 KiB
Rust
use codex_extension_api::ToolOutput;
|
|
use codex_extension_api::ToolPayload;
|
|
use codex_protocol::models::FunctionCallOutputContentItem;
|
|
use codex_protocol::models::FunctionCallOutputPayload;
|
|
use codex_protocol::models::ResponseInputItem;
|
|
|
|
pub(crate) struct EncryptedSearchOutput {
|
|
encrypted_output: String,
|
|
}
|
|
|
|
impl EncryptedSearchOutput {
|
|
pub(crate) fn new(encrypted_output: String) -> Self {
|
|
Self { encrypted_output }
|
|
}
|
|
}
|
|
|
|
impl ToolOutput for EncryptedSearchOutput {
|
|
fn log_preview(&self) -> String {
|
|
"[encrypted standalone web search output]".to_string()
|
|
}
|
|
|
|
fn success_for_logging(&self) -> bool {
|
|
true
|
|
}
|
|
|
|
fn to_response_item(&self, call_id: &str, _payload: &ToolPayload) -> ResponseInputItem {
|
|
// TODO: Make standalone search honor memories.disable_on_external_context,
|
|
// as hosted web search does.
|
|
ResponseInputItem::FunctionCallOutput {
|
|
call_id: call_id.to_string(),
|
|
output: FunctionCallOutputPayload::from_content_items(vec![
|
|
FunctionCallOutputContentItem::EncryptedContent {
|
|
encrypted_content: self.encrypted_output.clone(),
|
|
},
|
|
]),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use codex_extension_api::ToolPayload;
|
|
use codex_protocol::models::FunctionCallOutputContentItem;
|
|
use codex_protocol::models::FunctionCallOutputPayload;
|
|
use codex_protocol::models::ResponseInputItem;
|
|
use pretty_assertions::assert_eq;
|
|
|
|
use super::EncryptedSearchOutput;
|
|
use super::ToolOutput;
|
|
|
|
#[test]
|
|
fn emits_encrypted_function_call_output() {
|
|
let output = EncryptedSearchOutput::new("encrypted-search-output".to_string());
|
|
|
|
assert_eq!(
|
|
output.to_response_item(
|
|
"call-1",
|
|
&ToolPayload::Function {
|
|
arguments: "{}".to_string(),
|
|
},
|
|
),
|
|
ResponseInputItem::FunctionCallOutput {
|
|
call_id: "call-1".to_string(),
|
|
output: FunctionCallOutputPayload::from_content_items(vec![
|
|
FunctionCallOutputContentItem::EncryptedContent {
|
|
encrypted_content: "encrypted-search-output".to_string(),
|
|
},
|
|
]),
|
|
}
|
|
);
|
|
}
|
|
}
|