akribes-sdk 0.22.2

Rust client SDK for the Akribes workflow server
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use std::sync::Arc;

use crate::client::{AkribesClient, Inner};
use crate::error::Result;
use crate::models::ConvertResult;

/// Sub-client for document conversion via Docling. Obtained via [`AkribesClient::convert()`].
#[derive(Clone, Debug)]
pub struct ConvertClient {
    pub(crate) inner: Arc<Inner>,
}

impl ConvertClient {
    pub(crate) fn new(inner: Arc<Inner>) -> Self {
        Self { inner }
    }

    fn c(&self) -> AkribesClient {
        AkribesClient {
            inner: Arc::clone(&self.inner),
        }
    }

    /// Convert a document file to Markdown via the server's Docling integration.
    ///
    /// The server supports: PDF, DOCX, DOC, PPTX, XLSX, HTML, PNG, JPG, TIFF.
    pub async fn convert_file(&self, filename: &str, data: Vec<u8>) -> Result<ConvertResult> {
        let url = format!("{}/convert", self.inner.base_url);
        self.post_convert(&url, filename, data).await
    }

    /// Project-scoped convert. Prefer this over [`convert_file`](Self::convert_file)
    /// when uploading for a specific project so the server can enforce scope.
    pub async fn convert_file_for_project(
        &self,
        project_id: i64,
        filename: &str,
        data: Vec<u8>,
    ) -> Result<ConvertResult> {
        let url = format!("{}/projects/{}/convert", self.inner.base_url, project_id);
        self.post_convert(&url, filename, data).await
    }

    async fn post_convert(
        &self,
        url: &str,
        filename: &str,
        data: Vec<u8>,
    ) -> Result<ConvertResult> {
        let part = reqwest::multipart::Part::bytes(data)
            .file_name(filename.to_string())
            .mime_str(mime_for(filename))?;
        let form = reqwest::multipart::Form::new().part("file", part);
        self.c().post_multipart(url, form).await
    }
}

fn mime_for(filename: &str) -> &'static str {
    let ext = filename.rsplit('.').next().unwrap_or("").to_lowercase();
    match ext.as_str() {
        "pdf" => "application/pdf",
        "docx" => "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
        "doc" => "application/msword",
        "pptx" => "application/vnd.openxmlformats-officedocument.presentationml.presentation",
        "xlsx" => "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        "html" | "htm" => "text/html",
        "png" => "image/png",
        "jpg" | "jpeg" => "image/jpeg",
        "tiff" | "tif" => "image/tiff",
        _ => "application/octet-stream",
    }
}