diff options
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/input.rs | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/config/input.rs b/src/config/input.rs index eaa567a..b7b896c 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -2,10 +2,9 @@ use super::role::Role; use super::session::Session; use crate::client::{ImageUrl, MessageContent, MessageContentPart, ModelCapabilities}; -use crate::utils::sha256sum; +use crate::utils::{base64_encode, sha256}; use anyhow::{bail, Context, Result}; -use base64::{self, engine::general_purpose::STANDARD, Engine}; use fancy_regex::Regex; use lazy_static::lazy_static; use mime_guess::from_path; @@ -56,7 +55,7 @@ impl Input { if is_image { let data_url = read_media_to_data_url(&file_path) .with_context(|| format!("Unable to read media file '{file_item}'"))?; - data_urls.insert(sha256sum(&data_url), file_path.display().to_string()); + data_urls.insert(sha256(&data_url), file_path.display().to_string()); medias.push(data_url) } else { let text = read_file(&file_path) @@ -211,7 +210,7 @@ impl InputContext { pub fn resolve_data_url(data_urls: &HashMap<String, String>, data_url: String) -> String { if data_url.starts_with("data:") { - let hash = sha256sum(&data_url); + let hash = sha256(&data_url); if let Some(path) = data_urls.get(&hash) { return path.to_string(); } @@ -251,7 +250,7 @@ fn read_media_to_data_url<P: AsRef<Path>>(image_path: P) -> Result<String> { let mut buffer = Vec::new(); file.read_to_end(&mut buffer)?; - let encoded_image = STANDARD.encode(buffer); + let encoded_image = base64_encode(buffer); let data_url = format!("data:{};base64,{}", mime_type, encoded_image); Ok(data_url) |