fix: skip main branch for cortexso model source if has metadata.yml (#2042)

Co-authored-by: sangjanai <[email protected]>

Author: vansangpfiev

engine/services/model_source_service.cc

@@ -432,9 +432,10 @@ cpp::result<bool, std::string> ModelSourceService::AddCortexsoRepo(
   }
 
   auto author = hub_author;
+  auto model_author = hu::GetModelAuthorCortexsoHub(model_name);
   if (auto model_author = hu::GetModelAuthorCortexsoHub(model_name);
-      model_author.has_value() && !model_author->empty()) {
-    author = *model_author;
+      model_author.has_value() && !model_author.value().empty()) {
+    author = model_author.value();
   }
 
   // Get models from db
@@ -443,6 +444,10 @@ cpp::result<bool, std::string> ModelSourceService::AddCortexsoRepo(
   std::unordered_set<std::string> updated_model_list;
   std::vector<std::future<std::string>> tasks;
   for (auto const& [branch, _] : branches.value()) {
+    if (!model_author.has_error() && branch == "main") {
+      CTL_DBG("Skip main branch");
+      continue;
+    }
     CTL_DBG(branch);
     tasks.push_back(std::async(std::launch::async, [&, branch = branch] {
       return AddCortexsoRepoBranch(model_source, author, model_name, branch,

engine/utils/huggingface_utils.h

@@ -312,23 +312,23 @@ inline std::optional<std::string> GetDefaultBranch(
   }
 }
 
-inline std::optional<std::string> GetModelAuthorCortexsoHub(
+inline cpp::result<std::string, std::string> GetModelAuthorCortexsoHub(
     const std::string& model_name) {
   try {
     auto remote_yml = curl_utils::ReadRemoteYaml(GetMetadataUrl(model_name));
 
     if (remote_yml.has_error()) {
-      return std::nullopt;
+      return cpp::fail(remote_yml.error());
     }
 
     auto metadata = remote_yml.value();
     auto author = metadata["author"];
     if (author.IsDefined()) {
       return author.as<std::string>();
     }
-    return std::nullopt;
+    return "";
   } catch (const std::exception& e) {
-    return std::nullopt;
+    return "";
   }
 }
 }  // namespace huggingface_utils