synapse-sdk 1.0.0b4__py3-none-any.whl → 1.0.0b6__py3-none-any.whl

synapse_sdk/plugins/categories/export/actions/export.py

@@ -1,6 +1,7 @@
 import json
 from abc import ABC, abstractmethod
 from datetime import datetime
+from itertools import tee
 from typing import Annotated, Any, Literal
 
 from pydantic import AfterValidator, BaseModel, field_validator
@@ -324,16 +325,18 @@ class ExportAction(Action):
         handler = TargetHandlerFactory.get_handler(target)
 
         self.params['results'], self.params['count'] = self.get_filtered_results(filters, handler)
-        export_items = handler.get_export_item(self.params['results'])
 
         # For the 'ground_truth' target, retrieve project information from the first result and add configuration
         if target == 'ground_truth':
             try:
-                first_result = next(iter(self.params['results']))
+                # Split generator into two using tee()
+                peek_iter, main_iter = tee(self.params['results'])
+                first_result = next(peek_iter)  # Peek first value only
                 project_pk = first_result['project']
                 project_info = self.client.get_project(project_pk)
                 self.params['configuration'] = project_info.get('configuration', {})
-            except StopIteration:
+                self.params['results'] = main_iter  # Keep original generator intact
+            except (StopIteration, KeyError):
                 self.params['configuration'] = {}
         # For the 'assignment' and 'task' targets, retrieve the project from the filter as before
         elif target in ['assignment', 'task'] and 'project' in self.params['filter']:
@@ -341,6 +344,7 @@ class ExportAction(Action):
             project_info = self.client.get_project(project_pk)
             self.params['configuration'] = project_info.get('configuration', {})
 
+        export_items = handler.get_export_item(self.params['results'])
         storage = self.client.get_storage(self.params['storage'])
         pathlib_cwd = get_pathlib(storage, self.params['path'])
         return self.entrypoint(self.run, export_items, pathlib_cwd, **self.params)

synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py

@@ -8,6 +8,7 @@ from pydantic import AfterValidator, BaseModel, field_validator
 from pydantic_core import PydanticCustomError
 
 from synapse_sdk.clients.backend import BackendClient
+from synapse_sdk.clients.backend.models import JobStatus
 from synapse_sdk.clients.exceptions import ClientError
 from synapse_sdk.plugins.categories.base import Action
 from synapse_sdk.plugins.categories.decorators import register_action
@@ -35,6 +36,14 @@ class CriticalError(Exception):
         super().__init__(self.message)
 
 
+class PreAnnotationToTaskFailed(Exception):
+    """Pre-annotation to task failed."""
+
+    def __init__(self, message: str = 'Pre-annotation to task failed'):
+        self.message = message
+        super().__init__(self.message)
+
+
 class ToTaskRun(Run):
     class AnnotateTaskEventLog(BaseModel):
         """Annotate task event log model."""
@@ -255,6 +264,25 @@ class ToTaskParams(BaseModel):
         return value
 
 
+class ToTaskResult(BaseModel):
+    """Result model for ToTaskAction.start method.
+
+    Args:
+        status (JobStatus): The job status from the action execution.
+        message (str): A descriptive message about the action result.
+    """
+
+    status: JobStatus
+    message: str
+
+    def model_dump(self, **kwargs):
+        """Override model_dump to return status as enum value."""
+        data = super().model_dump(**kwargs)
+        if 'status' in data and isinstance(data['status'], JobStatus):
+            data['status'] = data['status'].value
+        return data
+
+
 @register_action
 class ToTaskAction(Action):
     """ToTask action for pre-annotation data processing.
@@ -306,14 +334,20 @@ class ToTaskAction(Action):
         }
     }
 
-    def start(self):
+    def start(self) -> dict:
         """Start to_task action.
 
         * Generate tasks.
         * Annotate data to tasks.
