gemini-client-api 7.5.2

Library to use Google Gemini API. Automatic context management, schema generation, function calling and more.
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

use gemini_client_api::gemini::embed::GeminiEmbedding;
use gemini_client_api::gemini::types::embedding::TaskType;
use std::env;

#[tokio::main]
async fn main() {
    // 1. Create the Gemini Embedding client
    // Get your API key from https://aistudio.google.com/app/apikey
    let api_key = env::var("GEMINI_API_KEY").expect("GEMINI_API_KEY must be set");
    let embedder = GeminiEmbedding::new(api_key, "gemini-embedding-001")
        .set_task_type(TaskType::RetrievalDocument)
        // Optional: reduce dimension for Matryoshka Representation Learning
        .set_output_dimensionality(256);

    // 2. Generate embedding for a single text
    let prompt = "Rust is a blazing fast and memory-efficient systems programming language.";
    let response = embedder.embed_text(prompt).await.unwrap();

    // 3. Print the embedding information
    let embedding = response.embedding();
    println!("Embedding generated for: {:?}", prompt);
    println!("Total Dimensions: {}", embedding.dimension());
    println!(
        "First 5 values: {:?}",
        &embedding.values()[..usize::min(embedding.dimension(), 5)]
    );
}