ai-registry 0.1.0

AI Registry on Solana - On-chain agent identity and knowledge management
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

use ai_registry_api::{
    consts::{AGENT, FLAG_KNOWLEDGE_FINALIZED},
    error::AgentError,
    event::KnowledgeFinalizedEvent,
    state::Agent,
};
use steel::*;

/// Process a finalize knowledge instruction
pub fn process_finalize_knowledge(
    accounts: &[AccountInfo<'_>],
    _data: &[u8],
) -> ProgramResult {
    // Load accounts (signer first, following Solana best practices)
    let [controller_info, agent_info] = accounts else {
        return Err(ProgramError::NotEnoughAccountKeys);
    };

    // Validate accounts
    agent_info.is_writable()?.has_seeds(
        &[AGENT, agent_info.as_account::<Agent>(&ai_registry_api::ID)?.creator.as_ref()],
        &ai_registry_api::ID,
    )?;
    controller_info.is_signer()?;

    // Load agent and verify controller
    let agent = agent_info.as_account::<Agent>(&ai_registry_api::ID)?;
    if controller_info.key != &agent.controller {
        return Err(AgentError::UnauthorizedController.into());
    }

    // Check if already finalized
    if agent.state_flags & FLAG_KNOWLEDGE_FINALIZED != 0 {
        return Err(AgentError::KnowledgeFinalized.into());
    }

    // Finalize knowledge
    let clock = Clock::get()?;
    let agent = agent_info.as_account_mut::<Agent>(&ai_registry_api::ID)?;
    agent.state_flags |= FLAG_KNOWLEDGE_FINALIZED;
    agent.finalized_at = clock.unix_timestamp;
    agent.updated_at = clock.unix_timestamp;

    // Emit event
    KnowledgeFinalizedEvent {
        creator: agent.creator,
        agent: *agent_info.key,
        timestamp: clock.unix_timestamp,
    }
    .log();

    Ok(())
}