Skip to main content

objectiveai_cli/command/agents/mcp/servers/
mod.rs

1//! `agents mcp servers` — CLI-side dispatch for listing the agent's
2//! connected MCP servers. One leaf, `list`, does a socket round-trip to the
3//! agent's per-`response_id` MCP listener and returns the proxy-local
4//! `ListServersResult`.
5
6use std::pin::Pin;
7
8use futures::Stream;
9use objectiveai_sdk::cli::command::agents::mcp::servers::{Request, ResponseItem};
10
11use crate::context::Context;
12use crate::error::Error;
13
14pub mod list;
15
16type ItemStream = Pin<Box<dyn Stream<Item = Result<ResponseItem, Error>> + Send>>;
17
18fn once<T: Send + 'static>(
19    item: Result<T, Error>,
20) -> Pin<Box<dyn Stream<Item = Result<T, Error>> + Send>> {
21    Box::pin(futures::stream::once(async move { item }))
22}
23
24pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
25    let stream: ItemStream = match request {
26        Request::List(req) => {
27            let value = list::execute(ctx, req).await?;
28            once(Ok(ResponseItem::List(value)))
29        }
30        Request::ListRequestSchema(req) => {
31            let value = list::request_schema::execute(ctx, req).await?;
32            once(Ok(ResponseItem::ListRequestSchema(value)))
33        }
34        Request::ListResponseSchema(req) => {
35            let value = list::response_schema::execute(ctx, req).await?;
36            once(Ok(ResponseItem::ListResponseSchema(value)))
37        }
38    };
39    Ok(stream)
40}