mentra 0.6.0

An agent runtime for tool-using LLM applications
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

use serde_json::Value;

use crate::ContentBlock;

use super::{
    RuntimeToolDescriptor, ToolApprovalCategory, ToolCapability, ToolDurability,
    ToolExecutionCategory, ToolResult, ToolSideEffectLevel,
};

pub(crate) struct RuntimeDescriptorParts {
    pub(crate) name: String,
    pub(crate) description: String,
    pub(crate) input_schema: Value,
    pub(crate) capabilities: Vec<ToolCapability>,
    pub(crate) side_effect_level: ToolSideEffectLevel,
    pub(crate) durability: ToolDurability,
    pub(crate) execution_category: ToolExecutionCategory,
    pub(crate) approval_category: ToolApprovalCategory,
}

pub(crate) fn build_runtime_descriptor(parts: RuntimeDescriptorParts) -> RuntimeToolDescriptor {
    RuntimeToolDescriptor::builder(parts.name)
        .description(parts.description)
        .input_schema(parts.input_schema)
        .capabilities(parts.capabilities)
        .side_effect_level(parts.side_effect_level)
        .durability(parts.durability)
        .execution_category(parts.execution_category)
        .approval_category(parts.approval_category)
        .build()
}

pub(crate) fn content_block_to_tool_result(surface: &str, block: ContentBlock) -> ToolResult {
    match block {
        ContentBlock::ToolResult {
            content, is_error, ..
        } => {
            if is_error {
                Err(content.to_display_string())
            } else {
                Ok(content.to_display_string())
            }
        }
        _ => Err(format!("{surface} returned an unexpected content block")),
    }
}