objectiveai-cli 2.2.9

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

//! `agents logs token-usage get` — read an agent's current stored
//! `total_tokens` snapshot. `total_tokens` is null when no
//! agent-completion usage has been recorded for the AIH yet.

use objectiveai_sdk::cli::command::agents::logs::token_usage::get::{Request, Response};

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

pub async fn execute(ctx: &Context, request: Request) -> Result<Response, Error> {
    let db = ctx.db_client().await?.clone();
    let total_tokens =
        crate::db::logs::get_agent_token_usage(&db, &request.agent_instance_hierarchy).await?;
    Ok(Response {
        agent_instance_hierarchy: request.agent_instance_hierarchy,
        total_tokens,
    })
}

pub mod request_schema {
    use objectiveai_sdk::cli::command::agents::logs::token_usage::get as sdk;
    use objectiveai_sdk::cli::command::agents::logs::token_usage::get::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::logs::token_usage::get as sdk;
    use objectiveai_sdk::cli::command::agents::logs::token_usage::get::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)))
    }
}