xberg 1.0.5

High-performance document intelligence library for Rust. Extract text, metadata, and structured data from PDFs, Office documents, images, and 98 formats and 306 programming languages via tree-sitter code intelligence with async/sync APIs.
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

#![cfg(feature = "classification")]

use xberg::ClassificationLabel;

#[test]
fn classification_label_with_confidence() {
    let label = ClassificationLabel {
        label: "invoice".to_string(),
        confidence: Some(0.85),
    };

    assert_eq!(label.label, "invoice");
    assert_eq!(label.confidence, Some(0.85));
}

#[test]
fn classification_label_without_confidence() {
    let label = ClassificationLabel {
        label: "memo".to_string(),
        confidence: None,
    };

    assert_eq!(label.label, "memo");
    assert_eq!(label.confidence, None);
}

#[test]
fn page_classification_structure() {
    use xberg::types::classification::PageClassification;

    let labels = vec![ClassificationLabel {
        label: "invoice".to_string(),
        confidence: Some(0.92),
    }];

    let classification = PageClassification { page_number: 1, labels };

    assert_eq!(classification.page_number, 1);
    assert_eq!(classification.labels.len(), 1);
    assert_eq!(classification.labels[0].label, "invoice");
}

#[test]
fn page_classification_config_creation() {
    use xberg::core::config::{LlmConfig, PageClassificationConfig};

    let config = PageClassificationConfig {
        labels: vec!["invoice".to_string(), "memo".to_string()],
        llm: LlmConfig::default(),
        prompt_template: None,
        multi_label: false,
    };

    assert_eq!(config.labels.len(), 2);
    assert_eq!(config.labels[0], "invoice");
    assert!(!config.multi_label);
}

#[test]
fn page_classification_multi_label_mode() {
    use xberg::core::config::{LlmConfig, PageClassificationConfig};

    let config = PageClassificationConfig {
        labels: vec!["invoice".to_string(), "memo".to_string(), "report".to_string()],
        llm: LlmConfig::default(),
        prompt_template: None,
        multi_label: true,
    };

    assert!(config.multi_label);
    assert_eq!(config.labels.len(), 3);
}

#[test]
fn classification_label_serialization() {
    let label = ClassificationLabel {
        label: "document".to_string(),
        confidence: Some(0.75),
    };

    let json = serde_json::to_string(&label).expect("serialization should succeed");
    assert!(json.contains("document"));
    assert!(json.contains("0.75"));

    let deserialized: ClassificationLabel = serde_json::from_str(&json).expect("deserialization should succeed");
    assert_eq!(deserialized.label, "document");
    assert_eq!(deserialized.confidence, Some(0.75));
}