+
+        Returns:
+            dict: Validated result with status and message.
         """
         if not self.run or not self.params:
-            raise ValueError('Run instance or parameters not properly initialized')
+            result = ToTaskResult(
+                status=JobStatus.FAILED, message='Run instance or parameters not properly initialized'
+            )
+            raise PreAnnotationToTaskFailed(result.message)
 
         # Type assertion to help the linter
         assert isinstance(self.run, ToTaskRun)
@@ -325,20 +359,23 @@ class ToTaskAction(Action):
         if isinstance(project_response, str):
             self.run.log_message_with_code('INVALID_PROJECT_RESPONSE')
             self.run.end_log()
-            return {}
+            result = ToTaskResult(status=JobStatus.FAILED, message='Invalid project response received')
+            raise PreAnnotationToTaskFailed(result.message)
         project: Dict[str, Any] = project_response
 
         data_collection_id = project.get('data_collection')
         if not data_collection_id:
             self.run.log_message_with_code('NO_DATA_COLLECTION')
             self.run.end_log()
-            return {}
+            result = ToTaskResult(status=JobStatus.FAILED, message='Project does not have a data collection')
+            raise PreAnnotationToTaskFailed(result.message)
 
         data_collection_response = client.get_data_collection(data_collection_id)
         if isinstance(data_collection_response, str):
             self.run.log_message_with_code('INVALID_DATA_COLLECTION_RESPONSE')
             self.run.end_log()
-            return {}
+            result = ToTaskResult(status=JobStatus.FAILED, message='Invalid data collection response received')
+            raise PreAnnotationToTaskFailed(result.message)
         data_collection: Dict[str, Any] = data_collection_response
 
         # Generate tasks if provided project is empty.
@@ -355,7 +392,8 @@ class ToTaskAction(Action):
         if not task_ids_count:
             self.run.log_message_with_code('NO_TASKS_FOUND')
             self.run.end_log()
-            return {}
+            result = ToTaskResult(status=JobStatus.FAILED, message='No tasks found to annotate')
+            raise PreAnnotationToTaskFailed(result.message)
 
         # Annotate data to tasks.
         method = self.params.get('method')
@@ -365,23 +403,39 @@ class ToTaskAction(Action):
             if not target_specification_name:
                 self.run.log_message_with_code('TARGET_SPEC_REQUIRED')
                 self.run.end_log()
-                return {}
+                result = ToTaskResult(
+                    status=JobStatus.FAILED, message='Target specification name is required for file annotation method'
+                )
+                raise PreAnnotationToTaskFailed(result.message)
 
             file_specifications = data_collection.get('file_specifications', [])
             target_spec_exists = any(spec.get('name') == target_specification_name for spec in file_specifications)
             if not target_spec_exists:
                 self.run.log_message_with_code('TARGET_SPEC_NOT_FOUND', target_specification_name)
                 self.run.end_log()
-                return {}
+                result = ToTaskResult(
+                    status=JobStatus.FAILED,
+                    message=f"Target specification name '{target_specification_name}' not found in file specifications",
+                )
+                raise PreAnnotationToTaskFailed(result.message)
             self._handle_annotate_data_from_files(task_ids, target_specification_name)
         elif method == AnnotationMethod.INFERENCE:
             self._handle_annotate_data_with_inference(task_ids)
         else:
             self.run.log_message_with_code('UNSUPPORTED_METHOD', method)
             self.run.end_log()
-            return {}
+            result = ToTaskResult(status=JobStatus.FAILED, message=f'Unsupported annotation method: {method}')
+            raise PreAnnotationToTaskFailed(result.message)
+
+        current_progress = self.run.logger.get_current_progress()
+        if current_progress['overall'] != 100:
+            result = ToTaskResult(
+                status=JobStatus.FAILED, message='Pre-annotation to task failed. Current progress is not 100%'
+            )
+            raise PreAnnotationToTaskFailed(result.message)
 
-        return {}
+        result = ToTaskResult(status=JobStatus.SUCCEEDED, message='Pre-annotation to task completed successfully')
+        return result.model_dump()
 
     def _handle_annotate_data_from_files(self, task_ids: List[int], target_specification_name: str):
         """Handle annotate data from files to tasks.
@@ -757,21 +811,21 @@ class ToTaskAction(Action):
         except Exception as e:
             return {'success': False, 'error': f'Failed to restart pre-processor: {str(e)}'}
 
-    def _extract_primary_file_url(self, task: Dict[str, Any]) -> Tuple[str, str]:
+    def _extract_primary_file_url(self, task: Dict[str, Any]) -> Tuple[Optional[str], Optional[str]]:
         """Extract the primary file URL from task data.
 
         Args:
             task (Dict[str, Any]): The task data.
 
         Returns:
-            Tuple[str, str]: The primary file URL and original name.
+            Tuple[Optional[str], Optional[str]]: The primary file URL and original name.
         """
         data_unit = task.get('data_unit', {})
         files = data_unit.get('files', {})
 
         for file_info in files.values():
             if isinstance(file_info, dict) and file_info.get('is_primary') and file_info.get('url'):
-                return file_info['url'], file_info['file_name_original']
+                return file_info['url'], file_info.get('file_name_original')
 
         return None, None
 

synapse_sdk/plugins/categories/upload/actions/upload.py

@@ -18,6 +18,7 @@ from synapse_sdk.i18n import gettext as _
 from synapse_sdk.plugins.categories.base import Action
 from synapse_sdk.plugins.categories.decorators import register_action
 from synapse_sdk.plugins.enums import PluginCategory, RunMethod
+from synapse_sdk.plugins.exceptions import ActionError
 from synapse_sdk.plugins.models import Run
 from synapse_sdk.shared.enums import Context
 from synapse_sdk.utils.pydantic.validators import non_blank
@@ -684,30 +685,26 @@ class UploadAction(Action):
         # Validate the organized files
         if not self._validate_organized_files(organized_files, file_specification_template):
             self.run.log_message('Validation failed.', context=Context.ERROR.value)
