From 16a5c1c7b64c4a8656e559160544d40b222bd835 Mon Sep 17 00:00:00 2001
From: vansangpfiev <sang@jan.ai>
Date: Fri, 13 Dec 2024 11:48:53 +0700
Subject: [PATCH] fix: correct stop inferencing condition

---
 engine/controllers/server.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/engine/controllers/server.cc b/engine/controllers/server.cc
index a9920e8aa..4c6bcaf82 100644
--- a/engine/controllers/server.cc
+++ b/engine/controllers/server.cc
@@ -140,7 +140,9 @@ void server::ProcessStreamRes(std::function<void(const HttpResponsePtr&)> cb,
                                       std::size_t buf_size) -> std::size_t {
     if (buf == nullptr) {
       LOG_TRACE << "Buf is null";
-      inference_svc_->StopInferencing(engine_type, model_id);
+      if (!(*err_or_done)) {
+        inference_svc_->StopInferencing(engine_type, model_id);
+      }
       return 0;
     }