mentra 0.5.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

use serde_json::Value;

use crate::tool::ToolExecutionCategory;

use super::schema::{FileOperation, FilesInput};

pub(crate) fn parse_files_input(input: &Value) -> Result<FilesInput, String> {
    serde_json::from_value::<FilesInput>(input.clone())
        .map_err(|error| format!("Invalid files input: {error}"))
}

pub(crate) fn ensure_files_have_operations(input: &FilesInput) -> Result<(), String> {
    if input.operations.is_empty() {
        Err("At least one file operation is required".to_string())
    } else {
        Ok(())
    }
}

pub(crate) fn file_execution_category(input: &Value) -> ToolExecutionCategory {
    let Ok(input) = parse_files_input(input) else {
        return ToolExecutionCategory::ExclusiveLocalMutation;
    };

    if input.operations.iter().all(FileOperation::is_read_only) {
        ToolExecutionCategory::ReadOnlyParallel
    } else {
        ToolExecutionCategory::ExclusiveLocalMutation
    }
}