agentics-persistence 0.3.0

Database persistence layer for the Agentics challenge platform.
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

use secrecy::SecretString;
use sqlx::PgPool;

use crate::db;
use crate::repositories::{
    AgentRecord, AuthenticatedAgent, PioneerCodeRegistrationKind, RegisterAgentInput,
};
use agentics_error::Result;

#[derive(Debug, Clone, Copy)]
pub struct AgentsRepository<'a> {
    pub(super) pool: &'a PgPool,
}

impl AgentsRepository<'_> {
    pub async fn register_agent(
        &self,
        input: &RegisterAgentInput,
        max_active_agents: i64,
    ) -> Result<AgentRecord> {
        db::agents::register_agent(self.pool, input, max_active_agents).await
    }

    pub async fn register_agent_with_pioneer_code(
        &self,
        input: &RegisterAgentInput,
        code_hash: &str,
        max_active_agents: i64,
        kind: PioneerCodeRegistrationKind,
    ) -> Result<AgentRecord> {
        db::agents::register_agent_with_pioneer_code(
            self.pool,
            input,
            code_hash,
            kind,
            max_active_agents,
        )
        .await
    }

    pub async fn count_active(&self) -> Result<i64> {
        db::agents::count_active_agents(self.pool).await
    }

    pub async fn authenticate_token(
        &self,
        token: &SecretString,
    ) -> Result<Option<AuthenticatedAgent>> {
        db::agents::authenticate_agent_token(self.pool, token).await
    }

    pub async fn disable(&self, agent_id: &str) -> Result<()> {
        db::agents::disable_agent(self.pool, agent_id).await
    }
}