feat: add ModelScope message (#637)

Signed-off-by: Gaius <[email protected]>

Author: gaius-qi

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "dragonfly-api"
-version = "2.2.22"
+version = "2.2.23"
 authors = ["Gaius <[email protected]>"]
 edition = "2021"
 license = "Apache-2.0"

Cargo.toml

@@ -709,6 +709,8 @@ message Download {
   bool enable_task_id_based_blob_digest = 30;
   // HuggingFace protocol information.
   optional HuggingFace hugging_face = 31;
+  // ModelScope protocol information.
+  optional ModelScope model_scope = 32;
 
   reserved 21;
   reserved "load_to_cache";
@@ -770,11 +772,23 @@ message HuggingFace {
     min_len: 1
     ignore_empty: true
   }];
-  // Insecure skip verify indicates whether to skip verifying the server's certificate chain.
-  optional bool insecure_skip_verify = 2;
   // Revision of the HuggingFace model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
   // it defaults to the repository's default main branch.
-  string revision = 3 [(validate.rules).string = {
+  string revision = 2 [(validate.rules).string = {
+    min_len: 1
+  }];
+}
+
+// ModelScope related information.
+message ModelScope {
+  // Access token for ModelScope Hub.
+  optional string token = 1 [(validate.rules).string = {
+    min_len: 1
+    ignore_empty: true
+  }];
+  // Revision of the ModelScope model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
+  // it defaults to the repository's default main branch.
+  string revision = 2 [(validate.rules).string = {
     min_len: 1
   }];
 }

pkg/apis/common/v2/common.pb.go

@@ -651,6 +651,8 @@ message Download {
   bool enable_task_id_based_blob_digest = 30;
   // HuggingFace protocol information.
   optional HuggingFace hugging_face = 31;
+  // ModelScope protocol information.
+  optional ModelScope model_scope = 32;
 
   reserved 21;
   reserved "load_to_cache";
@@ -688,11 +690,18 @@ message HDFS {
 message HuggingFace {
   // Access token for HuggingFace Hub.
   optional string token = 1;
-  // Insecure skip verify indicates whether to skip verifying the server's certificate chain.
-  optional bool insecure_skip_verify = 2;
   // Revision of the HuggingFace model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
   // it defaults to the repository's default main branch.
-  string revision = 3;
+  string revision = 2;
+}
+
+// ModelScope related information.
+message ModelScope {
+  // Access token for ModelScope Hub.
+  optional string token = 1;
+  // Revision of the ModelScope model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
+  // it defaults to the repository's default main branch.
+  string revision = 2;
 }
 
 // Range represents download range.

pkg/apis/common/v2/common.pb.validate.go

@@ -777,6 +777,9 @@ pub struct Download {
     /// HuggingFace protocol information.
     #[prost(message, optional, tag = "31")]
     pub hugging_face: ::core::option::Option<HuggingFace>,
+    /// ModelScope protocol information.
+    #[prost(message, optional, tag = "32")]
+    pub model_scope: ::core::option::Option<ModelScope>,
 }
 /// Object Storage related information.
 #[derive(serde::Serialize, serde::Deserialize)]
@@ -828,12 +831,22 @@ pub struct HuggingFace {
     /// Access token for HuggingFace Hub.
     #[prost(string, optional, tag = "1")]
     pub token: ::core::option::Option<::prost::alloc::string::String>,
-    /// Insecure skip verify indicates whether to skip verifying the server's certificate chain.
-    #[prost(bool, optional, tag = "2")]
-    pub insecure_skip_verify: ::core::option::Option<bool>,
     /// Revision of the HuggingFace model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
     /// it defaults to the repository's default main branch.
-    #[prost(string, tag = "3")]
+    #[prost(string, tag = "2")]
+    pub revision: ::prost::alloc::string::String,
+}
+/// ModelScope related information.
+#[derive(serde::Serialize, serde::Deserialize)]
+#[allow(clippy::derive_partial_eq_without_eq)]
+#[derive(Clone, PartialEq, ::prost::Message)]
+pub struct ModelScope {
+    /// Access token for ModelScope Hub.
+    #[prost(string, optional, tag = "1")]
+    pub token: ::core::option::Option<::prost::alloc::string::String>,
+    /// Revision of the ModelScope model, dataset, or space. It can be a branch name, tag name, or commit hash. If not specified,
+    /// it defaults to the repository's default main branch.
+    #[prost(string, tag = "2")]
     pub revision: ::prost::alloc::string::String,
 }
 /// Range represents download range.