use crate::cli::command::CommandRequest;
pub mod attach;
pub mod config;
pub mod create;
pub mod delete;
pub mod detach;
pub mod kill;
pub mod list;
pub mod spawn;
#[derive(clap::Subcommand)]
pub enum Command {
Attach(attach::Command),
Config {
#[command(subcommand)]
command: config::Command,
},
Create(create::Command),
Delete(delete::Command),
Detach(detach::Command),
Kill(kill::Command),
List(list::Command),
Spawn(spawn::Command),
}
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[serde(untagged)]
#[schemars(rename = "cli.command.laboratories.Request")]
pub enum Request {
#[schemars(title = "Attach")]
Attach(attach::Request),
#[schemars(title = "AttachRequestSchema")]
AttachRequestSchema(attach::request_schema::Request),
#[schemars(title = "AttachResponseSchema")]
AttachResponseSchema(attach::response_schema::Request),
#[schemars(title = "Config")]
Config(config::Request),
#[schemars(title = "Create")]
Create(create::Request),
#[schemars(title = "CreateRequestSchema")]
CreateRequestSchema(create::request_schema::Request),
#[schemars(title = "CreateResponseSchema")]
CreateResponseSchema(create::response_schema::Request),
#[schemars(title = "Delete")]
Delete(delete::Request),
#[schemars(title = "DeleteRequestSchema")]
DeleteRequestSchema(delete::request_schema::Request),
#[schemars(title = "DeleteResponseSchema")]
DeleteResponseSchema(delete::response_schema::Request),
#[schemars(title = "Detach")]
Detach(detach::Request),
#[schemars(title = "DetachRequestSchema")]
DetachRequestSchema(detach::request_schema::Request),
#[schemars(title = "DetachResponseSchema")]
DetachResponseSchema(detach::response_schema::Request),
#[schemars(title = "Kill")]
Kill(kill::Request),
#[schemars(title = "KillRequestSchema")]
KillRequestSchema(kill::request_schema::Request),
#[schemars(title = "KillResponseSchema")]
KillResponseSchema(kill::response_schema::Request),
#[schemars(title = "List")]
List(list::Request),
#[schemars(title = "ListRequestSchema")]
ListRequestSchema(list::request_schema::Request),
#[schemars(title = "ListResponseSchema")]
ListResponseSchema(list::response_schema::Request),
#[schemars(title = "Spawn")]
Spawn(spawn::Request),
#[schemars(title = "SpawnRequestSchema")]
SpawnRequestSchema(spawn::request_schema::Request),
#[schemars(title = "SpawnResponseSchema")]
SpawnResponseSchema(spawn::response_schema::Request),
}
#[objectiveai_sdk_macros::json_schema_ignore]
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.laboratories.ResponseItem")]
#[serde(untagged)]
pub enum ResponseItem {
#[schemars(title = "Attach")]
Attach(attach::Response),
#[schemars(title = "AttachRequestSchema")]
AttachRequestSchema(attach::request_schema::Response),
#[schemars(title = "AttachResponseSchema")]
AttachResponseSchema(attach::response_schema::Response),
#[schemars(title = "Config")]
Config(config::Response),
#[schemars(title = "Create")]
Create(create::Response),
#[schemars(title = "CreateRequestSchema")]
CreateRequestSchema(create::request_schema::Response),
#[schemars(title = "CreateResponseSchema")]
CreateResponseSchema(create::response_schema::Response),
#[schemars(title = "Delete")]
Delete(delete::Response),
#[schemars(title = "DeleteRequestSchema")]
DeleteRequestSchema(delete::request_schema::Response),
#[schemars(title = "DeleteResponseSchema")]
DeleteResponseSchema(delete::response_schema::Response),
#[schemars(title = "Detach")]
Detach(detach::Response),
#[schemars(title = "DetachRequestSchema")]
DetachRequestSchema(detach::request_schema::Response),
#[schemars(title = "DetachResponseSchema")]
DetachResponseSchema(detach::response_schema::Response),
#[schemars(title = "Kill")]
Kill(kill::Response),
#[schemars(title = "KillRequestSchema")]
KillRequestSchema(kill::request_schema::Response),
#[schemars(title = "KillResponseSchema")]
KillResponseSchema(kill::response_schema::Response),
#[schemars(title = "List")]
List(list::ResponseItem),
#[schemars(title = "ListRequestSchema")]
ListRequestSchema(list::request_schema::Response),
#[schemars(title = "ListResponseSchema")]
ListResponseSchema(list::response_schema::Response),
#[schemars(title = "Spawn")]
Spawn(spawn::Response),
#[schemars(title = "SpawnRequestSchema")]
SpawnRequestSchema(spawn::request_schema::Response),
#[schemars(title = "SpawnResponseSchema")]
SpawnResponseSchema(spawn::response_schema::Response),
}
#[cfg(feature = "mcp")]
impl crate::cli::command::CommandResponse for ResponseItem {
fn into_mcp(self) -> crate::cli::command::McpResponseItem {
match self {
ResponseItem::Attach(v) => v.into_mcp(),
ResponseItem::AttachRequestSchema(v) => v.into_mcp(),
ResponseItem::AttachResponseSchema(v) => v.into_mcp(),
ResponseItem::Config(v) => v.into_mcp(),
ResponseItem::Create(v) => v.into_mcp(),
ResponseItem::CreateRequestSchema(v) => v.into_mcp(),
ResponseItem::CreateResponseSchema(v) => v.into_mcp(),
ResponseItem::Delete(v) => v.into_mcp(),
ResponseItem::DeleteRequestSchema(v) => v.into_mcp(),
ResponseItem::DeleteResponseSchema(v) => v.into_mcp(),
ResponseItem::Detach(v) => v.into_mcp(),
ResponseItem::DetachRequestSchema(v) => v.into_mcp(),
ResponseItem::DetachResponseSchema(v) => v.into_mcp(),
ResponseItem::Kill(v) => v.into_mcp(),
ResponseItem::KillRequestSchema(v) => v.into_mcp(),
ResponseItem::KillResponseSchema(v) => v.into_mcp(),
ResponseItem::List(v) => v.into_mcp(),
ResponseItem::ListRequestSchema(v) => v.into_mcp(),
ResponseItem::ListResponseSchema(v) => v.into_mcp(),
ResponseItem::Spawn(v) => v.into_mcp(),
ResponseItem::SpawnRequestSchema(v) => v.into_mcp(),
ResponseItem::SpawnResponseSchema(v) => v.into_mcp(),
}
}
}
impl TryFrom<Command> for Request {
type Error = crate::cli::command::FromArgsError;
fn try_from(command: Command) -> Result<Self, Self::Error> {
match command {
Command::Attach(cmd) => match cmd.schema {
None => Ok(Request::Attach(attach::Request::try_from(cmd.args)?)),
Some(attach::Schema::RequestSchema(args)) => Ok(Request::AttachRequestSchema(
attach::request_schema::Request::try_from(args)?,
)),
Some(attach::Schema::ResponseSchema(args)) => Ok(Request::AttachResponseSchema(
attach::response_schema::Request::try_from(args)?,
)),
},
Command::Config { command } => {
Ok(Request::Config(config::Request::try_from(command)?))
}
Command::Create(cmd) => match cmd.schema {
None => Ok(Request::Create(create::Request::try_from(cmd.args)?)),
Some(create::Schema::RequestSchema(args)) => Ok(Request::CreateRequestSchema(
create::request_schema::Request::try_from(args)?,
)),
Some(create::Schema::ResponseSchema(args)) => Ok(Request::CreateResponseSchema(
create::response_schema::Request::try_from(args)?,
)),
},
Command::Delete(cmd) => match cmd.schema {
None => Ok(Request::Delete(delete::Request::try_from(cmd.args)?)),
Some(delete::Schema::RequestSchema(args)) => Ok(Request::DeleteRequestSchema(
delete::request_schema::Request::try_from(args)?,
)),
Some(delete::Schema::ResponseSchema(args)) => Ok(Request::DeleteResponseSchema(
delete::response_schema::Request::try_from(args)?,
)),
},
Command::Detach(cmd) => match cmd.schema {
None => Ok(Request::Detach(detach::Request::try_from(cmd.args)?)),
Some(detach::Schema::RequestSchema(args)) => Ok(Request::DetachRequestSchema(
detach::request_schema::Request::try_from(args)?,
)),
Some(detach::Schema::ResponseSchema(args)) => Ok(Request::DetachResponseSchema(
detach::response_schema::Request::try_from(args)?,
)),
},
Command::Kill(cmd) => match cmd.schema {
None => Ok(Request::Kill(kill::Request::try_from(cmd.args)?)),
Some(kill::Schema::RequestSchema(args)) => Ok(Request::KillRequestSchema(
kill::request_schema::Request::try_from(args)?,
)),
Some(kill::Schema::ResponseSchema(args)) => Ok(Request::KillResponseSchema(
kill::response_schema::Request::try_from(args)?,
)),
},
Command::Spawn(cmd) => match cmd.schema {
None => Ok(Request::Spawn(spawn::Request::try_from(cmd.args)?)),
Some(spawn::Schema::RequestSchema(args)) => Ok(Request::SpawnRequestSchema(
spawn::request_schema::Request::try_from(args)?,
)),
Some(spawn::Schema::ResponseSchema(args)) => Ok(Request::SpawnResponseSchema(
spawn::response_schema::Request::try_from(args)?,
)),
},
Command::List(cmd) => match cmd.schema {
None => Ok(Request::List(list::Request::try_from(cmd.args)?)),
Some(list::Schema::RequestSchema(args)) => Ok(Request::ListRequestSchema(
list::request_schema::Request::try_from(args)?,
)),
Some(list::Schema::ResponseSchema(args)) => Ok(Request::ListResponseSchema(
list::response_schema::Request::try_from(args)?,
)),
},
}
}
}
impl CommandRequest for Request {
fn request_base(&self) -> &crate::cli::command::RequestBase {
match self {
Request::Attach(inner) => inner.request_base(),
Request::AttachRequestSchema(inner) => inner.request_base(),
Request::AttachResponseSchema(inner) => inner.request_base(),
Request::Config(inner) => inner.request_base(),
Request::Kill(inner) => inner.request_base(),
Request::KillRequestSchema(inner) => inner.request_base(),
Request::KillResponseSchema(inner) => inner.request_base(),
Request::Spawn(inner) => inner.request_base(),
Request::SpawnRequestSchema(inner) => inner.request_base(),
Request::SpawnResponseSchema(inner) => inner.request_base(),
Request::Create(inner) => inner.request_base(),
Request::CreateRequestSchema(inner) => inner.request_base(),
Request::CreateResponseSchema(inner) => inner.request_base(),
Request::Delete(inner) => inner.request_base(),
Request::DeleteRequestSchema(inner) => inner.request_base(),
Request::DeleteResponseSchema(inner) => inner.request_base(),
Request::Detach(inner) => inner.request_base(),
Request::DetachRequestSchema(inner) => inner.request_base(),
Request::DetachResponseSchema(inner) => inner.request_base(),
Request::List(inner) => inner.request_base(),
Request::ListRequestSchema(inner) => inner.request_base(),
Request::ListResponseSchema(inner) => inner.request_base(),
}
}
fn request_base_mut(&mut self) -> Option<&mut crate::cli::command::RequestBase> {
match self {
Request::Attach(inner) => inner.request_base_mut(),
Request::AttachRequestSchema(inner) => inner.request_base_mut(),
Request::AttachResponseSchema(inner) => inner.request_base_mut(),
Request::Config(inner) => inner.request_base_mut(),
Request::Kill(inner) => inner.request_base_mut(),
Request::KillRequestSchema(inner) => inner.request_base_mut(),
Request::KillResponseSchema(inner) => inner.request_base_mut(),
Request::Spawn(inner) => inner.request_base_mut(),
Request::SpawnRequestSchema(inner) => inner.request_base_mut(),
Request::SpawnResponseSchema(inner) => inner.request_base_mut(),
Request::Create(inner) => inner.request_base_mut(),
Request::CreateRequestSchema(inner) => inner.request_base_mut(),
Request::CreateResponseSchema(inner) => inner.request_base_mut(),
Request::Delete(inner) => inner.request_base_mut(),
Request::DeleteRequestSchema(inner) => inner.request_base_mut(),
Request::DeleteResponseSchema(inner) => inner.request_base_mut(),
Request::Detach(inner) => inner.request_base_mut(),
Request::DetachRequestSchema(inner) => inner.request_base_mut(),
Request::DetachResponseSchema(inner) => inner.request_base_mut(),
Request::List(inner) => inner.request_base_mut(),
Request::ListRequestSchema(inner) => inner.request_base_mut(),
Request::ListResponseSchema(inner) => inner.request_base_mut(),
}
}
}
#[cfg(feature = "cli-executor")]
pub async fn execute<E: crate::cli::command::CommandExecutor>(
executor: &E,
request: Request,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<
std::pin::Pin<Box<dyn futures::Stream<Item = Result<ResponseItem, E::Error>> + Send>>,
E::Error,
> {
use futures::StreamExt as _;
let stream: std::pin::Pin<
Box<dyn futures::Stream<Item = Result<ResponseItem, E::Error>> + Send>,
> = match request {
Request::Attach(req) => {
let value = attach::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Attach(value),
)))
}
Request::AttachRequestSchema(req) => {
let value = attach::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::AttachRequestSchema(value),
)))
}
Request::AttachResponseSchema(req) => {
let value = attach::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::AttachResponseSchema(value),
)))
}
Request::Config(req) => {
let inner = config::execute(executor, req, agent_arguments).await?;
Box::pin(inner.map(|r| r.map(ResponseItem::Config)))
}
Request::Kill(req) => {
let value = kill::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Kill(value),
)))
}
Request::KillRequestSchema(req) => {
let value = kill::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::KillRequestSchema(value),
)))
}
Request::KillResponseSchema(req) => {
let value = kill::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::KillResponseSchema(value),
)))
}
Request::Spawn(req) => {
let value = spawn::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Spawn(value),
)))
}
Request::SpawnRequestSchema(req) => {
let value = spawn::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::SpawnRequestSchema(value),
)))
}
Request::SpawnResponseSchema(req) => {
let value = spawn::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::SpawnResponseSchema(value),
)))
}
Request::Create(req) => {
let value = create::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Create(value),
)))
}
Request::CreateRequestSchema(req) => {
let value = create::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::CreateRequestSchema(value),
)))
}
Request::CreateResponseSchema(req) => {
let value = create::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::CreateResponseSchema(value),
)))
}
Request::Delete(req) => {
let value = delete::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Delete(value),
)))
}
Request::DeleteRequestSchema(req) => {
let value = delete::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::DeleteRequestSchema(value),
)))
}
Request::DeleteResponseSchema(req) => {
let value = delete::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::DeleteResponseSchema(value),
)))
}
Request::Detach(req) => {
let value = detach::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::Detach(value),
)))
}
Request::DetachRequestSchema(req) => {
let value = detach::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::DetachRequestSchema(value),
)))
}
Request::DetachResponseSchema(req) => {
let value = detach::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::DetachResponseSchema(value),
)))
}
Request::List(req) => {
let inner = list::execute(executor, req, agent_arguments).await?;
Box::pin(inner.map(|r| r.map(ResponseItem::List)))
}
Request::ListRequestSchema(req) => {
let value = list::request_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::ListRequestSchema(value),
)))
}
Request::ListResponseSchema(req) => {
let value = list::response_schema::execute(executor, req, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(
ResponseItem::ListResponseSchema(value),
)))
}
};
Ok(stream)
}
#[cfg(feature = "cli-executor")]
pub async fn execute_transform<E: crate::cli::command::CommandExecutor>(
executor: &E,
request: Request,
transform: crate::cli::command::Transform,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<
std::pin::Pin<Box<dyn futures::Stream<Item = Result<serde_json::Value, E::Error>> + Send>>,
E::Error,
> {
let stream: std::pin::Pin<
Box<dyn futures::Stream<Item = Result<serde_json::Value, E::Error>> + Send>,
> = match request {
Request::Attach(req) => {
let value = attach::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::AttachRequestSchema(req) => {
let value = attach::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::AttachResponseSchema(req) => {
let value = attach::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::Config(req) => {
let inner = config::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(inner)
}
Request::Kill(req) => {
let value = kill::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::KillRequestSchema(req) => {
let value = kill::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::KillResponseSchema(req) => {
let value = kill::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::Spawn(req) => {
let value = spawn::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::SpawnRequestSchema(req) => {
let value = spawn::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::SpawnResponseSchema(req) => {
let value = spawn::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::Create(req) => {
let value = create::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::CreateRequestSchema(req) => {
let value = create::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::CreateResponseSchema(req) => {
let value = create::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::Delete(req) => {
let value = delete::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::DeleteRequestSchema(req) => {
let value = delete::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::DeleteResponseSchema(req) => {
let value = delete::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::Detach(req) => {
let value = detach::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::DetachRequestSchema(req) => {
let value = detach::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::DetachResponseSchema(req) => {
let value = detach::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::List(req) => {
let inner = list::execute_transform(executor, req, transform, agent_arguments).await?;
Box::pin(inner)
}
Request::ListRequestSchema(req) => {
let value = list::request_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
Request::ListResponseSchema(req) => {
let value = list::response_schema::execute_transform(
executor,
req,
transform,
agent_arguments,
)
.await?;
Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
}
};
Ok(stream)
}
#[cfg(feature = "cli-listener")]
pub enum ListenerExecution {
Attach(attach::ListenerExecution),
AttachRequestSchema(attach::request_schema::ListenerExecution),
AttachResponseSchema(attach::response_schema::ListenerExecution),
Config(config::ListenerExecution),
Kill(kill::ListenerExecution),
KillRequestSchema(kill::request_schema::ListenerExecution),
KillResponseSchema(kill::response_schema::ListenerExecution),
Spawn(spawn::ListenerExecution),
SpawnRequestSchema(spawn::request_schema::ListenerExecution),
SpawnResponseSchema(spawn::response_schema::ListenerExecution),
Create(create::ListenerExecution),
CreateRequestSchema(create::request_schema::ListenerExecution),
CreateResponseSchema(create::response_schema::ListenerExecution),
Delete(delete::ListenerExecution),
DeleteRequestSchema(delete::request_schema::ListenerExecution),
DeleteResponseSchema(delete::response_schema::ListenerExecution),
Detach(detach::ListenerExecution),
DetachRequestSchema(detach::request_schema::ListenerExecution),
DetachResponseSchema(detach::response_schema::ListenerExecution),
List(list::ListenerExecution),
ListRequestSchema(list::request_schema::ListenerExecution),
ListResponseSchema(list::response_schema::ListenerExecution),
}