fix: use a certifi ssl context for r2 uploads

horde_worker_regen/process_management/process_manager.py

@@ -8,6 +8,7 @@
 import os
 import queue
 import random
+import ssl
 import sys
 import time
 from asyncio import CancelledError, Task
@@ -22,6 +23,7 @@
 
 import aiohttp
 import aiohttp.client_exceptions
+import certifi
 import PIL
 import PIL.Image
 import psutil
@@ -89,6 +91,8 @@
 )
 from horde_worker_regen.process_management.worker_entry_points import start_inference_process, start_safety_process
 
+sslcontext = ssl.create_default_context(cafile=certifi.where())
+
 # This is due to Linux/Windows differences in the multiprocessing module
 try:
     from multiprocessing.connection import PipeConnection as Connection  # type: ignore
@@ -2469,6 +2473,7 @@ async def _do_upload(new_submit: PendingSubmitJob, image_in_buffer_bytes: bytes)
                     data=image_in_buffer_bytes,
                     skip_auto_headers=["content-type"],
                     timeout=aiohttp.ClientTimeout(total=10),
+                    ssl=sslcontext,
                 ) as response:
                     if response.status != 200:
                         logger.error(f"Failed to upload image to R2: {response}")

requirements.txt

@@ -1,6 +1,8 @@
 numpy==1.26.4
 torch==2.3.1
-qrcode==7.4.2
+qrcode==7.4.2 # >8 breaks horde-engine 2.15.3 via the qr code generation nodes
+
+certifi # Required for SSL cert resolution
 
 horde_sdk~=0.14.11
 horde_safety~=0.2.3