add wait_for_job_completion

Author: fernandogrd

README.md

@@ -33,6 +33,11 @@ create_job_info = client.create_job(job_name='your-job-name', workflow_id='your-
 job_id = create_job_info['id']
 print('Job Created:', job_id)
 
+# Wait for job to complete
+job_info = client.wait_for_job_completion(job_id)
+print('Job Status:', job_info['status'])
+print('Job Result:', job_info['result'])
+
 # Get job info
 job_info = client.get_job(job_id=job_id)
 print('Job Status:', job_info['status'])

musicai_sdk/__pycache__/client.cpython-311.pyc

@@ -1,10 +1,13 @@
+import time
 import requests
 from requests.exceptions import HTTPError
 
+
 class MusicAiClient:
-    def __init__(self, api_key):
+    def __init__(self, api_key, job_monitor_interval=2):
         self.api_key = api_key
         self.base_url = 'https://api.music.ai/api'
+        self.job_monitor_interval = job_monitor_interval
 
     def get_headers(self):
         return {
@@ -28,6 +31,12 @@ def get_job(self, job_id):
             raise HTTPError(f'Error getting job: {response.status_code} {response.text}')
         return response.json()
 
+    def get_job_status(self, job_id):
+        response = requests.get(f'{self.base_url}/job/{job_id}/status', headers=self.get_headers())
+        if response.status_code // 100 != 2:
+            raise HTTPError(f'Error getting job: {response.status_code} {response.text}')
+        return response.json()
+
     def get_jobs(self):
         response = requests.get(f'{self.base_url}/job', headers=self.get_headers())
         if response.status_code // 100 != 2:
@@ -51,8 +60,15 @@ def delete_job(self, job_id):
             raise HTTPError(f'Error deleting job: {response.status_code} {response.text}')
         return response.json()
 
+    def wait_for_job_completion(self, id):
+        while True:
+            job = self.get_job_status(id)
+            if job['status'] in ['SUCCEEDED', 'FAILED']:
+                return self.get_job(id)
+            time.sleep(self.job_monitor_interval),
+
     def get_application_info(self):
         response = requests.get(f'{self.base_url}/application', headers=self.get_headers())
         if response.status_code // 100 != 2:
             raise HTTPError(f'Error getting application info: {response.status_code} {response.text}')
-        return response.json()
+        return response.json()