orbitkit 0.8.71__tar.gz → 0.8.73__tar.gz

{orbitkit-0.8.71/orbitkit.egg-info → orbitkit-0.8.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orbitkit
-Version: 0.8.71
+Version: 0.8.73
 Summary: This project is only for Orbit Tech internal use.
 Home-page: https://github.com/clown-0726/orbitkit
 Author: Lilu Cao

orbitkit-0.8.73/orbitkit/VERSION

@@ -0,0 +1 @@
+0.8.73

{orbitkit-0.8.71 → orbitkit-0.8.73}/orbitkit/airflow_handler/file_flow_entry_process.py

@@ -217,10 +217,10 @@ class FilingOfficialProcessor:
                            query: dict, tags: list[str], priority: str,
                            is_important: bool = False, custom_step: Optional[list[str]] = None, important_level = None, db_name: str = None, queue_type: str = 'Default'):
 
-        if not important_level or not isinstance(important_level, int):
-            important_level = 0
+        if source == 'filing_data':
+            raise ValueError(f"Cannot update documents from 'filing_data' source. ")
 
-        if important_level == 0:
+        if important_level is None or not isinstance(important_level, int) or important_level <= 0:
             raise ValueError(f'important_level must be an integer (int) greater than 0. {important_level}')
 
         allowed_steps = {"convert", "extract", "embedding"}

{orbitkit-0.8.71 → orbitkit-0.8.73}/orbitkit/airflow_handler/file_flow_exit_process.py

@@ -96,7 +96,11 @@ class FlowUpdater:
             update_params.update({self.kafka_ignore.X_OTHERS_K_IGNORE_KEY: self.kafka_ignore.get_k_ignore_val()})
         update_params.update({"x_updated_date": datetime.now(tz=pytz.timezone('UTC')).strftime("%Y-%m-%dT%H:%M:%S%z")})
         collection = self._check_and_create_collection(data_source)
-        result = collection.update_one({'_id': report_id}, {'$set': update_params})
+        # 2026-03-02 Status rollback. In special cases, it is necessary to reprocess attachments when the original success has now failed, addressing database constraint exceptions.
+        if update_params.get('x_status_list.status_convert.status') == 'convert_failed':
+            result = collection.update_one({'_id': report_id}, {'$set': update_params, '$unset': {'x_attachments_pdf': ''}})
+        else:
+            result = collection.update_one({'_id': report_id}, {'$set': update_params})
         if result.matched_count == 0:
             raise LookupError(f"No document found with id {report_id} to update. {data_source}")
         logger.info(f"Document with id {report_id} update attempted. Modified: {result.modified_count}")
@@ -133,11 +137,7 @@ class FlowUpdater:
         db_doc = self._check_and_create_collection(data_source).find_one({'_id': report_id},
                                                                          {'_id': 1, 'x_attachments': 1,
                                                                           'x_status_list': 1})
-        if not db_doc and data_source == 'filing_data':
-            data_source = 'filing_data_staging'
-            db_doc = self._check_and_create_collection(data_source).find_one({'_id': report_id},
-                                                                             {'_id': 1, 'x_attachments': 1,
-                                                                              'x_status_list': 1})
+
         if not db_doc:
             logger.warning(f"No document found with id {report_id}.")
             return

{orbitkit-0.8.71 → orbitkit-0.8.73/orbitkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orbitkit
-Version: 0.8.71
+Version: 0.8.73
 Summary: This project is only for Orbit Tech internal use.
 Home-page: https://github.com/clown-0726/orbitkit
 Author: Lilu Cao

orbitkit-0.8.71/orbitkit/VERSION

@@ -1 +0,0 @@
-0.8.71