Skip to main content

objectiveai_cli/command/laboratories/
mod.rs

1//! `laboratories` — top-level CLI dispatch for laboratory containers (podman
2//! containers the conduit dials as client-side MCP servers). Distinct from
3//! `agents laboratories` (attachments). `create` creates + starts a
4//! laboratory container; `list` reads them back from podman.
5
6use std::pin::Pin;
7
8use futures::{Stream, StreamExt};
9use objectiveai_sdk::cli::command::laboratories::{Request, ResponseItem};
10
11use crate::context::Context;
12use crate::error::Error;
13
14pub mod create;
15pub mod list;
16
17type ItemStream = Pin<Box<dyn Stream<Item = Result<ResponseItem, Error>> + Send>>;
18
19fn once<T: Send + 'static>(
20    item: Result<T, Error>,
21) -> Pin<Box<dyn Stream<Item = Result<T, Error>> + Send>> {
22    Box::pin(futures::stream::once(async move { item }))
23}
24
25pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
26    let stream: ItemStream = match request {
27        Request::Create(req) => {
28            let value = create::execute(ctx, req).await?;
29            once(Ok(ResponseItem::Create(value)))
30        }
31        Request::CreateRequestSchema(req) => {
32            let value = create::request_schema::execute(ctx, req).await?;
33            once(Ok(ResponseItem::CreateRequestSchema(value)))
34        }
35        Request::CreateResponseSchema(req) => {
36            let value = create::response_schema::execute(ctx, req).await?;
37            once(Ok(ResponseItem::CreateResponseSchema(value)))
38        }
39        Request::List(req) => {
40            let inner = list::execute(ctx, req).await?;
41            Box::pin(inner.map(|r| r.map(ResponseItem::List)))
42        }
43        Request::ListRequestSchema(req) => {
44            let value = list::request_schema::execute(ctx, req).await?;
45            once(Ok(ResponseItem::ListRequestSchema(value)))
46        }
47        Request::ListResponseSchema(req) => {
48            let value = list::response_schema::execute(ctx, req).await?;
49            once(Ok(ResponseItem::ListResponseSchema(value)))
50        }
51    };
52    Ok(stream)
53}