From 0d320d8a819b4af0c1acb800d8ddf2f08c6c1e10 Mon Sep 17 00:00:00 2001
From: Thien Tran <gau.nernst@yahoo.com.sg>
Date: Wed, 19 Feb 2025 18:15:13 +0800
Subject: [PATCH] chore: remove hard-coded engine value

---
 engine/cli/commands/ps_cmd.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/engine/cli/commands/ps_cmd.cc b/engine/cli/commands/ps_cmd.cc
index 24ef497c6..4f83f4f42 100644
--- a/engine/cli/commands/ps_cmd.cc
+++ b/engine/cli/commands/ps_cmd.cc
@@ -27,7 +27,8 @@ void PsCmd::Exec(const std::string& host, int port) {
     for (const auto& item : res.value()["data"]) {
       ModelLoadedStatus model_status;
       // TODO(sang) hardcode for now
-      model_status.engine = kLlamaEngine;
+      model_status.engine = item["engine"].isNull()
+                            ? kLlamaEngine : item["engine"].asString();
       model_status.model = item["id"].asString();
       model_status.ram = item["ram"].asUInt64();
       model_status.start_time = item["start_time"].asUInt64();