tirea-agentos 0.5.0

Agent runtime with streaming LLM integration, sub-agent orchestration, and context window 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

mod agent;
mod behavior;
mod model;
mod provider;
mod stop_policy;
mod tool;
pub mod traits;

use std::collections::HashMap;
use std::sync::Arc;

#[cfg(feature = "skills")]
use tirea_extension_skills::SkillRegistry;

pub use stop_policy::StopPolicyRegistry;
pub use traits::{AgentRegistry, BehaviorRegistry, ModelRegistry, ProviderRegistry, ToolRegistry};

pub use agent::{CompositeAgentRegistry, InMemoryAgentRegistry};
pub use behavior::{CompositeBehaviorRegistry, InMemoryBehaviorRegistry};
pub use model::{CompositeModelRegistry, InMemoryModelRegistry};
pub use provider::{CompositeProviderRegistry, InMemoryProviderRegistry};
pub use stop_policy::{
    CompositeStopPolicyRegistry, InMemoryStopPolicyRegistry, StopPolicyRegistryError,
};
pub use tool::{CompositeToolRegistry, InMemoryToolRegistry};

pub use traits::{
    AgentRegistryError, BehaviorRegistryError, ModelDefinition, ModelRegistryError,
    ProviderRegistryError, RegistryBundle, ToolRegistryError,
};

pub(crate) fn sorted_registry_ids<T>(entries: &HashMap<String, T>) -> Vec<String> {
    let mut ids: Vec<String> = entries.keys().cloned().collect();
    ids.sort();
    ids
}

/// Aggregated registry set used by [`crate::runtime::AgentOs`] after build-time composition.
#[derive(Clone)]
pub struct RegistrySet {
    pub agents: Arc<dyn AgentRegistry>,
    pub tools: Arc<dyn ToolRegistry>,
    pub behaviors: Arc<dyn BehaviorRegistry>,
    pub providers: Arc<dyn ProviderRegistry>,
    pub models: Arc<dyn ModelRegistry>,
    pub stop_policies: Arc<dyn StopPolicyRegistry>,
    #[cfg(feature = "skills")]
    pub skills: Option<Arc<dyn SkillRegistry>>,
}

impl RegistrySet {
    pub fn new(
        agents: Arc<dyn AgentRegistry>,
        tools: Arc<dyn ToolRegistry>,
        behaviors: Arc<dyn BehaviorRegistry>,
        providers: Arc<dyn ProviderRegistry>,
        models: Arc<dyn ModelRegistry>,
        stop_policies: Arc<dyn StopPolicyRegistry>,
        #[cfg(feature = "skills")] skills: Option<Arc<dyn SkillRegistry>>,
    ) -> Self {
        Self {
            agents,
            tools,
            behaviors,
            providers,
            models,
            stop_policies,
            #[cfg(feature = "skills")]
            skills,
        }
    }
}