uqa-execution 0.4.5

Volcano physical operators with row-batch pipelines
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

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

use super::*;
use uqa_storage::diskann_index::{
    build::DiskANNTemporaryBudget, DiskANNIndexOptions, DiskANNMemoryIndex,
};
use uqa_storage::vector_index::DiskANNIndexParams;

#[test]
fn diskann_calibration_runtime_validation_uses_actual_selected_versions() {
    let control = StorageReadControl::with_limit(1 << 20);
    let mut index = DiskANNMemoryIndex::new(
        2,
        DiskANNIndexOptions::for_parameters(DiskANNIndexParams::for_dimensions(2).unwrap()),
        &DiskANNTemporaryBudget::new(1 << 20),
        &control,
    )
    .unwrap();
    index.add(1, vec![1.0, 0.0]).unwrap();
    let target = diskann_target(
        &index,
        "public.docs",
        "embedding",
        ("fixture", "1"),
        3,
        &control,
    )
    .unwrap();
    validate_index(&index, &target, &control).unwrap();
    let model = VectorCalibrationModel::new(
        uqa_scoring::VectorProbabilityTransform::new(0.0, 1.0, 1.0, 0.5).unwrap(),
        uqa_scoring::VectorCalibrationProvenance {
            model_version: "fixed".into(),
            target: target.clone(),
            fit_sample_count: 100,
        },
    )
    .unwrap();
    validate_names(&model, &target, "public.docs", "embedding").unwrap();
    assert!(validate_names(&model, &target, "public.other", "embedding").is_err());
    assert!(validate_names(&model, &target, "public.docs", "other").is_err());
    let mut wrong = target.clone();
    wrong.index_kind = "hnsw".into();
    assert!(validate_index(&index, &wrong, &control).is_err());
    wrong = target.clone();
    wrong.dimensions = 3;
    assert!(validate_index(&index, &wrong, &control).is_err());
    index.add(2, vec![0.0, 1.0]).unwrap();
    assert!(validate_index(&index, &target, &control)
        .unwrap_err()
        .to_string()
        .contains("target mismatch"));
    let fresh = diskann_target(
        &index,
        "public.docs",
        "embedding",
        ("fixture", "1"),
        3,
        &control,
    )
    .unwrap();
    assert_ne!(target.corpus_version, fresh.corpus_version);
    assert_eq!(target.index_version, fresh.index_version);
    assert!(model.validate_for(&fresh).is_err());
}