rucora-core 0.1.2

Core abstractions and types for the rucora LLM agent framework
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

use async_trait::async_trait;
use rucora_core::error::ToolError;
use rucora_core::tool::Tool;
use serde_json::{Value, json};

struct EchoTool;

#[async_trait]
impl Tool for EchoTool {
    fn name(&self) -> &str {
        "echo"
    }

    fn input_schema(&self) -> Value {
        json!({
            "type": "object",
            "properties": {
                "text": {"type": "string"}
            },
            "required": ["text"]
        })
    }

    async fn call(&self, input: Value) -> Result<Value, ToolError> {
        let text = input
            .get("text")
            .and_then(|v| v.as_str())
            .ok_or_else(|| ToolError::Message("missing field: text".to_string()))?;
        Ok(json!({"success": true, "output": text}))
    }
}

#[tokio::test]
async fn tool_contract_should_have_non_empty_name() {
    let t = EchoTool;
    assert!(!t.name().trim().is_empty());
}

#[tokio::test]
async fn tool_contract_call_should_return_json_value_or_tool_error() {
    let t = EchoTool;

    let ok = t.call(json!({"text": "hi"})).await.unwrap();
    assert_eq!(ok.get("success").and_then(|v| v.as_bool()), Some(true));

    let err = t.call(json!({})).await.unwrap_err();
    match err {
        ToolError::Message(msg) => assert!(msg.contains("text")),
        _ => panic!("unexpected ToolError variant"),
    }
}