documente_shared 0.1.97__tar.gz → 0.1.99__tar.gz

{documente_shared-0.1.97 → documente_shared-0.1.99}/PKG-INFO

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

documente_shared-0.1.97/documente_shared/domain/entities/in_memory_result.py → documente_shared-0.1.99/documente_shared/domain/entities/in_memory_document.py

@@ -1,10 +1,12 @@
-from dataclasses import dataclass
+import base64
 from typing import Optional
+from dataclasses import dataclass
 
 from documente_shared.application.files import (
     remove_slash_from_path,
     get_filename_from_path,
 )
+from documente_shared.domain.exceptions import InMemoryDocumentContentError
 
 
 @dataclass
@@ -37,17 +39,25 @@ class InMemoryDocument(object):
 
     @property
     def to_dict(self) -> dict:
-        data = dict()
-        if self.file_path:
-            data['file_path'] = remove_slash_from_path(self.file_path)
-        if self.file_bytes:
-            data['file_bytes'] = self.file_bytes.decode('utf-8')
-        if self.file_base64:
-            data['file_base64'] = self.file_base64
-        return data
+        return {
+            "file_path": self.file_path,
+            "file_base64": self.file_base64,
+        }
 
     @classmethod
     def from_dict(cls, data: dict):
+        has_bytes_content = data.get('file_bytes') and isinstance(data['file_bytes'], bytes)
+        has_base64_content = data.get('file_base64') and isinstance(data['file_base64'], str)
+
+        if not has_bytes_content and not has_base64_content:
+            raise InMemoryDocumentContentError
+
+        if has_bytes_content and not data.get('file_base64'):
+            data['file_base64'] = base64.b64encode(data.get('file_bytes')).decode("utf-8")
+
+        if has_base64_content and not data.get('file_bytes'):
+            data['file_bytes'] = base64.b64decode(data.get('file_base64'))
+
         return cls(
             file_path=data.get('file_path'),
             file_bytes=data.get('file_bytes'),

{documente_shared-0.1.97 → documente_shared-0.1.99}/documente_shared/domain/entities/processing_case.py

@@ -88,9 +88,10 @@ class ProcessingCase(object):
     def to_dict(self) -> dict:
         return {
             'uuid': self.uuid,
-            'label': self.name,
+            'tenant_slug': self.tenant_slug,
+            'name': self.name,
             'status': str(self.status),
-            'category': (
+            'case_type': (
                 str(self.case_type)
                 if self.case_type else None
             ),

{documente_shared-0.1.97 → documente_shared-0.1.99}/documente_shared/domain/entities/processing_case_item.py

@@ -5,7 +5,7 @@ from typing import Optional, List
 
 from documente_shared.application.time_utils import get_datetime_from_data
 from documente_shared.domain.constants import la_paz_tz
-from documente_shared.domain.entities.in_memory_result import InMemoryDocument
+from documente_shared.domain.entities.in_memory_document import InMemoryDocument
 from documente_shared.domain.enums.common import ProcessingStatus, ProcessingSource
 from documente_shared.domain.enums.processing_case import ProcessingDocumentType
 

documente_shared-0.1.99/documente_shared/domain/exceptions.py

@@ -0,0 +1,5 @@
+
+
+
+class InMemoryDocumentContentError(Exception):
+    pass

{documente_shared-0.1.97 → documente_shared-0.1.99}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.97"
+version = "0.1.99"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"
@@ -15,9 +15,10 @@ pytz = "^2025.2"
 requests = "^2.32.3"
 loguru = "^0.7.3"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 pytest = "^8.3.4"
 ipdb = "^0.13.13"
+expects = "^0.9.0"
 
 
 [tool.pytest.ini_options]