From b514c8df1004af4587ffedf6d417cf31031e9082 Mon Sep 17 00:00:00 2001 From: pakrym-oai Date: Wed, 11 Mar 2026 19:04:37 -0700 Subject: [PATCH] Fix code mode tool export name --- codex-rs/core/src/tools/code_mode_description.rs | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/codex-rs/core/src/tools/code_mode_description.rs b/codex-rs/core/src/tools/code_mode_description.rs index 2a3ba815cc..20015b1b84 100644 --- a/codex-rs/core/src/tools/code_mode_description.rs +++ b/codex-rs/core/src/tools/code_mode_description.rs @@ -75,9 +75,16 @@ fn append_code_mode_sample( output_type: String, ) -> String { let reference = code_mode_tool_reference(tool_name); + let local_name = code_mode_local_name(&reference.tool_key); + let import_binding = if reference.tool_key != "tools" && local_name == reference.tool_key { + local_name.as_str() + } else { + "tools" + }; + format!( - "{description}\n\nCode mode declaration:\n```ts\nimport {{ {} }} from \"{}\";\ndeclare function {}({input_name}: {input_type}): Promise<{output_type}>;\n```", - reference.tool_key, reference.module_path, reference.tool_key + "{description}\n\nCode mode declaration:\n```ts\nimport {{ {import_binding} }} from \"{}\";\ndeclare function {local_name}({input_name}: {input_type}): Promise<{output_type}>;\n```", + reference.module_path, ) }