57ec3a8277
### Summary
* Add `requestUserInput` tool that the model can use for gather
feedback/asking question mid turn.
### Tool input schema
```
{
"$schema": "http://json-schema.org/draft-07/schema#",
"title": "requestUserInput input",
"type": "object",
"additionalProperties": false,
"required": ["questions"],
"properties": {
"questions": {
"type": "array",
"description": "Questions to show the user (1-3). Prefer 1 unless multiple independent decisions block progress.",
"minItems": 1,
"maxItems": 3,
"items": {
"type": "object",
"additionalProperties": false,
"required": ["id", "header", "question"],
"properties": {
"id": {
"type": "string",
"description": "Stable identifier for mapping answers (snake_case)."
},
"header": {
"type": "string",
"description": "Short header label shown in the UI (12 or fewer chars)."
},
"question": {
"type": "string",
"description": "Single-sentence prompt shown to the user."
},
"options": {
"type": "array",
"description": "Optional 2-3 mutually exclusive choices. Put the recommended option first and suffix its label with \"(Recommended)\". Only include \"Other\" option if we want to include a free form option. If the question is free form in nature, do not include any option.",
"minItems": 2,
"maxItems": 3,
"items": {
"type": "object",
"additionalProperties": false,
"required": ["value", "label", "description"],
"properties": {
"value": {
"type": "string",
"description": "Machine-readable value (snake_case)."
},
"label": {
"type": "string",
"description": "User-facing label (1-5 words)."
},
"description": {
"type": "string",
"description": "One short sentence explaining impact/tradeoff if selected."
}
}
}
}
}
}
}
}
}
```
### Tool output schema
```
{
"$schema": "http://json-schema.org/draft-07/schema#",
"title": "requestUserInput output",
"type": "object",
"additionalProperties": false,
"required": ["answers"],
"properties": {
"answers": {
"type": "object",
"description": "Map of question id to user answer.",
"additionalProperties": {
"type": "object",
"additionalProperties": false,
"required": ["selected"],
"properties": {
"selected": {
"type": "array",
"items": { "type": "string" }
},
"other": {
"type": ["string", "null"]
}
}
}
}
}
}
```
85 lines
2.9 KiB
Rust
85 lines
2.9 KiB
Rust
use core_test_support::responses;
|
|
use serde_json::json;
|
|
use std::path::Path;
|
|
|
|
pub fn create_shell_command_sse_response(
|
|
command: Vec<String>,
|
|
workdir: Option<&Path>,
|
|
timeout_ms: Option<u64>,
|
|
call_id: &str,
|
|
) -> anyhow::Result<String> {
|
|
// The `arguments` for the `shell_command` tool is a serialized JSON object.
|
|
let command_str = shlex::try_join(command.iter().map(String::as_str))?;
|
|
let tool_call_arguments = serde_json::to_string(&json!({
|
|
"command": command_str,
|
|
"workdir": workdir.map(|w| w.to_string_lossy()),
|
|
"timeout_ms": timeout_ms
|
|
}))?;
|
|
Ok(responses::sse(vec![
|
|
responses::ev_response_created("resp-1"),
|
|
responses::ev_function_call(call_id, "shell_command", &tool_call_arguments),
|
|
responses::ev_completed("resp-1"),
|
|
]))
|
|
}
|
|
|
|
pub fn create_final_assistant_message_sse_response(message: &str) -> anyhow::Result<String> {
|
|
Ok(responses::sse(vec![
|
|
responses::ev_response_created("resp-1"),
|
|
responses::ev_assistant_message("msg-1", message),
|
|
responses::ev_completed("resp-1"),
|
|
]))
|
|
}
|
|
|
|
pub fn create_apply_patch_sse_response(
|
|
patch_content: &str,
|
|
call_id: &str,
|
|
) -> anyhow::Result<String> {
|
|
Ok(responses::sse(vec![
|
|
responses::ev_response_created("resp-1"),
|
|
responses::ev_apply_patch_shell_command_call_via_heredoc(call_id, patch_content),
|
|
responses::ev_completed("resp-1"),
|
|
]))
|
|
}
|
|
|
|
pub fn create_exec_command_sse_response(call_id: &str) -> anyhow::Result<String> {
|
|
let (cmd, args) = if cfg!(windows) {
|
|
("cmd.exe", vec!["/d", "/c", "echo hi"])
|
|
} else {
|
|
("/bin/sh", vec!["-c", "echo hi"])
|
|
};
|
|
let command = std::iter::once(cmd.to_string())
|
|
.chain(args.into_iter().map(str::to_string))
|
|
.collect::<Vec<_>>();
|
|
let tool_call_arguments = serde_json::to_string(&json!({
|
|
"cmd": command.join(" "),
|
|
"yield_time_ms": 500
|
|
}))?;
|
|
Ok(responses::sse(vec![
|
|
responses::ev_response_created("resp-1"),
|
|
responses::ev_function_call(call_id, "exec_command", &tool_call_arguments),
|
|
responses::ev_completed("resp-1"),
|
|
]))
|
|
}
|
|
|
|
pub fn create_request_user_input_sse_response(call_id: &str) -> anyhow::Result<String> {
|
|
let tool_call_arguments = serde_json::to_string(&json!({
|
|
"questions": [{
|
|
"id": "confirm_path",
|
|
"header": "Confirm",
|
|
"question": "Proceed with the plan?",
|
|
"options": [{
|
|
"label": "Yes (Recommended)",
|
|
"description": "Continue the current plan."
|
|
}, {
|
|
"label": "No",
|
|
"description": "Stop and revisit the approach."
|
|
}]
|
|
}]
|
|
}))?;
|
|
|
|
Ok(responses::sse(vec![
|
|
responses::ev_response_created("resp-1"),
|
|
responses::ev_function_call(call_id, "request_user_input", &tool_call_arguments),
|
|
responses::ev_completed("resp-1"),
|
|
]))
|
|
}
|