synapse-sdk 1.0.0b10__py3-none-any.whl → 1.0.0b12__py3-none-any.whl

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)}'
@@ -851,7 +862,7 @@ class ToTaskAction(Action):
             pre_processor_params['image_path'] = primary_file_url
 
             inference_payload = {
-                'agent': 1,
+                'agent': self.params['agent'],
                 'action': 'inference',
                 'version': pre_processor_version,
                 'params': {
@@ -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.dist-info → synapse_sdk-1.0.0b12.dist-info}/METADATA

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

{synapse_sdk-1.0.0b10.dist-info → synapse_sdk-1.0.0b12.dist-info}/RECORD

@@ -141,7 +141,7 @@ synapse_sdk/plugins/categories/post_annotation/templates/plugin/post_annotation.
 synapse_sdk/plugins/categories/pre_annotation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/pre_annotation/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/pre_annotation/actions/pre_annotation.py,sha256=6ib3RmnGrjpsQ0e_G-mRH1lfFunQ3gh2M831vuDn7HU,344
-synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py,sha256=trdlqnBlUSRrAZisnGw-aSxaiMZECSi05zuRyJxfbLQ,40516
+synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py,sha256=WSoLkQWKfLdS6qaCJEO780_fUNMi3--7zPf84YJET-Y,42006
 synapse_sdk/plugins/categories/pre_annotation/templates/config.yaml,sha256=VREoCp9wsvZ8T2E1d_MEKlR8TC_herDJGVQtu3ezAYU,589
 synapse_sdk/plugins/categories/pre_annotation/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/pre_annotation/templates/plugin/pre_annotation.py,sha256=HBHxHuv2gMBzDB2alFfrzI_SZ1Ztk6mo7eFbR5GqHKw,106
@@ -211,9 +211,9 @@ synapse_sdk/utils/storage/providers/gcp.py,sha256=i2BQCu1Kej1If9SuNr2_lEyTcr5M_n
 synapse_sdk/utils/storage/providers/http.py,sha256=2DhIulND47JOnS5ZY7MZUex7Su3peAPksGo1Wwg07L4,5828
 synapse_sdk/utils/storage/providers/s3.py,sha256=ZmqekAvIgcQBdRU-QVJYv1Rlp6VHfXwtbtjTSphua94,2573
 synapse_sdk/utils/storage/providers/sftp.py,sha256=_8s9hf0JXIO21gvm-JVS00FbLsbtvly4c-ETLRax68A,1426
-synapse_sdk-1.0.0b10.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0b10.dist-info/METADATA,sha256=SbK-qGKAO8WGswV_JGf0qJVFBPfC_f6dx0I6o46MDlQ,3745
-synapse_sdk-1.0.0b10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synapse_sdk-1.0.0b10.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0b10.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0b10.dist-info/RECORD,,
+synapse_sdk-1.0.0b12.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0b12.dist-info/METADATA,sha256=pecWR-58KXEkIky8tZLrFknRIX3YKTDAVwYPzONzz38,3745
+synapse_sdk-1.0.0b12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synapse_sdk-1.0.0b12.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0b12.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0b12.dist-info/RECORD,,