objectiveai-cli 2.2.8

ObjectiveAI command-line interface and embeddable library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

//! `agents laboratories list` — stream the laboratory ids attached to an
//! agent target. Read-only: no locks. Resolves the target to its DB key
//! directly (tag → tag rows; instance → AIH rows; ref → error) and
//! streams `{ "id": "<id>" }` per attached laboratory.

use std::pin::Pin;

use futures::Stream;
use objectiveai_sdk::cli::command::agents::laboratories::list::{Request, ResponseItem};
use objectiveai_sdk::cli::command::agents::selector::AgentSelector;

use crate::context::Context;
use crate::db::laboratory_attachments::Target;
use crate::error::Error;

type ItemStream = Pin<Box<dyn Stream<Item = Result<ResponseItem, Error>> + Send>>;

pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
    let target = match &request.selector {
        AgentSelector::Ref { .. } => return Err(Error::LaboratoryRefTarget),
        AgentSelector::Tag { agent_tag } => Target::Tag(agent_tag.clone()),
        AgentSelector::Instance {
            parent_agent_instance_hierarchy,
            agent_instance,
        } => {
            let parent = parent_agent_instance_hierarchy
                .as_deref()
                .unwrap_or(&ctx.config.agent_instance_hierarchy);
            Target::Aih(format!("{parent}/{agent_instance}"))
        }
    };
    let pool = ctx.db_client().await?.clone();
    let stream = async_stream::stream! {
        match crate::db::laboratory_attachments::list(&pool, &target).await {
            Ok(ids) => {
                for id in ids {
                    yield Ok(ResponseItem { id });
                }
            }
            Err(e) => yield Err(Error::from(e)),
        }
    };
    Ok(Box::pin(stream))
}

pub mod request_schema {
    use objectiveai_sdk::cli::command::agents::laboratories::list as sdk;
    use objectiveai_sdk::cli::command::agents::laboratories::list::request_schema::{
        Request, Response,
    };

    use crate::context::Context;
    use crate::error::Error;

    pub async fn execute(_ctx: &Context, _request: Request) -> Result<Response, Error> {
        Ok(objectiveai_sdk::cli::command::ResponseSchema(
            schemars::schema_for!(sdk::Request),
        ))
    }
}

pub mod response_schema {
    use objectiveai_sdk::cli::command::agents::laboratories::list as sdk;
    use objectiveai_sdk::cli::command::agents::laboratories::list::response_schema::{
        Request, Response,
    };

    use crate::context::Context;
    use crate::error::Error;

    pub async fn execute(_ctx: &Context, _request: Request) -> Result<Response, Error> {
        Ok(objectiveai_sdk::cli::command::ResponseSchema(
            schemars::schema_for!(sdk::ResponseItem),
        ))
    }
}