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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use serde::{Deserialize, Serialize};
use serde_json::Value;

use super::types::TaskStatus;

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub(crate) struct TaskCreateInput {
    pub(crate) subject: String,
    #[serde(default)]
    pub(crate) description: String,
    #[serde(default)]
    pub(crate) owner: String,
    #[serde(default)]
    pub(crate) working_directory: Option<String>,
    #[serde(default)]
    pub(crate) blocked_by: Vec<u64>,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub(crate) struct TaskClaimInput {
    #[serde(default)]
    pub(crate) task_id: Option<u64>,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub(crate) struct TaskUpdateInput {
    pub(crate) task_id: u64,
    #[serde(default)]
    pub(crate) subject: Option<String>,
    #[serde(default)]
    pub(crate) description: Option<String>,
    #[serde(default)]
    pub(crate) owner: Option<String>,
    #[serde(default)]
    pub(crate) working_directory: Option<Option<String>>,
    #[serde(default)]
    pub(crate) status: Option<TaskStatus>,
    #[serde(default)]
    pub(crate) add_blocked_by: Vec<u64>,
    #[serde(default)]
    pub(crate) remove_blocked_by: Vec<u64>,
    #[serde(default)]
    pub(crate) add_blocks: Vec<u64>,
    #[serde(default)]
    pub(crate) remove_blocks: Vec<u64>,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub(crate) struct TaskGetInput {
    pub(crate) task_id: u64,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub(crate) struct TaskListInput {}

pub(crate) fn parse_task_create_input(input: Value) -> Result<TaskCreateInput, String> {
    let parsed = serde_json::from_value::<TaskCreateInput>(input)
        .map_err(|error| format!("Invalid task_create input: {error}"))?;

    if parsed.subject.trim().is_empty() {
        return Err("Task subject must not be empty".to_string());
    }

    Ok(TaskCreateInput {
        working_directory: normalize_optional_path(parsed.working_directory),
        ..parsed
    })
}

pub(crate) fn parse_task_update_input(input: Value) -> Result<TaskUpdateInput, String> {
    let parsed = serde_json::from_value::<TaskUpdateInput>(input)
        .map_err(|error| format!("Invalid task_update input: {error}"))?;

    if matches!(parsed.subject.as_deref(), Some(subject) if subject.trim().is_empty()) {
        return Err("Task subject must not be empty".to_string());
    }

    Ok(TaskUpdateInput {
        working_directory: parsed.working_directory.map(normalize_optional_path),
        ..parsed
    })
}

pub(crate) fn parse_task_claim_input(input: Value) -> Result<TaskClaimInput, String> {
    serde_json::from_value::<TaskClaimInput>(input)
        .map_err(|error| format!("Invalid task_claim input: {error}"))
}

pub(crate) fn parse_task_get_input(input: Value) -> Result<TaskGetInput, String> {
    serde_json::from_value::<TaskGetInput>(input)
        .map_err(|error| format!("Invalid task_get input: {error}"))
}

pub(crate) fn parse_task_list_input(input: Value) -> Result<(), String> {
    serde_json::from_value::<TaskListInput>(input)
        .map(|_| ())
        .map_err(|error| format!("Invalid task_list input: {error}"))
}

fn normalize_optional_path(value: Option<String>) -> Option<String> {
    value.and_then(|value| {
        let trimmed = value.trim();
        (!trimmed.is_empty()).then(|| trimmed.to_string())
    })
}