kova-sdk 0.1.0

Async-first Rust library for building LLM-powered agents with tool calling, streaming, MCP, and multi-agent orchestration
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
34
35
36
37

pub mod approval;
pub mod registry;

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

use crate::error::KovaError;
use crate::models::ToolResult;

/// A callable tool that the LLM can invoke to perform actions.
#[async_trait]
pub trait Tool: Send + Sync {
    /// Unique name for this tool.
    fn name(&self) -> &str;

    /// Human-readable description for the LLM.
    fn description(&self) -> &str;

    /// JSON Schema describing the tool's parameters.
    fn parameters_schema(&self) -> Value;

    /// Execute the tool with the given JSON arguments.
    async fn execute(&self, args: Value) -> Result<ToolResult, KovaError>;
}

/// Optional lifecycle hook for observing tool execution.
///
/// Implement this to display progress, log, or apply custom
/// side-effects around each tool call without modifying tools.
#[async_trait]
pub trait ToolLifecycleHook: Send + Sync {
    /// Called immediately before a tool is executed.
    async fn on_tool_start(&self, tool_name: &str, args: &Value);

    /// Called immediately after a tool returns its result.
    async fn on_tool_end(&self, tool_name: &str, result: &str, is_error: bool);
}