Tools

use picrust::{Tool, ToolResult};
use async_trait::async_trait;
use serde_json::Value;

pub struct CustomTool;

#[async_trait]
impl Tool for CustomTool {
    fn name(&self) -> &str {
        "custom_tool"
    }

    fn description(&self) -> &str {
        "Does something custom"
    }

    fn parameters(&self) -> Value {
        serde_json::json!({
            "type": "object",
            "properties": {
                "input": {
                    "type": "string",
                    "description": "Input parameter"
                }
            },
            "required": ["input"]
        })
    }

    async fn execute(&self, params: Value) -> ToolResult {
        let input = params["input"].as_str().unwrap();

        // Your custom logic here
        let result = format!("Processed: {}", input);

        Ok(result)
    }
}

agent.add_tool(CustomTool);

use picrust::{Agent, tools::*};

#[tokio::main]
async fn main() -> Result<()> {
    let mut agent = Agent::new("api-key").await?;

    // Add file tools
    agent.add_tool(ReadFile::new());
    agent.add_tool(Glob::new());

    // Ask agent to analyze project structure
    let response = agent.execute(
        "List all Rust files in the src directory and \
         summarize the main modules"
    ).await?;

    println!("{}", response);
    Ok(())
}

let mut agent = Agent::new("api-key").await?;
agent.add_tool(BashExecutor::new());

let response = agent.execute(
    "Run the test suite and report any failures"
).await?;