Struct rig::agent::AgentBuilder

pub struct AgentBuilder<M: CompletionModel> { /* private fields */ }

A builder for creating an agent

Example

use rig::{providers::openai, agent::AgentBuilder};

let openai = openai::Client::from_env();

let gpt4o = openai.completion_model("gpt-4o");

// Configure the agent
let agent = AgentBuilder::new(model)
    .preamble("System prompt")
    .context("Context document 1")
    .context("Context document 2")
    .tool(tool1)
    .tool(tool2)
    .temperature(0.8)
    .additional_params(json!({"foo": "bar"}))
    .build();

Implementations

impl<M: CompletionModel> AgentBuilder<M>

pub fn new(model: M) -> Self

pub fn preamble(self, doc: &str) -> Self

Set the preamble of the agent

pub fn append_preamble(self, doc: &str) -> Self

Append to the preamble of the agent

pub fn context(self, doc: &str) -> Self

Add a static context document to the agent

pub fn tool(self, tool: impl Tool + 'static) -> Self

Add a static tool to the agent

pub fn temperature(self, temperature: f64) -> Self

Set the temperature of the model

pub fn additional_params(self, params: Value) -> Self

Set additional parameters to be passed to the model

pub fn build(self) -> Agent<M>

Build the agent