Merge pull request #231 from runpod/job-take-bug

Job take bug

Author: justinmerrell

runpod/serverless/modules/rp_job.py

@@ -71,7 +71,7 @@ async def get_job(session: ClientSession, retry=True) -> Optional[Dict[str, Any]
                     continue
 
                 received_request = await response.json()
-                log.debug("Request Received", {next_job})
+                log.debug(f"Request Received | {received_request}")
 
                 # Check if the job is valid
                 job_id = received_request.get("id", None)
@@ -92,7 +92,7 @@ async def get_job(session: ClientSession, retry=True) -> Optional[Dict[str, Any]
             err_type = type(err).__name__
             err_message = str(err)
             err_traceback = traceback.format_exc()
-            log.error(f"Failed to get job, error type: {err_type}, error message: {err_message}")
+            log.error(f"Failed to get job. | Error Type: {err_type} | Error Message: {err_message}")
             log.error(f"Traceback: {err_traceback}")
 
         if next_job is None:

runpod/serverless/modules/rp_logger.py

@@ -12,6 +12,7 @@
 
 import os
 import json
+from typing import Optional
 
 
 LOG_LEVELS = ['NOTSET', 'DEBUG', 'INFO', 'WARN', 'ERROR']
@@ -95,25 +96,25 @@ def secret(self, secret_name, secret):
         redacted_secret = secret[0] + '*' * (len(secret)-2) + secret[-1]
         self.info(f"{secret_name}: {redacted_secret}")
 
-    def debug(self, message, request_id=None):
+    def debug(self, message, request_id: Optional[str] = None):
         '''
         debug log
         '''
         self.log(message, 'DEBUG', request_id)
 
-    def info(self, message, request_id=None):
+    def info(self, message, request_id: Optional[str] = None):
         '''
         info log
         '''
         self.log(message, 'INFO', request_id)
 
-    def warn(self, message, request_id=None):
+    def warn(self, message, request_id: Optional[str] = None):
         '''
         warn log
         '''
         self.log(message, 'WARN', request_id)
 
-    def error(self, message, request_id=None):
+    def error(self, message, request_id: Optional[str] = None):
         '''
         error log
         '''