添加 All-MiniLM-L6-v2-Embedding 模型

src/models/core/registry.rs

@@ -1,15 +1,16 @@
 use anyhow::Result;
 
 use crate::models::{
-    ModelInstance, WhichModel, load_model_legacy, qwen3::generate::Qwen3GenerateModel,
-    qwen3_5::generate::Qwen3_5GenerateModel, qwen3_embedding::generate::Qwen3EmbeddingModel,
-    qwen3_reranker::generate::Qwen3RerankerModel,
+    ModelInstance, WhichModel, all_minilm_l6_v2::generate::AllMiniLML6V2Model, load_model_legacy,
+    qwen3::generate::Qwen3GenerateModel, qwen3_5::generate::Qwen3_5GenerateModel,
+    qwen3_embedding::generate::Qwen3EmbeddingModel, qwen3_reranker::generate::Qwen3RerankerModel,
 };
 
 use super::artifact::LoadSpec;
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
 enum ModelLoaderFamily {
+    AllMiniLML6V2,
     Qwen3,
     Qwen3Embedding,
     Qwen3Reranker,
@@ -19,6 +20,7 @@ enum ModelLoaderFamily {
 
 fn resolve_model_loader_family(model: WhichModel) -> ModelLoaderFamily {
     match model {
+        WhichModel::AllMiniLML6V2 => ModelLoaderFamily::AllMiniLML6V2,
         WhichModel::Qwen3_0_6B => ModelLoaderFamily::Qwen3,
         WhichModel::Qwen3Embedding0_6B
         | WhichModel::Qwen3Embedding4B
@@ -44,6 +46,9 @@ fn resolve_model_loader_family(model: WhichModel) -> ModelLoaderFamily {
 pub fn load_model_from_spec<'a>(spec: &LoadSpec) -> Result<ModelInstance<'a>> {
     spec.validate()?;
     let model = match resolve_model_loader_family(spec.model) {
+        ModelLoaderFamily::AllMiniLML6V2 => {
+            ModelInstance::AllMiniLML6V2(AllMiniLML6V2Model::init_from_spec(spec, None, None)?)
+        }
         ModelLoaderFamily::Qwen3 => {
             ModelInstance::Qwen3(Qwen3GenerateModel::init_from_spec(spec, None, None)?)
         }
@@ -73,6 +78,10 @@ mod tests {
 
     #[test]
     fn registry_routes_qwen3_family_models() {
+        assert_eq!(
+            resolve_model_loader_family(WhichModel::AllMiniLML6V2),
+            ModelLoaderFamily::AllMiniLML6V2
+        );
         assert_eq!(
             resolve_model_loader_family(WhichModel::Qwen3_0_6B),
             ModelLoaderFamily::Qwen3