markhor_extensions 0.1.0-alpha.0.2

Extensions for Markhor, a project connecting AI models, documents, and workflows for knowledge work
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use std::sync::Arc;

use async_trait::async_trait;
use markhor_core::{convert::{ConversionError, Converter}, storage::Content};
use mime::Mime;
use tokio::io::AsyncRead;

use super::client::{MistralClientInner, URI};


pub struct MistralOcr(pub(crate) Arc<MistralClientInner>);

#[async_trait]
impl Converter for MistralOcr {
    async fn convert(&self, input: Content, output_type: Mime) -> Result<Vec<Box<dyn AsyncRead + Unpin>>, ConversionError> {
        // Check if markdown is expected
        if output_type != "text/markdown" {
            return Err(ConversionError::UnsupportedMimeType(output_type));
        }

        // just yolo the error conversions for now
        // TODO: fix this once `ConversionError` is implemented properly
        let dir = tempfile::tempdir().map_err(|e| ConversionError::Other(Box::new(e)))?;
        let output_path = dir.path().join("output.md");

        self.0.ocr_file_to_markdown(input.path(), &*output_path).await.map_err(|e| ConversionError::Other(Box::new(e)))?;

        let file = tokio::fs::File::open(&output_path).await.map_err(|e| ConversionError::IoError(e))?;
        let reader = tokio::io::BufReader::new(file);

        Ok(vec![Box::new(reader)])
    }
}