fix: validate request after style apply

Author: db0

CHANGELOG.md

@@ -6,6 +6,11 @@ SPDX-License-Identifier: AGPL-3.0-or-later
 
 # Changelog
 
+# 4.44.3
+
+* Fix image validation warnings being sent to the wrong requests
+* Validate request with styles, only after style is applied
+
 # 4.44.2
 
 * Allow trusted users to also create styles

horde/apis/v2/kobold.py

@@ -95,8 +95,6 @@ def post(self):
         return (ret_dict, 202)
 
     def initiate_waiting_prompt(self):
-        self.prompt = self.args.prompt
-        self.apply_style()
         self.wp = TextWaitingPrompt(
             worker_ids=self.workers,
             models=self.models,
@@ -169,6 +167,8 @@ def get_size_too_big_message(self):
         )
 
     def validate(self):
+        self.prompt = self.args.prompt
+        self.apply_style()
         super().validate()
         param_validator = ParamValidator(self.args.prompt, self.args.models, self.params, self.user)
         self.warnings = param_validator.validate_text_params()

horde/apis/v2/stable.py

@@ -115,6 +115,8 @@ def get_size_too_big_message(self):
         )
 
     def validate(self):
+        self.prompt = self.args.prompt
+        self.apply_style()
         super().validate()
         param_validator = ParamValidator(prompt=self.args.prompt, models=self.args.models, params=self.params, user=self.user)
         self.warnings = param_validator.validate_image_params()
@@ -223,8 +225,6 @@ def initiate_waiting_prompt(self):
             shared = True
         else:
             shared = False
-        self.prompt = self.args.prompt
-        self.apply_style()
         self.wp = ImageWaitingPrompt(
             worker_ids=self.workers,
             models=self.models,

horde/consts.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: AGPL-3.0-or-later
 
-HORDE_VERSION = "4.44.2"
+HORDE_VERSION = "4.44.3"
 HORDE_API_VERSION = "2.5"
 
 WHITELISTED_SERVICE_IPS = {

horde/validation.py

@@ -24,6 +24,7 @@ def __init__(self, prompt, models, params, user):
         self.models = models
         self.params = params
         self.user = user
+        self.warnings = set()
 
     def validate_base_params(self):
         pass