ai-agents-runtime 1.0.0

Runtime agent and builder for AI Agents framework
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

//! Runtime agent and builder for AI Agents framework

mod builder;
pub mod optimization;
mod runtime;
mod streaming;
mod turn_context;

pub mod orchestration;
pub mod spawner;
pub mod spec;

pub use builder::AgentBuilder;
pub use optimization::{
    AwaitBeforeNextTurn, BackgroundOverflowPolicy, MainResponseDraft, MaintenanceMode,
    MaintenanceTaskPolicy, PostTurnOptimizationConfig, RuntimeBranch, RuntimeBranchOutcome,
    RuntimeBranchResult, RuntimeBranchStatus, RuntimeCommitBehavior, RuntimeConfig,
    RuntimeOptimizationConfig, RuntimeOptimizationKind, RuntimeTaskPriority, RuntimeTaskPurpose,
    ScheduledBranchSet, SkillCandidate, StreamBranchBuffer, StreamingDraftResult,
    StreamingOptimizationPolicy, TurnBranchScheduler, TurnOptimizationContext,
};
pub use runtime::{RuntimeAgent, RuntimeControlHandle};
pub use streaming::{StreamChunk, StreamingConfig};
pub use turn_context::TurnActorContext;

pub use ai_agents_core::{AgentInfo, AgentResponse, Result, ToolCall};

use async_trait::async_trait;
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ParallelToolsConfig {
    #[serde(default = "default_parallel_enabled")]
    pub enabled: bool,
    #[serde(default = "default_max_parallel")]
    pub max_parallel: usize,
}

fn default_parallel_enabled() -> bool {
    true
}

fn default_max_parallel() -> usize {
    5
}

impl Default for ParallelToolsConfig {
    fn default() -> Self {
        Self {
            enabled: true,
            max_parallel: 5,
        }
    }
}

#[async_trait]
pub trait Agent: Send + Sync {
    async fn chat(&self, input: &str) -> Result<AgentResponse>;
    fn info(&self) -> AgentInfo;
    async fn reset(&self) -> Result<()>;
}