Print info about available models when model is not found

lanterndata · Feb 4, 2024 · c7f3af0 · c7f3af0
1 parent a83507e
commit c7f3af0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lantern_embeddings_core/src/ort_runtime.rs b/lantern_embeddings_core/src/ort_runtime.rs
@@ -607,7 +607,11 @@ impl<'a> OrtRuntime<'a> {
             let model_info = map.get(model_name);
 
             if model_info.is_none() {
-                anyhow::bail!("Model \"{}\" not found", model_name)
+                anyhow::bail!(
+                    "Model \"{}\" not found.\nAvailable models: {}",
+                    model_name,
+                    map.keys().join(", ")
+                )
             }
 
             let model_info = model_info.unwrap();