Skip to content

Commit 80c327b

Browse files
committed
fix: cortexso model download
1 parent a21c224 commit 80c327b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

engine/services/model_service.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,8 @@ ModelService::DownloadModelFromCortexsoAsync(
618618
.branch_name = branch,
619619
.path_to_model_yaml = rel.string(),
620620
.model_alias = unique_model_id,
621-
.status = cortex::db::ModelStatus::Downloaded};
621+
.status = cortex::db::ModelStatus::Downloaded,
622+
.engine = mc.engine};
622623
auto result = db_service_->AddModelEntry(model_entry);
623624

624625
if (result.has_error()) {
@@ -627,6 +628,7 @@ ModelService::DownloadModelFromCortexsoAsync(
627628
} else {
628629
if (auto m = db_service_->GetModelInfo(unique_model_id); m.has_value()) {
629630
auto upd_m = m.value();
631+
upd_m.path_to_model_yaml = rel.string();
630632
upd_m.status = cortex::db::ModelStatus::Downloaded;
631633
if (auto r = db_service_->UpdateModelEntry(unique_model_id, upd_m);
632634
r.has_error()) {

0 commit comments

Comments
 (0)