objectiveai_cli/command/agents/mcp/resources/
mod.rs1use std::pin::Pin;
6
7use futures::Stream;
8use objectiveai_sdk::cli::command::agents::mcp::resources::{Request, ResponseItem};
9
10use crate::context::Context;
11use crate::error::Error;
12
13pub mod list;
14pub mod read;
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 Request::Read(req) => {
39 let value = read::execute(ctx, req).await?;
40 once(Ok(ResponseItem::Read(value)))
41 }
42 Request::ReadRequestSchema(req) => {
43 let value = read::request_schema::execute(ctx, req).await?;
44 once(Ok(ResponseItem::ReadRequestSchema(value)))
45 }
46 Request::ReadResponseSchema(req) => {
47 let value = read::response_schema::execute(ctx, req).await?;
48 once(Ok(ResponseItem::ReadResponseSchema(value)))
49 }
50 };
51 Ok(stream)
52}