streamlit-octostar-utils 0.4.2.dev23__tar.gz → 0.4.2.dev24__tar.gz

{streamlit_octostar_utils-0.4.2.dev23 → streamlit_octostar_utils-0.4.2.dev24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.4.2.dev23
+Version: 0.4.2.dev24
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.4.2.dev23 → streamlit_octostar_utils-0.4.2.dev24}/pyproject.toml

@@ -5,7 +5,7 @@ include = '\.pyi?$'
 
 [tool.poetry]
 name = "streamlit-octostar-utils"
-version = "0.4.2-dev.23"
+version = "0.4.2-dev.24"
 description = ""
 license = "MIT"
 authors = ["Octostar"]

{streamlit_octostar_utils-0.4.2.dev23 → streamlit_octostar_utils-0.4.2.dev24}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -25,7 +25,7 @@ from ..core.dict import recursive_update_dict, travel_dict, jsondict_hash
 from ..core.timestamp import now, string_to_datetime
 from .fastapi import DefaultErrorRoute, Route
 from ..ontology.inheritance import is_child_concept as is_child_concept_fn, get_label, get_label_keys
-from .contents import Contents, WorkspaceAttachmentContents, TemporaryAttachmentContents
+from .contents import Contents, WorkspaceAttachmentContents, TemporaryAttachmentContents, MemoryContents
 
 
 class RelationshipName(BaseModel):
@@ -49,6 +49,8 @@ OS_RESERVED_FIELDS = [
     "entity_label", "os_created_at", "os_created_by",
     "os_last_updated_at", "os_last_updated_by",
 ]
+MAX_IN_MEMORY_SIZE_BYTES = 5_242_880
+
 
 class NifiProxyEntityModel(BaseModel):
     entity_id: str
@@ -957,6 +959,12 @@ class NifiEntity(object):
         )
         if isinstance(file, Contents):
             child_entity._contents = file
+        elif len(file) <= MAX_IN_MEMORY_SIZE_BYTES:
+            child_entity._contents = MemoryContents(
+                entity_type=os_entity_type,
+                filetype=filetype,
+                initial_data=file,
+            )
         else:
             temp_filename = f"tmp_{child_entity.record['os_entity_uid']}"
             temp_contents = TemporaryAttachmentContents(
@@ -965,6 +973,7 @@ class NifiEntity(object):
                 filename=temp_filename,
                 client=self.context.client,
             )
+            temp_contents.truncate(0)
             temp_contents.write(file)
             temp_contents.flush()
             child_entity._contents = temp_contents