fix: don't reset new_response_dict in _get_source_images

Haidra-Org · Dec 27, 2023 · d173e6d · d173e6d
1 parent 941d27d
commit d173e6d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py
@@ -1767,7 +1767,7 @@ async def _get_source_images(self, job_pop_response: ImageGenerateJobPopResponse
 
         # TODO: Move this into horde_sdk
         for field_name in image_fields:
-            field_value = getattr(job_pop_response, field_name)
+            field_value = new_response_dict[field_name]
             if field_value is not None and "https://" in field_value:
                 fail_count = 0
                 while True:
@@ -1785,7 +1785,6 @@ async def _get_source_images(self, job_pop_response: ImageGenerateJobPopResponse
                             timeout=aiohttp.ClientTimeout(total=10),
                         )
                         response.raise_for_status()
-                        new_response_dict = job_pop_response.model_dump(by_alias=True)
 
                         content = await response.content.read()