From ba1ec7b6ebb79d1836b34911d6f37d68e220b166 Mon Sep 17 00:00:00 2001
From: Sam Kolamkanny <mcmalayalam@gmail.com>
Date: Tue, 9 Jun 2026 21:58:22 +0100
Subject: [PATCH] Add /health endpoint that returns HTTP 200 once model is
 fully loaded

---
 ds4_server.c | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/ds4_server.c b/ds4_server.c
index 7aa7bc0e5..b569630b0 100644
--- a/ds4_server.c
+++ b/ds4_server.c
@@ -7708,6 +7708,7 @@ struct server {
     visible_live_state thinking_live;
     bool disable_exact_dsml_tool_replay;
     bool enable_cors;
+    volatile bool model_ready;
     pthread_mutex_t tool_mu;
     pthread_mutex_t mu;
     pthread_cond_t cv;
@@ -11147,6 +11148,16 @@ static void *client_main(void *arg) {
         goto done;
     }
 
+    if (!strcmp(hr.method, "GET") && !strcmp(hr.path, "/health")) {
+        if (s->model_ready) {
+            http_response(fd, s->enable_cors, 200, "text/plain", "ok");
+        } else {
+            http_error(fd, s->enable_cors, 503, "model not yet loaded");
+        }
+        http_request_free(&hr);
+        goto done;
+    }
+
     if (!strcmp(hr.method, "GET") && !strcmp(hr.path, "/v1/models")) {
         send_models(s, fd);
         http_request_free(&hr);
@@ -11625,6 +11636,7 @@ int main(int argc, char **argv) {
     s.disable_exact_dsml_tool_replay = cfg.disable_exact_dsml_tool_replay;
     s.tool_mem.max_entries = cfg.tool_memory_max_ids;
     s.enable_cors = cfg.enable_cors;
+    s.model_ready = true;
     if (cfg.kv_disk_dir) {
         kv_cache_open(&s.kv, cfg.kv_disk_dir, cfg.kv_disk_space_mb,
                       cfg.kv_cache_reject_different_quant, cfg.kv_cache);