UncountablePythonSDK 0.0.44__py3-none-any.whl → 0.0.46__py3-none-any.whl

{UncountablePythonSDK-0.0.44.dist-info → UncountablePythonSDK-0.0.46.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.44
+Version: 0.0.46
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.44.dist-info → UncountablePythonSDK-0.0.46.dist-info}/RECORD

@@ -50,7 +50,7 @@ pkgs/type_spec/config.py,sha256=IQyo2Vj11uNt7_d6jQxvominAOU-oPB8ldEmuGzJLpU,4644
 pkgs/type_spec/emit_io_ts.py,sha256=Ghd8XYqyNYldHQDepwa9GLfHXcoi48ztBw84K28ETic,5707
 pkgs/type_spec/emit_open_api.py,sha256=92POd3j4nrrROrw9M-bgEmK4ukYbI0TW6E9IigJzoTE,24512
 pkgs/type_spec/emit_open_api_util.py,sha256=y2slouAflUJmyTPH_d4CbXql9zpOoD1uTOELqL3NX-M,2448
-pkgs/type_spec/emit_python.py,sha256=R5CZcApOdTzbA4R5klFOxXt4AQtTI72GTxp4Non8q2E,46509
+pkgs/type_spec/emit_python.py,sha256=bQUqVmfVgAXw8rGcqlg93YbUsUvawgR2o_NrV_c9Zio,46873
 pkgs/type_spec/emit_typescript.py,sha256=cdr5h8N70PuwORcvhURUujzwH9r1LVwJB8V2EoipGkw,17917
 pkgs/type_spec/emit_typescript_util.py,sha256=sR7ys3Ilnh6SQiXJbfYk4pxfOu0bDjbUFTEYEW-ud6c,863
 pkgs/type_spec/load_types.py,sha256=BOLyndtxPqqhUqZAh-lIbN5IZBaW_m-bdYpKGsbPyXM,3654
@@ -72,25 +72,25 @@ pkgs/type_spec/value_spec/types.py,sha256=a2zxbbCRWepY1l8OtjeCDKgBKFPFHVgV99oP6p
 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=Zo02TICZ7AK81Qa02NQ8fp8uaijQph1FEta28K3cKb0,749
-uncountable/core/client.py,sha256=50vR11GwwyTc3qOIahUxCJK77zCM7LOXwbZdHoZCZgY,10168
+uncountable/core/async_batch.py,sha256=Gur0VOS0AH2ugwvk65hwoX-iqwQAAyJaejY_LyAZZPo,1210
+uncountable/core/client.py,sha256=2QRLy0GcSamQDqUABQq8R7D_wGofYewSoTQiZzOyZXk,10168
 uncountable/core/file_upload.py,sha256=TkQ0fKbbYrPgns1Jh51JU35DUqZHB3ljOaVgjSlBx9Y,3149
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
 uncountable/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/construct_client.py,sha256=y8x65wzG-_84uLlnSVQG4WuMhyVGUfdJVmhy8gs8v4A,1082
-uncountable/integration/cron.py,sha256=czW-RY2T5lD7yJwyVZcFg9240mXURrzeExd1EL6zHeg,1179
-uncountable/integration/entrypoint.py,sha256=Hs-XUTTxjrxsb31lytN7yVViiBYc10x96W_Vy6iq95A,1382
-uncountable/integration/job.py,sha256=ZaVm77v2Pkhyy0FA8PJPp4hd7yNshXs1T3NlZjNi03U,1398
-uncountable/integration/server.py,sha256=OlRVxrpnpJ_QXJ27kQt4z_U5HnI0pfcm5OkxhN_Sa-Y,3347
+uncountable/integration/construct_client.py,sha256=9ZsfkA0oZIE04mPnwGlt6EcrlnpaUPA93K100FS1F7A,1546
+uncountable/integration/cron.py,sha256=n4HjW-KNruYQ_09iwNBd9i4-Juy_9ddzbf71_wnydHA,1674
+uncountable/integration/entrypoint.py,sha256=9rk06gBTsCqytIs8Shsnlf6ir_4Uq5d5rfP1veiSLzc,1437
+uncountable/integration/job.py,sha256=5HJcjtwj_pGFYAZ6SK7xa9RqDXCoBb5rAb_OHdTR61Q,1463
+uncountable/integration/server.py,sha256=85yzQrrpp8OEG37CTN9wtgB9UrzWnDLuDkY0nWcp--k,3548
 uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/db/connect.py,sha256=YtQHJ1DBGPhxKFRCfiXqohOYUceKSxMVOJ88aPI48Ug,181
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/executors/executors.py,sha256=v5ClGVUlvrZcMdmGQa8Ll668G_HGTnKpGOnTM7UMZCQ,956
-uncountable/integration/executors/generic_upload_executor.py,sha256=H_pfseSATBvY44xVNFC8VP6wL9pqoXG2qcReA76hl0E,8790
+uncountable/integration/executors/generic_upload_executor.py,sha256=axGpIfF0IMHilGzFawDizWoDhYffeqYXuhfyYkQhWF4,8754
 uncountable/integration/executors/script_executor.py,sha256=OmSBOtU48G3mqza9c2lCm84pGGyaDk-ZBJCx3RsdJXc,846
 uncountable/integration/secret_retrieval/__init__.py,sha256=3QXVj35w8rRMxVvmmsViFYDi3lcb3g70incfalOEm6o,87
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=M0qXVJpD8hMYIFypHFeyh598sqmIDX8ZOyXK23CluF0,1323
-uncountable/types/__init__.py,sha256=wXpqHGgD4czLWYk3NtuyH-Pcj1orzXRtxmm1Jp04yDY,8090
+uncountable/types/__init__.py,sha256=95iOd3WXWoI_4a461IS2ieWRic3zRyNaCYzfTpX764o,8162
 uncountable/types/async_batch.py,sha256=ihCv5XWSTTPmuO-GMPn1EACGI2CBUIJTATZ3aPgsNBA,523
 uncountable/types/async_batch_processor.py,sha256=BNEWdRIcuJHGeY4KVCudf4gG-kV6_dZR7aeSWupCPjk,8581
 uncountable/types/async_batch_t.py,sha256=9jp9rOyetRdD5aQVyijzQggTyYU4021PBVGXk0ooBCQ,1911
