datalab-api 0.5.0__tar.gz → 0.5.1__tar.gz

{datalab_api-0.5.0/src/datalab_api.egg-info → datalab_api-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datalab-api
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python API for instances of the datalab data management platform (https://github.com/datalab-org/datalab).
 Author-email: Matthew Evans <matthew@ml-evs.science>
 License-Expression: MIT

{datalab_api-0.5.0 → datalab_api-0.5.1}/src/datalab_api/__init__.py

@@ -273,7 +273,13 @@ class DatalabClient(BaseDatalabClient):
                     for chunk in response.iter_bytes(chunk_size=1024):
                         file.write(chunk)
 
-    def get_block(self, item_id: str, block_id: str, block_data: dict[str, Any]) -> dict[str, Any]:
+    def get_block(
+        self,
+        item_id: str,
+        block_id: str,
+        block_data: dict[str, Any] | None = None,
+        event_data: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
         """Get a block with a given ID and block data.
         Should be used in conjunction with `get_item` to load an existing block.
 
@@ -281,17 +287,25 @@ class DatalabClient(BaseDatalabClient):
             item_id: The ID of the item to search for.
             block_id: The ID of the block to search for.
             block_data: Any other block data required by the request.
+            event_data: Any block events to trigger.
 
         Returns:
             A dictionary of block data for the block with the given ID.
 
         """
-        block_url = f"{self.datalab_api_url}/update-block/"
+        block_url = f"{self.datalab_api_url}/blocks/"
+
+        if not block_data:
+            block_data = {}
+
+        block_data["item_id"] = item_id
+        block_data["block_id"] = block_id
+        if "blocktype" not in block_data:
+            raise ValueError("block_data must contain a 'blocktype' key.")
+
         block_request = {
             "block_data": block_data,
-            "item_id": item_id,
-            "block_id": block_id,
-            "save_to_db": False,
+            "event_data": event_data,
         }
         block = self._post(block_url, json=block_request)
         return block["new_block_data"]
@@ -424,10 +438,10 @@ class DatalabClient(BaseDatalabClient):
 
         if file_ids:
             if len(file_ids) > 1:
-                raise RuntimeError(
-                    "API does not currently support attaching multiple files in a block."
-                )
-            block_data["file_id"] = file_ids[0]
+                block_data["file_ids"] = file_ids
+            else:
+                block_data.pop("file_ids", None)
+                block_data["file_id"] = file_ids[0]
 
         blocks_url = f"{self.datalab_api_url}/update-block/"
         payload = {

{datalab_api-0.5.0 → datalab_api-0.5.1/src/datalab_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datalab-api
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python API for instances of the datalab data management platform (https://github.com/datalab-org/datalab).
 Author-email: Matthew Evans <matthew@ml-evs.science>
 License-Expression: MIT