Re-export client types and model const

This commit is contained in:
Collin Brittain 2025-04-11 11:50:29 -05:00
parent bffb245eed
commit 329805c9ee
2 changed files with 5 additions and 2 deletions

View File

@ -1,6 +1,6 @@
use std::env;
use rig::{completion::Prompt, providers::inception::client::ClientBuilder};
use rig::{completion::Prompt, providers::inception::{ClientBuilder, MERCURY_CODER_SMALL}};
#[tokio::main]
async fn main() -> Result<(), anyhow::Error> {
@ -11,7 +11,7 @@ async fn main() -> Result<(), anyhow::Error> {
// Create agent with a single context prompt
let agent = client
.agent("mercury-coder-small")
.agent(MERCURY_CODER_SMALL)
.preamble("You are a helpful AI assistant.")
.temperature(0.0)
.build();

View File

@ -1,2 +1,5 @@
pub mod client;
pub mod completion;
pub use client::{Client, ClientBuilder};
pub use completion::MERCURY_CODER_SMALL;