synapse-sdk 1.0.0b10__tar.gz → 1.0.0b11__tar.gz

{synapse_sdk-1.0.0b10 → synapse_sdk-1.0.0b11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0b10
+Version: 1.0.0b11
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0b10 → synapse_sdk-1.0.0b11}/synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py

@@ -163,6 +163,10 @@ class ToTaskRun(Run):
             'message': 'Inference annotation completed. Success: {}, Failed: {}',
             'level': None,
         },
+        'INFERENCE_PREPROCESSOR_FAILED': {
+            'message': 'Inference pre processor failed for task {}: {}',
+            'level': Context.DANGER,
+        },
     }
 
     def log_message_with_code(self, code: str, *args, level: Optional[Context] = None):
@@ -658,6 +662,8 @@ class ToTaskAction(Action):
             # Get pre-processor information
             pre_processor_info = self._get_pre_processor_info(client, pre_processor_id)
             if not pre_processor_info['success']:
+                error_msg = pre_processor_info.get('error', 'Failed to get pre-processor info')
+                self.run.log_annotate_task_event('INFERENCE_PREPROCESSOR_FAILED', task_id, error_msg)
                 return pre_processor_info
 
             pre_processor_code = pre_processor_info['code']
@@ -666,6 +672,8 @@ class ToTaskAction(Action):
             # Ensure pre-processor is running
             pre_processor_status = self._ensure_pre_processor_running(client, pre_processor_code)
             if not pre_processor_status['success']:
+                error_msg = pre_processor_status.get('error', 'Failed to ensure pre-processor running')
+                self.run.log_annotate_task_event('INFERENCE_PREPROCESSOR_FAILED', task_id, error_msg)
                 return pre_processor_status
 
             # Extract primary file URL from task data
@@ -679,20 +687,23 @@ class ToTaskAction(Action):
             # Run inference
             inference_result = self._run_inference(client, pre_processor_code, pre_processor_version, primary_file_url)
             if not inference_result['success']:
+                error_msg = inference_result.get('error', 'Failed to run inference')
+                self.run.log_annotate_task_event('INFERENCE_PREPROCESSOR_FAILED', task_id, error_msg)
                 return inference_result
 
             # Convert and submit inference data
-            annotation_to_task = self.entrypoint(self.run)
-            converted_result = annotation_to_task.convert_data_from_inference(inference_result['data'])
+            try:
+                annotation_to_task = self.entrypoint(self.run)
+                converted_result = annotation_to_task.convert_data_from_inference(inference_result['data'])
+            except Exception as e:
+                error_msg = f'Failed to convert inference data: {str(e)}'
+                self.run.log_annotate_task_event('INFERENCE_PREPROCESSOR_FAILED', task_id, error_msg)
+                return {'success': False, 'error': error_msg}
 
             # Submit inference annotation data
             client.annotate_task_data(task_id, data={'action': 'submit', 'data': converted_result})
 
-            # Log success
-            self.run.log_annotate_task_data(
-                {'task_id': task_id, 'pre_processor_id': pre_processor_id}, AnnotateTaskDataStatus.SUCCESS
-            )
-            return {'success': True}
+            return {'success': True, 'pre_processor_id': pre_processor_id}
 
         except Exception as e:
             error_msg = f'Failed to process inference for task {task_id}: {str(e)}'
@@ -926,8 +937,17 @@ class ToTaskAction(Action):
                 result = self._process_single_task(client, task_id, task_params, None, AnnotationMethod.INFERENCE)
                 if result['success']:
                     success_count += 1
+                    pre_processor_id = result.get('pre_processor_id')
+                    task_data = {'task_id': task_id}
+                    if pre_processor_id:
+                        task_data['pre_processor_id'] = pre_processor_id
+                    self.run.log_annotate_task_data(task_data, AnnotateTaskDataStatus.SUCCESS)
                 else:
                     failed_count += 1
+                    error_msg = result.get('error', 'Unknown error')
+                    self.run.log_annotate_task_data(
+                        {'task_id': task_id, 'error': error_msg}, AnnotateTaskDataStatus.FAILED
+                    )
 
                 current_progress += 1
                 self._update_metrics(total_tasks, success_count, failed_count)

{synapse_sdk-1.0.0b10 → synapse_sdk-1.0.0b11}/synapse_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0b10
+Version: 1.0.0b11
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT