monothink 0.6.0

A totally non-agentic coding assistant
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

/*
* This file is part of monothink.
*
* monothink is free software: you can redistribute it and/or modify it under the terms
* of the GNU General Public License as published by the Free Software Foundation,
* either version 3 of the License, or (at your option) any later version.
*
* monothink is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with monothink.
* If not, see <https://www.gnu.org/licenses/>.

*/

use crate::llm::compactors::SharedObservations;
use rig::{
    message::ToolResultContent,
    tool::{Tool, ToolContext},
};
use serde::{Deserialize, Serialize};
use serde_json::json;

#[derive(Debug, Deserialize, Serialize)]
pub struct RetrieveObservationArgs {
    id: u64,
}

#[derive(Debug)]
pub struct RetrieveObservation {
    pub observations: SharedObservations,
}

impl RetrieveObservation {
    pub fn new(observations: SharedObservations) -> Self {
        Self { observations }
    }
}

impl Tool for RetrieveObservation {
    const NAME: &'static str = "retrieve_observation";

    type Error = super::ToolError;
    type Args = RetrieveObservationArgs;
    type Output = Vec<ToolResultContent>;

    #[tracing::instrument(skip(self, _context))]
    async fn call(
        &self,
        _context: &mut ToolContext,
        args: Self::Args,
    ) -> Result<Self::Output, Self::Error> {
        self.observations
            .lock()
            .await
            .get(&args.id)
            .ok_or_else(|| super::ToolError::ObservationNotFound {
                identifier: args.id.to_string(),
            })
            .cloned()
    }

    fn description(&self) -> String {
        "Retrieve the full content of a masked tool result from conversation history. \
     When conversation history is compacted to save context space, older tool results \
     are replaced with placeholders like '[Observation:3 - masked]'. \
     Use this tool when you need to see the full content of a masked result. \
     The number inside the brackets (e.g., 3 in '[Observation:3 - masked]') is the \
     id you should pass to this tool. Each tool result in the history is assigned a \
     unique observation number starting from 1."
            .to_string()
    }

    fn parameters(&self) -> serde_json::Value {
        let id_description = "The integer ID `N` from a masked tool result placeholder (e.g., `[Observation:N - masked]`). \
            For example, if you see '[Observation:3 - masked]', pass 3 as this parameter. \
            **Only** use this tool to recover outputs that are explicitly masked in the *current* conversation history. \
            Do not guess IDs or attempt to retrieve observations that are not currently visible as placeholders.";

        json!({
            "type": "object",
            "properties": {
                "id": {
                    "type": "integer",
                    "description": id_description
                 },
            },
            "required": ["id"]
        })
    }

    fn map_error(&self, error: Self::Error) -> rig::tool::ToolExecutionError {
        rig::tool::ToolExecutionError::from_error(error)
    }
}