sipp-rs 0.1.2

Unified Rust library for extensible Sipp inference
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

//! Small helpers used by `ModelStore`: path comparison, asset classification,
//! pairing-state projection, and fingerprinting.

use serde_json::json;

use crate::collection::sorted_values;
use crate::lifecycle::backend_policy::BackendPlan;
use crate::lifecycle::util::{sha256_hex, sorted_model_asset_ids};
use crate::lifecycle::{ModelEntry, ModelError, PairingPlan};

/////////////////////////////////////////////////////////////////////////////////
/// TESTS
/////////////////////////////////////////////////////////////////////////////////

#[cfg(test)]
#[path = "../../tests/lifecycle/service/helpers_tests.rs"]
mod helpers_tests;

/////////////////////////////////////////////////////////////////////////////////
/// SRC
/////////////////////////////////////////////////////////////////////////////////

pub(super) fn model_id_from_plan(plan: &PairingPlan) -> String {
    let ids = sorted_model_asset_ids(&plan.model_asset_ids, plan.projector_asset_id.as_ref());
    format!("model-{}", sha256_hex(ids.join("\n").as_bytes()))
}

pub(super) fn runtime_fingerprint(
    entry: &ModelEntry,
    backend_plan: &BackendPlan,
) -> Result<String, ModelError> {
    let runtime = serde_json::to_value(&backend_plan.config)?;
    let value = json!({
        "modelAssetIds": sorted_values(entry.model_asset_ids.clone()),
        "projectorAssetId": entry.projector_asset_id,
        "backend": backend_plan.selection.selected,
        "runtime": runtime,
    });
    Ok(sha256_hex(value.to_string().as_bytes()))
}