llm-chain-tools 0.4.0

A library for providing Large Language Models with tools (also known as 'actions') that they can trigger
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::description::ToolDescription;

macro_rules! gen_invoke_function {
    () => {
        fn invoke(&self, input: serde_yaml::Value) -> Result<serde_yaml::Value, String> {
            let input = serde_yaml::from_value(input).unwrap();
            let output = self.invoke_typed(&input).unwrap();
            Ok(serde_yaml::to_value(output).unwrap())
        }
    };
}
pub(crate) use gen_invoke_function;

pub trait Tool {
    fn description(&self) -> ToolDescription;
    fn invoke(&self, input: serde_yaml::Value) -> Result<serde_yaml::Value, String>;
    fn matches(&self, name: &str) -> bool {
        self.description().name == name
    }
}