From 3f2bcff9486060962dada20d6300f92c39ad1e4f Mon Sep 17 00:00:00 2001
From: db0 <mail@dbzer0.com>
Date: Thu, 22 Aug 2024 19:17:32 +0200
Subject: [PATCH] fix: job_ttl on pop

---
 CHANGELOG.md                         | 4 ++++
 horde/classes/base/waiting_prompt.py | 3 ++-
 horde/consts.py                      | 2 +-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 0222aa0a..15aa7b90 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,10 @@ SPDX-License-Identifier: AGPL-3.0-or-later
 
 # Changelog
 
+# 4.40.3
+
+* Ensure jobs don't expire soon after being picked up
+
 # 4.40.2
 
 * Prevents too high max context from being used for rewards
diff --git a/horde/classes/base/waiting_prompt.py b/horde/classes/base/waiting_prompt.py
index b3789348..9075c2c9 100644
--- a/horde/classes/base/waiting_prompt.py
+++ b/horde/classes/base/waiting_prompt.py
@@ -239,7 +239,8 @@ def start_generation(self, worker, amount=1):
         current_n = self.n
         self.n -= safe_amount
         payload = self.get_job_payload(current_n)
-        db.session.commit()
+        # This does a commit as well
+        self.refresh()
         procgen_class = procgen_classes[self.wp_type]
         gens_list = []
         model = None
diff --git a/horde/consts.py b/horde/consts.py
index b62f1a29..d1d8f106 100644
--- a/horde/consts.py
+++ b/horde/consts.py
@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: AGPL-3.0-or-later
 
-HORDE_VERSION = "4.40.2git "
+HORDE_VERSION = "4.40.3 "
 
 WHITELISTED_SERVICE_IPS = {
     "212.227.227.178",  # Turing Bot