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    ///# Panics
11    /// `regex` must have a Regex with atleast 1 capture group with file URL as first capture group, else it PANICS.
12    /// # Arguments
13    /// `mime_type_guess` is used to detect mimi_type of URL pointing to file system or web resource
14    /// with no "Content-Type" header.
15    /// # Example
16    /// ```rust
17    /// from_regex("Your markdown string...", Regex::new(r"(?s)!\[.*?].?\((.*?)\)").unwrap(), |_| "image/png".to_string())
18    /// ```
19    pub async fn from_regex(
20        markdown: &'a str,
21        regex: Regex,
22        guess_mime_type: fn(url: &str) -> String,
23    ) -> Self {
24        Self {
25            base64s: utils::get_file_base64s(markdown, regex, guess_mime_type).await,
26            markdown,
27        }
28    }
29    ///Converts markdown to parts considering `![image](link)` means Gemini will be see the images too. `link` can be URL or file path.  
30    /// `mime_type_guess` is used to detect mimi_type of URL pointing to file system or web resource
31    /// with no "Content-Type" header.
32    /// # Example
33    /// ```rust
34    /// new("Your markdown string...", |_| "image/png".to_string())
35    /// ```
36    pub async fn new(markdown: &'a str, guess_mime_type: fn(url: &str) -> String) -> Self {
37        let image_regex = Regex::new(r"(?s)!\[.*?].?\((.*?)\)").unwrap();
38        Self {
39            base64s: utils::get_file_base64s(markdown, image_regex, guess_mime_type).await,
40            markdown,
41        }
42    }
43    pub fn process(mut self) -> Vec<Part> {
44        let mut parts: Vec<Part> = Vec::new();
45        let mut removed_length = 0;
46        for file in self.base64s {
47            if let Some(file) = file {
48                let end = file.index + file.length - removed_length;
49                let text = &self.markdown[..end];
50                parts.push(Part::text(text.to_string()));
51                parts.push(Part::inline_data(InlineData::new(
52                    file.mime_type,
53                    file.base64,
54                )));
55
56                self.markdown = &self.markdown[end..];
57                removed_length += end;
58            }
59        }
60        if self.markdown.len() != 0 {
61            parts.push(Part::text(self.markdown.to_string()));
62        }
63        parts
64    }
65}
gemini_client_api/gemini/utils.rs

gemini_client_api/gemini/
utils.rs
