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
30
31
32
33
34
35
36
37

use abu_agent::{hook::ConsoleLoggerHook, model::ChatModel, AgentBuilder};
use tracing::{debug, info, level_filters::LevelFilter};

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt()
        .with_target(false)
        .with_max_level(LevelFilter::INFO)
        .with_level(true)
        .init();

    info!("start main");
    if let Err(e) = result_main().await {
        eprint!("Err: {:?}", e);
    }
}

async fn result_main() -> anyhow::Result<()> {
    dotenv::from_filename(".env")?;
    let model = std::env::var("CHAT_MODEL")?;
    let llm = ChatModel::deepseek(model)?;
    let mut agent = AgentBuilder::new(llm)
        .with_builtin_tools(true)
        .system_prompt("You are an autonomous task-solving agent.")
        .with_mcpserver("python3", ["./mcp/weather.py"])
        .with_mcpserver("python3", ["./mcp/username.py"])
        .with_hook(ConsoleLoggerHook::new())
        .build()
        .await?;

    debug!("{:#?}", agent.tool_list());
    
    agent.run("帮我查询上海的天气").await?;
    agent.run("我的名字是？").await?;

    Ok(())
}