save code

worker.py

@@ -8,7 +8,7 @@ import threading
 import json
 from datetime import datetime, timedelta
 
-from config import REDIS_HOST
+from config import REDIS_HOST, DEFAULT_TIMEOUT
 from config import REDIS_PORT
 from config import REDIS_DB
 from config import QUEUE_NAME
@@ -201,7 +201,7 @@ def call_remote_gpu_server(task_data_str, server=None):
             if data['output_format'] == 'base64':
                 base64 = True
 
-            response = requests.post(url, headers=headers, json=data, timeout=30)
+            response = requests.post(url, headers=headers, json=data, timeout=DEFAULT_TIMEOUT)
             
             logger.info(f"call finshed {response.status_code} {response.headers} {response.text[:128]}")
             if response.status_code == 200:
@@ -260,9 +260,9 @@ def call_remote_gpu_server(task_data_str, server=None):
         result_status_code = '500'
     
     result_key = f"result_{key}"
-    redis_conn.set(result_key, result_str, ex=60)
+    redis_conn.set(result_key, result_str, ex=DEFAULT_TIMEOUT)
     result_status_key = f"result_status_code_{key}"
-    redis_conn.set(result_status_key, result_status_code, ex=60)
+    redis_conn.set(result_status_key, result_status_code, ex=DEFAULT_TIMEOUT)
     
 
 def get_task_queue_key():
@@ -315,7 +315,6 @@ if __name__ == '__main__':
     print(f"[Worker] Starting with PID: {os.getpid()}")
     redis_conn.set(GPU_SERVER_LIST, "[]")
 
-    servers = regServer("localhost")
-    for s in servers:
-        regServer(s)
+    regServer("localhost")
+
     main_worker()