@@ -101,10 +101,12 @@ uncountable/types/calculations_t.py,sha256=7GTSi2L8NYjzjUJJx3cmtVkK9uD-uhfYvIFK-
 uncountable/types/chemical_structure.py,sha256=E-LnikTFDoVQ1b2zKaVUIO_PAKm-7aZZYJi8I8SDSic,302
 uncountable/types/chemical_structure_t.py,sha256=aFsTkkbzy6Gvyde3qrrEYD95gcYhxkgKMiDRaRE0o-Y,760
 uncountable/types/client_base.py,sha256=Z3tQl_RPOd6SCWY2emeYrdguKGEFyoo__wqXzj2eipM,65171
+uncountable/types/client_config.py,sha256=4h5Liko9uKCo9_0gdbPhoK6Jr2Kv7tioLiQ8iKeq-_4,301
+uncountable/types/client_config_t.py,sha256=_HdS37gMSTIiD4qLnW9dIgt8_Rt5A6xhwMGGga7vnLg,625
 uncountable/types/curves.py,sha256=W6uMpG5SyW1MS82szNpxkFEn1MnxNpBFyFbQb2Ysfng,366
 uncountable/types/curves_t.py,sha256=TDpsThz4lKmiBmS9b4ItUSCp64TGv8-qDkxb4B2RoTo,1314
 uncountable/types/entity.py,sha256=3XhLteFDRDZvHejDuYh-KvB65hpwrBygljFfiUcOAM8,315
-uncountable/types/entity_t.py,sha256=eNSDGCa6pZ13evJ1ptMbkjNRTysSOFF30HYLaFILBMc,11631
+uncountable/types/entity_t.py,sha256=HhKmcH5Op8rEkDhpw7SXQ7QakbFdvG5FpB6O4e7cMY0,14439
 uncountable/types/experiment_groups.py,sha256=_0OXcPzSAbkE-rfKt5tPx178YJ4pcEKZvrCxUHgDnvw,309
 uncountable/types/experiment_groups_t.py,sha256=0IGAXwkYiwdjj6aFjLMihxwauACQTyuRU_1usJTeUg4,593
 uncountable/types/field_values.py,sha256=uuIWX-xmfvcinYPdfkWJeb56zzQY01mc9rmotMPMh24,503
@@ -121,8 +123,8 @@ uncountable/types/input_attributes.py,sha256=IrIKQnHqHdS1Utdfzr9GnOe17a8riaqYcO1
 uncountable/types/input_attributes_t.py,sha256=wE1ekiQfb72Z9VpF5SHipKJkgaJFUHJrNkkJdmuhF9w,820
 uncountable/types/inputs.py,sha256=6RIEFfCxLqpeHEGOpu63O4i8zPogjGeB7wiV_rPBw_g,404
 uncountable/types/inputs_t.py,sha256=RW7gF9zTOwByu-nMTMVuBabLOuWKx4O1nvfgvx_R55o,1611
