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

use mime::Mime;
use snafu::ResultExt;
use std::sync::Arc;
use tracing::instrument;

use super::*;
use crate::client::GeminiClient;

/// A builder for creating a file resource.
pub struct FileBuilder {
    client: Arc<GeminiClient>,
    file_bytes: Vec<u8>,
    display_name: Option<String>,
    mime_type: Option<Mime>,
}

impl FileBuilder {
    pub(crate) fn new<B: Into<Vec<u8>>>(client: Arc<GeminiClient>, file_bytes: B) -> Self {
        Self { client, file_bytes: file_bytes.into(), display_name: None, mime_type: None }
    }

    /// The display name of the file.
    pub fn display_name(mut self, display_name: impl Into<String>) -> Self {
        self.display_name = Some(display_name.into());
        self
    }

    /// The MIME type of the file.
    pub fn with_mime_type(mut self, mime_type: Mime) -> Self {
        self.mime_type = Some(mime_type);
        self
    }

    /// Upload the file.
    #[instrument(skip_all, fields(
        file.size = self.file_bytes.len(),
        mime.type = self.mime_type.as_ref().map(|m| m.to_string()),
        file.display_name = self.display_name,
    ))]
    pub async fn upload(self) -> Result<super::handle::FileHandle, super::Error> {
        let mime_type = self.mime_type.unwrap_or(mime::APPLICATION_OCTET_STREAM);

        let file = self
            .client
            .upload_file(self.display_name, self.file_bytes, mime_type)
            .await
            .context(ClientSnafu)?;

        Ok(super::handle::FileHandle::new(self.client, file))
    }
}