UncountablePythonSDK 0.0.111__py3-none-any.whl → 0.0.113__py3-none-any.whl

uncountable/types/api/recipes/lock_recipes.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 from enum import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -34,7 +35,7 @@ class RecipeLockType(StrEnum):
 @serial_class(
     named_type_path="sdk.api.recipes.lock_recipes.RecipeLockBase",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeLockBase:
     recipe: identifier_t.IdentifierKey
 
@@ -43,7 +44,7 @@ class RecipeLockBase:
 @serial_class(
     named_type_path="sdk.api.recipes.lock_recipes.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     type: RecipeLockType = RecipeLockType.ALL
     recipes: list[identifier_t.IdentifierKey]
@@ -56,7 +57,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.lock_recipes.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/remove_recipe_from_project.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -25,7 +26,7 @@ ENDPOINT_PATH = "api/external/recipes/remove_recipe_from_project"
 @serial_class(
     named_type_path="sdk.api.recipes.remove_recipe_from_project.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     recipe_key: identifier_t.IdentifierKey
     project_key: identifier_t.IdentifierKey
@@ -35,7 +36,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.remove_recipe_from_project.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_inputs.py

@@ -27,7 +27,7 @@ ENDPOINT_PATH = "api/external/recipes/external_set_recipe_inputs"
     named_type_path="sdk.api.recipes.set_recipe_inputs.RecipeInputValue",
     to_string_values={"value_numeric"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeInputValue:
     recipe_id: base_t.ObjectId
     input_id: base_t.ObjectId
@@ -43,7 +43,7 @@ class RecipeInputValue:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_inputs.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     input_data: list[RecipeInputValue]
 
@@ -52,7 +52,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_inputs.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_metadata.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import identifier_t
 from ... import recipe_metadata_t
 
@@ -26,7 +27,7 @@ ENDPOINT_PATH = "api/external/recipes/set_recipe_metadata"
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_metadata.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     recipe_key: identifier_t.IdentifierKey
     recipe_metadata: list[recipe_metadata_t.MetadataValue]
@@ -36,7 +37,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_metadata.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_output_annotations.py

@@ -35,7 +35,7 @@ ENDPOINT_PATH = "api/external/recipes/set_recipe_output_annotations"
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.AnnotationEdit",
     to_string_values={"lower_value", "upper_value"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AnnotationEdit:
     annotation_type_key: identifier_t.IdentifierKey
     lower_value: Decimal | None = None
@@ -52,7 +52,7 @@ class AnnotationUpdateType(StrEnum):
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.RecipeOutputEditBase",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputEditBase:
     recipe_id: base_t.ObjectId
     output_id: base_t.ObjectId
@@ -66,7 +66,7 @@ class RecipeOutputEditBase:
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.RecipeOutputMergeAnnotations",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputMergeAnnotations(RecipeOutputEditBase):
     type: typing.Literal[AnnotationUpdateType.MERGE] = AnnotationUpdateType.MERGE
 
@@ -76,7 +76,7 @@ class RecipeOutputMergeAnnotations(RecipeOutputEditBase):
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.RecipeOutputReplaceAnnotations",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputReplaceAnnotations(RecipeOutputEditBase):
     type: typing.Literal[AnnotationUpdateType.REPLACE] = AnnotationUpdateType.REPLACE
 
@@ -99,7 +99,7 @@ RecipeOutputUpdateAnnotations = typing.Annotated[
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     updates: list[RecipeOutputUpdateAnnotations]
 
@@ -108,7 +108,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_annotations.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_output_file.py

@@ -27,7 +27,7 @@ ENDPOINT_PATH = "api/external/recipes/external_set_recipe_output_file"
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_file.RecipeOutputFileValue",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputFileValue:
     recipe_id: base_t.ObjectId
     output_id: base_t.ObjectId
@@ -40,7 +40,7 @@ class RecipeOutputFileValue:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_file.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     output_file_data: RecipeOutputFileValue
 
@@ -49,7 +49,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_output_file.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data(response_t.Response):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_outputs.py

@@ -33,7 +33,7 @@ ENDPOINT_PATH = "api/external/recipes/external_set_recipe_outputs"
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_outputs.CurveValues",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class CurveValues:
     x_values: list[Decimal | None]
     y_values: list[Decimal | None]
@@ -50,7 +50,7 @@ class NullBehavior(StrEnum):
     named_type_path="sdk.api.recipes.set_recipe_outputs.RecipeOutputValue",
     to_string_values={"value_numeric"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputValue:
     recipe_id: base_t.ObjectId
     output_id: base_t.ObjectId
@@ -60,7 +60,7 @@ class RecipeOutputValue:
     value_numeric: Decimal | None = None
     value_str: str | None = None
     value_curve: CurveValues | None = None
-    value_color: data_t.RecipeOutputColor | None = None
+    value_color: data_t.SupportedColorFormatColor | None = None
     formatting: recipes_t.RecipeAttributeFormatting | None = None
     field_values: list[field_values_t.ArgumentValueRefName | field_values_t.ArgumentValueId] | None = None
 
@@ -69,7 +69,7 @@ class RecipeOutputValue:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_outputs.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     output_data: list[RecipeOutputValue]
 
@@ -78,7 +78,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_outputs.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data(response_t.Response):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/set_recipe_tags.py

@@ -41,7 +41,7 @@ class RecipeTagUpdateType(StrEnum):
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_tags.RecipeTagUpdateBase",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeTagUpdateBase:
     type: RecipeTagUpdateType
 
@@ -51,7 +51,7 @@ class RecipeTagUpdateBase:
     named_type_path="sdk.api.recipes.set_recipe_tags.RecipeTagAppend",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeTagAppend(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.APPEND] = RecipeTagUpdateType.APPEND
     recipe_tag_ids: list[base_t.ObjectId]
@@ -62,7 +62,7 @@ class RecipeTagAppend(RecipeTagUpdateBase):
     named_type_path="sdk.api.recipes.set_recipe_tags.RecipeTagRemove",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeTagRemove(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.REMOVE] = RecipeTagUpdateType.REMOVE
     recipe_tag_ids: list[base_t.ObjectId]
@@ -73,7 +73,7 @@ class RecipeTagRemove(RecipeTagUpdateBase):
     named_type_path="sdk.api.recipes.set_recipe_tags.RecipeTagOverride",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeTagOverride(RecipeTagUpdateBase):
     type: typing.Literal[RecipeTagUpdateType.OVERRIDE] = RecipeTagUpdateType.OVERRIDE
     recipe_tag_ids: list[base_t.ObjectId]
@@ -92,7 +92,7 @@ RecipeTagUpdate = typing.Annotated[
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_tags.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     recipe_key: identifier_t.IdentifierKey
     recipe_tag_update: RecipeTagUpdate
@@ -102,7 +102,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.set_recipe_tags.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/unarchive_recipes.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -25,7 +26,7 @@ ENDPOINT_PATH = "api/external/recipes/unarchive"
 @serial_class(
     named_type_path="sdk.api.recipes.unarchive_recipes.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     recipes: list[identifier_t.IdentifierKey]
 
@@ -34,7 +35,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.unarchive_recipes.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/unlock_recipes.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 from enum import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -32,7 +33,7 @@ class RecipeUnlockType(StrEnum):
 @serial_class(
     named_type_path="sdk.api.recipes.unlock_recipes.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     type: RecipeUnlockType = RecipeUnlockType.STANDARD
     recipes: list[identifier_t.IdentifierKey]
@@ -43,7 +44,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.recipes.unlock_recipes.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/runsheet/__init__.py

@@ -0,0 +1 @@
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/runsheet/complete_async_upload.py

@@ -0,0 +1,41 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# 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 base_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/runsheet/complete_async_upload"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.runsheet.complete_async_upload.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    async_job_id: base_t.ObjectId
+    file_id: base_t.ObjectId
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.runsheet.complete_async_upload.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data:
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/triggers/run_trigger.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from ... import base_t
 from ... import entity_t
 from ... import response_t
 
@@ -26,7 +27,7 @@ ENDPOINT_PATH = "api/external/triggers/run_trigger"
 @serial_class(
     named_type_path="sdk.api.triggers.run_trigger.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     trigger_ref_name: str
     entity: entity_t.Entity | None = None
@@ -36,7 +37,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.triggers.run_trigger.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data(response_t.Response):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/uploader/invoke_uploader.py

@@ -28,7 +28,7 @@ ENDPOINT_PATH = "api/external/uploader/invoke_uploader"
 @serial_class(
     named_type_path="sdk.api.uploader.invoke_uploader.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     uploader_key: identifier_t.IdentifierKey
     destination: generic_upload_t.UploadDestination
@@ -40,7 +40,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.uploader.invoke_uploader.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch_processor.py

@@ -13,6 +13,7 @@ import uncountable.types.api.recipes.associate_recipe_as_lot as associate_recipe
 from uncountable.types import async_batch_t
 from uncountable.types import base_t
 import uncountable.types.api.recipes.clear_recipe_outputs as clear_recipe_outputs_t
+import uncountable.types.api.runsheet.complete_async_upload as complete_async_upload_t
 import uncountable.types.api.entity.create_or_update_entity as create_or_update_entity_t
 import uncountable.types.api.recipes.create_recipe as create_recipe_t
 import uncountable.types.api.recipes.edit_recipe_inputs as edit_recipe_inputs_t
@@ -183,6 +184,39 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def complete_async_upload(
+        self,
+        *,
+        async_job_id: base_t.ObjectId,
+        file_id: base_t.ObjectId,
+        depends_on: list[str] | None = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Processes an file id with a given async job id to be uploaded asynchronously
+
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = complete_async_upload_t.Arguments(
+            async_job_id=async_job_id,
+            file_id=file_id,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.COMPLETE_ASYNC_UPLOAD,
+            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 create_or_update_entity(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -43,6 +43,7 @@ class AsyncBatchRequestPath(StrEnum):
     LOOKUP_ENTITY = "entity/lookup_entity"
     CREATE_RECIPE_LINK = "recipe_links/create_recipe_link"
     UPSERT_CONDITION_MATCH = "condition_parameters/upsert_condition_match"
+    COMPLETE_ASYNC_UPLOAD = "runsheet/complete_async_upload"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -50,7 +51,7 @@ class AsyncBatchRequestPath(StrEnum):
     named_type_path="sdk.async_batch.AsyncBatchRequest",
     unconverted_values={"data"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AsyncBatchRequest:
     path: AsyncBatchRequestPath
     data: base_t.JsonValue
@@ -63,7 +64,7 @@ class AsyncBatchRequest:
     named_type_path="sdk.async_batch.AsyncBatchResultValue",
     unconverted_values={"value"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AsyncBatchResultValue:
     data_key: str
     value: base_t.JsonValue
@@ -73,7 +74,7 @@ class AsyncBatchResultValue:
 @serial_class(
     named_type_path="sdk.async_batch.AsyncBatchActionReturn",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AsyncBatchActionReturn:
     modification_made: bool
     result_id: base_t.ObjectId | None = None
@@ -85,7 +86,7 @@ class AsyncBatchActionReturn:
 @serial_class(
     named_type_path="sdk.async_batch.SavedAsyncBatchActionReturn",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class SavedAsyncBatchActionReturn:
     identifier: str
     result_data: AsyncBatchActionReturn
@@ -95,7 +96,7 @@ class SavedAsyncBatchActionReturn:
 @serial_class(
     named_type_path="sdk.async_batch.QueuedAsyncBatchRequest",
 )
-@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True, frozen=True, eq=True)  # type: ignore[literal-required]
 class QueuedAsyncBatchRequest:
     path: AsyncBatchRequestPath
     batch_reference: str

uncountable/types/auth_retrieval_t.py

@@ -10,6 +10,7 @@ from enum import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
+from . import base_t
 from . import secret_retrieval_t
 
 __all__: list[str] = [
@@ -31,7 +32,7 @@ class AuthRetrievalType(StrEnum):
 @serial_class(
     named_type_path="sdk.auth_retrieval.AuthRetrievalBase",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AuthRetrievalBase:
     type: AuthRetrievalType
 
@@ -41,7 +42,7 @@ class AuthRetrievalBase:
     named_type_path="sdk.auth_retrieval.AuthRetrievalOAuth",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AuthRetrievalOAuth(AuthRetrievalBase):
     type: typing.Literal[AuthRetrievalType.OAUTH] = AuthRetrievalType.OAUTH
     refresh_token_secret: secret_retrieval_t.SecretRetrieval
@@ -52,7 +53,7 @@ class AuthRetrievalOAuth(AuthRetrievalBase):
     named_type_path="sdk.auth_retrieval.AuthRetrievalBasic",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AuthRetrievalBasic(AuthRetrievalBase):
     type: typing.Literal[AuthRetrievalType.BASIC] = AuthRetrievalType.BASIC
     api_id_secret: secret_retrieval_t.SecretRetrieval

uncountable/types/base_t.py

@@ -44,6 +44,10 @@ REF_NAME_STRICT_REGEX = rf"{REF_NAME_STRICT_REGEX_STRING}"
 ID_REGEX = r"-?[1-9][0-9]{0,20}"
 
 
+# ENABLE_SLOTS should be removed after slots have been tested locally
+import os
+ENABLE_SLOTS = os.environ.get("UNC_ENABLE_DATACLASS_SLOTS") == "true"
+
 if TYPE_CHECKING:
     JsonValue = Union[JsonScalar, Mapping[str, "JsonValue"], Sequence["JsonValue"]]
     ExtJsonValue = JsonValue

uncountable/types/calculations_t.py

@@ -19,7 +19,7 @@ __all__: list[str] = [
 @serial_class(
     named_type_path="sdk.calculations.Calculation",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Calculation:
     id: base_t.ObjectId
     name: str

uncountable/types/chemical_structure_t.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from . import base_t
 
 __all__: list[str] = [
     "ChemicalStructure",
@@ -19,7 +20,7 @@ __all__: list[str] = [
     named_type_path="sdk.chemical_structure.ChemicalStructure",
     to_string_values={"molecular_weight"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class ChemicalStructure:
     ketcher_file: str | None = None
     gross_formula: str | None = None

uncountable/types/client_base.py

@@ -17,6 +17,7 @@ import uncountable.types.api.recipes.associate_recipe_as_lot as associate_recipe
 from uncountable.types import async_batch_t
 from uncountable.types import base_t
 import uncountable.types.api.recipes.clear_recipe_outputs as clear_recipe_outputs_t
+import uncountable.types.api.runsheet.complete_async_upload as complete_async_upload_t
 import uncountable.types.api.chemical.convert_chemical_formats as convert_chemical_formats_t
 import uncountable.types.api.entity.create_entities as create_entities_t
 import uncountable.types.api.entity.create_entity as create_entity_t
@@ -266,6 +267,26 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=clear_recipe_outputs_t.Data)
 
+    def complete_async_upload(
+        self,
+        *,
+        async_job_id: base_t.ObjectId,
+        file_id: base_t.ObjectId,
+    ) -> complete_async_upload_t.Data:
+        """Processes an file id with a given async job id to be uploaded asynchronously
+
+        """
+        args = complete_async_upload_t.Arguments(
+            async_job_id=async_job_id,
+            file_id=file_id,
+        )
+        api_request = APIRequest(
+            method=complete_async_upload_t.ENDPOINT_METHOD,
+            endpoint=complete_async_upload_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=complete_async_upload_t.Data)
+
     def convert_chemical_formats(
         self,
         *,

uncountable/types/client_config_t.py

@@ -8,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from . import base_t
 
 __all__: list[str] = [
     "ClientConfigOptions",
@@ -18,7 +19,7 @@ __all__: list[str] = [
 @serial_class(
     named_type_path="sdk.client_config.ClientConfigOptions",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class ClientConfigOptions:
     allow_insecure_tls: bool = False
     extra_headers: dict[str, str] | None = None