clone/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-04-24 22:54:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Preserve web search process items

Browse Source
This commit is contained in:
Ahmed Ibrahim
2025-11-01 14:05:39 -07:00
parent d9118c04bf
commit 3459829062
1 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
codex-rs/core/src/response_processing.rs
View File

@@ -22,6 +22,9 @@ pub(crate) async fn process_items(
// If the model returned a message, we need to record it.
items_to_record_in_conversation_history.push(item);
}
(ResponseItem::WebSearchCall { .. }, _) => {
items_to_record_in_conversation_history.push(item);
}
(
ResponseItem::LocalShellCall { .. },
Some(ResponseInputItem::FunctionCallOutput { call_id, output }),
@@ -102,3 +105,45 @@ pub(crate) async fn process_items(
}
(responses, items_to_record_in_conversation_history)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::codex::ProcessedResponseItem;
use crate::codex::make_session_and_context;
use codex_protocol::models::ResponseInputItem;
use codex_protocol::models::ResponseItem;
use codex_protocol::models::WebSearchAction;
use pretty_assertions::assert_eq;
#[tokio::test]
async fn records_web_search_call_without_id() {
let (session, turn_context) = make_session_and_context();
let status = Some("completed".to_string());
let query = "search query".to_string();
let processed_items = vec![ProcessedResponseItem {
item: ResponseItem::WebSearchCall {
id: None,
status: status.clone(),
action: WebSearchAction::Search {
query: query.clone(),
},
},
response: None,
}];
let (responses, recorded_items) =
process_items(processed_items, &session, &turn_context).await;
assert_eq!(responses, Vec::<ResponseInputItem>::new());
assert_eq!(
recorded_items,
vec![ResponseItem::WebSearchCall {
id: None,
status,
action: WebSearchAction::Search { query },
}]
);
}
}