sgr-agent-tools 0.1.0

11 reusable file-system tools for sgr-agent based AI agents
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

//! MkDirTool — create a directory.

use std::sync::Arc;

use schemars::JsonSchema;
use serde::Deserialize;
use serde_json::Value;
use sgr_agent_core::agent_tool::{Tool, ToolError, ToolOutput, parse_args};
use sgr_agent_core::context::AgentContext;
use sgr_agent_core::schema::json_schema_for;

use crate::backend::FileBackend;
use crate::helpers::backend_err;

pub struct MkDirTool<B: FileBackend>(pub Arc<B>);

#[derive(Deserialize, JsonSchema)]
struct MkDirArgs {
    /// Directory path to create
    path: String,
}

#[async_trait::async_trait]
impl<B: FileBackend> Tool for MkDirTool<B> {
    fn name(&self) -> &str { "mkdir" }
    fn description(&self) -> &str { "Create a directory" }
    fn parameters_schema(&self) -> Value { json_schema_for::<MkDirArgs>() }
    async fn execute(&self, args: Value, _ctx: &mut AgentContext) -> Result<ToolOutput, ToolError> {
        let a: MkDirArgs = parse_args(&args)?;
        self.0.mkdir(&a.path).await.map_err(backend_err)?;
        Ok(ToolOutput::text(format!("Created directory {}", a.path)))
    }
}