mirror of
https://github.com/openai/codex.git
synced 2026-04-24 06:35:50 +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})
109 lines
4.4 KiB
Rust
109 lines
4.4 KiB
Rust
use super::*;
|
|
use crate::JsonSchema;
|
|
use pretty_assertions::assert_eq;
|
|
use std::collections::BTreeMap;
|
|
|
|
#[test]
|
|
fn list_dir_tool_matches_expected_spec() {
|
|
assert_eq!(
|
|
create_list_dir_tool(),
|
|
ToolSpec::Function(ResponsesApiTool {
|
|
name: "list_dir".to_string(),
|
|
description:
|
|
"Lists entries in a local directory with 1-indexed entry numbers and simple type labels."
|
|
.to_string(),
|
|
strict: false,
|
|
defer_loading: None,
|
|
parameters: JsonSchema::object(BTreeMap::from([
|
|
(
|
|
"depth".to_string(),
|
|
JsonSchema::number(Some(
|
|
"The maximum directory depth to traverse. Must be 1 or greater."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"dir_path".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Absolute path to the directory to list.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"limit".to_string(),
|
|
JsonSchema::number(Some(
|
|
"The maximum number of entries to return.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"offset".to_string(),
|
|
JsonSchema::number(Some(
|
|
"The entry number to start listing from. Must be 1 or greater."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
]), Some(vec!["dir_path".to_string()]), Some(false.into())),
|
|
output_schema: None,
|
|
})
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_sync_tool_matches_expected_spec() {
|
|
assert_eq!(
|
|
create_test_sync_tool(),
|
|
ToolSpec::Function(ResponsesApiTool {
|
|
name: "test_sync_tool".to_string(),
|
|
description: "Internal synchronization helper used by Codex integration tests."
|
|
.to_string(),
|
|
strict: false,
|
|
defer_loading: None,
|
|
parameters: JsonSchema::object(BTreeMap::from([
|
|
(
|
|
"barrier".to_string(),
|
|
JsonSchema::object(
|
|
BTreeMap::from([
|
|
(
|
|
"id".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Identifier shared by concurrent calls that should rendezvous"
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"participants".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Number of tool calls that must arrive before the barrier opens"
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"timeout_ms".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Maximum time in milliseconds to wait at the barrier"
|
|
.to_string(),
|
|
)),
|
|
),
|
|
]),
|
|
Some(vec!["id".to_string(), "participants".to_string()]),
|
|
Some(false.into()),
|
|
),
|
|
),
|
|
(
|
|
"sleep_after_ms".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Optional delay in milliseconds after completing the barrier"
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"sleep_before_ms".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Optional delay in milliseconds before any other action".to_string(),
|
|
)),
|
|
),
|
|
]), /*required*/ None, Some(false.into())),
|
|
output_schema: None,
|
|
})
|
|
);
|
|
}
|