molo 0.3.0

An embeddable Rust agent runtime and harness framework for building safe, extensible tool-calling agents
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

use molo::{
    LoadSkillReferenceTool, RunContext, SharedState, Skill, SkillLayer, SkillRegistry,
    SkillResourceStore, ToolRegistry,
};
use std::sync::Arc;

#[tokio::main(flavor = "current_thread")]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let registry = Arc::new(SkillRegistry::new());
    registry.add(Skill::parse(
        "---\n\
         name: code-review\n\
         description: Review code changes for bugs and regressions\n\
         allowed-tools: read_file search_repo\n\
         ---\n\
         Read the diff first, then inspect changed call sites.",
    )?);

    let layer = SkillLayer::new(Arc::clone(&registry));
    let assembly = layer.assemble();

    let mut tools = ToolRegistry::new();
    if let Some(load_skill) = assembly.load_skill_tool {
        tools.register_with_source(load_skill, layer.load_skill_source())?;
    }
    tools.register_with_source(
        LoadSkillReferenceTool::new(
            Arc::clone(&registry),
            None,
            layer.activation_state(),
            SkillResourceStore::default(),
        ),
        LoadSkillReferenceTool::source("skills", molo::ToolTrustLevel::Project),
    )?;

    println!("prompt fragment:\n{}", assembly.prompt_fragment);
    println!("tools: {:?}", tools.names());

    let loaded = tools
        .call_named(
            "load_skill",
            r#"{"name":"code-review"}"#,
            &RunContext::new("skill-layer-example"),
            &SharedState::new(),
        )
        .await?;
    println!("loaded skill bytes: {}", loaded.to_string().len());
    Ok(())
}