lattice-embed 0.9.0

SIMD-accelerated vector operations and embedding generation
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
124
125
126

//! Argument-validation tests for `embed`'s `--image` mode.
//!
//! Drives the compiled binary directly (no model loading, no network) so
//! these cover the hand-rolled arg loop's validation branches without
//! needing a real vision-language checkpoint.

use std::process::Command;

fn embed_cmd() -> Command {
    Command::new(env!("CARGO_BIN_EXE_embed"))
}

#[test]
fn image_and_text_are_mutually_exclusive() {
    let output = embed_cmd()
        .args(["--image", "x.png", "--text", "hello"])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(stderr.contains("mutually exclusive"), "stderr: {stderr}");
}

#[test]
fn vision_model_dir_without_image_is_a_usage_error() {
    let output = embed_cmd()
        .args(["--vision-model-dir", "/some/dir", "--text", "hello"])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        stderr.contains("--vision-model-dir requires --image"),
        "stderr: {stderr}"
    );
}

#[test]
fn image_requires_vision_model_dir() {
    let output = embed_cmd()
        .args(["--image", "x.png"])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        stderr.contains("--vision-model-dir is required"),
        "stderr: {stderr}"
    );
}

#[test]
fn image_rejects_unknown_pooling_strategy() {
    let output = embed_cmd()
        .args([
            "--image",
            "x.png",
            "--vision-model-dir",
            "/nonexistent",
            "--pooling",
            "max",
        ])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        stderr.contains("--pooling must be 'mean_visual' or 'last_token'"),
        "stderr: {stderr}"
    );
}

#[test]
fn image_with_missing_checkpoint_directory_fails_closed() {
    let output = embed_cmd()
        .args([
            "--image",
            "x.png",
            "--vision-model-dir",
            "/nonexistent-dir-xyz",
        ])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        stderr.contains("failed to load vision-language checkpoint"),
        "stderr: {stderr}"
    );
}

#[test]
fn image_with_unreadable_file_fails_closed() {
    let dir =
        std::env::temp_dir().join(format!("embed-cli-image-args-test-{}", std::process::id()));
    std::fs::create_dir_all(&dir).expect("temp dir must create");
    // Not a real checkpoint, but present -- config.json load fails before
    // this test's actual target (the missing image file) is ever reached,
    // so this exercises the same "checkpoint failed to load" fail-closed
    // path as the directory-missing case above from a different starting
    // point (a dir that exists but is not a checkpoint).
    let output = embed_cmd()
        .args([
            "--image",
            "does-not-exist.png",
            "--vision-model-dir",
            dir.to_str().unwrap(),
        ])
        .output()
        .expect("embed binary must run");
    assert!(!output.status.success());
    std::fs::remove_dir_all(&dir).ok();
}

#[test]
fn help_still_works_with_no_arguments_related_to_image_mode() {
    let output = embed_cmd()
        .args(["--help"])
        .output()
        .expect("embed binary must run");
    assert!(output.status.success());
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(stderr.contains("--image"), "stderr: {stderr}");
    assert!(stderr.contains("--vision-model-dir"), "stderr: {stderr}");
    assert!(stderr.contains("--metal"), "stderr: {stderr}");
}