UncountablePythonSDK 0.0.86__py3-none-any.whl → 0.0.88__py3-none-any.whl

{UncountablePythonSDK-0.0.86.dist-info → UncountablePythonSDK-0.0.88.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.86
+Version: 0.0.88
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -37,6 +37,7 @@ Requires-Dist: flask==3.*
 Requires-Dist: simplejson==3.*
 Requires-Dist: grpcio==1.67.1
 Requires-Dist: protobuf>=4.21.1
+Requires-Dist: azure-storage-blob==12.*
 Provides-Extra: test
 Requires-Dist: mypy==1.*; extra == "test"
 Requires-Dist: ruff==0.*; extra == "test"

{UncountablePythonSDK-0.0.86.dist-info → UncountablePythonSDK-0.0.88.dist-info}/RECORD

@@ -33,15 +33,16 @@ pkgs/argument_parser/_is_namedtuple.py,sha256=Rjc1bKanIPPogl3qG5JPBxglG1TqWYOo1n
 pkgs/argument_parser/argument_parser.py,sha256=XjWQdcWanfaCGdLx7c_yQtqbZp7db-KAWzw8sTpXOsY,17713
 pkgs/argument_parser/case_convert.py,sha256=NuJLJUJRbyVb6_Slen4uqaStEHbcOS1d-hBBfDrrw-c,605
 pkgs/filesystem_utils/__init__.py,sha256=NSsQrUCoGISBCqCCyq6_583sYHTVEQeDjDO8hvZn3ag,1261
+pkgs/filesystem_utils/_blob_session.py,sha256=CtoB7PIocuZo8vvFIS_Rc-YR6KwzFB0rHUVPKFEbRAI,4862
 pkgs/filesystem_utils/_gdrive_session.py,sha256=GJuZYJq1W4QQ_7OLvZIMK99FgRq8FxJHg6cMUx9prtA,11077
 pkgs/filesystem_utils/_local_session.py,sha256=xFEYhAvNqrOYqwt4jrEYOuYkjJn0zclZhTelW_Q1-rw,2325
 pkgs/filesystem_utils/_s3_session.py,sha256=_jLK5C-UElT5Sl5teM2pFR7fQe11NlhUd04EgwN9FRs,3962
 pkgs/filesystem_utils/_sftp_session.py,sha256=6zoF7YsEUp0GpyFb-BeIhUAWvbTK7IUjvPNJ1B0vEyI,4743
-pkgs/filesystem_utils/file_type_utils.py,sha256=Xd-mg35mAENUgNJVz5uK8nEfrUp-NQld_gnXFEq3K-8,1487
+pkgs/filesystem_utils/file_type_utils.py,sha256=Au79J66Rh_Yyt_dXctsygPAg1Y4QRawqwRhbvnCpDIY,1933
 pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsres6XrOQ8RoiEpcE,1045
 pkgs/serialization/__init__.py,sha256=LifasRW0a50A3qRFmo2bf3FQ6TXhZWOTz2-CVTgPjcQ,753
 pkgs/serialization/missing_sentry.py,sha256=aM_9KxbCk9dVvXvcOKgkIQBqFWvLhv8QlIUCiuFEXMo,806
-pkgs/serialization/opaque_key.py,sha256=FIfXEE0DA1U8R_taFbQ1RCoTSgehrPjP06-qvo-GeNQ,177
+pkgs/serialization/opaque_key.py,sha256=8ak7aMCGWkKDjnG374yqy8gtnCCUzG2DSJEBfoPgi0c,194
 pkgs/serialization/serial_class.py,sha256=fEceYVIV8dc3Ium-okYxp9cCNFg05HcliXqoCcV5Wcc,6151
 pkgs/serialization/serial_union.py,sha256=xpdeqCrRd0sNCaUwBQRzje6V40ndCbJpZrLX2K0d5xo,2741
 pkgs/serialization/yaml.py,sha256=yoJtu7_ixnJV6uTxA_U1PpK5F_ixT08AKVh5ocyYwXM,1466
@@ -58,9 +59,9 @@ pkgs/type_spec/config.py,sha256=pMUNnpZRuS8hybgcPZiVYXz79qZcO9i_ZoiFAgbtVq4,4813
 pkgs/type_spec/emit_io_ts.py,sha256=CUvBs0boB_X-Kndh66yYcqFfq3oC_LGs8YffLkJ0ZXA,5707
 pkgs/type_spec/emit_open_api.py,sha256=_oBTuYixZUp3DT2cJaiY55VnCi_jGjqV_1vBx2dYdsw,24596
 pkgs/type_spec/emit_open_api_util.py,sha256=x4GCiZSGdypJ9Qtm6I5W_3UvwdJyMs8_OGhJ8_THznA,2401
-pkgs/type_spec/emit_python.py,sha256=GC-HSMTS0E0QgREjv8DbMRA9NJ1RkYywTOTqZ5ZHmkM,47375
+pkgs/type_spec/emit_python.py,sha256=iJgRfke0rCFKY_7nGWLCPR_SQ4gfRuudC6b7r5ftKPE,47456
 pkgs/type_spec/emit_typescript.py,sha256=PNeXHoeMA8UWgxsgJisIhJeFX5CNXn0NIz8UtgTvxes,8838
-pkgs/type_spec/emit_typescript_util.py,sha256=lG3Wtm5vN6etE0NeYrKfnrxj1vVWGzQBSDcDItaszWk,10319
+pkgs/type_spec/emit_typescript_util.py,sha256=sugxboeMwXiIdk4RDyUR_CxU31Le3OQ7i5HmuHeSLvc,10447
 pkgs/type_spec/load_types.py,sha256=vO8VLI7aTKzzHQIla-WO-5Z_mfTuwUqH4ZSKN9E9n5U,3688
 pkgs/type_spec/open_api_util.py,sha256=IGh-_snGPST_P_8FdYtO8MTEa9PUxRW6Rzg9X9EgQik,7114
 pkgs/type_spec/test.py,sha256=4ueujBq-pEgnX3Z69HyPmD-bullFXmpixcpVzfOkhP4,489
@@ -81,7 +82,7 @@ uncountable/__init__.py,sha256=8l8XWNCKsu7TG94c-xa2KHpDegvxDC2FyQISdWC763Y,89
 uncountable/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/core/__init__.py,sha256=RFv0kO6rKFf1PtBPu83hCGmxqkJamRtsgQ9_-ztw7tA,341
 uncountable/core/async_batch.py,sha256=9pYGFzVCQXt8059qFHgutweGIFPquJ5Xfq6NT5P-1K0,1206
-uncountable/core/client.py,sha256=qTM61IJV4DTE2LsTZyv4kePBZAv55vncWt2rtBf-SLQ,10632
+uncountable/core/client.py,sha256=SVBaoXfbV-sj7py3JHOFrQs9_i3E_7P8LqGSDeTAV5Q,11217
 uncountable/core/environment.py,sha256=K2TtE52JbW5UOBkBSc2Ee2l9rDIoRNoFDXDqRha1fJI,1036
 uncountable/core/file_upload.py,sha256=L1tjW-zw0vu6y7ytqSWR_aVF8OIsqlQYHCa-24kbAf4,3408
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
@@ -103,7 +104,7 @@ uncountable/integration/executors/executors.py,sha256=Kzisp1eKufGCWrHIw4mmAj-l1U
 uncountable/integration/executors/generic_upload_executor.py,sha256=0vs9NVk1UL2FBhiMCH6o8p4KtVXNFNvv861QCOD3UU4,10375
 uncountable/integration/executors/script_executor.py,sha256=BBQ9f0l7uH2hgKf60jtm-pONzwk-EeOhM2qBAbv_URo,846
 uncountable/integration/queue_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/queue_runner/job_scheduler.py,sha256=lhvcl11jyk_wg9BSc1Xyh5u8iEtsdqQFW83FPLupjQI,6189
+uncountable/integration/queue_runner/job_scheduler.py,sha256=2acmeWscG87MZsifxwMRMGJ4sQgQVNJNr5CqIPUg75E,6288
 uncountable/integration/queue_runner/queue_runner.py,sha256=0BmYu5zHdothTevGsB-nXg6MBd1UD-WkP3h1WCKMdQg,710
 uncountable/integration/queue_runner/types.py,sha256=8qTq29BTSa5rmW6CBlBntP0pNIiDcwu1wHa78pjroS0,219
 uncountable/integration/queue_runner/worker.py,sha256=5bcfg_1tVlAenRDCD89gJI3Ru7ev-Sk7Gok5wjLYxnI,4588
@@ -123,10 +124,10 @@ uncountable/integration/queue_runner/datastore/model.py,sha256=8-RI5A2yPZVGBLWIN
 uncountable/integration/secret_retrieval/__init__.py,sha256=3QXVj35w8rRMxVvmmsViFYDi3lcb3g70incfalOEm6o,87
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=9iz9N8Z-B68QwFCXsx8hTYbgDbk06ejkJ3RQ9mCLMyM,3000
 uncountable/integration/webhook_server/entrypoint.py,sha256=yQWQq_k3kbJkSsEEt6k22YwhXekezJZfV0rnn-hP-Yo,5516
-uncountable/types/__init__.py,sha256=OAuDaFSXysVPw0Y8slVJuBviSNfAXNzPKKG2eGHeeM8,9028
-uncountable/types/async_batch.py,sha256=_OhT25_dEVts_z_n1kqfJH3xlZg3btLqR6TNkfFLlXE,609
-uncountable/types/async_batch_processor.py,sha256=DJn9KdgUv_l7ojCVJ_d9wCS3GUNc21b5cOrpunty2KU,13129
-uncountable/types/async_batch_t.py,sha256=niXgIM7FQXb_1RLX8CBXiGaYSa8nqd-jqX68p7gMgJo,2558
+uncountable/types/__init__.py,sha256=HaqDPQiYcRk_xbKtOebSskAHBHQ_3rBd08ghBO1ZCI4,9120
+uncountable/types/async_batch.py,sha256=-qOtydEQDfZLkuDvQPH6PxnE_JBFeO1Ro1eaxLyiWD4,683
+uncountable/types/async_batch_processor.py,sha256=WQqEFi9azyE7Gmk2KtD7F5xMtNZOIxHLj0is67WA60s,14420
+uncountable/types/async_batch_t.py,sha256=JZY4t6GuriSCDr1cWu8vs-2awGrw8Hsr5OP-SFJ5V0E,3065
 uncountable/types/async_jobs.py,sha256=gFtEPBZot72sR5zrMWc5zD5QrgNmFmGyD67Y-KhyAmo,343
 uncountable/types/async_jobs_t.py,sha256=sEh1Ev46ff4n1v8mMV3rEbH3MGmaXnLuvYf0UhDdzeU,1425
 uncountable/types/auth_retrieval.py,sha256=FY8Vr_BWD4O8PsauPNt_7_08YZSHFaUlTT72L5XJ-4o,570
@@ -137,17 +138,17 @@ uncountable/types/calculations.py,sha256=FFO_D3BbKoGDZnqWvTKpW4KF359i2vrKjpdFCLY
 uncountable/types/calculations_t.py,sha256=157qD0VqijD5kNDF5BRsfGli3WaPGnNjoo2o2CPX-Ik,669
 uncountable/types/chemical_structure.py,sha256=E-LnikTFDoVQ1b2zKaVUIO_PAKm-7aZZYJi8I8SDSic,302
 uncountable/types/chemical_structure_t.py,sha256=zDJ6WkeT3YwWZRZT21znQn2ZYelv3L7yv7kJiGoNZCw,824
-uncountable/types/client_base.py,sha256=fpVHWPifRToIiaBPKcM_YT8WQFrK-SW58muGnCTX3d0,68780
+uncountable/types/client_base.py,sha256=7AA4rHQ0N1o22Grqij3ckyy5QbjsrGl5X6YTRfbH_fU,69635
 uncountable/types/client_config.py,sha256=4h5Liko9uKCo9_0gdbPhoK6Jr2Kv7tioLiQ8iKeq-_4,301
 uncountable/types/client_config_t.py,sha256=6dStfR0IEHiPW8f9_aF3DD_tHmXXw2rEVrgpebzq8Fg,747
 uncountable/types/curves.py,sha256=W6uMpG5SyW1MS82szNpxkFEn1MnxNpBFyFbQb2Ysfng,366
 uncountable/types/curves_t.py,sha256=lKhRM-2cZ_sFaW7pa_I_Ipz_pJhm3_yTFehRXI79pKk,1416
-uncountable/types/entity.py,sha256=ECvhswTj9xp4gUEKTZoZYyxHvx1oyvE5FNiGNfSyUgk,528
-uncountable/types/entity_t.py,sha256=MPLE0CCrROH89JUE9qyI3MLL0DtVzsuXlJ8bG-x-K4A,16731
+uncountable/types/entity.py,sha256=Y7r_xuYIsY_v1NfIil_d-1_4-lUDMmOSKnMBYCJ20jM,587
+uncountable/types/entity_t.py,sha256=iyf8BKII5Jy072RguoJDAg-o0ZKa8ZrVUrKPrpWLR0E,17155
 uncountable/types/experiment_groups.py,sha256=_0OXcPzSAbkE-rfKt5tPx178YJ4pcEKZvrCxUHgDnvw,309
 uncountable/types/experiment_groups_t.py,sha256=qEs8YW0eJOJ_sCOObT5v9QRx9wsjLYpJqJhCJXa-vNA,721
-uncountable/types/field_values.py,sha256=uuIWX-xmfvcinYPdfkWJeb56zzQY01mc9rmotMPMh24,503
-uncountable/types/field_values_t.py,sha256=WIGXJKtQbHroCgaHhb15H6Rzi00liSkhoi2JBMVEgv0,1935
+uncountable/types/field_values.py,sha256=x3oNoMTgT9k1knWVxAVaX4krj7V8palj90FZuFtFbRw,1202
+uncountable/types/field_values_t.py,sha256=7yJcaamU6x6ZkCVI73lyoh_TKUBwi-8plsc6_kzY3wE,5903
 uncountable/types/fields.py,sha256=GUY5ne8Zp2_Lalikr0zcbdJrin8dG81eyS8fKWJ9yf8,266
 uncountable/types/fields_t.py,sha256=LZBEfBHDn2thc1u4i8BulMEcFfDxK4JA-VzahTjivNA,665
 uncountable/types/generic_upload.py,sha256=n6hue9BX_rLSXeEt_DcGwL2ckxfNXg1wEPR9JNEGQxQ,879
@@ -218,6 +219,7 @@ uncountable/types/api/entity/grant_entity_permissions.py,sha256=YAXyJ3I_nqDQYFYG
 uncountable/types/api/entity/list_entities.py,sha256=ykbdq4DD31uiRz4i8LH-8LLeA2Lpp_5fWfb5fdyx248,2000
 uncountable/types/api/entity/lock_entity.py,sha256=mMZx2tWOtuYg0sIftdPsFWgZO5LCav2ubqTw97dCtDU,1197
 uncountable/types/api/entity/resolve_entity_ids.py,sha256=GnQjeoTdzL0PIubrLay-PpaRsYFFWVGrTxhzSmP4hhw,1387
+uncountable/types/api/entity/set_entity_values.py,sha256=irihNNbUHScKhOWpC0D8R9b3QXPT8yuSJVBnat1Y42o,1182
 uncountable/types/api/entity/set_values.py,sha256=O_LpcYeBXFfxxUVOL2FDDYQwU7La-IBM_uEQLgtPrVo,1125
 uncountable/types/api/entity/transition_entity_phase.py,sha256=J0lO_dubmTCO9s7P2DZgC-Zq6m0VagpXfg0jZqxz5XM,2160
 uncountable/types/api/entity/unlock_entity.py,sha256=8UGffqqV8eiSbz_-7y0W2CXLsz4IvM496RFz6L0Y23o,1150
@@ -286,7 +288,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=-oZgPyn43xEKSCs81DVNzwaYMCdRJxbM9GY6fsqKwf4,1090
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=6mwVG136oLp9JcbB2I-kZnrcm3aeZzYZB-SFjEImY2o,1314
-UncountablePythonSDK-0.0.86.dist-info/METADATA,sha256=M-eS-Ah9j0F9pCNkwlJufJM7CCpGxzqWeo_Q-Dk6SkE,2024
-UncountablePythonSDK-0.0.86.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-UncountablePythonSDK-0.0.86.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.86.dist-info/RECORD,,
+UncountablePythonSDK-0.0.88.dist-info/METADATA,sha256=ffYG155Wxi8gwlAe-T91tlE8En-elS89jOL5ZgBHCQY,2064
+UncountablePythonSDK-0.0.88.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+UncountablePythonSDK-0.0.88.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.88.dist-info/RECORD,,

pkgs/filesystem_utils/_blob_session.py

@@ -0,0 +1,137 @@
+from io import BytesIO
+
+from azure.storage.blob import BlobServiceClient, ContainerClient
+
+from pkgs.filesystem_utils.file_type_utils import (
+    FileObjectData,
+    FileSystemBlobConfig,
+    FileSystemFileReference,
+    FileSystemObject,
+    FileTransfer,
+    IncompatibleFileReference,
+)
+
+from .filesystem_session import FileSystemSession
+
+
+def _add_slash(prefix: str) -> str:
+    if len(prefix) > 0 and prefix[-1] != "/":
+        prefix = prefix + "/"
+    return prefix
+
+
+class BlobSession(FileSystemSession):
+    config: FileSystemBlobConfig
+
+    def __init__(self, blob_config: FileSystemBlobConfig) -> None:
+        super().__init__()
+        self.config = blob_config
+
+    def start(self) -> None:
+        self.service_client: BlobServiceClient | None = BlobServiceClient(
+            self.config.account_url, credential=self.config.credential
+        )
+        self.container_client: ContainerClient | None = (
+            self.service_client.get_container_client(self.config.container)
+        )
+
+    def __enter__(self) -> "BlobSession":
+        self.start()
+        return self
+
+    def __exit__(self, exc_type: object, exc_val: object, exc_tb: object) -> None:
+        self.service_client = None
+        self.container_client = None
+
+    def list_files(
+        self,
+        dir_path: FileSystemObject,
+        *,
+        recursive: bool = False,
+        valid_extensions: list[str] | None = None,
+    ) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReference()
+
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to list_files on uninitialized blob session"
+        )
+
+        filesystem_file_references: list[FileSystemObject] = []
+        prefix = _add_slash(dir_path.filepath)
+        for blob in self.container_client.list_blobs(name_starts_with=prefix):
+            if not recursive and (
+                blob.name == prefix or "/" in blob.name[len(prefix) :]
+            ):
+                continue
+            if valid_extensions is None or any(
+                blob.name.endswith(valid_extension)
+                for valid_extension in valid_extensions
+            ):
+                filesystem_file_references.append(
+                    FileSystemFileReference(
+                        filepath=blob.name,
+                    )
+                )
+
+        return filesystem_file_references
+
+    def download_files(
+        self,
+        filepaths: list[FileSystemObject],
+    ) -> list[FileObjectData]:
+        downloaded_files: list[FileObjectData] = []
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to download_files on uninitialized blob session"
+        )
+
+        for file_object in filepaths:
+            if (
+                not isinstance(file_object, FileSystemFileReference)
+                or file_object.filename is None
+            ):
+                raise IncompatibleFileReference()
+
+            blob_client = self.container_client.get_blob_client(file_object.filepath)
+            download_stream = blob_client.download_blob()
+            file_data = download_stream.readall()
+            downloaded_files.append(
+                FileObjectData(
+                    file_data=file_data,
+                    file_IO=BytesIO(file_data),
+                    filename=file_object.filename,
+                    filepath=file_object.filepath,
+                )
+            )
+
+        return downloaded_files
+
+    def move_files(self, file_mappings: list[FileTransfer]) -> None:
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to move_files on uninitialized blob session"
+        )
+
+        for src_file, dest_file in file_mappings:
+            if not isinstance(src_file, FileSystemFileReference) or not isinstance(
+                dest_file, FileSystemFileReference
+            ):
+                raise IncompatibleFileReference()
+
+            source_blob_client = self.container_client.get_blob_client(
+                src_file.filepath
+            )
+            dest_blob_client = self.container_client.get_blob_client(dest_file.filepath)
+
+            dest_blob_client.start_copy_from_url(source_blob_client.url)
+            source_blob_client.delete_blob()
+
+    def delete_files(self, filepaths: list[FileSystemObject]) -> None:
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to delete_files on uninitialized blob session"
+        )
+        for file_object in filepaths:
+            if not isinstance(file_object, FileSystemFileReference):
+                raise IncompatibleFileReference()
+
+            blob_client = self.container_client.get_blob_client(file_object.filepath)
+            blob_client.delete_blob()