-uncountable/types/job_definition.py,sha256=ImmUw8y4_nEsz_pBEceZdbnJDhFZRfm9BtW4hHhe9ik,1584
-uncountable/types/job_definition_t.py,sha256=oqF988ub1UZDRdMrg2cndTMn7oml102LQQKZ-vLUAVI,5406
+uncountable/types/job_definition.py,sha256=W2wth9vPRBVL0temwnSmvfxxPMKFxZqbQtsMXn8BU3U,1637
+uncountable/types/job_definition_t.py,sha256=1VZO3l5oGnvtG-ikQB6YuukblHhEwtmJfQpGmo0Gi40,5735
 uncountable/types/outputs.py,sha256=sUZx_X-TKCZtLm1YCEH8OISX9DdPlv9ZuUfM3-askCc,281
 uncountable/types/outputs_t.py,sha256=2aORUOr0ls1ZYo-ddkWax3D1ZndmQsWtHfJxpYozlhg,656
 uncountable/types/permissions.py,sha256=1mRnSsmRgjuLgp6pylTwwACD_YRIcmlqxHkufwZtMns,297
@@ -136,7 +138,7 @@ uncountable/types/recipe_identifiers_t.py,sha256=suumLdp_8SZD7oqIxSEaV9D1irMeLJ5
 uncountable/types/recipe_inputs.py,sha256=5ThNFEOGbADqqfj1V3hNvZUcO5pn1Gm17LmzQkWDrtI,351
 uncountable/types/recipe_inputs_t.py,sha256=t5nFzuF1AU6SUHpya6xKHSlcckmt3XxAuk9ofjZ2oGU,746
 uncountable/types/recipe_links.py,sha256=YRiu6t7FWr7ZWycIaOPXBtQFqbY_q5unaP6KQzh1LzE,343
-uncountable/types/recipe_links_t.py,sha256=RPS-Q4FAvKlio7pwufIdERQhlNr27IZGVxMPaoTVqrI,1400
+uncountable/types/recipe_links_t.py,sha256=ZaTeP8AsIT9-gDRi_uQreDsAzEmqF34YTesh9uydYa0,1417
 uncountable/types/recipe_metadata.py,sha256=Zpcrupq_mlT2UhXpMJup5XjtubmvgZ8SbJNq7da2pCs,441
 uncountable/types/recipe_metadata_t.py,sha256=y2O2rIanS6WG0j1UDSsRSwyEKYNDeNaixlR3PmkYnkU,1455
 uncountable/types/recipe_output_metadata.py,sha256=83jKZCvogG9QjZeJpQptdSam_MnnUj-tqh9EVIrTWjE,322
@@ -235,7 +237,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=_Rpha9nxXI3Xr17CrGDtofg4HZ81x2lt0rMZ6As0qfE,893
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=2RCvnccwEm6J5dt7UJVLUKxfk49vfpGXjBLCWIBD3Ck,1006
-UncountablePythonSDK-0.0.44.dist-info/METADATA,sha256=pq3HyRKI8gVZ_HmK6Z7WQekcPSlTMWShM9CTwNpotGE,1707
-UncountablePythonSDK-0.0.44.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-UncountablePythonSDK-0.0.44.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.44.dist-info/RECORD,,
+UncountablePythonSDK-0.0.46.dist-info/METADATA,sha256=T5qGdsLkqIsqutdNrO1eUtVrnZwQ-gCfKcaaDpimbrQ,1707
+UncountablePythonSDK-0.0.46.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+UncountablePythonSDK-0.0.46.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.46.dist-info/RECORD,,

