Struct ModelDownloadTracker 

pub struct ModelDownloadTracker { /* private fields */ }

Tracks the status of model downloads using SQLite for persistence

Implementations

impl ModelDownloadTracker

pub fn new() -> Self

pub fn get_status(&self, model_name: &str) -> Option<ModelStatus>

Gets the status of a model from the database If the model is not in the database, it returns None

pub fn set_status_and_notify( &self, model_name: String, status: ModelStatus, provider: ModelProvider, message: Option<String>, )

Sets the status of a model and notifies all waiting channels

pub fn set_status( &self, model_name: String, status: ModelStatus, provider: ModelProvider, )

Sets the status of a model

pub fn add_waiting_channel( &self, model_name: &str, tx: Sender<Result<ModelStatusUpdate, Status>>, )

Adds a channel to wait for updates on a specific model

pub fn delete_status(&self, model_name: &str)

Deletes the status of a model from the database This is used when a model is removed from the tracker

pub async fn ensure_model_downloaded( &self, model_name: &str, provider: ModelProvider, tx: &Sender<Result<ModelStatusUpdate, Status>>, ignore_weights: bool, ) -> ModelStatus

Initiates a download for a model and streams status updates

Trait Implementations

impl Clone for ModelDownloadTracker

fn clone(&self) -> ModelDownloadTracker

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelDownloadTracker

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ModelDownloadTracker

fn default() -> Self

Returns the “default value” for a type.