objectiveai-cli 2.1.2

ObjectiveAI command-line interface and embeddable library
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
77
78
79
80
81
82

//! `agents queue delete` — bare-naked handler. Atomically
//! drops one queued message by its `message_queue.id` and returns the
//! deleted row's metadata + reconstructed content body. Cascade on
//! `message_queue_contents.message_queue_id` sweeps every per-kind content row
//! inside the same transaction.

use objectiveai_sdk::cli::command::agents::queue::delete::{Request, Response};

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

pub async fn execute(ctx: &Context, request: Request) -> Result<Response, Error> {
    use db::message_queue::DeleteOutcome;

    let outcome = db::message_queue::delete_by_id(
        &ctx.db,
        request.id,
        &ctx.config.agent_instance_hierarchy,
    )
    .await?;
    let item = match outcome {
        DeleteOutcome::Deleted(item) => item,
        DeleteOutcome::NotFound => {
            return Err(Error::Filesystem(crate::filesystem::Error::NotFound(
                format!("message-queue prompt id {}", request.id),
            )));
        }
        DeleteOutcome::Unauthorized {
            sender_agent_instance_hierarchy,
        } => {
            return Err(Error::QueueDeleteUnauthorized {
                id: request.id,
                sender_agent_instance_hierarchy,
                caller_agent_instance_hierarchy: ctx
                    .config
                    .agent_instance_hierarchy
                    .clone(),
            });
        }
    };
    Ok(Response {
        id: request.id,
        agent_instance_hierarchy: item.agent_instance_hierarchy,
        agent_tag: item.agent_tag,
        key: item.key,
        enqueued_at: item.enqueued_at,
        content: item.content,
    })
}

pub mod request_schema {
    use objectiveai_sdk::cli::command::agents::queue::delete as sdk;
    use objectiveai_sdk::cli::command::agents::queue::delete::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::queue::delete as sdk;
    use objectiveai_sdk::cli::command::agents::queue::delete::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::Response),
        ))
    }
}