streamlit-octostar-utils 0.5.0.dev3__tar.gz → 0.5.0.dev5__tar.gz

{streamlit_octostar_utils-0.5.0.dev3 → streamlit_octostar_utils-0.5.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.5.0.dev3
+Version: 0.5.0.dev5
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.5.0.dev3 → streamlit_octostar_utils-0.5.0.dev5}/pyproject.toml

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

{streamlit_octostar_utils-0.5.0.dev3 → streamlit_octostar_utils-0.5.0.dev5}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -73,7 +73,7 @@ class NifiEntityModel(BaseModel):
             relationships: List[str]
             label_keys: List[str]
 
-        jwt: str
+        jwt: Optional[str] = None
         ontology_name: str
         ontology_info: OntologyInfoModel
         entity_timestamp: Optional[str]
@@ -624,7 +624,8 @@ class NifiContextManager(object):
             {
                 "entity_id": e["os_entity_uid"],
                 "entity_type": e["os_concept"],
-                "do_not_reindex_at": long_expiry,
+                "do_not_reindex_before": long_expiry,
+                "do_not_update_before": long_expiry,
             }
             for e in entities
         ]
@@ -637,7 +638,8 @@ class NifiContextManager(object):
                 {
                     "entity_id": e["os_entity_uid"],
                     "entity_type": e["os_concept"],
-                    "do_not_reindex_at": short_expiry,
+                    "do_not_reindex_before": short_expiry,
+                    "do_not_update_before": short_expiry,
                 }
                 for e in entities
             ]
@@ -722,7 +724,6 @@ class NifiContextManager(object):
                 {
                     "entity_type": entity.record["os_concept"],
                     "os_entity_uid": entity.record["os_entity_uid"],
-                    "os_last_updated_at": entity.request["entity_timestamp"],
                     "fields": {k: entity.record.get(k) for k in fields},
                 }
                 for entity, fields in entities_to_upsert
@@ -1186,6 +1187,48 @@ class NifiEntity(object):
         self.children.append(child_entity_proxy)
         return child_entity_proxy
 
+    def prepare_contents(
+        self,
+        file: Union[Contents, bytes],
+        entity_type: str,
+        filetype: str,
+        temp_blob_key: Optional[str] = None,
+    ) -> Contents:
+        """Build a Contents object from raw bytes, mirroring the same
+        MemoryContents / TemporaryAttachmentContents logic used internally.
+
+        Thread-safe: each call creates independent objects keyed by
+        *temp_blob_key* (for large blobs) and does not mutate the entity tree.
+        The returned Contents can later be passed to ``add_child_fragment``
+        or ``add_child_file`` to skip the redundant bytes→Contents conversion.
+
+        Args:
+            file: A Contents object (returned as-is) or raw bytes.
+            entity_type: Concept / entity type for the contents metadata.
+            filetype: MIME type for the contents metadata.
+            temp_blob_key: Unique key for the temporary blob when the data
+                exceeds the in-memory threshold.  Defaults to a random UUID.
+        """
+        if isinstance(file, Contents):
+            return file
+        if len(file) <= MAX_IN_MEMORY_SIZE_BYTES:
+            return MemoryContents(
+                entity_type=entity_type,
+                filetype=filetype,
+                initial_data=file,
+            )
+        temp_filename = temp_blob_key or f"tmp_{uuid.uuid4()}"
+        temp_contents = TemporaryAttachmentContents(
+            entity_type=entity_type,
+            filetype=filetype,
+            filename=temp_filename,
+            client=self.context.client,
+        )
+        temp_contents.truncate(0)
+        temp_contents.write(file)
+        temp_contents.flush()
+        return temp_contents
+
     def _add_child_attachable(
         self,
         os_workspace,
@@ -1213,24 +1256,11 @@ 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(
-                entity_type=os_entity_type,
-                filetype=filetype,
-                filename=temp_filename,
-                client=self.context.client,
+            child_entity._contents = self.prepare_contents(
+                file, os_entity_type, filetype,
+                temp_blob_key=f"tmp_{child_entity.record['os_entity_uid']}",
             )
-            temp_contents.truncate(0)
-            temp_contents.write(file)
-            temp_contents.flush()
-            child_entity._contents = temp_contents
         child_entity.sync_params = recursive_update_dict(
             child_entity.sync_params,
             {NifiContextManager.SyncFlag.WRITE_CONTENTS: True},