feat: simplify download management, model file should be able to indi…

MODEL_SPEC.md

@@ -1,5 +1,7 @@
 # Tabby Model Specification (Unstable)
 
+> [!WARNING] **Since v0.5.0** This document is intended exclusively for local models. For remote models, we rely on the `tabby-registry` repository within each organization or user. You can refer to https://github.com/TabbyML/registry-tabby/blob/main/models.json for an example.
+
 Tabby organizes the model within a directory. This document provides an explanation of the necessary contents for supporting model serving. An example model directory can be found at https://huggingface.co/TabbyML/StarCoder-1B
 
 The minimal Tabby model directory should include the following contents:

crates/tabby-common/Cargo.toml

@@ -14,6 +14,7 @@ reqwest = { workspace = true, features = [ "json" ] }
 tokio = { workspace = true, features = ["rt", "macros"] }
 uuid = { version = "1.4.1", features = ["v4"] }
 tantivy.workspace = true
+anyhow.workspace = true
 
 [features]
 testutils = []

crates/tabby-common/src/lib.rs

@@ -3,6 +3,7 @@ pub mod events;
 pub mod index;
 pub mod languages;
 pub mod path;
+pub mod registry;
 pub mod usage;
 
 use std::{

crates/tabby-common/src/path.rs

@@ -51,38 +51,4 @@ pub fn events_dir() -> PathBuf {
     tabby_root().join("events")
 }
 
-pub struct ModelDir(PathBuf);
-
-impl ModelDir {
-    pub fn new(model: &str) -> Self {
-        Self(models_dir().join(model))
-    }
-
-    pub fn from(path: &str) -> Self {
-        Self(PathBuf::from(path))
-    }
-
-    pub fn path(&self) -> &PathBuf {
-        &self.0
-    }
-
-    pub fn path_string(&self, name: &str) -> String {
-        self.0.join(name).display().to_string()
-    }
-
-    pub fn cache_info_file(&self) -> String {
-        self.path_string(".cache_info.json")
-    }
-
-    pub fn metadata_file(&self) -> String {
-        self.path_string("tabby.json")
-    }
-
-    pub fn ggml_q8_0_file(&self) -> String {
-        self.path_string("ggml/q8_0.gguf")
-    }
-
-    pub fn ggml_q8_0_v2_file(&self) -> String {
-        self.path_string("ggml/q8_0.v2.gguf")
-    }
-}
+mod registry {}

crates/tabby-common/src/registry.rs

@@ -0,0 +1,85 @@
+use std::{fs, path::PathBuf};
+
+use anyhow::Result;
+use serde::{Deserialize, Serialize};
+
+use crate::path::models_dir;
+
+#[derive(Serialize, Deserialize)]
+pub struct ModelInfo {
+    pub name: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub prompt_template: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub chat_template: Option<String>,
+    pub urls: Vec<String>,
+    pub sha256: String,
+}
+
+fn models_json_file(registry: &str) -> PathBuf {
+    models_dir().join(registry).join("models.json")
+}
+
+async fn load_remote_registry(registry: &str) -> Result<Vec<ModelInfo>> {
+    let value = reqwest::get(format!(
+        "https://raw.githubusercontent.com/{}/registry-tabby/main/models.json",
+        registry
+    ))
+    .await?
+    .json()
+    .await?;
+    fs::create_dir_all(models_dir().join(registry))?;
+    serdeconv::to_json_file(&value, models_json_file(registry))?;
+    Ok(value)
+}
+
+fn load_local_registry(registry: &str) -> Result<Vec<ModelInfo>> {
+    Ok(serdeconv::from_json_file(models_json_file(registry))?)
+}
+
+#[derive(Default)]
+pub struct ModelRegistry {
+    pub name: String,
+    pub models: Vec<ModelInfo>,
+}
+
+impl ModelRegistry {
+    pub async fn new(registry: &str) -> Self {
+        Self {
+            name: registry.to_owned(),
+            models: load_remote_registry(registry).await.unwrap_or_else(|err| {
+                load_local_registry(registry).unwrap_or_else(|_| {
+                    panic!(
+                        "Failed to fetch model organization <{}>: {:?}",
+                        registry, err
+                    )
+                })
+            }),
+        }
+    }
+
+    pub fn get_model_path(&self, name: &str) -> PathBuf {
+        models_dir()
+            .join(&self.name)
+            .join(name)
+            .join(GGML_MODEL_RELATIVE_PATH)
+    }
+
+    pub fn get_model_info(&self, name: &str) -> &ModelInfo {
+        self.models
+            .iter()
+            .find(|x| x.name == name)
+            .unwrap_or_else(|| panic!("Invalid model_id <{}/{}>", self.name, name))
+    }
+}
+
+pub fn parse_model_id(model_id: &str) -> (&str, &str) {
+    let parts: Vec<_> = model_id.split('/').collect();
+    if parts.len() != 2 {
+        panic!("Invalid model id {}", model_id);
+    }
+
+    (parts[0], parts[1])
+}
+
+pub static GGML_MODEL_RELATIVE_PATH: &str = "ggml/q8_0.v2.gguf";

crates/tabby-download/Cargo.toml

@@ -17,3 +17,4 @@ urlencoding = "2.1.3"
 serde_json = { workspace = true }
 cached = { version = "0.46.0", features = ["async", "proc_macro"] }
 async-trait = { workspace = true }
+sha256 = "1.4.0"