mineru-sdk 0.1.0

Rust SDK for MinerU API - document extraction services
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use std::fmt;

use crate::{error::MineruError, types::*};

#[derive(Clone)]
pub struct MineruClient {
    base_url: String,
    token: String,
    client: hpx::Client,
}

impl fmt::Debug for MineruClient {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("MineruClient")
            .field("base_url", &self.base_url)
            .field("token", &"***")
            .finish()
    }
}

impl MineruClient {
    pub fn new(token: String) -> Self {
        Self {
            base_url: "https://mineru.net".to_string(),
            token,
            client: hpx::Client::new(),
        }
    }

    pub async fn create_extract_task(
        &self,
        request: ExtractTaskRequest,
    ) -> Result<ExtractTaskResponse, MineruError> {
        let url = format!("{}/api/v4/extract/task", self.base_url);
        let body = serde_json::to_vec(&request)?;
        let resp = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.token))
            .header("Content-Type", "application/json")
            .body(body)
            .send()
            .await?;

        let bytes = resp.bytes().await?;
        let result: ExtractTaskResponse = serde_json::from_slice(&bytes)?;
        Ok(result)
    }

    pub async fn get_task_result(&self, task_id: &str) -> Result<TaskResultResponse, MineruError> {
        let url = format!("{}/api/v4/extract/task/{}", self.base_url, task_id);
        let resp = self
            .client
            .get(&url)
            .header("Authorization", format!("Bearer {}", self.token))
            .send()
            .await?;

        let bytes = resp.bytes().await?;
        let result: TaskResultResponse = serde_json::from_slice(&bytes)?;
        Ok(result)
    }

    pub async fn batch_file_upload_urls(
        &self,
        request: BatchFileRequest,
    ) -> Result<BatchFileResponse, MineruError> {
        let url = format!("{}/api/v4/file-urls/batch", self.base_url);
        let body = serde_json::to_vec(&request)?;
        let resp = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.token))
            .header("Content-Type", "application/json")
            .body(body)
            .send()
            .await?;

        let bytes = resp.bytes().await?;
        let result: BatchFileResponse = serde_json::from_slice(&bytes)?;
        Ok(result)
    }

    pub async fn batch_url_upload(
        &self,
        request: BatchUrlRequest,
    ) -> Result<BatchUrlResponse, MineruError> {
        let url = format!("{}/api/v4/extract/task/batch", self.base_url);
        let body = serde_json::to_vec(&request)?;
        let resp = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.token))
            .header("Content-Type", "application/json")
            .body(body)
            .send()
            .await?;

        let bytes = resp.bytes().await?;
        let result: BatchUrlResponse = serde_json::from_slice(&bytes)?;
        Ok(result)
    }

    pub async fn get_batch_results(
        &self,
        batch_id: &str,
    ) -> Result<BatchResultResponse, MineruError> {
        let url = format!(
            "{}/api/v4/extract-results/batch/{}",
            self.base_url, batch_id
        );
        let resp = self
            .client
            .get(&url)
            .header("Authorization", format!("Bearer {}", self.token))
            .send()
            .await?;

        let bytes = resp.bytes().await?;
        let result: BatchResultResponse = serde_json::from_slice(&bytes)?;
        Ok(result)
    }
}