From 93e8afbfb332f0d5d76be91b73ec0a3596032d84 Mon Sep 17 00:00:00 2001
From: Matti Pulkkinen <pulkkinen@ultimatium.com>
Date: Mon, 8 Dec 2025 16:33:46 +0200
Subject: [PATCH 1/2] Remove reduntant start parameters

- No longer exits if non-required parameters are missing
---
 examples/server/main.cpp | 38 --------------------------------------
 1 file changed, 38 deletions(-)

diff --git a/examples/server/main.cpp b/examples/server/main.cpp
index d29d78646..f0d3750e5 100644
--- a/examples/server/main.cpp
+++ b/examples/server/main.cpp
@@ -649,44 +649,6 @@ void parse_args(int argc, const char** argv, SDParams& params) {
         params.ctxParams.n_threads = sd_get_num_physical_cores();
     }
 
-    if (params.lastRequest.prompt.length() == 0) {
-        fprintf(stderr, "error: the following arguments are required: prompt\n");
-        print_usage(argc, argv, options);
-        exit(1);
-    }
-
-    if (params.ctxParams.model_path.length() == 0 && params.ctxParams.diffusion_model_path.length() == 0) {
-        fprintf(stderr, "error: the following arguments are required: model_path/diffusion_model\n");
-        print_usage(argc, argv, options);
-        exit(1);
-    }
-
-    if (params.output_path.length() == 0) {
-        fprintf(stderr, "error: the following arguments are required: output_path\n");
-        print_usage(argc, argv, options);
-        exit(1);
-    }
-
-    if (params.lastRequest.height <= 0) {
-        fprintf(stderr, "error: the height must be greater than 0\n");
-        exit(1);
-    }
-
-    if (params.lastRequest.width <= 0) {
-        fprintf(stderr, "error: the width must be greater than 0\n");
-        exit(1);
-    }
-
-    if (params.lastRequest.sample_params.sample_steps <= 0) {
-        fprintf(stderr, "error: the sample_steps must be greater than 0\n");
-        exit(1);
-    }
-
-    if (params.lastRequest.strength < 0.f || params.lastRequest.strength > 1.f) {
-        fprintf(stderr, "error: can only work with strength in [0.0, 1.0]\n");
-        exit(1);
-    }
-
     if (params.lastRequest.seed < 0) {
         srand((int)time(nullptr));
         params.lastRequest.seed = rand();

From 3bd760d1ff3bbe69f4ef122bd667f7b8ac312e22 Mon Sep 17 00:00:00 2001
From: Matti Pulkkinen <pulkkinen@ultimatium.com>
Date: Mon, 8 Dec 2025 16:36:07 +0200
Subject: [PATCH 2/2] Don't erase task result too early

- Only erase when client polls the result and task is failed or done
---
 examples/server/main.cpp | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/examples/server/main.cpp b/examples/server/main.cpp
index f0d3750e5..f3482cd63 100644
--- a/examples/server/main.cpp
+++ b/examples/server/main.cpp
@@ -2579,9 +2579,12 @@ void start_server(SDParams& params) {
             if (g_task_results.find(task_id) != g_task_results.end()) {
                 json result = g_task_results[task_id];
                 res.set_content(result.dump(), "application/json");
-                // Erase data after sending
-                result["data"]          = json::array();
-                g_task_results[task_id] = result;
+
+                if (g_task_results[task_id]["status"] == "Completed" ||
+                   g_task_results[task_id]["status"] == "Failed") {
+                    // Remove completed or failed tasks from the results map to free memory
+                    g_task_results.erase(task_id);
+                }
             } else {
                 res.set_content("Cannot find task " + task_id + " in queue", "text/plain");
                 res.status = 404;