gglib_core::services::model_verification

Struct ModelVerificationService

pub struct ModelVerificationService {
    model_repo: Arc<dyn ModelRepository>,
    model_files_repo: Arc<dyn ModelFilesReaderPort>,
    hf_client: Arc<dyn HfClientPort>,
    download_trigger: Arc<dyn DownloadTriggerPort>,
    operation_lock: ModelOperationLock,
}

Expand description

Model verification service.

Fields§

§model_repo: Arc<dyn ModelRepository>

Repository for model metadata.

§model_files_repo: Arc<dyn ModelFilesReaderPort>

Repository for model file metadata.

§hf_client: Arc<dyn HfClientPort>

HuggingFace client for update checks.

§download_trigger: Arc<dyn DownloadTriggerPort>

Download trigger for repairs.

§operation_lock: ModelOperationLock

Concurrency control.

Implementations§

Source §

impl ModelVerificationService

Source

pub fn new( model_repo: Arc<dyn ModelRepository>, model_files_repo: Arc<dyn ModelFilesReaderPort>, hf_client: Arc<dyn HfClientPort>, download_trigger: Arc<dyn DownloadTriggerPort>, ) -> Self

Create a new verification service.

Source

pub async fn verify_model_integrity( &self, model_id: i64, ) -> Result<(Receiver<VerificationProgress>, JoinHandle<Result<VerificationReport, RepositoryError>>), String>

Verify the integrity of a model by computing SHA256 hashes.

Returns a channel for progress updates and a handle to the verification task.

§Arguments

model_id - ID of the model to verify

§Returns

receiver - Channel for receiving progress updates
handle - Join handle for the verification task

Source

async fn verify_shard( file: &ModelFile, resolved_path: &Path, model_id: i64, index: usize, total_shards: usize, tx: &Sender<VerificationProgress>, ) -> ShardHealth

Verify a single shard by computing its SHA256 and comparing with OID.

Source

pub async fn check_for_updates( &self, model_id: i64, ) -> Result<UpdateCheckResult, RepositoryError>

Check if updates are available for a model.

Compares local OIDs with remote OIDs from HuggingFace.

Source

pub async fn repair_model( &self, model_id: i64, shard_indices: Option<Vec<usize>>, ) -> Result<String, String>

Repair a model by re-downloading corrupt or missing shards.

§Arguments

model_id - ID of the model to repair
shard_indices - Optional list of specific shard indices to repair. If None, all unhealthy shards will be repaired.