-            self.run.log_message('Upload is aborted due to validation errors.', context=Context.ERROR.value)
-            return result
+            raise ActionError('Upload is aborted due to validation errors.')
 
         # Upload files to synapse-backend.
         if not organized_files:
             self.run.log_message('Files not found on the path.', context=Context.WARNING.value)
-            self.run.log_message('Upload is aborted due to missing files.', context=Context.ERROR.value)
-            return result
+            raise ActionError('Upload is aborted due to missing files.')
         uploaded_files = self._upload_files(organized_files)
         result['uploaded_files_count'] = len(uploaded_files)
 
         # Generate data units for the uploaded data.
         if not uploaded_files:
             self.run.log_message('No files were uploaded.', context=Context.WARNING.value)
-            self.run.log_message('Upload is aborted due to no uploaded files.', context=Context.ERROR.value)
-            return result
+            raise ActionError('Upload is aborted due to no uploaded files.')
         generated_data_units = self._generate_data_units(uploaded_files)
         result['generated_data_units_count'] = len(generated_data_units)
 
         # Setup task with uploaded synapse-backend data units.
         if not generated_data_units:
             self.run.log_message('No data units were generated.', context=Context.WARNING.value)
-            self.run.log_message('Upload is aborted due to no generated data units.', context=Context.ERROR.value)
-            return result
+            raise ActionError('Upload is aborted due to no generated data units.')
 
         self.run.log_message('Import completed.')
         return result

{synapse_sdk-1.0.0b4.dist-info → synapse_sdk-1.0.0b6.dist-info}/METADATA

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

{synapse_sdk-1.0.0b4.dist-info → synapse_sdk-1.0.0b6.dist-info}/RECORD

@@ -113,7 +113,7 @@ synapse_sdk/plugins/categories/data_validation/templates/plugin/validation.py,sh
 synapse_sdk/plugins/categories/export/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/export/enums.py,sha256=gtyngvQ1DKkos9iKGcbecwTVQQ6sDwbrBPSGPNb5Am0,127
 synapse_sdk/plugins/categories/export/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/export/actions/export.py,sha256=7OCU3kwkYqYnelG6vfZ4ClyyMj_FHT-WQoUmgcfH080,12012
+synapse_sdk/plugins/categories/export/actions/export.py,sha256=XxhFIebt9tGTNcj0N6guVhAPP4d2YZ5Lvsx22-va72Y,12263
 synapse_sdk/plugins/categories/export/templates/config.yaml,sha256=N7YmnFROb3s3M35SA9nmabyzoSb5O2t2TRPicwFNN2o,56
 synapse_sdk/plugins/categories/export/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/export/templates/plugin/export.py,sha256=GDb6Ucodsr5aBPMU4alr68-DyFoLR5TyhC_MCaJrkF0,6411
@@ -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=M4EUOkCCSchxcdyfkhxW0eQZ80I55poVuW6od8hiuxI,37541
+synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py,sha256=x6KPK89OWS7wsmYZ-wFFPhUl_fXEO74NIVB4bUChm6Q,40323
 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
@@ -154,7 +154,7 @@ synapse_sdk/plugins/categories/smart_tool/templates/plugin/__init__.py,sha256=47
 synapse_sdk/plugins/categories/smart_tool/templates/plugin/auto_label.py,sha256=eevNg0nOcYFR4z_L_R-sCvVOYoLWSAH1jwDkAf3YCjY,320
 synapse_sdk/plugins/categories/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/upload/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/upload/actions/upload.py,sha256=7oV8kcJ_JTkHtRHHf0e1s3dka_CKf2engMkIioLTdek,38291
+synapse_sdk/plugins/categories/upload/actions/upload.py,sha256=Lhc_ezB_FvkDI7LTe50tadOQ2W9vp2ppTWYNJSWYKfY,38114
 synapse_sdk/plugins/categories/upload/templates/config.yaml,sha256=6_dRa0_J2aS8NSUfO4MKbPxZcdPS2FpJzzp51edYAZc,281
 synapse_sdk/plugins/categories/upload/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/upload/templates/plugin/upload.py,sha256=IZU4sdSMSLKPCtlNqF7DP2howTdYR6hr74HCUZsGdPk,1559
@@ -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.0b4.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0b4.dist-info/METADATA,sha256=5fqI0sn3Ro1CRZQVGAputa73_iZGd360Cxww_o6_K3Y,3718
-synapse_sdk-1.0.0b4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synapse_sdk-1.0.0b4.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0b4.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0b4.dist-info/RECORD,,
+synapse_sdk-1.0.0b6.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0b6.dist-info/METADATA,sha256=mU8-WwXY45NYGlXdk_mS1P9P8Ip21Q_9RqLZ9joe9hc,3718
+synapse_sdk-1.0.0b6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synapse_sdk-1.0.0b6.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0b6.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0b6.dist-info/RECORD,,