pkgs/filesystem_utils/file_type_utils.py

@@ -4,6 +4,12 @@ from io import BytesIO
 from typing import Optional, Union
 
 import paramiko
+from azure.core.credentials import (
+    AzureNamedKeyCredential,
+    AzureSasCredential,
+    TokenCredential,
+)
+from azure.storage.blob import ContainerProperties
 
 
 @dataclass
@@ -69,3 +75,17 @@ class FileSystemS3Config:
     access_key_id: Optional[str]
     secret_access_key: Optional[str]
     session_token: Optional[str]
+
+
+@dataclass(kw_only=True)
+class FileSystemBlobConfig:
+    account_url: str
+    credential: (
+        str
+        | dict[str, str]
+        | AzureNamedKeyCredential
+        | AzureSasCredential
+        | TokenCredential
+        | None
+    )
+    container: ContainerProperties | str

pkgs/serialization/opaque_key.py

@@ -1,4 +1,4 @@
 # Blocks a string key value from being interpreted for case conversion
-class OpaqueKey(str):
+class OpaqueKey(str):  # noqa: FURB189
     def __new__(cls, key: str) -> "OpaqueKey":
         return str.__new__(cls, key)

pkgs/type_spec/emit_python.py

@@ -1148,13 +1148,15 @@ def _create_api_function(
 
     ruff_requires_wrap = len(endpoint.path_basename) > 14
 
+    account_type = endpoint.root if endpoint.root != "external" else "materials"
+
     api_out.write(
         f"""import {api_import} as api
 from main.db.session import Session, SessionMaker
 from main.site.decorators import APIError, APIResponse, {validated_method}
 
 
-@{validated_method}(api.ENDPOINT_PATH, "{endpoint.root}", api.Arguments)
+@{validated_method}(api.ENDPOINT_PATH, "{account_type}", api.Arguments)
 def {endpoint.path_basename}({WRAP_ARGS_START if ruff_requires_wrap else ""}args: api.Arguments, client_sm: SessionMaker{WRAP_ARGS_END if ruff_requires_wrap else ""}) -> APIResponse[api.Data]:
     with Session(client_sm) as session:
         # return APIResponse(data=api.Data())

pkgs/type_spec/emit_typescript_util.py

@@ -82,9 +82,11 @@ def emit_value_ts(
             return (
                 "{\n\t"
                 + ",\n\t".join(
-                    "["
-                    + emit_value_ts(ctx, key_type, dkey)
-                    + "]: "
+                    (
+                        f"[{emit_value_ts(ctx, key_type, dkey)}]: "
+                        if not key_type.is_base_type(builder.BaseTypeName.s_string)
+                        else f"{dkey}: "
+                    )
                     + emit_value_ts(ctx, value_type, dvalue)
                     for dkey, dvalue in value.items()
                 )

uncountable/core/client.py

@@ -71,16 +71,24 @@ class APIResponseError(Exception):
     extra_details: dict[str, JsonValue] | None
 
     def __init__(
-        self, status_code: int, message: str, extra_details: dict[str, JsonValue] | None
+        self,
+        status_code: int,
+        message: str,
+        extra_details: dict[str, JsonValue] | None,
+        request_id: str,
     ) -> None:
         super().__init__(status_code, message, extra_details)
         self.status_code = status_code
         self.message = message
         self.extra_details = extra_details
+        self.request_id = request_id
 
     @classmethod
     def construct_error(
-        cls, status_code: int, extra_details: dict[str, JsonValue] | None
+        cls,
+        status_code: int,
+        extra_details: dict[str, JsonValue] | None,
+        request_id: str,
     ) -> "APIResponseError":
         message: str
         match status_code:
@@ -103,9 +111,21 @@ class APIResponseError(Exception):
             case _:
                 message = "unknown error"
         return APIResponseError(
-            status_code=status_code, message=message, extra_details=extra_details
+            status_code=status_code,
+            message=message,
+            extra_details=extra_details,
+            request_id=request_id,
         )
 
+    def __str__(self) -> str:
+        details_obj = {
+            "request_id": self.request_id,
+            "status_code": self.status_code,
+            "extra_details": self.extra_details,
+        }
+        details = json.dumps(details_obj)
+        return f"API response error ({self.status_code}): '{self.message}'. Details: {details}"
+
 
 class SDKError(Exception):
     message: str
@@ -174,7 +194,9 @@ class Client(ClientMethods):
             except JSONDecodeError:
                 pass
             raise APIResponseError.construct_error(
-                status_code=response.status_code, extra_details=extra_details
+                status_code=response.status_code,
+                extra_details=extra_details,
+                request_id=request_id,
             )
         try:
             return typing.cast(dict[str, JsonValue], response.json())

uncountable/integration/queue_runner/job_scheduler.py

@@ -106,7 +106,10 @@ async def start_scheduler(command_queue: CommandQueue) -> None:
         ) -> str:
             if isinstance(
                 payload.invocation_context,
-                queued_job_t.InvocationContextCron,
+                (
+                    queued_job_t.InvocationContextCron,
+                    queued_job_t.InvocationContextManual,
+                ),
             ):
                 existing_queued_jobs = datastore.load_job_queue()
                 duplicate_job = next(

uncountable/types/__init__.py

@@ -85,6 +85,7 @@ from . import response_t as response_t
 from .api.triggers import run_trigger as run_trigger_t
 from . import secret_retrieval_t as secret_retrieval_t
 from .api.permissions import set_core_permissions as set_core_permissions_t
+from .api.entity import set_entity_values as set_entity_values_t
 from .api.inputs import set_input_attribute_values as set_input_attribute_values_t
 from .api.inputs import set_input_category as set_input_category_t
 from .api.inputs import set_input_subcategories as set_input_subcategories_t
@@ -191,6 +192,7 @@ __all__: list[str] = [
     "run_trigger_t",
     "secret_retrieval_t",
     "set_core_permissions_t",
+    "set_entity_values_t",
     "set_input_attribute_values_t",
     "set_input_category_t",
     "set_input_subcategories_t",

uncountable/types/api/entity/set_entity_values.py

@@ -0,0 +1,44 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import entity_t
+from ... import field_values_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/entity/set_entity_values"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_entity_values.Arguments",
+)
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    entity_identifier: entity_t.EntityIdentifier
+    field_values: list[field_values_t.FieldArgumentValue]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_entity_values.Data",
+)
+@dataclasses.dataclass(kw_only=True)
+class Data(async_batch_t.AsyncBatchActionReturn):
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch.py

@@ -6,6 +6,7 @@
 # Kept only for SDK backwards compatibility
 from .async_batch_t import AsyncBatchRequestPath as AsyncBatchRequestPath
 from .async_batch_t import AsyncBatchRequest as AsyncBatchRequest
+from .async_batch_t import AsyncBatchResultValue as AsyncBatchResultValue
 from .async_batch_t import AsyncBatchActionReturn as AsyncBatchActionReturn
 from .async_batch_t import SavedAsyncBatchActionReturn as SavedAsyncBatchActionReturn
 from .async_batch_t import QueuedAsyncBatchRequest as QueuedAsyncBatchRequest

uncountable/types/async_batch_processor.py

@@ -16,6 +16,7 @@ import uncountable.types.api.recipes.clear_recipe_outputs as clear_recipe_output
 import uncountable.types.api.recipes.create_recipe as create_recipe_t
 import uncountable.types.api.recipes.edit_recipe_inputs as edit_recipe_inputs_t
 from uncountable.types import entity_t
+from uncountable.types import field_values_t
 from uncountable.types import generic_upload_t
 import uncountable.types.api.entity.grant_entity_permissions as grant_entity_permissions_t
 from uncountable.types import identifier_t
@@ -23,6 +24,7 @@ import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
 from uncountable.types import recipe_identifiers_t
 from uncountable.types import recipe_metadata_t
 from uncountable.types import recipe_workflow_steps_t
+import uncountable.types.api.entity.set_entity_values as set_entity_values_t
 import uncountable.types.api.recipes.set_recipe_metadata as set_recipe_metadata_t
 import uuid
 from abc import ABC, abstractmethod
@@ -311,6 +313,41 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def set_entity_values(
+        self,
+        *,
+        entity_identifier: entity_t.EntityIdentifier,
+        field_values: list[field_values_t.FieldArgumentValue],
+        depends_on: typing.Optional[list[str]] = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Sets field values for an entity
+
+        :param entity_identifier: Entity to update
+        :param field_values: Field values to set
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = set_entity_values_t.Arguments(
+            entity_identifier=entity_identifier,
+            field_values=field_values,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.SET_ENTITY_VALUES,
+            data=json_data,
+            depends_on=depends_on,
+            batch_reference=batch_reference,
+        )
+
+        self._enqueue(req)
+
+        return async_batch_t.QueuedAsyncBatchRequest(
+            path=req.path,
+            batch_reference=req.batch_reference,
+        )
+
     def set_recipe_metadata(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -11,11 +11,13 @@ from pkgs.strenum_compat import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
+from . import entity_t
 
 __all__: list[str] = [
     "AsyncBatchActionReturn",
     "AsyncBatchRequest",
     "AsyncBatchRequestPath",
+    "AsyncBatchResultValue",
     "QueuedAsyncBatchRequest",
     "SavedAsyncBatchActionReturn",
 ]
@@ -35,6 +37,7 @@ class AsyncBatchRequestPath(StrEnum):
     ASSOCIATE_RECIPE_AS_INPUT = "recipes/associate_recipe_as_input"
     CLEAR_RECIPE_OUTPUTS = "recipes/clear_recipe_outputs"
     GRANT_ENTITY_PERMISSIONS = "entity/grant_entity_permissions"
+    SET_ENTITY_VALUES = "entity/set_entity_values"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -50,6 +53,17 @@ class AsyncBatchRequest:
     depends_on: typing.Optional[list[str]] = None
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.async_batch.AsyncBatchResultValue",
+    unconverted_values={"value"},
+)
+@dataclasses.dataclass(kw_only=True)
+class AsyncBatchResultValue:
+    data_key: str
+    value: base_t.JsonValue
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.async_batch.AsyncBatchActionReturn",
@@ -58,6 +72,8 @@ class AsyncBatchRequest:
 class AsyncBatchActionReturn:
     modification_made: bool
     result_id: typing.Optional[base_t.ObjectId] = None
+    entity: typing.Optional[entity_t.Entity] = None
+    result_values: typing.Optional[list[AsyncBatchResultValue]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -69,6 +69,7 @@ import uncountable.types.api.entity.resolve_entity_ids as resolve_entity_ids_t
 import uncountable.types.api.outputs.resolve_output_conditions as resolve_output_conditions_t
 import uncountable.types.api.triggers.run_trigger as run_trigger_t
 import uncountable.types.api.permissions.set_core_permissions as set_core_permissions_t
+import uncountable.types.api.entity.set_entity_values as set_entity_values_t
 import uncountable.types.api.inputs.set_input_attribute_values as set_input_attribute_values_t
 import uncountable.types.api.inputs.set_input_category as set_input_category_t
 import uncountable.types.api.inputs.set_input_subcategories as set_input_subcategories_t
@@ -1232,6 +1233,28 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=set_core_permissions_t.Data)
 
+    def set_entity_values(
+        self,
+        *,
+        entity_identifier: entity_t.EntityIdentifier,
+        field_values: list[field_values_t.FieldArgumentValue],
+    ) -> set_entity_values_t.Data:
+        """Sets field values for an entity
+
+        :param entity_identifier: Entity to update
+        :param field_values: Field values to set
+        """
+        args = set_entity_values_t.Arguments(
+            entity_identifier=entity_identifier,
+            field_values=field_values,
+        )
+        api_request = APIRequest(
+            method=set_entity_values_t.ENDPOINT_METHOD,
+            endpoint=set_entity_values_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=set_entity_values_t.Data)
+
     def set_input_attribute_values(
         self,
         *,

uncountable/types/entity.py

@@ -9,4 +9,5 @@ from .entity_t import LimitedEntityType as LimitedEntityType
 from .entity_t import EntityPermissionType as EntityPermissionType
 from .entity_t import GrantableEntityPermissionType as GrantableEntityPermissionType
 from .entity_t import Entity as Entity
+from .entity_t import EntityIdentifier as EntityIdentifier
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/entity_t.py

@@ -12,9 +12,11 @@ import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_string_enum
 from . import base_t
+from . import identifier_t
 
 __all__: list[str] = [
     "Entity",
+    "EntityIdentifier",
     "EntityPermissionType",
     "EntityType",
     "GrantableEntityPermissionType",
@@ -47,7 +49,7 @@ __all__: list[str] = [
         "doe_run": "Suggest Experiments",
         "email_record_external": "External Emails",
         "email_relay_server": "Email Relay Server",
-        "enter_recipe_view_preference": "Enter Recipe View Preference",
+        "enter_recipe_view_preference": "Enter Experiment View Preference",
         "entities_viewed": "Entities Viewed",
         "entity_annotation": "Annotation",
         "entity_field": "Field",
@@ -90,7 +92,7 @@ __all__: list[str] = [
         "license": "License",
         "maintenance_schedule": "Maintenance Schedule",
         "material_family": "Material Family",
-        "measurement_group_recipe": "Measurement Group Recipe",
+        "measurement_group_recipe": "Measurement Group Experiment",
         "ml_job": "Batch Processing Job",
         "naming_scheme": "Naming Scheme",
         "naming_counter": "Naming Counter",
@@ -106,7 +108,7 @@ __all__: list[str] = [
         "output_condition_parameter_analytical_method": "Output Condition Parameter Analytical Method",
         "output_group": "Output Group",
         "permission": "Permission",
-        "permissions_recipe_link_inheritance": "Permissions Recipe Link Inheritance",
+        "permissions_recipe_link_inheritance": "Permissions Experiment Link Inheritance",
         "phase_workflow": "Phase Workflow",
         "platform_config": "Platform Config",
         "predictive_model": "Predictive Model",
@@ -115,21 +117,21 @@ __all__: list[str] = [
         "project_workbook_comment": "Notebook Cell Comment",
         "recipe": "Experiment",
         "recipe_audit_log": "Experiment Audit Log",
-        "recipe_calculation": "Recipe Calculation",
+        "recipe_calculation": "Experiment Calculation",
         "recipe_check": "Experiment Check",
-        "recipe_export": "Recipe Export",
+        "recipe_export": "Experiment Export",
         "recipe_goal": "Experiment Goal",
-        "recipe_ingredient": "Recipe Ingredient",
-        "recipe_ingredient_actual": "Recipe Ingredient Actual",
-        "recipe_ingredients_compounded": "Recipe Ingredients Compounded",
-        "recipe_ingredients_compounded_calculation": "Recipe Ingredients Compounded Caclulation",
-        "recipe_output": "Recipe Output",
-        "recipe_output_metadata": "Recipe Output Metadata",
-        "recipe_permission_view": "Recipe Permission View",
-        "recipe_project": "Recipe Project",
-        "recipe_step": "Recipe Step",
-        "recipe_workflow_step": "Recipe Workflow Step",
-        "recipe_tag": "Recipe Tag",
+        "recipe_ingredient": "Experiment Ingredient",
+        "recipe_ingredient_actual": "Experiment Ingredient Actual",
+        "recipe_ingredients_compounded": "Experiment Ingredients Compounded",
+        "recipe_ingredients_compounded_calculation": "Experiment Ingredients Compounded Caclulation",
+        "recipe_output": "Experiment Output",
+        "recipe_output_metadata": "Experiment Output Metadata",
+        "recipe_permission_view": "Experiment Permission View",
+        "recipe_project": "Experiment Project",
+        "recipe_step": "Experiment Step",
+        "recipe_workflow_step": "Experiment Workflow Step",
+        "recipe_tag": "Experiment Tag",
         "recipe_metadata": "Experiment Metadata",
         "recipe_metadata_mapping": "Experiment Metadata Mapping",
         "metadata_value": "Metadata Value",
@@ -343,7 +345,7 @@ class EntityType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-LimitedEntityType = typing.Union[typing.Literal[EntityType.LAB_REQUEST], typing.Literal[EntityType.APPROVAL], typing.Literal[EntityType.CUSTOM_ENTITY], typing.Literal[EntityType.INVENTORY_AMOUNT], typing.Literal[EntityType.TASK], typing.Literal[EntityType.PROJECT], typing.Literal[EntityType.EQUIPMENT], typing.Literal[EntityType.INV_LOCAL_LOCATIONS], typing.Literal[EntityType.FIELD_OPTION_SET], typing.Literal[EntityType.WEBHOOK], typing.Literal[EntityType.SPECS], typing.Literal[EntityType.GOAL], typing.Literal[EntityType.INGREDIENT_TAG_MAP], typing.Literal[EntityType.INGREDIENT_TAG], typing.Literal[EntityType.CONDITION_PARAMETER], typing.Literal[EntityType.OUTPUT], typing.Literal[EntityType.ASYNC_JOB]]
+LimitedEntityType = typing.Union[typing.Literal[EntityType.LAB_REQUEST], typing.Literal[EntityType.APPROVAL], typing.Literal[EntityType.CUSTOM_ENTITY], typing.Literal[EntityType.INVENTORY_AMOUNT], typing.Literal[EntityType.TASK], typing.Literal[EntityType.PROJECT], typing.Literal[EntityType.EQUIPMENT], typing.Literal[EntityType.INV_LOCAL_LOCATIONS], typing.Literal[EntityType.FIELD_OPTION_SET], typing.Literal[EntityType.WEBHOOK], typing.Literal[EntityType.SPECS], typing.Literal[EntityType.GOAL], typing.Literal[EntityType.INGREDIENT_TAG_MAP], typing.Literal[EntityType.INGREDIENT_TAG], typing.Literal[EntityType.CONDITION_PARAMETER], typing.Literal[EntityType.OUTPUT], typing.Literal[EntityType.OUTPUT_CONDITION_PARAMETER], typing.Literal[EntityType.ASYNC_JOB]]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -367,4 +369,14 @@ GrantableEntityPermissionType = typing.Union[typing.Literal[EntityPermissionType
 class Entity:
     id: base_t.ObjectId
     type: EntityType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.entity.EntityIdentifier",
+)
+@dataclasses.dataclass(kw_only=True)
+class EntityIdentifier:
+    identifier_key: identifier_t.IdentifierKey
+    type: EntityType
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/field_values.py

@@ -8,4 +8,15 @@ from .field_values_t import FieldRefNameValue as FieldRefNameValue
 from .field_values_t import FieldRefIdNameValue as FieldRefIdNameValue
 from .field_values_t import ArgumentValueRefName as ArgumentValueRefName
 from .field_values_t import ArgumentValueId as ArgumentValueId
+from .field_values_t import FieldValueType as FieldValueType
+from .field_values_t import FieldValueBase as FieldValueBase
+from .field_values_t import FieldValueFiles as FieldValueFiles
+from .field_values_t import FieldValueId as FieldValueId
+from .field_values_t import FieldValueIds as FieldValueIds
+from .field_values_t import FieldValueText as FieldValueText
+from .field_values_t import FieldValueBoolean as FieldValueBoolean
+from .field_values_t import FieldValueNumeric as FieldValueNumeric
+from .field_values_t import FieldValueBatchReference as FieldValueBatchReference
+from .field_values_t import FieldValue as FieldValue
+from .field_values_t import FieldArgumentValue as FieldArgumentValue
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/field_values_t.py

@@ -7,15 +7,30 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
 from . import base_t
+from . import entity_t
+from . import identifier_t
 
 __all__: list[str] = [
     "ArgumentValueId",
     "ArgumentValueRefName",
+    "FieldArgumentValue",
     "FieldRefIdNameValue",
     "FieldRefNameValue",
+    "FieldValue",
+    "FieldValueBase",
+    "FieldValueBatchReference",
+    "FieldValueBoolean",
+    "FieldValueFiles",
+    "FieldValueId",
+    "FieldValueIds",
+    "FieldValueNumeric",
+    "FieldValueText",
+    "FieldValueType",
 ]
 
 
@@ -66,4 +81,124 @@ class ArgumentValueId:
     field_id: base_t.ObjectId
     value: base_t.JsonValue
     row_index: typing.Optional[typing.Optional[int]] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class FieldValueType(StrEnum):
+    FILES = "files"
+    TEXT = "text"
+    ID = "id"
+    IDS = "ids"
+    BOOLEAN = "boolean"
+    NUMERIC = "numeric"
+    BATCH_REFERENCE = "batch_reference"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueBase",
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueBase:
+    type: FieldValueType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueFiles",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueFiles(FieldValueBase):
+    type: typing.Literal[FieldValueType.FILES] = FieldValueType.FILES
+    file_ids: list[base_t.ObjectId]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueId",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueId(FieldValueBase):
+    type: typing.Literal[FieldValueType.ID] = FieldValueType.ID
+    entity_type: entity_t.EntityType
+    identifier_key: identifier_t.IdentifierKey
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueIds",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueIds(FieldValueBase):
+    type: typing.Literal[FieldValueType.IDS] = FieldValueType.IDS
+    entity_type: entity_t.EntityType
+    identifier_keys: list[identifier_t.IdentifierKey]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueText",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueText(FieldValueBase):
+    type: typing.Literal[FieldValueType.TEXT] = FieldValueType.TEXT
+    value: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueBoolean",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueBoolean(FieldValueBase):
+    type: typing.Literal[FieldValueType.BOOLEAN] = FieldValueType.BOOLEAN
+    value: bool
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueNumeric",
+    to_string_values={"value"},
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueNumeric(FieldValueBase):
+    type: typing.Literal[FieldValueType.NUMERIC] = FieldValueType.NUMERIC
+    value: Decimal
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldValueBatchReference",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldValueBatchReference(FieldValueBase):
+    type: typing.Literal[FieldValueType.BATCH_REFERENCE] = FieldValueType.BATCH_REFERENCE
+    reference_key: identifier_t.IdentifierKeyBatchReference
+    data_key: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+FieldValue = typing.Annotated[
+    typing.Union[FieldValueFiles, FieldValueId, FieldValueIds, FieldValueText, FieldValueBoolean, FieldValueNumeric, FieldValueBatchReference],
+    serial_union_annotation(
+    named_type_path="sdk.field_values.FieldValue",
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.field_values.FieldArgumentValue",
+)
+@dataclasses.dataclass(kw_only=True)
+class FieldArgumentValue:
+    field_key: identifier_t.IdentifierKey
+    value: FieldValue
 # DO NOT MODIFY -- This file is generated by type_spec