meerkat-tools 0.5.1

Tool validation and dispatch for Meerkat
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

//! Skill discovery and activation tools.
//!
//! Provides skill discovery/activation/resource/function builtin tools for skill
//! discovery and per-turn activation.

pub mod browse;
pub mod functions;
pub mod load;
pub mod resources;

pub use browse::BrowseSkillsTool;
pub use functions::SkillInvokeFunctionTool;
pub use load::LoadSkillTool;
pub use resources::{SkillListResourcesTool, SkillReadResourceTool};

use std::sync::Arc;

use meerkat_core::skills::SkillRuntime;

use crate::builtin::BuiltinTool;

/// Bundles skill tools for registration in the dispatcher.
pub struct SkillToolSet {
    pub browse: BrowseSkillsTool,
    pub load: LoadSkillTool,
    pub list_resources: SkillListResourcesTool,
    pub read_resource: SkillReadResourceTool,
    pub invoke_function: SkillInvokeFunctionTool,
}

impl SkillToolSet {
    pub fn new(engine: Arc<SkillRuntime>) -> Self {
        Self {
            browse: BrowseSkillsTool::new(Arc::clone(&engine)),
            load: LoadSkillTool::new(Arc::clone(&engine)),
            list_resources: SkillListResourcesTool::new(Arc::clone(&engine)),
            read_resource: SkillReadResourceTool::new(Arc::clone(&engine)),
            invoke_function: SkillInvokeFunctionTool::new(engine),
        }
    }

    pub fn tools(&self) -> Vec<&dyn BuiltinTool> {
        vec![
            &self.browse as &dyn BuiltinTool,
            &self.load as &dyn BuiltinTool,
            &self.list_resources as &dyn BuiltinTool,
            &self.read_resource as &dyn BuiltinTool,
            &self.invoke_function as &dyn BuiltinTool,
        ]
    }
}