robotnik/src/main.rs

use color_eyre::{
    Result,
    eyre::WrapErr,
};
use human_panic::setup_panic;
use tracing::{
    Level,
    info,
};
use tracing_subscriber::FmtSubscriber;

mod chat;
mod command;
mod qna;
mod setup;

const DEFAULT_INSTRUCT: &'static str =
"You are a shady, yet helpful IRC bot.  You try to give responses that can
 be sent in a single IRC response according to the specification.";

#[tokio::main]
async fn main() -> Result<()> {
    // Some error sprucing.
    better_panic::install();
    setup_panic!();

    let subscriber = FmtSubscriber::builder()
        .with_max_level(Level::TRACE)
        .finish();

    tracing::subscriber::set_global_default(subscriber)
        .wrap_err("Failed to setup trace logging.")?;

    info!("Starting");

    let settings = setup::init().await.wrap_err("Failed to initialize.")?;
    let config = settings.config;

    let handle = qna::new(
        config.get_string("api-key").wrap_err("API missing.")?,
        config
            .get_string("base-url")
            .wrap_err("base-url missing.")?,
        config
            .get_string("model")
            .wrap_err("model string missing.")?,
        config.get_string("instruct").unwrap_or_else(|_| DEFAULT_INSTRUCT.to_string()),
    )
    .wrap_err("Couldn't initialize LLM handle.")?;
    let mut c = chat::new(&config, &handle).await?;

    c.run().await.unwrap();

    Ok(())
}