mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
Vivian Fang
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})
120 lines
5.1 KiB
Rust
120 lines
5.1 KiB
Rust
use super::*;
|
|
use crate::JsonSchema;
|
|
use pretty_assertions::assert_eq;
|
|
use std::collections::BTreeMap;
|
|
|
|
#[test]
|
|
fn spawn_agents_on_csv_tool_requires_csv_and_instruction() {
|
|
assert_eq!(
|
|
create_spawn_agents_on_csv_tool(),
|
|
ToolSpec::Function(ResponsesApiTool {
|
|
name: "spawn_agents_on_csv".to_string(),
|
|
description: "Process a CSV by spawning one worker sub-agent per row. The instruction string is a template where `{column}` placeholders are replaced with row values. Each worker must call `report_agent_job_result` with a JSON object (matching `output_schema` when provided); missing reports are treated as failures. This call blocks until all rows finish and automatically exports results to `output_csv_path` (or a default path)."
|
|
.to_string(),
|
|
strict: false,
|
|
defer_loading: None,
|
|
parameters: JsonSchema::object(BTreeMap::from([
|
|
(
|
|
"csv_path".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Path to the CSV file containing input rows.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"instruction".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Instruction template to apply to each CSV row. Use {column_name} placeholders to inject values from the row."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"id_column".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Optional column name to use as stable item id.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"output_csv_path".to_string(),
|
|
JsonSchema::string(Some(
|
|
"Optional output CSV path for exported results.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"max_concurrency".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Maximum concurrent workers for this job. Defaults to 16 and is capped by config."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"max_workers".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Alias for max_concurrency. Set to 1 to run sequentially.".to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"max_runtime_seconds".to_string(),
|
|
JsonSchema::number(Some(
|
|
"Maximum runtime per worker before it is failed. Defaults to 1800 seconds."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
(
|
|
"output_schema".to_string(),
|
|
JsonSchema::object(
|
|
BTreeMap::new(),
|
|
/*required*/ None,
|
|
/*additional_properties*/ None,
|
|
),
|
|
),
|
|
]), Some(vec!["csv_path".to_string(), "instruction".to_string()]), Some(false.into())),
|
|
output_schema: None,
|
|
})
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn report_agent_job_result_tool_requires_result_payload() {
|
|
assert_eq!(
|
|
create_report_agent_job_result_tool(),
|
|
ToolSpec::Function(ResponsesApiTool {
|
|
name: "report_agent_job_result".to_string(),
|
|
description:
|
|
"Worker-only tool to report a result for an agent job item. Main agents should not call this."
|
|
.to_string(),
|
|
strict: false,
|
|
defer_loading: None,
|
|
parameters: JsonSchema::object(BTreeMap::from([
|
|
(
|
|
"job_id".to_string(),
|
|
JsonSchema::string(Some("Identifier of the job.".to_string())),
|
|
),
|
|
(
|
|
"item_id".to_string(),
|
|
JsonSchema::string(Some("Identifier of the job item.".to_string())),
|
|
),
|
|
(
|
|
"result".to_string(),
|
|
JsonSchema::object(
|
|
BTreeMap::new(),
|
|
/*required*/ None,
|
|
/*additional_properties*/ None,
|
|
),
|
|
),
|
|
(
|
|
"stop".to_string(),
|
|
JsonSchema::boolean(Some(
|
|
"Optional. When true, cancels the remaining job items after this result is recorded."
|
|
.to_string(),
|
|
)),
|
|
),
|
|
]), Some(vec![
|
|
"job_id".to_string(),
|
|
"item_id".to_string(),
|
|
"result".to_string(),
|
|
]), Some(false.into())),
|
|
output_schema: None,
|
|
})
|
|
);
|
|
}
|