objectiveai_cli/command/agents/logs/read/
mod.rs

1//! `agents read` sub-tier. `all`, `pending`, `subscribe` are streaming;
2//! `id` is unary.
3
4use std::pin::Pin;
5
6use futures::{Stream, StreamExt};
7use objectiveai_sdk::cli::command::agents::logs::read::{Request, ResponseItem};
8
9use crate::context::Context;
10use crate::error::Error;
11
12pub mod all;
13pub mod id;
14pub mod pending;
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::All(req) => {
28            let inner = all::execute(ctx, req).await?;
29            Box::pin(inner.map(|r| r.map(ResponseItem::All)))
30        }
31        Request::AllRequestSchema(req) => {
32            let value = all::request_schema::execute(ctx, req).await?;
33            once(Ok(ResponseItem::AllRequestSchema(value)))
34        }
35        Request::AllResponseSchema(req) => {
36            let value = all::response_schema::execute(ctx, req).await?;
37            once(Ok(ResponseItem::AllResponseSchema(value)))
38        }
39        Request::Id(req) => {
40            let value = id::execute(ctx, req).await?;
41            once(Ok(ResponseItem::Id(value)))
42        }
43        Request::IdRequestSchema(req) => {
44            let value = id::request_schema::execute(ctx, req).await?;
45            once(Ok(ResponseItem::IdRequestSchema(value)))
46        }
47        Request::IdResponseSchema(req) => {
48            let value = id::response_schema::execute(ctx, req).await?;
49            once(Ok(ResponseItem::IdResponseSchema(value)))
50        }
51        Request::Pending(req) => {
52            let inner = pending::execute(ctx, req).await?;
53            Box::pin(inner.map(|r| r.map(ResponseItem::Pending)))
54        }
55        Request::PendingRequestSchema(req) => {
56            let value = pending::request_schema::execute(ctx, req).await?;
57            once(Ok(ResponseItem::PendingRequestSchema(value)))
58        }
59        Request::PendingResponseSchema(req) => {
60            let value = pending::response_schema::execute(ctx, req).await?;
61            once(Ok(ResponseItem::PendingResponseSchema(value)))
62        }
63        Request::Subscribe(req) => {
64            let inner = subscribe::execute(ctx, req).await?;
65            Box::pin(inner.map(|r| r.map(ResponseItem::Subscribe)))
66        }
67        Request::SubscribeRequestSchema(req) => {
68            let value = subscribe::request_schema::execute(ctx, req).await?;
69            once(Ok(ResponseItem::SubscribeRequestSchema(value)))
70        }
71        Request::SubscribeResponseSchema(req) => {
72            let value = subscribe::response_schema::execute(ctx, req).await?;
73            once(Ok(ResponseItem::SubscribeResponseSchema(value)))
74        }
75    };
76    Ok(stream)
77}
objectiveai_cli/command/agents/logs/read/mod.rs

objectiveai_cli/command/agents/logs/read/
mod.rs
