Skip to main content

ToolBuilder

Struct ToolBuilder 

Source
pub struct ToolBuilder { /* private fields */ }
Expand description

Fluent builder for defining tools with both schemas and handlers.

ToolBuilder combines tool definitions (sent to the API) with tool handlers (for local execution), similar to the TypeScript/Go SDK patterns.

§Example

use modelrelay::{ToolBuilder, ToolCall};
use schemars::JsonSchema;
use serde::Deserialize;
use serde_json::json;

#[derive(JsonSchema, Deserialize)]
struct WeatherArgs {
    /// City name
    location: String,
    /// Temperature unit (optional)
    unit: Option<String>,
}

let tools = ToolBuilder::new()
    .add_sync::<WeatherArgs, _>("get_weather", "Get weather for a location", |args, _call| {
        Ok(json!({ "temp": 72, "location": args.location }))
    })
    .add_async::<SearchArgs, _>("search", "Search the web", |args, _call| {
        Box::pin(async move {
            Ok(json!({ "results": ["result1", "result2"] }))
        })
    });

// Extract definitions for API request
let (definitions, registry) = tools.build();

Implementations§

Source§

impl ToolBuilder

Source

pub fn new() -> Self

Creates a new empty tool builder.

Source

pub fn add_sync<T, F>( self, name: impl Into<String>, description: impl Into<String>, handler: F, ) -> Self
where T: JsonSchema + DeserializeOwned + Send + 'static, F: Fn(T, ToolCall) -> Result<Value, String> + Send + Sync + 'static,

Adds a synchronous tool handler with type-safe arguments.

The argument type must implement JsonSchema (for generating the schema) and DeserializeOwned (for parsing incoming arguments).

§Example
use schemars::JsonSchema;
use serde::Deserialize;
use modelrelay::ToolBuilder;

#[derive(JsonSchema, Deserialize)]
struct Args { path: String }

let tools = ToolBuilder::new()
    .add_sync::<Args, _>("read_file", "Read a file", |args, _call| {
        Ok(serde_json::json!({ "content": "file contents" }))
    });
Source

pub fn add_async<T, F>( self, name: impl Into<String>, description: impl Into<String>, handler: F, ) -> Self
where T: JsonSchema + DeserializeOwned + Send + 'static, F: Fn(T, ToolCall) -> BoxFuture<'static, Result<Value, String>> + Send + Sync + 'static,

Adds an asynchronous tool handler with type-safe arguments.

The handler returns a boxed future for async execution.

§Example
use schemars::JsonSchema;
use serde::Deserialize;
use modelrelay::ToolBuilder;

#[derive(JsonSchema, Deserialize)]
struct Args { query: String }

let tools = ToolBuilder::new()
    .add_async::<Args>("search", "Search the web", |args, _call| {
        Box::pin(async move {
            Ok(serde_json::json!({ "results": ["result"] }))
        })
    });
Source

pub fn add_raw( self, name: impl Into<String>, description: impl Into<String>, parameters: Option<Value>, handler: ToolHandler, ) -> Self

Adds a raw tool handler without type-safe argument parsing.

Use this when you want to handle argument parsing yourself or need more control over the handler behavior.

Source

pub fn definitions(&self) -> Vec<Tool>

Returns just the tool definitions (for API requests).

Source

pub fn registry(&self) -> &ToolRegistry

Returns just the tool registry (for execution).

Source

pub fn build(self) -> (Vec<Tool>, ToolRegistry)

Consumes the builder and returns both definitions and registry.

This is typically called when passing tools to an agent loop.

Trait Implementations§

Source§

impl Default for ToolBuilder

Source§

fn default() -> Self

Returns the “default value” for a type. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more