Struct rig::model::ModelBuilder

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

A builder for creating a model

Example

use rig::{providers::openai, model::ModelBuilder};

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

let gpt4 = openai_client.completion_model("gpt-4");

// Configure the model
let model = ModelBuilder::new(model)
    .temperature(0.8)
    .build();

Implementations

impl<M: CompletionModel> ModelBuilder<M>

pub fn new(model: M) -> Self

Create a new model builder

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

Set the temperature of the model

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

Set the temperature of the model (set to None to use the default temperature of the model)

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

Build the model