mirror of https://github.com/0xplaygrounds/rig
Re-export client types and model const
This commit is contained in:
parent
bffb245eed
commit
329805c9ee
|
@ -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();
|
||||
|
|
|
@ -1,2 +1,5 @@
|
|||
pub mod client;
|
||||
pub mod completion;
|
||||
|
||||
pub use client::{Client, ClientBuilder};
|
||||
pub use completion::MERCURY_CODER_SMALL;
|
||||
|
|
Loading…
Reference in New Issue