adk_gemini/files/
builder.rs

1use mime::Mime;
2use snafu::ResultExt;
3use std::sync::Arc;
4use tracing::instrument;
5
6use super::*;
7use crate::client::GeminiClient;
8
9/// A builder for creating a file resource.
10pub struct FileBuilder {
11    client: Arc<GeminiClient>,
12    file_bytes: Vec<u8>,
13    display_name: Option<String>,
14    mime_type: Option<Mime>,
15}
16
17impl FileBuilder {
18    pub(crate) fn new<B: Into<Vec<u8>>>(client: Arc<GeminiClient>, file_bytes: B) -> Self {
19        Self { client, file_bytes: file_bytes.into(), display_name: None, mime_type: None }
20    }
21
22    /// The display name of the file.
23    pub fn display_name(mut self, display_name: impl Into<String>) -> Self {
24        self.display_name = Some(display_name.into());
25        self
26    }
27
28    /// The MIME type of the file.
29    pub fn with_mime_type(mut self, mime_type: Mime) -> Self {
30        self.mime_type = Some(mime_type);
31        self
32    }
33
34    /// Upload the file.
35    #[instrument(skip_all, fields(
36        file.size = self.file_bytes.len(),
37        mime.type = self.mime_type.as_ref().map(|m| m.to_string()),
38        file.display_name = self.display_name,
39    ))]
40    pub async fn upload(self) -> Result<super::handle::FileHandle, super::Error> {
41        let mime_type = self.mime_type.unwrap_or(mime::APPLICATION_OCTET_STREAM);
42
43        let file = self
44            .client
45            .upload_file(self.display_name, self.file_bytes, mime_type)
46            .await
47            .context(ClientSnafu)?;
48
49        Ok(super::handle::FileHandle::new(self.client, file))
50    }
51}
adk_gemini/files/builder.rs

adk_gemini/files/
builder.rs
