rig-core 0.42.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
33

//! Embeds an image with Cohere Embed v3.
//!
//! Set `COHERE_API_KEY`, then run:
//!
//! ```text
//! cargo run -p rig-core --example cohere_image_embeddings -- path/to/image.png
//! ```

use anyhow::{Context, Result};
use rig_core::{
    client::ProviderClient, embeddings::ImageEmbeddingModel, providers::cohere::Client,
};

#[tokio::main]
async fn main() -> Result<()> {
    let path = std::env::args_os()
        .nth(1)
        .context("pass the path to a PNG, JPEG, WebP, or GIF image")?;
    let image = std::fs::read(&path)
        .with_context(|| format!("failed to read image at {}", path.to_string_lossy()))?;

    let client = Client::from_env()?;
    let model = client.image_embedding_model();
    let embedding = model.embed_image(&image).await?;

    println!(
        "embedded {} bytes into {} dimensions",
        image.len(),
        embedding.vec.len()
    );

    Ok(())
}