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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use super::policy::is_scope_allowed;
use super::AgentOs;
use crate::composition::AgentCatalog;
use crate::contracts::runtime::tool_call::{Tool, ToolDescriptor, ToolResult};
use crate::contracts::scope::ScopeDomain;
use crate::contracts::thread::{Message, Role, ToolCall};
use crate::contracts::{AgentEvent, Suspension};
#[cfg(feature = "permission")]
use tirea_extension_permission::ToolPermissionBehavior;

#[cfg(not(feature = "permission"))]
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[allow(dead_code)]
pub(crate) enum ToolPermissionBehavior {
    Allow,
    Ask,
    Deny,
}
use crate::runtime::loop_runner::RunCancellationToken;
use async_trait::async_trait;
use futures::StreamExt;
use serde_json::{json, Value};
use std::collections::{HashMap, HashSet, VecDeque};
use std::sync::Arc;
use tokio::sync::Mutex;
use types::{
    SubAgent, SubAgentAction, SubAgentExecutionRef, SubAgentRemoteProtocol, SubAgentState,
    SubAgentStatus,
};

pub(crate) const AGENT_TOOLS_PLUGIN_ID: &str = "agent_tools";
pub(crate) const AGENT_RECOVERY_PLUGIN_ID: &str = "agent_recovery";
pub(crate) const AGENT_RUN_TOOL_ID: &str = "agent_run";
pub(crate) const AGENT_STOP_TOOL_ID: &str = "agent_stop";
pub(crate) const AGENT_OUTPUT_TOOL_ID: &str = "agent_output";
#[cfg(feature = "handoff")]
pub(crate) const AGENT_HANDOFF_TOOL_ID: &str = "agent_handoff";
pub(crate) const AGENT_RECOVERY_INTERACTION_ACTION: &str = "recover_agent_run";
pub(crate) const AGENT_RECOVERY_INTERACTION_PREFIX: &str = "agent_recovery_";

fn collect_descendant_run_ids(
    children_by_parent: &HashMap<String, Vec<String>>,
    root_run_id: &str,
    include_root: bool,
) -> Vec<String> {
    let mut queue = VecDeque::from([root_run_id.to_string()]);
    let mut seen: HashSet<String> = HashSet::new();
    let mut out = Vec::new();
    while let Some(id) = queue.pop_front() {
        if !seen.insert(id.clone()) {
            continue;
        }
        if include_root || id != root_run_id {
            out.push(id.clone());
        }
        if let Some(children) = children_by_parent.get(&id) {
            for child_id in children {
                queue.push_back(child_id.clone());
            }
        }
    }
    out
}

mod backend;
mod manager;
mod plugins;
mod remote_a2a;
mod state;
mod tools;
mod types;

use backend::{
    resolve_backend_for_execution, resolve_backend_for_target, AgentBackend,
    AgentBackendOutputSyncRequest, AgentBackendRefreshRequest, AgentBackendStartRequest,
};
#[cfg(test)]
use manager::SubAgentCompletion;
use manager::{execute_sub_agent, SubAgentExecutionRequest};
pub(crate) use manager::{SubAgentHandleTable, SubAgentSummary};
pub(super) use plugins::{AgentRecoveryPlugin, AgentToolsPlugin};
use state::*;
#[cfg(feature = "handoff")]
pub(super) use tools::AgentHandoffTool;
pub(super) use tools::{AgentOutputTool, AgentRunTool, AgentStopTool};

#[cfg(test)]
mod tests;