From fbc7a0cfa1810f4dced87e7ce5984f45bb90fd03 Mon Sep 17 00:00:00 2001
From: Matthew Evans <git@ml-evs.science>
Date: Thu, 14 May 2026 23:09:46 +0100
Subject: [PATCH] Switch between legacy file_id and file_ids field when
 updating blocks

---
 src/datalab_api/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/datalab_api/__init__.py b/src/datalab_api/__init__.py
index 3487b67..d25cfed 100644
--- a/src/datalab_api/__init__.py
+++ b/src/datalab_api/__init__.py
@@ -424,10 +424,10 @@ def _update_data_block(
 
         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 = {