ModelRegistrar

Struct ModelRegistrar 

Source
pub struct ModelRegistrar {
    model_repo: Arc<dyn ModelRepository>,
    gguf_parser: Arc<dyn GgufParserPort>,
    model_files_repo: Option<Arc<dyn ModelFilesRepositoryPort>>,
}
Expand description

Implementation of the model registrar port.

This service composes over ModelRepository for persistence and GgufParserPort for metadata extraction.

Fields§

§model_repo: Arc<dyn ModelRepository>

Repository for persisting models.

§gguf_parser: Arc<dyn GgufParserPort>

Parser for extracting GGUF metadata.

§model_files_repo: Option<Arc<dyn ModelFilesRepositoryPort>>

Repository for persisting model file metadata.

Implementations§

Source§

impl ModelRegistrar

Source

pub fn new( model_repo: Arc<dyn ModelRepository>, gguf_parser: Arc<dyn GgufParserPort>, model_files_repo: Option<Arc<dyn ModelFilesRepositoryPort>>, ) -> Self

Create a new model registrar.

§Arguments
  • model_repo - Repository for persisting models
  • gguf_parser - Parser for extracting GGUF metadata
  • model_files_repo - Optional repository for persisting model file metadata
Source

fn filter_hf_tags(tags: &[String]) -> Vec<String>

Filter HuggingFace tags using a blocklist.

Removes noisy tags like gguf, arxiv:*, region:*, license:*, dataset:*.

Source

fn merge_tags(gguf_tags: Vec<String>, hf_tags: &[String]) -> Vec<String>

Merge GGUF-derived tags with filtered HF tags, removing duplicates.

GGUF-derived tags are prioritized (appear first in the result).

Trait Implementations§

Source§

impl ModelRegistrarPort for ModelRegistrar

Source§

fn register_model<'life0, 'life1, 'async_trait>( &'life0 self, download: &'life1 CompletedDownload, ) -> Pin<Box<dyn Future<Output = Result<Model, RepositoryError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Register a downloaded model in the database. Read more
Source§

fn register_model_from_path<'life0, 'life1, 'life2, 'life3, 'life4, 'async_trait>( &'life0 self, repo_id: &'life1 str, commit_sha: &'life2 str, file_path: &'life3 Path, quantization: &'life4 str, ) -> Pin<Box<dyn Future<Output = Result<Model, RepositoryError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait,

Register a model using raw path parameters. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

§

impl<T> Instrument for T

§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more
§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
§

impl<T> WithSubscriber for T

§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more
§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more