initial commit WIP

Author: agrimk

ads/aqua/modeldeployment/deployment.py

@@ -1309,4 +1309,25 @@ def list_shapes(self, **kwargs) -> List[ComputeShapeSummary]:
                 or gpu_specs.shapes.get(oci_shape.name.upper()),
             )
             for oci_shape in oci_shapes
-        ]
+        ]
+    
+        
+    def get_deployment_status(self,model_deployment_id: str, work_request_id : str) : 
+    #     category= "aqua/{model_type}/deployment/status", action= "FAILED/SUCCEEDED", detail="Error message from Work request", value= {"ocid": md_ocid[:8]}
+    # # tracks unique evaluation that were created for the given evaluation source
+    #     self.telemetry.record_event_async(
+    #         category="aqua/evaluation",
+    #         action="create",
+    #         detail=self._get_service_model_name(evaluation_source),
+    #     )
+    
+    
+        return 
+    
+    def get_deployment_status_async(self,model_deployment_id: str, work_request_id : str) : 
+        # tracks unique evaluation that were created for the given evaluation source
+        self.telemetry.record_event_async(
+            category="aqua/evaluation",
+            action="create",
+            detail=self._get_service_model_name(evaluation_source),
+        )

ads/common/work_request.py

@@ -38,6 +38,7 @@ def __init__(
         config: dict = None, 
         signer: Signer = None, 
         client_kwargs: dict = None, 
+        _error_message: str = None,
         **kwargs
     ) -> None:
         """Initializes ADSWorkRequest object.
@@ -65,6 +66,7 @@ def __init__(
         self._description = description
         self._percentage = 0
         self._status = None
+        _error_message = _error_message
         super().__init__(config, signer, client_kwargs, **kwargs)
 
 
@@ -78,6 +80,7 @@ def _sync(self):
         self._percentage= work_request.percent_complete
         self._status = work_request.status
         self._description = work_request_logs[-1].message if work_request_logs else "Processing"
+        if work_request.status == 'FAILED' : self._error_message = self.client.list_work_request_errors
 
     def watch(
         self, 

ads/model/service/oci_datascience_model_deployment.py

@@ -185,9 +185,9 @@ def activate(
                 self.id,
             )
 
+           
+            self.workflow_req_id = response.headers.get("opc-work-request-id", None)
             if wait_for_completion:
-                self.workflow_req_id = response.headers.get("opc-work-request-id", None)
-
                 try:
                     DataScienceWorkRequest(self.workflow_req_id).wait_work_request(
                         progress_bar_description="Activating model deployment",

ads/telemetry/client.py

@@ -119,4 +119,4 @@ def record_event_async(
         Thread
             A started thread to send a head request to generate an event record.
         """
-        thread_pool.submit(self.record_event, args=(category, action, detail), kwargs=kwargs)
+        thread_pool.submit(self.record_event, category, action, detail, **kwargs)