mirror of
https://github.com/openai/codex.git
synced 2026-04-23 22:24:57 +00:00
ea516f9a40
This brings us into better alignment with the JSON schema subset that is supported in <https://developers.openai.com/api/docs/guides/structured-outputs#supported-schemas>, and also allows us to render richer function signatures in code mode (e.g., anyOf{null, OtherObjectType})
142 lines
6.3 KiB
Rust
142 lines
6.3 KiB
Rust
use super::*;
|
|
use crate::JsonSchema;
|
|
use codex_protocol::config_types::ModeKind;
|
|
use pretty_assertions::assert_eq;
|
|
use std::collections::BTreeMap;
|
|
|
|
#[test]
|
|
fn request_user_input_tool_includes_questions_schema() {
|
|
assert_eq!(
|
|
create_request_user_input_tool("Ask the user to choose.".to_string()),
|
|
ToolSpec::Function(ResponsesApiTool {
|
|
name: "request_user_input".to_string(),
|
|
description: "Ask the user to choose.".to_string(),
|
|
strict: false,
|
|
defer_loading: None,
|
|
parameters: JsonSchema::object(BTreeMap::from([(
|
|
"questions".to_string(),
|
|
JsonSchema::array(
|
|
JsonSchema::object(
|
|
BTreeMap::from([
|
|
(
|
|
"header".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Short header label shown in the UI (12 or fewer chars)."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"id".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Stable identifier for mapping answers (snake_case)."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"options".to_string(),
|
|
JsonSchema::array(
|
|
JsonSchema::object(
|
|
BTreeMap::from([
|
|
(
|
|
"description".to_string(),
|
|
JsonSchema::string(Some(
|
|
"One short sentence explaining impact/tradeoff if selected."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"label".to_string(),
|
|
JsonSchema::string(Some(
|
|
"User-facing label (1-5 words)."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
]),
|
|
Some(vec![
|
|
"label".to_string(),
|
|
"description".to_string(),
|
|
]),
|
|
Some(false.into()),
|
|
),
|
|
Some(
|
|
"Provide 2-3 mutually exclusive choices. Put the recommended option first and suffix its label with \"(Recommended)\". Do not include an \"Other\" option in this list; the client will add a free-form \"Other\" option automatically."
|
|
.to_string(),
|
|
),
|
|
),
|
|
),
|
|
(
|
|
"question".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Single-sentence prompt shown to the user.".to_string(),
|
|
)),
|
|
),
|
|
]),
|
|
Some(vec![
|
|
"id".to_string(),
|
|
"header".to_string(),
|
|
"question".to_string(),
|
|
"options".to_string(),
|
|
]),
|
|
Some(false.into()),
|
|
),
|
|
Some(
|
|
"Questions to show the user. Prefer 1 and do not exceed 3".to_string(),
|
|
),
|
|
),
|
|
)]), Some(vec!["questions".to_string()]), Some(false.into())),
|
|
output_schema: None,
|
|
})
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn request_user_input_unavailable_messages_respect_default_mode_feature_flag() {
|
|
assert_eq!(
|
|
request_user_input_unavailable_message(
|
|
ModeKind::Plan,
|
|
/*default_mode_request_user_input*/ false
|
|
),
|
|
None
|
|
);
|
|
assert_eq!(
|
|
request_user_input_unavailable_message(
|
|
ModeKind::Default,
|
|
/*default_mode_request_user_input*/ false
|
|
),
|
|
Some("request_user_input is unavailable in Default mode".to_string())
|
|
);
|
|
assert_eq!(
|
|
request_user_input_unavailable_message(
|
|
ModeKind::Default,
|
|
/*default_mode_request_user_input*/ true
|
|
),
|
|
None
|
|
);
|
|
assert_eq!(
|
|
request_user_input_unavailable_message(
|
|
ModeKind::Execute,
|
|
/*default_mode_request_user_input*/ false
|
|
),
|
|
Some("request_user_input is unavailable in Execute mode".to_string())
|
|
);
|
|
assert_eq!(
|
|
request_user_input_unavailable_message(
|
|
ModeKind::PairProgramming,
|
|
/*default_mode_request_user_input*/ false
|
|
),
|
|
Some("request_user_input is unavailable in Pair Programming mode".to_string())
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn request_user_input_tool_description_mentions_available_modes() {
|
|
assert_eq!(
|
|
request_user_input_tool_description(/*default_mode_request_user_input*/ false),
|
|
"Request user input for one to three short questions and wait for the response. This tool is only available in Plan mode.".to_string()
|
|
);
|
|
assert_eq!(
|
|
request_user_input_tool_description(/*default_mode_request_user_input*/ true),
|
|
"Request user input for one to three short questions and wait for the response. This tool is only available in Default or Plan mode.".to_string()
|
|
);
|
|
}
|