use crate::cli::command::CommandRequest;
use crate::cli::command::agents::spawn::AgentSpec;
use crate::functions::inventions::state::ParamsState;
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.functions.inventions.recursive.create.remote.Request")]
pub struct Request {
pub path_type: Path,
pub state: RequestState,
pub agent: AgentSpec,
pub continuation: Option<String>,
pub seed: Option<i64>,
pub dangerous_advanced: Option<RequestDangerousAdvanced>,
pub jq: Option<String>,
}
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.functions.inventions.recursive.create.remote.Path")]
pub enum Path {
#[serde(rename = "functions/inventions/recursive/create/remote")]
FunctionsInventionsRecursiveCreateRemote,
}
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.functions.inventions.recursive.create.remote.RequestState")]
pub enum RequestState {
#[schemars(title = "Inline")]
Inline(ParamsState),
#[schemars(title = "Ref")]
Ref(String),
}
impl RequestState {
fn push_flags(&self, out: &mut Vec<String>) {
match self {
RequestState::Inline(v) => {
out.push("--state-inline".to_string());
out.push(serde_json::to_string(v).expect("state serializes"));
}
RequestState::Ref(r) => {
out.push("--state".to_string());
out.push(r.clone());
}
}
}
}
impl CommandRequest for Request {
fn into_command(&self) -> Vec<String> {
let mut argv = vec![
"functions".to_string(),
"inventions".to_string(),
"recursive".to_string(),
"create".to_string(),
"remote".to_string(),
];
self.state.push_flags(&mut argv);
argv.push("--agent-inline".to_string());
argv.push(serde_json::to_string(&self.agent).expect("agent serializes"));
if let Some(c) = &self.continuation {
argv.push("--continuation".to_string());
argv.push(c.clone());
}
if let Some(seed) = self.seed {
argv.push("--seed".to_string());
argv.push(seed.to_string());
}
if let Some(advanced) = &self.dangerous_advanced {
argv.push("--dangerous-advanced".to_string());
argv.push(
serde_json::to_string(advanced)
.expect("RequestDangerousAdvanced serializes"),
);
}
if let Some(jq) = &self.jq {
argv.push("--jq".to_string());
argv.push(jq.clone());
}
argv
}
}
#[derive(Debug, Clone, Default, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.functions.inventions.recursive.create.remote.RequestDangerousAdvanced")]
pub struct RequestDangerousAdvanced {
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub stream: Option<bool>,
}
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[serde(untagged)]
#[schemars(rename = "cli.command.functions.inventions.recursive.create.remote.ResponseItem")]
pub enum ResponseItem {
#[schemars(title = "Chunk")]
Chunk(crate::functions::inventions::recursive::response::streaming::FunctionInventionRecursiveChunk),
#[schemars(title = "Id")]
Id(String),
}
pub type Response = String;
#[derive(clap::Args)]
#[group(id = "state", required = true, multiple = false)]
pub struct Args {
#[arg(long, group = "state")]
pub state: Option<String>,
#[arg(long, group = "state")]
pub state_inline: Option<String>,
#[arg(long)]
pub agent_inline: String,
#[arg(long)]
pub continuation: Option<String>,
#[arg(long)]
pub seed: Option<i64>,
#[arg(long)]
pub dangerous_advanced: Option<String>,
#[arg(long)]
pub jq: Option<String>,
}
#[derive(clap::Args)]
#[command(args_conflicts_with_subcommands = true)]
pub struct Command {
#[command(flatten)]
pub args: Args,
#[command(subcommand)]
pub schema: Option<Schema>,
}
#[derive(clap::Subcommand)]
pub enum Schema {
RequestSchema(request_schema::Args),
ResponseSchema(response_schema::Args),
}
impl TryFrom<Args> for Request {
type Error = crate::cli::command::FromArgsError;
fn try_from(args: Args) -> Result<Self, Self::Error> {
let state = if let Some(s) = args.state_inline {
let mut de = serde_json::Deserializer::from_str(&s);
let v = serde_path_to_error::deserialize(&mut de).map_err(|source| {
crate::cli::command::FromArgsError {
field: "state_inline",
source: source.into(),
}
})?;
RequestState::Inline(v)
} else {
RequestState::Ref(args.state.unwrap())
};
let agent = {
let mut de = serde_json::Deserializer::from_str(&args.agent_inline);
serde_path_to_error::deserialize(&mut de).map_err(|source| {
crate::cli::command::FromArgsError {
field: "agent_inline",
source: source.into(),
}
})?
};
let dangerous_advanced = if let Some(s) = args.dangerous_advanced {
let mut de = serde_json::Deserializer::from_str(&s);
let v = serde_path_to_error::deserialize(&mut de).map_err(|source| {
crate::cli::command::FromArgsError {
field: "dangerous_advanced",
source: source.into(),
}
})?;
Some(v)
} else {
None
};
Ok(Self { path_type: Path::FunctionsInventionsRecursiveCreateRemote,
state,
agent,
continuation: args.continuation,
seed: args.seed,
dangerous_advanced,
jq: args.jq,
})
}
}
#[cfg(feature = "cli-executor")]
pub async fn execute_streaming<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<E::Stream<ResponseItem>, E::Error> {
request.jq = None;
let mut advanced = request.dangerous_advanced.unwrap_or_default();
advanced.stream = Some(true);
request.dangerous_advanced = Some(advanced);
executor.execute(request, agent_arguments).await
}
#[cfg(feature = "cli-executor")]
pub async fn execute_streaming_jq<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
jq: String,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<E::Stream<serde_json::Value>, E::Error> {
request.jq = Some(jq);
let mut advanced = request.dangerous_advanced.unwrap_or_default();
advanced.stream = Some(true);
request.dangerous_advanced = Some(advanced);
executor.execute(request, agent_arguments).await
}
#[cfg(feature = "cli-executor")]
pub async fn execute<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<Response, E::Error> {
request.jq = None;
if let Some(advanced) = request.dangerous_advanced.as_mut() {
advanced.stream = None;
}
executor.execute_one(request, agent_arguments).await
}
#[cfg(feature = "cli-executor")]
pub async fn execute_jq<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
jq: String,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<serde_json::Value, E::Error> {
request.jq = Some(jq);
if let Some(advanced) = request.dangerous_advanced.as_mut() {
advanced.stream = None;
}
executor.execute_one(request, agent_arguments).await
}
#[cfg(feature = "mcp")]
impl crate::cli::command::CommandResponse for ResponseItem {
fn into_mcp(self) -> crate::cli::command::McpResponseItem {
crate::cli::command::McpResponseItem::JSONL(serde_json::to_value(self).unwrap())
}
}
pub mod request_schema;
pub mod response_schema;