monothink 0.4.7

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

/*
* 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 Wavething.
* If not, see <https://www.gnu.org/licenses/>.

*/

use rig::tool::{Tool, ToolContext};
use serde::{Deserialize, Serialize};
use serde_json::json;
use std::sync::Arc;
use tokio::sync::Mutex;

use crate::llm::package_resolver::PackageResolver;

#[derive(Deserialize, Serialize)]
pub struct FindPackageArgs {
    pattern: String,
}

pub struct FindPackage {
    dependencies: Arc<Mutex<PackageResolver>>,
}

impl FindPackage {
    pub fn new(dependencies: Arc<Mutex<PackageResolver>>) -> Self {
        Self { dependencies }
    }
}
impl Tool for FindPackage {
    const NAME: &'static str = "find_package";

    type Error = super::ToolError;
    type Args = FindPackageArgs;
    type Output = Vec<String>;

    async fn call(
        &self,
        _context: &mut ToolContext,
        args: Self::Args,
    ) -> Result<Self::Output, Self::Error> {
        let dependencies = self.dependencies.lock().await;

        Ok(dependencies.search_dependency(&args.pattern))
    }

    fn description(&self) -> String {
        // @todo check for multiple package manager when supported
        "Find all packages/dependencies whose name matches the provided pattern. \
            Returns a list of identifier for the packages that match the pattern."
            .to_string()
    }

    fn parameters(&self) -> serde_json::Value {
        // @todo support pacman parameter once multiple package managers are supported
        json!({
            "type": "object",
            "properties": {
                "pattern": {
                    "type": "string",
                    "description": "Pattern to match agains the names of packages. \
                     Matches all packages that contain the pattern text as part of their name."
                },
            },
            "required": ["pattern"]
        })
    }

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