rig 0.37.0

An opinionated library for building LLM powered applications.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

#[path = "../../common/support.rs"]
mod support;

use rig::bedrock::{client::Client, completion, embedding, image as bedrock_image};
use rig::client::ProviderClient;

pub(crate) const BEDROCK_COMPLETION_MODEL: &str = completion::AMAZON_NOVA_LITE;
pub(crate) const BEDROCK_EMBEDDING_MODEL: &str = embedding::AMAZON_TITAN_EMBED_TEXT_V2_0;
pub(crate) const BEDROCK_IMAGE_MODEL: &str = bedrock_image::AMAZON_NOVA_CANVAS;

pub(crate) fn anthropic_adaptive_model() -> String {
    std::env::var("BEDROCK_ANTHROPIC_ADAPTIVE_MODEL")
        .unwrap_or_else(|_| "us.anthropic.claude-sonnet-4-6".to_string())
}

pub(crate) fn anthropic_signature_only_model() -> String {
    std::env::var("BEDROCK_ANTHROPIC_SIGNATURE_ONLY_MODEL")
        .unwrap_or_else(|_| "global.anthropic.claude-opus-4-7".to_string())
}

pub(crate) fn client() -> Client {
    Client::from_env().expect("client should build")
}

mod adaptive_thinking;
mod agent;
mod documents;
mod embeddings;
mod extractor;
mod image_generation;
mod image_prompt;
mod streaming;