starred/minne
1
0
Fork 0
mirror of https://github.com/perstarkse/minne.git synced 2026-04-23 17:28:34 +02:00
Code Issues Packages Projects Releases 10 Wiki Activity

chore: clippy helps out

Browse Source
This commit is contained in:
Per Stark
2025-09-06 21:00:39 +02:00
parent 153efd1a98
commit 69954cf78e
9 changed files with 28 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ingestion-pipeline/src/types/mod.rs
View File

@@ -22,7 +22,9 @@ use std::io::{Seek, SeekFrom};
use tempfile::NamedTempFile;
use tracing::{error, info};
use crate::utils::{audio_transcription::transcribe_audio_file, image_parsing::extract_text_from_image};
use crate::utils::{
audio_transcription::transcribe_audio_file, image_parsing::extract_text_from_image,
};
pub async fn to_text_content(
ingestion_payload: IngestionPayload,
@@ -37,7 +39,7 @@ pub async fn to_text_content(
category,
user_id,
} => {
let (article, file_info) = fetch_article_from_url(&url, db, &user_id, &config).await?;
let (article, file_info) = fetch_article_from_url(&url, db, &user_id, config).await?;
Ok(TextContent::new(
article.text_content.into(),
Some(context),
@@ -179,7 +181,7 @@ async fn fetch_article_from_url(
};
// Store screenshot
let file_info = FileInfo::new(field_data, db, user_id, &config).await?;
let file_info = FileInfo::new(field_data, db, user_id, config).await?;
// Parse content...
let config = dom_smoothie::Config {
@@ -231,8 +233,8 @@ async fn extract_text_from_file(
let content = tokio::fs::read_to_string(&file_info.path).await?;
Ok(content)
}
"audio/mpeg" | "audio/mp3" | "audio/wav" | "audio/x-wav" | "audio/webm" | "audio/mp4" | "audio/ogg" | "audio/flac" => {
"audio/mpeg" | "audio/mp3" | "audio/wav" | "audio/x-wav" | "audio/webm" | "audio/mp4"
| "audio/ogg" | "audio/flac" => {
transcribe_audio_file(&file_info.path, db_client, openai_client).await
}
// Handle other MIME types as needed

2
ingestion-pipeline/src/utils/image_parsing.rs
View File

@@ -48,7 +48,7 @@ pub async fn extract_text_from_image(
let description = response
.choices
.get(0)
.first()
.and_then(|c| c.message.content.as_ref())
.cloned()
.unwrap_or_else(|| "No description found.".to_string());

20
ingestion-pipeline/src/utils/mod.rs
View File

@@ -1,6 +1,6 @@
pub mod audio_transcription;
pub mod image_parsing;
pub mod llm_instructions;
pub mod audio_transcription;
use common::error::AppError;
use std::collections::HashMap;
@@ -33,16 +33,12 @@ impl GraphMapper {
}
// If parsing fails, look it up in the map.
self.key_to_id
.get(key)
.map(|id| *id) // Dereference the &Uuid to get Uuid
// If `get` returned None, create and return an error.
.ok_or_else(|| {
AppError::GraphMapper(format!(
"Key '{}' is not a valid UUID and was not found in the map.",
key
))
})
self.key_to_id.get(key).copied().ok_or_else(|| {
AppError::GraphMapper(format!(
"Key '{}' is not a valid UUID and was not found in the map.",
key
))
})
}
/// Assigns a new UUID for a given key. (No changes needed here)
@@ -56,7 +52,7 @@ impl GraphMapper {
pub fn get_id(&self, key: &str) -> Result<Uuid, AppError> {
self.key_to_id
.get(key)
.map(|id| *id)
.copied()
.ok_or_else(|| AppError::GraphMapper(format!("Key '{}' not found in map.", key)))
}
}