address comments

Author: Bihan Jiang

nucleus/__init__.py

@@ -1171,14 +1171,7 @@ def create_custom_index(
             f"indexing/{dataset_id}",
             requests_command=requests.post,
         )
-
-        job = AsyncJob(
-            job_id=response_objects[JOB_ID_KEY],
-            job_last_known_status=response_objects[JOB_STATUS_KEY],
-            job_type=response_objects[JOB_TYPE_KEY],
-            job_creation_time=response_objects[JOB_CREATION_TIME_KEY],
-            client=self,
-        )
+        job = AsyncJob.from_json(response_objects, self)
         dataset_id = response_objects[DATASET_ID_KEY]
         message = response_objects[MESSAGE_KEY]
 

nucleus/job.py

@@ -21,11 +21,13 @@ class AsyncJob:
     client: "NucleusClient"  # type: ignore # noqa: F821
 
     def status(self) -> Dict[str, str]:
-        return self.client.make_request(
+        response = self.client.make_request(
             payload={},
             route=f"job/{self.job_id}",
             requests_command=requests.get,
         )
+        self.job_last_known_status = response["job_status"]
+        return response
 
     def errors(self) -> List[str]:
         return self.client.make_request(