gemini_client_api/gemini/
utils.rs

1use super::types::request::*;
2use crate::utils::{self, MatchedFiles};
3use regex::Regex;
4
5pub struct MarkdownToParts<'a> {
6    base64s: Vec<Option<MatchedFiles>>,
7    markdown: &'a str,
8}
9impl<'a> MarkdownToParts<'a> {
10    pub async fn from_regex(
11        markdown: &'a str,
12        regex: Regex,
13        guess_mime_type: fn(url: &str) -> String,
14    ) -> Self {
15        Self {
16            base64s: utils::get_file_base64s(markdown, regex, guess_mime_type).await,
17            markdown,
18        }
19    }
20    ///Converts markdown to parts considering `![image](link)` means Gemini will be see the images too. `link` can be URL or file path.
21    pub async fn new(markdown: &'a str, guess_mime_type: fn(url: &str) -> String) -> Self {
22        let image_regex = Regex::new(r"(?s)!\[.*?].?\((.*?)\)").unwrap();
23        Self {
24            base64s: utils::get_file_base64s(markdown, image_regex, guess_mime_type).await,
25            markdown,
26        }
27    }
28    pub fn process(mut self) -> Vec<Part> {
29        let mut parts: Vec<Part> = Vec::new();
30        for file in self.base64s {
31            if let Some(file) = file {
32                let end = file.index + file.length;
33                let text = &self.markdown[..end];
34                parts.push(Part::text(text.to_string()));
35                parts.push(Part::inline_data(InlineData::new(
36                    file.mime_type,
37                    file.base64,
38                )));
39
40                self.markdown = &self.markdown[end..];
41            }
42        }
43        if self.markdown.len() != 0 {
44            parts.push(Part::text(self.markdown.to_string()));
45        }
46        parts
47    }
48}
gemini_client_api/gemini/utils.rs

gemini_client_api/gemini/
utils.rs
