fix: job_ttl on pop

Haidra-Org · Aug 22, 2024 · 3f2bcff · 3f2bcff
1 parent 932bc13
commit 3f2bcff
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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