adk-gemini 0.3.2

Rust client for Google Gemini API
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

//! An example of uploading a file, downloading it, and verifying the content.
use adk_gemini::Gemini;
use display_error_chain::DisplayErrorChain;
use futures::TryStreamExt;
use std::process::ExitCode;
use tracing::{error, info};

const TEST_CONTENT: &str = "Hello, world! This is a test file for Gemini-Rust.";

#[tokio::main]
async fn main() -> ExitCode {
    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::builder()
                .with_default_directive(tracing::level_filters::LevelFilter::INFO.into())
                .from_env_lossy(),
        )
        .init();

    match do_main().await {
        Ok(()) => ExitCode::SUCCESS,
        Err(e) => {
            let error_chain = DisplayErrorChain::new(e.as_ref());
            tracing::error!(error.debug = ?e, error.chained = %error_chain, "execution failed");
            ExitCode::FAILURE
        }
    }
}

async fn do_main() -> Result<(), Box<dyn std::error::Error>> {
    let api_key = std::env::var("GEMINI_API_KEY").expect("GEMINI_API_KEY must be set");
    let gemini = Gemini::new(&api_key)?;

    info!("uploading a file with known content");

    // 1. Upload a file
    let file_handle = gemini
        .create_file(TEST_CONTENT)
        .display_name("files-lifecycle-test.bin")
        .with_mime_type(mime::APPLICATION_OCTET_STREAM)
        .upload()
        .await?;

    info!(file_name = file_handle.name(), "file uploaded successfully");

    // 2. Find remote file
    let files = gemini.list_files(None).try_collect::<Vec<_>>().await?;
    let remote_file = files
        .iter()
        .find(|file| file.name() == file_handle.name())
        .expect("uploaded file should be present");

    assert!(
        remote_file.download().await.is_err(),
        "Download of user-uploaded files is not supported by the Gemini API; only generated files can be downloaded"
    );

    // 3. Assert that the byte size is identical
    info!("verifying file content");
    assert_eq!(
        remote_file.get_file_meta().size_bytes.unwrap(),
        TEST_CONTENT.len() as i64,
        "File size does not match expected size!"
    );

    // 4. Delete the file
    info!("cleaning up by deleting the file");
    match file_handle.delete().await {
        Ok(_) => info!("file deleted successfully"),
        Err((_, e)) => error!(error = %e, "failed to delete file"),
    }

    Ok(())
}