fix: remove extra env vars

Author: Justin Merrell

docs/serverless-worker.md

@@ -7,10 +7,8 @@ To convert a Pod to a Worker, you need to add the following annotations to the P
 ```bash
 # Development
 DEBUG= # Set to 'true' to enable debug mode, otherwise leave blank
-TEST_LOCAL= # Set to 'true' to test locally, otherwise leave blank
 
 # Configurations
-MODEL= # Model to use (Default: stable_diffusion)
 EXECUTION_TIMEOUT= # Timeout for execution in milliseconds (Default: 300000)
 TERMINATE_IDLE_TIME= # Time in milliseconds to wait before terminating idle pods (Default: 60000)
 
@@ -78,4 +76,4 @@ To test locally, create the file `test_inputs.json` in the root directory that c
 }
 ```
 
-The inputs should match the inputs your model would expect to see from the API. Then set `TEST_LOCAL` to `true` in the .env file and run the worker.
+If the required webhook environment variables are not set, the worker will default to local testing.

runpod/serverless/modules/job.py

@@ -75,7 +75,7 @@ def post(worker_id, job_id, job_output, job_time):
     '''
     Complete the job.
     '''
-    if os.environ.get('TEST_LOCAL', 'false') == 'true':
+    if os.environ.get('WEBHOOK_POST_OUTPUT', None) is None:
         return
 
     job_data = {

runpod/serverless/modules/lifecycle.py

@@ -72,7 +72,7 @@ def check_worker_ttl_thread(self):
         Check worker TTL.
         Reduce TTL or timeout by 1 every second.
         '''
-        if os.environ.get('TEST_LOCAL', 'false') != 'true':
+        if os.environ.get('RUNPOD_API_KEY', None) is not None:
             if self.ttl <= 0 or self.work_timeout <= 0:
                 self.seppuku()
 

runpod/serverless/modules/upload.py

@@ -21,13 +21,16 @@
     }
 )
 
-boto_client = bucket_session.client(
-    's3',
-    endpoint_url=os.environ.get('BUCKET_ENDPOINT_URL', None),
-    aws_access_key_id=os.environ.get('BUCKET_ACCESS_KEY_ID', None),
-    aws_secret_access_key=os.environ.get('BUCKET_SECRET_ACCESS_KEY', None),
-    config=boto_config
-)
+if os.environ.get('BUCKET_ENDPOINT_URL', None) is not None:
+    boto_client = bucket_session.client(
+        's3',
+        endpoint_url=os.environ.get('BUCKET_ENDPOINT_URL', None),
+        aws_access_key_id=os.environ.get('BUCKET_ACCESS_KEY_ID', None),
+        aws_secret_access_key=os.environ.get('BUCKET_SECRET_ACCESS_KEY', None),
+        config=boto_config
+    )
+else:
+    boto_client = None
 
 
 # ---------------------------------------------------------------------------- #
@@ -37,7 +40,7 @@ def upload_image(job_id, job_results):
     '''
     Upload image to bucket storage.
     '''
-    if os.environ.get('TEST_LOCAL', 'false') == 'true':
+    if boto_client is None:
         # Save the output to a file
         for index, result in enumerate(job_results):
             output = BytesIO()

setup.cfg

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