Skip to main content

resolve_model_id

Function resolve_model_id 

Source
pub fn resolve_model_id(model: &str) -> Result<(String, String), CodememError>
Expand description

Resolve the HuggingFace repo ID and local directory name from a model identifier.

Accepts:

  • Full HF repo: "BAAI/bge-base-en-v1.5" → repo="BAAI/bge-base-en-v1.5", dir="bge-base-en-v1.5"
  • Short name: "bge-small-en-v1.5" → repo="BAAI/bge-small-en-v1.5", dir="bge-small-en-v1.5"

Returns Err if the model identifier is a bare name without an org prefix and isn’t a recognized bge-* shorthand — HuggingFace requires org/repo format.