openai/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-09-10 20:26:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Preserve rendered code mode history

Browse Source
This commit is contained in:
starr-openai
2026-05-19 23:17:00 -07:00
parent a4bb775c0e
commit 560e339d86
3 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
codex-rs/core/src/tools/code_mode/execute_handler.rs
View File

@@ -10,7 +10,7 @@ use codex_tools::ToolSpec;
use super::ExecContext;
use super::PUBLIC_TOOL_NAME;
use super::code_mode_output_truncation_policy;
use super::code_mode_history_truncation_policy;
use super::handle_runtime_response;
use super::is_exec_tool_name;
@@ -124,12 +124,8 @@ impl CoreToolRuntime for CodeModeExecuteHandler {
fn history_truncation_policy(
&self,
invocation: &ToolInvocation,
_invocation: &ToolInvocation,
) -> Option<codex_utils_output_truncation::TruncationPolicy> {
let ToolPayload::Custom { input } = &invocation.payload else {
return None;
};
let args = codex_code_mode::parse_exec_source(input).ok()?;
Some(code_mode_output_truncation_policy(args.max_output_tokens))
Some(code_mode_history_truncation_policy())
}
}

4
codex-rs/core/src/tools/code_mode/mod.rs
View File

@@ -245,6 +245,10 @@ pub(super) fn code_mode_output_truncation_policy(
TruncationPolicy::Tokens(resolve_max_tokens(max_output_tokens))
}
pub(super) fn code_mode_history_truncation_policy() -> TruncationPolicy {
TruncationPolicy::Bytes(usize::MAX)
}
async fn call_nested_tool(
_exec: ExecContext,
tool_runtime: ToolCallRuntime,

10
codex-rs/core/src/tools/code_mode/wait_handler.rs
View File

@@ -15,7 +15,7 @@ use codex_tools::ToolSpec;
use super::DEFAULT_WAIT_YIELD_TIME_MS;
use super::ExecContext;
use super::WAIT_TOOL_NAME;
use super::code_mode_output_truncation_policy;
use super::code_mode_history_truncation_policy;
use super::handle_runtime_response;
use super::wait_spec::create_wait_tool;
@@ -117,13 +117,9 @@ impl ToolExecutor<ToolInvocation> for CodeModeWaitHandler {
impl CoreToolRuntime for CodeModeWaitHandler {
fn history_truncation_policy(
&self,
invocation: &ToolInvocation,
_invocation: &ToolInvocation,
) -> Option<codex_utils_output_truncation::TruncationPolicy> {
let ToolPayload::Function { arguments } = &invocation.payload else {
return None;
};
let args: ExecWaitArgs = parse_arguments(arguments).ok()?;
Some(code_mode_output_truncation_policy(args.max_tokens))
Some(code_mode_history_truncation_policy())
}
fn pre_tool_use_payload(&self, _invocation: &ToolInvocation) -> Option<PreToolUsePayload> {