objectiveai_cli/command/agents/logs/
mod.rs1use std::pin::Pin;
6
7use futures::{Stream, StreamExt};
8use objectiveai_sdk::cli::command::agents::logs::{Request, ResponseItem};
9
10use crate::context::Context;
11use crate::error::Error;
12
13pub mod list;
14pub mod open;
15pub mod subscribe;
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::Open(req) => {
28 let value = open::execute(ctx, req).await?;
29 once(Ok(ResponseItem::Open(value)))
30 }
31 Request::OpenRequestSchema(req) => {
32 let value = open::request_schema::execute(ctx, req).await?;
33 once(Ok(ResponseItem::OpenRequestSchema(value)))
34 }
35 Request::OpenResponseSchema(req) => {
36 let value = open::response_schema::execute(ctx, req).await?;
37 once(Ok(ResponseItem::OpenResponseSchema(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 Request::Subscribe(req) => {
52 let inner = subscribe::execute(ctx, req).await?;
53 Box::pin(inner.map(|r| r.map(ResponseItem::Subscribe)))
54 }
55 Request::SubscribeRequestSchema(req) => {
56 let value = subscribe::request_schema::execute(ctx, req).await?;
57 once(Ok(ResponseItem::SubscribeRequestSchema(value)))
58 }
59 Request::SubscribeResponseSchema(req) => {
60 let value = subscribe::response_schema::execute(ctx, req).await?;
61 once(Ok(ResponseItem::SubscribeResponseSchema(value)))
62 }
63 };
64 Ok(stream)
65}