fix: logging level, error to string, realtime url

docs/serverless/worker.md

@@ -16,6 +16,8 @@ RUNPOD_WEBHOOK_POST_OUTPUT= # URL to post output to
 RUNPOD_WEBHOOK_PING= # URL to ping
 RUNPOD_PING_INTERVAL= # Interval in milliseconds to ping the API (Default: 10000)
 
+RUNPOD_ENDPOINT_ID= # Endpoint ID
+
 # Realtime
 RUNPOD_REALTIME_PORT= # Port to listen on for realtime connections (Default: None)
 RUNPOD_REALTIME_CONCURRENCY= # Number of workers to spawn (Default: 1)

runpod/serverless/modules/job.py

@@ -47,7 +47,7 @@ def run_job(handler, job):
         if isinstance(job_output, bool):
             run_result = {"output": job_output}
         elif "error" in job_output:
-            run_result = {"error": job_output['error']}
+            run_result = {"error": str(job_output['error'])}
         else:
             run_result = {"output": job_output}
 

runpod/serverless/modules/logging.py

@@ -4,7 +4,7 @@
 from dotenv import load_dotenv
 
 LOG_LEVEL_ERROR = 'ERROR  '
-LOG_LEVEL_WARN = 'WARNING'
+LOG_LEVEL_WARN = 'WARN'
 LOG_LEVEL_INFO = 'INFO   '
 LOG_LEVEL_DEBUG = 'DEBUG  '
 
@@ -16,8 +16,22 @@ def log(message, level='INFO'):
     '''
     Log message to stdout if RUNPOD_DEBUG is true.
     '''
-    if os.environ.get('RUNPOD_DEBUG', 'true') == 'true':
-        print(f'{level} | {message}', flush=True)
+    set_level = os.environ.get('RUNPOD_DEBUG_LEVEL', 'DEBUG').upper()
+
+    if os.environ.get('RUNPOD_DEBUG', 'False') != 'true':
+        return
+
+    if set_level == 'ERROR' and level != 'ERROR':
+        return
+
+    if set_level == 'WARN' and level not in ['ERROR', 'WARN']:
+        return
+
+    if set_level == 'INFO' and level not in ['ERROR', 'WARN', 'INFO']:
+        return
+
+    print(f'{level} | {message}', flush=True)
+    return
 
 
 def log_secret(secret_name, secret, level='INFO'):

runpod/serverless/modules/rp_fastapi.py

@@ -34,7 +34,8 @@ def __init__(self):
 
         self.config = {"handler": None}
         self.rp_app = FastAPI()
-        self.rp_app.add_api_route("/run", self.run, methods=["POST"])
+        self.rp_app.add_api_route(f"{os.environ.get('RUNPOD_ENDPOINT_ID')}/realtime",
+                                  self.run, methods=["POST"])
 
     def start_uvicorn(self, api_port):
         '''

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = runpod
-version = 0.8.0
+version = 0.8.1
 description = Official Python library for RunPod API & SDK.
 long_description = file: README.md
 long_description_content_type = text/markdown