{UncountablePythonSDK-0.0.44.dist-info → UncountablePythonSDK-0.0.46.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (72.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

pkgs/type_spec/emit_python.py

@@ -719,6 +719,14 @@ def _emit_properties(
                 default = "None"
             elif prop.has_default:
                 default = _emit_value(ctx, prop.spec_type, prop.default)
+                if (
+                    isinstance(prop.spec_type, builder.SpecTypeInstance)
+                    and (
+                        prop.spec_type.defn_type.is_base_type(builder.BaseTypeName.s_list)
+                    )
+                    and default == "[]"
+                ):
+                    default = "dataclasses.field(default_factory=list)"
             class_out.write(f"{INDENT * num_indent}{py_name}: {ref_type}")
             if default:
                 class_out.write(f" = {default}")

uncountable/core/async_batch.py

@@ -4,19 +4,34 @@ from uncountable.types.async_batch import AsyncBatchRequest
 from uncountable.types.async_batch_processor import AsyncBatchProcessorBase
 
 
+class AsyncBatchSubmissionError(BaseException):
+    pass
+
+
 class AsyncBatchProcessor(AsyncBatchProcessorBase):
     _client: Client
     _queue: list[AsyncBatchRequest]
+    _submitted_job_ids: list[base_t.ObjectId]
 
     def __init__(self, *, client: Client) -> None:
         super().__init__()
         self._client = client
         self._queue = []
+        self._submitted_job_ids = []
 
     def _enqueue(self, req: async_batch_t.AsyncBatchRequest) -> None:
         self._queue.append(req)
 
+    def current_queue_size(self) -> int:
+        return len(self._queue)
+
     def send(self) -> base_t.ObjectId:
+        if len(self._queue) == 0:
+            raise AsyncBatchSubmissionError("queue is empty")
         job_id = self._client.execute_batch_load_async(requests=self._queue).job_id
+        self._submitted_job_ids.append(job_id)
         self._queue = []
         return job_id
+
+    def get_submitted_job_ids(self) -> list[base_t.ObjectId]:
+        return self._submitted_job_ids

uncountable/core/client.py

@@ -15,6 +15,7 @@ from pkgs.argument_parser import CachedParser
 from pkgs.serialization_util import serialize_for_api
 from pkgs.serialization_util.serialization_helpers import JsonValue
 from uncountable.types.client_base import APIRequest, ClientMethods
+from uncountable.types.client_config import ClientConfigOptions
 
 from .file_upload import FileUpload, FileUploader, UploadedFile
 from .types import AuthDetailsAll, AuthDetailsApiKey, AuthDetailsOAuth
@@ -58,9 +59,7 @@ HTTPRequest = HTTPPostRequest | HTTPGetRequest
 
 
 @dataclass(kw_only=True)
-class ClientConfig:
-    allow_insecure_tls: bool = False
-    extra_headers: dict[str, str] | None = None
+class ClientConfig(ClientConfigOptions):
     transform_request: typing.Callable[[requests.Request], requests.Request] | None = None
 
 

uncountable/integration/construct_client.py

@@ -1,7 +1,8 @@
 import os
-from typing import assert_never
 
 from uncountable.core import AuthDetailsApiKey, Client
+from uncountable.core.client import ClientConfig
+from uncountable.core.types import AuthDetailsAll
 from uncountable.types.job_definition_t import (
     AuthRetrievalEnv,
     ProfileMetadata,
@@ -15,7 +16,7 @@ def _get_env_var(name: str) -> str:
     return value
 
 
-def construct_uncountable_client(profile_meta: ProfileMetadata) -> Client:
+def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
     match profile_meta.auth_retrieval:
         case AuthRetrievalEnv():
             api_id = _get_env_var(f"UNC_PROFILE_{profile_meta.name.upper()}_API_ID")
@@ -26,10 +27,21 @@ def construct_uncountable_client(profile_meta: ProfileMetadata) -> Client:
             assert api_id is not None
             assert api_secret_key is not None
 
-            return Client(
-                base_url=profile_meta.base_url,
-                auth_details=AuthDetailsApiKey(
-                    api_id=api_id, api_secret_key=api_secret_key
-                ),
-            )
-    assert_never(profile_meta.auth_retrieval)
+            return AuthDetailsApiKey(api_id=api_id, api_secret_key=api_secret_key)
+
+
+def _construct_client_config(profile_meta: ProfileMetadata) -> ClientConfig | None:
+    if profile_meta.client_options is None:
+        return None
+    return ClientConfig(
+        allow_insecure_tls=profile_meta.client_options.allow_insecure_tls,
+        extra_headers=profile_meta.client_options.extra_headers,
+    )
+
+
+def construct_uncountable_client(profile_meta: ProfileMetadata) -> Client:
+    return Client(
+        base_url=profile_meta.base_url,
+        auth_details=_construct_auth_details(profile_meta),
+        config=_construct_client_config(profile_meta),
+    )

uncountable/integration/cron.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 
 from pkgs.argument_parser import CachedParser
+from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.integration.construct_client import construct_uncountable_client
 from uncountable.integration.executors.executors import resolve_executor
 from uncountable.integration.job import CronJobArguments, JobLogger
@@ -18,9 +19,12 @@ cron_args_parser = CachedParser(CronJobArgs)
 
 def cron_job_executor(**kwargs: dict) -> None:
     args_passed = cron_args_parser.parse_storage(kwargs)
+    client = construct_uncountable_client(profile_meta=args_passed.profile_metadata)
+    batch_processor = AsyncBatchProcessor(client=client)
     args = CronJobArguments(
         job_definition=args_passed.definition,
-        client=construct_uncountable_client(profile_meta=args_passed.profile_metadata),
+        client=client,
+        batch_processor=batch_processor,
         profile_metadata=args_passed.profile_metadata,
         logger=JobLogger(
             profile_metadata=args_passed.profile_metadata,
@@ -33,3 +37,11 @@ def cron_job_executor(**kwargs: dict) -> None:
     print(f"running job {args_passed.definition.name}")
 
     job.run(args=args)
+
+    if batch_processor.current_queue_size() != 0:
+        batch_processor.send()
+
+    print(f"completed job {args_passed.definition.name}")
+    submitted_batch_job_ids = batch_processor.get_submitted_job_ids()
+    if len(submitted_batch_job_ids) != 0:
+        print("submitted batch jobs", submitted_batch_job_ids)

uncountable/integration/entrypoint.py

@@ -32,6 +32,7 @@ def main(blocking: bool) -> None:
                 base_url=profile.base_url,
                 auth_retrieval=profile.auth_retrieval,
                 jobs=profile.jobs,
+                client_options=profile.client_options,
             )
 
         if blocking:

uncountable/integration/executors/generic_upload_executor.py

@@ -2,7 +2,6 @@ import io
 import os
 import re
 from datetime import datetime, timezone
-from tempfile import NamedTemporaryFile
 
 import paramiko
 
@@ -17,7 +16,7 @@ from pkgs.filesystem_utils import (
 from pkgs.filesystem_utils.filesystem_session import FileSystemSession
 from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.core.file_upload import DataFileUpload, FileUpload
-from uncountable.integration.job import Job, JobArguments, JobLogger, JobResult
+from uncountable.integration.job import Job, JobArguments, JobLogger
 from uncountable.integration.secret_retrieval import retrieve_secret
 from uncountable.types.generic_upload_t import (
     GenericRemoteDirectoryScope,
@@ -26,6 +25,7 @@ from uncountable.types.generic_upload_t import (
 from uncountable.types.job_definition_t import (
     GenericUploadDataSource,
     GenericUploadDataSourceSFTP,
+    JobResult,
 )
 
 

uncountable/integration/job.py

@@ -1,8 +1,9 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 
+from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.core.client import Client
-from uncountable.types.job_definition_t import JobDefinition, ProfileMetadata
+from uncountable.types.job_definition_t import JobDefinition, JobResult, ProfileMetadata
 
 
 class JobLogger:
@@ -25,6 +26,7 @@ class JobArgumentsBase:
     job_definition: JobDefinition
     profile_metadata: ProfileMetadata
     client: Client
+    batch_processor: AsyncBatchProcessor
     logger: JobLogger
 
 
@@ -37,11 +39,6 @@ class CronJobArguments(JobArgumentsBase):
 JobArguments = CronJobArguments
 
 
-@dataclass
-class JobResult:
-    success: bool
-
-
 class Job(ABC):
     _unc_job_registered: bool = False
 

uncountable/integration/server.py

@@ -11,6 +11,7 @@ from apscheduler.triggers.cron import CronTrigger
 from sqlalchemy.engine.base import Engine
 
 from uncountable.integration.cron import CronJobArgs, cron_job_executor
+from uncountable.types.client_config_t import ClientConfigOptions
 from uncountable.types.job_definition_t import (
     AuthRetrieval,
     CronJobDefinition,
@@ -40,10 +41,14 @@ class IntegrationServer:
         base_url: str,
         auth_retrieval: AuthRetrieval,
         jobs: list[JobDefinition],
+        client_options: Optional[ClientConfigOptions],
     ) -> None:
         for job_defn in jobs:
             profile_metadata = ProfileMetadata(
-                name=profile_name, auth_retrieval=auth_retrieval, base_url=base_url
+                name=profile_name,
+                auth_retrieval=auth_retrieval,
+                base_url=base_url,
+                client_options=client_options,
             )
             match job_defn:
                 case CronJobDefinition():

uncountable/types/__init__.py

@@ -12,6 +12,7 @@ from . import async_batch_t as async_batch_t
 from . import base_t as base_t
 from . import calculations_t as calculations_t
 from . import chemical_structure_t as chemical_structure_t
+from . import client_config_t as client_config_t
 from .api.chemical import convert_chemical_formats as convert_chemical_formats_t
 from .api.entity import create_entities as create_entities_t
 from .api.entity import create_entity as create_entity_t
@@ -107,6 +108,7 @@ __all__: list[str] = [
     "base_t",
     "calculations_t",
     "chemical_structure_t",
+    "client_config_t",
     "convert_chemical_formats_t",
     "create_entities_t",
     "create_entity_t",

uncountable/types/client_config.py

@@ -0,0 +1,8 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .client_config_t import ClientConfigOptions as ClientConfigOptions
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_config_t.py

@@ -0,0 +1,22 @@
+# 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
+
+__all__: list[str] = [
+    "ClientConfigOptions",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class ClientConfigOptions:
+    allow_insecure_tls: bool = False
+    extra_headers: typing.Optional[typing.Optional[dict[str, str]]] = None
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/entity_t.py

@@ -29,40 +29,53 @@ __all__: list[str] = [
         "calendar": "Calendar",
         "calendar_event": "Calendar Event",
         "certification": "Certification",
+        "characterization": "Characterization",
         "comment": "Comment",
         "comment_thread": "Comment Thread",
         "condition_parameter": "Condition Parameter",
         "condition_parameter_mat_family": "Condition Parameter Material Family",
+        "condition_parameter_value": "Condition Parameter Value",
         "constraint": "Constraint",
         "constraint_set": "Constraint Set",
         "curve_settings": "Curve Settings",
         "custom_entity": "Custom Entity",
         "doe_run": "Suggest Experiments",
+        "email_record_external": "External Emails",
+        "email_relay_server": "Email Relay Server",
+        "enter_recipe_view_preference": "Enter Recipe View Preference",
         "entities_viewed": "Entities Viewed",
+        "entity_annotation": "Annotation",
         "entity_field": "Field",
         "entity_field_audit_log": "Entity Field Audit Log",
+        "entity_link": "Entity Link",
         "entity_phase_event": "Entity Phase Event",
+        "entity_search": "Entity Search",
         "entity_tag": "Entity Tag",
         "equipment": "Equipment",
+        "equipment_maintenance": "Equipment Maintenance",
         "equipment_type": "Equipment Type",
         "experiment_group": "Experiment Group",
         "experiment_group_type": "Experiment Group Type",
         "favorite_link": "Favorite Link",
         "field_option_set": "Field Option Set",
+        "field_option": "Field Option",
         "file_folder": "File Folder",
         "file_folder_membership": "File Folder Membership",
         "goal": "Goal",
         "ingredient": "Ingredient",
         "ingredient_attribute": "Ingredient Attribute",
         "ingredient_attribute_category": "Ingredient Attribute Category",
-        "ingredient_category": "Ingredient Category",
+        "ingredient_category": "Ingredient Category (Project)",
         "ingredient_category_all": "Ingredient Category",
+        "ingredient_category_all_material_family": "Ingredient Category Material Family",
         "ingredient_lot": "Ingredient Lot",
         "ingredient_role": "Ingredient Role",
         "ingredient_tag": "Ingredient Subcategory",
         "input_group": "Input Group",
         "inv_local_locations": "Inventory Location",
         "inventory_amount": "Inventory Amount",
+        "inventory_limit": "Inventory Limit",
+        "inventory_history": "Inventory History",
         "lab_location": "Lab Location",
         "lab_request": "Lab Request",
         "license": "License",
@@ -82,46 +95,59 @@ __all__: list[str] = [
         "output_condition_parameter_analytical_method_category": "Output Condition Parameter Analytical Method Category",
         "output_condition_parameter_analytical_method": "Output Condition Parameter Analytical Method",
         "output_group": "Output Group",
-        "recipe_output_metadata": "Recipe Output Metadata",
         "permission": "Permission",
+        "permissions_recipe_link_inheritance": "Permissions Recipe Link Inheritance",
         "phase_workflow": "Phase Workflow",
+        "platform_config": "Platform Config",
         "predictive_model": "Predictive Model",
         "product": "Product",
         "project": "Project",
+        "project_workbook_comment": "Notebook Cell Comment",
         "recipe": "Experiment",
+        "recipe_audit_log": "Experiment Audit Log",
         "recipe_calculation": "Recipe Calculation",
-        "recipe_permission_view": "Recipe Permission View",
-        "recipe_project": "Recipe Project",
         "recipe_check": "Experiment Check",
         "recipe_export": "Recipe Export",
+        "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_tag": "Recipe Tag",
         "recipe_metadata": "Experiment Metadata",
+        "metadata_value": "Metadata Value",
         "review": "Review",
         "review_entity_user_status": "Review Entity User Status",
         "review_user": "Review User",
         "rlw_phase": "Phase",
+        "rlw_trigger": "Trigger",
         "rlw_trigger_set": "Trigger Set",
         "safety_data_sheet": "Safety Data Sheet",
         "scheduled_notification": "Scheduled Notification",
         "specs": "Spec",
         "split_template": "Split Template",
         "structured_loading_user_config": "List Config",
+        "table_builder_config": "Table Builder Config",
         "task": "Task",
         "task_board": "Task Board",
         "template": "Template",
         "timesheet_entry": "Timesheet Entry",
         "training_run": "Analyze Experiment",
+        "training_run_template": "Scheduled Training Run",
         "training_set": "Training Set",
         "time_series": "Time Series",
+        "time_series_stage": "Time Series Stage",
         "units": "Units",
         "user_favorite": "User Favorite",
         "user": "User",
         "user_certification": "User Certification",
         "user_group": "User Group",
         "user_list_set": "User List Set",
+        "value_spec_property": "Value Spec Property",
         "webhook": "Webhook",
         "webhook_event": "Webhook Event",
         "workflow": "Workflow",
@@ -135,6 +161,7 @@ __all__: list[str] = [
         "public_data_sourced_output": "ChemQuery Property",
         "public_data_sourced_project": "ChemQuery Category",
         "public_data_sourced_recipe": "ChemQuery Product",
+        "public_data_sourced_raw_material": "ChemQuery Raw Materials",
         "public_data_sourced_recipe_project": "ChemQuery Product Category",
         "public_data_sourced_recipe_metadata": "ChemQuery Product Information",
         "public_data_sourced_units": "ChemQuery Units",
@@ -152,26 +179,36 @@ class EntityType(StrEnum):
     CALENDAR = "calendar"
     CALENDAR_EVENT = "calendar_event"
     CERTIFICATION = "certification"
+    CHARACTERIZATION = "characterization"
     COMMENT = "comment"
     COMMENT_THREAD = "comment_thread"
     CONDITION_PARAMETER = "condition_parameter"
     CONDITION_PARAMETER_MAT_FAMILY = "condition_parameter_mat_family"
+    CONDITION_PARAMETER_VALUE = "condition_parameter_value"
     CONSTRAINT = "constraint"
     CONSTRAINT_SET = "constraint_set"
     CURVE_SETTINGS = "curve_settings"
     CUSTOM_ENTITY = "custom_entity"
     DOE_RUN = "doe_run"
+    EMAIL_RECORD_EXTERNAL = "email_record_external"
+    EMAIL_RELAY_SERVER = "email_relay_server"
+    ENTER_RECIPE_VIEW_PREFERENCE = "enter_recipe_view_preference"
     ENTITIES_VIEWED = "entities_viewed"
+    ENTITY_ANNOTATION = "entity_annotation"
     ENTITY_FIELD = "entity_field"
     ENTITY_FIELD_AUDIT_LOG = "entity_field_audit_log"
+    ENTITY_LINK = "entity_link"
     ENTITY_PHASE_EVENT = "entity_phase_event"
+    ENTITY_SEARCH = "entity_search"
     ENTITY_TAG = "entity_tag"
     EQUIPMENT = "equipment"
+    EQUIPMENT_MAINTENANCE = "equipment_maintenance"
     EQUIPMENT_TYPE = "equipment_type"
     EXPERIMENT_GROUP = "experiment_group"
     EXPERIMENT_GROUP_TYPE = "experiment_group_type"
     FAVORITE_LINK = "favorite_link"
     FIELD_OPTION_SET = "field_option_set"
+    FIELD_OPTION = "field_option"
     FILE_FOLDER = "file_folder"
     FILE_FOLDER_MEMBERSHIP = "file_folder_membership"
     GOAL = "goal"
@@ -180,12 +217,15 @@ class EntityType(StrEnum):
     INGREDIENT_ATTRIBUTE_CATEGORY = "ingredient_attribute_category"
     INGREDIENT_CATEGORY = "ingredient_category"
     INGREDIENT_CATEGORY_ALL = "ingredient_category_all"
+    INGREDIENT_CATEGORY_ALL_MATERIAL_FAMILY = "ingredient_category_all_material_family"
     INGREDIENT_LOT = "ingredient_lot"
     INGREDIENT_ROLE = "ingredient_role"
     INGREDIENT_TAG = "ingredient_tag"
     INPUT_GROUP = "input_group"
     INV_LOCAL_LOCATIONS = "inv_local_locations"
     INVENTORY_AMOUNT = "inventory_amount"
+    INVENTORY_LIMIT = "inventory_limit"
+    INVENTORY_HISTORY = "inventory_history"
     LAB_LOCATION = "lab_location"
     LAB_REQUEST = "lab_request"
     LICENSE = "license"
@@ -205,46 +245,59 @@ class EntityType(StrEnum):
     OUTPUT_CONDITION_PARAMETER_ANALYTICAL_METHOD_CATEGORY = "output_condition_parameter_analytical_method_category"
     OUTPUT_CONDITION_PARAMETER_ANALYTICAL_METHOD = "output_condition_parameter_analytical_method"
     OUTPUT_GROUP = "output_group"
-    RECIPE_OUTPUT_METADATA = "recipe_output_metadata"
     PERMISSION = "permission"
+    PERMISSIONS_RECIPE_LINK_INHERITANCE = "permissions_recipe_link_inheritance"
     PHASE_WORKFLOW = "phase_workflow"
+    PLATFORM_CONFIG = "platform_config"
     PREDICTIVE_MODEL = "predictive_model"
     PRODUCT = "product"
     PROJECT = "project"
+    PROJECT_WORKBOOK_COMMENT = "project_workbook_comment"
     RECIPE = "recipe"
+    RECIPE_AUDIT_LOG = "recipe_audit_log"
     RECIPE_CALCULATION = "recipe_calculation"
-    RECIPE_PERMISSION_VIEW = "recipe_permission_view"
-    RECIPE_PROJECT = "recipe_project"
     RECIPE_CHECK = "recipe_check"
     RECIPE_EXPORT = "recipe_export"
+    RECIPE_INGREDIENT = "recipe_ingredient"
+    RECIPE_INGREDIENT_ACTUAL = "recipe_ingredient_actual"
     RECIPE_INGREDIENTS_COMPOUNDED = "recipe_ingredients_compounded"
     RECIPE_INGREDIENTS_COMPOUNDED_CALCULATION = "recipe_ingredients_compounded_calculation"
+    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_TAG = "recipe_tag"
     RECIPE_METADATA = "recipe_metadata"
+    METADATA_VALUE = "metadata_value"
     REVIEW = "review"
     REVIEW_ENTITY_USER_STATUS = "review_entity_user_status"
     REVIEW_USER = "review_user"
     RLW_PHASE = "rlw_phase"
+    RLW_TRIGGER = "rlw_trigger"
     RLW_TRIGGER_SET = "rlw_trigger_set"
     SAFETY_DATA_SHEET = "safety_data_sheet"
     SCHEDULED_NOTIFICATION = "scheduled_notification"
     SPECS = "specs"
     SPLIT_TEMPLATE = "split_template"
     STRUCTURED_LOADING_USER_CONFIG = "structured_loading_user_config"
+    TABLE_BUILDER_CONFIG = "table_builder_config"
     TASK = "task"
     TASK_BOARD = "task_board"
     TEMPLATE = "template"
     TIMESHEET_ENTRY = "timesheet_entry"
     TRAINING_RUN = "training_run"
+    TRAINING_RUN_TEMPLATE = "training_run_template"
     TRAINING_SET = "training_set"
     TIME_SERIES = "time_series"
+    TIME_SERIES_STAGE = "time_series_stage"
     UNITS = "units"
     USER_FAVORITE = "user_favorite"
     USER = "user"
     USER_CERTIFICATION = "user_certification"
     USER_GROUP = "user_group"
     USER_LIST_SET = "user_list_set"
+    VALUE_SPEC_PROPERTY = "value_spec_property"
     WEBHOOK = "webhook"
     WEBHOOK_EVENT = "webhook_event"
     WORKFLOW = "workflow"
@@ -258,6 +311,7 @@ class EntityType(StrEnum):
     PUBLIC_DATA_SOURCED_OUTPUT = "public_data_sourced_output"
     PUBLIC_DATA_SOURCED_PROJECT = "public_data_sourced_project"
     PUBLIC_DATA_SOURCED_RECIPE = "public_data_sourced_recipe"
+    PUBLIC_DATA_SOURCED_RAW_MATERIAL = "public_data_sourced_raw_material"
     PUBLIC_DATA_SOURCED_RECIPE_PROJECT = "public_data_sourced_recipe_project"
     PUBLIC_DATA_SOURCED_RECIPE_METADATA = "public_data_sourced_recipe_metadata"
     PUBLIC_DATA_SOURCED_UNITS = "public_data_sourced_units"

uncountable/types/job_definition.py

@@ -23,4 +23,5 @@ from .job_definition_t import AuthRetrievalEnv as AuthRetrievalEnv
 from .job_definition_t import AuthRetrieval as AuthRetrieval
 from .job_definition_t import ProfileDefinition as ProfileDefinition
 from .job_definition_t import ProfileMetadata as ProfileMetadata
+from .job_definition_t import JobResult as JobResult
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/job_definition_t.py

@@ -11,6 +11,7 @@ from pkgs.strenum_compat import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
+from . import client_config_t
 from . import generic_upload_t
 from . import secret_retrieval_t
 
@@ -32,6 +33,7 @@ __all__: list[str] = [
     "JobExecutorGenericUpload",
     "JobExecutorScript",
     "JobExecutorType",
+    "JobResult",
     "ProfileDefinition",
     "ProfileMetadata",
 ]
@@ -192,6 +194,7 @@ class ProfileDefinition:
     auth_retrieval: AuthRetrieval
     base_url: str
     jobs: list[JobDefinition]
+    client_options: typing.Optional[client_config_t.ClientConfigOptions] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -200,4 +203,11 @@ class ProfileMetadata:
     name: str
     base_url: str
     auth_retrieval: AuthRetrieval
+    client_options: typing.Optional[client_config_t.ClientConfigOptions]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class JobResult:
+    success: bool
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/recipe_links_t.py

@@ -46,5 +46,5 @@ class RecipeLink:
     recipe_id_from: base_t.ObjectId
     recipe_id_to: base_t.ObjectId
     link_type: str
-    name: str
+    name: typing.Optional[str]
 # DO NOT MODIFY -- This file is generated by type_spec