agent-base 0.1.3

A lightweight Agent Runtime Kernel for building AI agents in Rust
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

use async_trait::async_trait;
use serde_json::Value;

use crate::types::{AgentResult, ApprovalRequest};
use super::{ToolContext, ToolOutput};

#[async_trait]
pub trait ToolPolicy: Send + Sync {
    async fn evaluate_approval(
        &self,
        tool_name: &str,
        args: &Value,
    ) -> Option<ApprovalRequest>;

    fn before_call(&self, tool_name: &str, args: &Value, ctx: &ToolContext) -> AgentResult<()> {
        let _ = (tool_name, args, ctx);
        Ok(())
    }

    fn after_call(&self, tool_name: &str, args: &Value, result: &ToolOutput, ctx: &ToolContext) -> AgentResult<()> {
        let _ = (tool_name, args, result, ctx);
        Ok(())
    }
}