abu-agent 0.2.0

Agent development library
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

mod error;
pub use error::*;
mod build;
pub use build::*;
mod lloop;

use abu_provider::ChatProvide;
use crate::context::ContextBuilder;
use crate::hook::HookManager;
use crate::memory::Memory;
use crate::middleware::MiddlewareManager;
use crate::model::ChatModel;
use crate::toolbox::ToolBox;

#[derive(Clone)]
pub struct AgentConfig {
    pub max_iteration: usize,
    pub temperature: f64,
}

pub struct Agent<P: ChatProvide, M: Memory> {
    pub config: AgentConfig,
    pub llm: ChatModel<P>,
    pub memory: M,
    pub context_builder: ContextBuilder,
    pub toolbox: ToolBox,
    pub hooks: HookManager,
    pub middlewares: MiddlewareManager,
}