documente_shared 0.1.131__tar.gz → 0.1.131b2__tar.gz

{documente_shared-0.1.131 → documente_shared-0.1.131b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.131
+Version: 0.1.131b2
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech

{documente_shared-0.1.131 → documente_shared-0.1.131b2}/documente_shared/domain/entities/document.py

@@ -49,6 +49,7 @@ class DocumentProcessing(object):
     feedback: Optional[list | dict] = None
     completed_at: Optional[datetime] = None
     metadata: Optional[dict] = None
+    document_size: Optional[Decimal] = None
     metadata_items: Optional[List[DocumentProcessingMetadata]] = None
 
     def __post_init__(self):
@@ -108,7 +109,6 @@ class DocumentProcessing(object):
             DocumentProcessingStatus.COMPLETED,
             DocumentProcessingStatus.FAILED,
         ]
-
     @property
     def file_key(self) -> str:
         return remove_slash_from_path(self.file_path)
@@ -220,6 +220,12 @@ class DocumentProcessing(object):
     
     def in_review(self):
         self.status = DocumentProcessingStatus.IN_REVIEW
+    
+    def is_size_gt(self, size: Decimal = Decimal(10)):
+        return self.document_size and self.document_size >= size
+    
+    def is_size_lt(self, size: Decimal = Decimal(10)):
+        return self.document_size and self.document_size <= size
 
     def __eq__(self, other: 'DocumentProcessing') -> bool:
         if not other:
@@ -279,6 +285,7 @@ class DocumentProcessing(object):
             'metadata': self.metadata,
             'completed_at': self.completed_at.isoformat() if self.completed_at else None,
             'metadata_items': [metadata.to_dict for metadata in self.metadata_items],
+            'document_size': self.document_size,
         }
 
     @property
@@ -324,6 +331,7 @@ class DocumentProcessing(object):
             'failed_reason',
             'feedback',
             'metadata',
+            'document_size',
             'completed_at',
         ]
         for _property in instance_properties:
@@ -377,5 +385,6 @@ class DocumentProcessing(object):
                 DocumentProcessingMetadata.from_dict(metadata)
                 for metadata in data.get('metadata_items', [])
             ],
+            document_size=data.get('document_size'),
         )
 

{documente_shared-0.1.131 → documente_shared-0.1.131b2}/documente_shared/domain/entities/scaling.py

@@ -4,14 +4,16 @@ from typing import Optional
 
 @dataclass
 class ScalingRequirements(object):
-    documents: Optional[int] = 0
+    lg_documents: Optional[int] = 0
+    md_documents: Optional[int] = 0
     processing_cases: Optional[int] = 0
     processing_case_items: Optional[int] = 0
 
     @property
     def to_dict(self):
         return {
-            "documents": self.documents,
+            "lg_documents": self.lg_documents,
+            "md_documents": self.md_documents,
             "processing_cases": self.processing_cases,
             "processing_case_items": self.processing_case_items,
         }
@@ -19,7 +21,8 @@ class ScalingRequirements(object):
     @classmethod
     def from_dict(cls, data: dict) -> "ScalingRequirements":
         return cls(
-            documents=data.get("documents", 0),
+            lg_documents=data.get("lg_documents", 0),
+            md_documents=data.get("md_documents", 0),
             processing_cases=data.get("processing_cases", 0),
             processing_case_items=data.get("processing_case_items", 0),
         )

{documente_shared-0.1.131 → documente_shared-0.1.131b2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.131"
+version = "0.1.131b2"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"