documente_shared 0.1.130__tar.gz → 0.1.131b0__tar.gz

{documente_shared-0.1.130 → documente_shared-0.1.131b0}/PKG-INFO

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

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

@@ -27,7 +27,7 @@ class DocumentProcessing(object):
     status: DocumentProcessingStatus
     category: DocumentProcessingCategory
     file_path: Optional[str] = None
-    file_pdf_path: Optional[str] = None
+    file_auxiliar_path: Optional[str] = None
     file_bytes: Optional[bytes] = None
     sub_category: Optional[DocumentProcessingSubCategory] = None
     uploaded_from: Optional[DocumentProcessingSource] = None
@@ -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):
@@ -229,7 +230,7 @@ class DocumentProcessing(object):
             self.digest == other.digest
             and self.status == other.status
             and self.file_path == other.file_path
-            and self.file_pdf_path == other.file_pdf_path
+            and self.file_auxiliar_path == other.file_auxiliar_path
             and self.issued_at == other.issued_at
             and self.uploaded_at == other.uploaded_at
             and self.enqueued_at == other.enqueued_at
@@ -244,7 +245,7 @@ class DocumentProcessing(object):
             'digest': self.digest,
             'status': str(self.status),
             'file_path': self.file_path,
-            'file_pdf_path': self.file_pdf_path,
+            'file_auxiliar_path': self.file_auxiliar_path,
             'category': (
                 str(self.category)
                 if self.category else None
@@ -279,6 +280,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
@@ -301,7 +303,7 @@ class DocumentProcessing(object):
             'status',
             'metadata',
             'file_path',
-            'file_pdf_path',
+            'file_auxiliar_path',
             'file_bytes',
             'category',
             'sub_category',
@@ -324,6 +326,7 @@ class DocumentProcessing(object):
             'failed_reason',
             'feedback',
             'metadata',
+            'document_size',
             'completed_at',
         ]
         for _property in instance_properties:
@@ -339,7 +342,7 @@ class DocumentProcessing(object):
             digest=data.get('digest'),
             status=DocumentProcessingStatus.from_value(data.get('status')),
             file_path=data.get('file_path'),
-            file_pdf_path=data.get('file_pdf_path'),
+            file_auxiliar_path=data.get('file_auxiliar_path'),
             category=(
                 DocumentProcessingCategory.from_value(data.get('category'))
                 if data.get('category') else None
@@ -377,5 +380,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.130 → documente_shared-0.1.131b0}/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.130 → documente_shared-0.1.131b0}/pyproject.toml

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