fix(download): loose download host check - when TABBY_DOWNLOAD_HOST i…

crates/tabby-download/src/lib.rs

@@ -14,8 +14,12 @@
 };
 use tracing::{info, warn};
 
-fn download_host() -> String {
-    std::env::var("TABBY_DOWNLOAD_HOST").unwrap_or("huggingface.co".to_owned())
+fn select_by_download_host(url: &String) -> bool {
+    if let Ok(host) = std::env::var("TABBY_DOWNLOAD_HOST") {
+        url.contains(&host)
+    } else {
+        true
+    }
 }
 
 async fn download_model_impl(
@@ -50,18 +54,13 @@
         return download_split_model(model_info, &model_path).await;
     }
 
-    let registry = download_host();
     let Some(model_url) = model_info
         .urls
         .iter()
         .flatten()
-        .find(|x| x.contains(&registry))
+        .find(|x| select_by_download_host(x))
     else {
-        return Err(anyhow!(
-            "Invalid mirror <{}> for model urls: {:?}",
-            registry,
-            model_info.urls
-        ));
+        return Err(anyhow!("No valid url for model <{}>", model_info.name));
     };
 
     let strategy = ExponentialBackoff::from_millis(100).map(jitter).take(2);
@@ -79,17 +78,12 @@
     }
     let mut paths = vec![];
     let partition_urls = model_info.partition_urls.clone().unwrap_or_default();
-    let mirror = download_host();
 
     let Some(urls) = partition_urls
         .iter()
-        .find(|urls| urls.iter().all(|url| url.contains(&mirror)))
+        .find(|urls| urls.iter().all(select_by_download_host))
     else {
-        return Err(anyhow!(
-            "Invalid mirror <{}> for model urls: {:?}",
-            mirror,
-            partition_urls
-        ));
+        return Err(anyhow!("No valid url for model <{}>", model_info.name));
     };
 
     for (index, url) in urls.iter().enumerate() {