UncountablePythonSDK 0.0.110__py3-none-any.whl → 0.0.112__py3-none-any.whl

uncountable/core/client.py

@@ -1,8 +1,9 @@
 import base64
+import datetime
 import re
 import typing
 from dataclasses import dataclass
-from datetime import datetime, timedelta
+from datetime import UTC, timedelta
 from enum import StrEnum
 from io import BytesIO
 from urllib.parse import unquote, urljoin
@@ -145,7 +146,7 @@ class SDKError(Exception):
 @dataclass(kw_only=True)
 class OAuthBearerTokenCache:
     token: str
-    expires_at: datetime
+    expires_at: datetime.datetime
 
 
 @dataclass(kw_only=True)
@@ -274,7 +275,8 @@ class Client(ClientMethods):
         if (
             self._oauth_bearer_token_cache is None
             or (
-                self._oauth_bearer_token_cache.expires_at - datetime.now()
+                self._oauth_bearer_token_cache.expires_at
+                - datetime.datetime.now(tz=UTC)
             ).total_seconds()
             < OAUTH_REFRESH_WINDOW_SECONDS
         ):
@@ -290,7 +292,8 @@ class Client(ClientMethods):
             token_data = oauth_bearer_token_data_parser.parse_storage(data)
             self._oauth_bearer_token_cache = OAuthBearerTokenCache(
                 token=token_data.access_token,
-                expires_at=datetime.now() + timedelta(seconds=token_data.expires_in),
+                expires_at=datetime.datetime.now(tz=UTC)
+                + timedelta(seconds=token_data.expires_in),
             )
 
         return self._oauth_bearer_token_cache.token

uncountable/integration/executors/generic_upload_executor.py

@@ -1,7 +1,8 @@
+import datetime
 import io
 import os
 import re
-from datetime import datetime, timezone
+from datetime import UTC
 
 import paramiko
 
@@ -164,7 +165,7 @@ def _move_files_post_upload(
     appended_text = ""
 
     if remote_directory_scope.prepend_date_on_archive:
-        appended_text = f"-{datetime.now(timezone.utc).timestamp()}"
+        appended_text = f"-{datetime.datetime.now(UTC).timestamp()}"
 
     for file_path in success_file_paths:
         filename = os.path.split(file_path)[-1]

uncountable/integration/job.py

@@ -6,8 +6,9 @@ from dataclasses import dataclass
 from pkgs.argument_parser import CachedParser
 from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.core.client import Client
+from uncountable.core.file_upload import FileUpload
 from uncountable.integration.telemetry import JobLogger
-from uncountable.types import base_t, webhook_job_t
+from uncountable.types import base_t, entity_t, webhook_job_t
 from uncountable.types.job_definition_t import JobDefinition, JobResult, ProfileMetadata
 
 
@@ -91,3 +92,25 @@ class RunsheetWebhookJob(WebhookJob[webhook_job_t.RunsheetWebhookPayload]):
     @property
     def webhook_payload_type(self) -> type:
         return webhook_job_t.RunsheetWebhookPayload
+
+    @abstractmethod
+    def build_runsheet(
+        self,
+        *,
+        args: JobArguments,
+        entities: list[entity_t.Entity],
+    ) -> FileUpload: ...
+
+    def run(
+        self, args: JobArguments, payload: webhook_job_t.RunsheetWebhookPayload
+    ) -> JobResult:
+        runsheet = self.build_runsheet(args=args, entities=payload.entities)
+
+        files = args.client.upload_files(file_uploads=[runsheet])
+        args.client.complete_async_upload(
+            async_job_id=payload.async_job_id, file_id=files[0].file_id
+        )
+
+        return JobResult(
+            success=True,
+        )

uncountable/integration/queue_runner/datastore/datastore_sqlite.py

@@ -1,5 +1,6 @@
+import datetime
 import uuid
-from datetime import datetime, timezone
+from datetime import UTC
 
 from sqlalchemy import delete, insert, select, update
 from sqlalchemy.engine import Engine
@@ -30,7 +31,7 @@ class DatastoreSqlite(Datastore):
             serialized_payload = serialize_for_storage(job_payload)
             queued_job_uuid = str(uuid.uuid4())
             num_attempts = 0
-            submitted_at = datetime.now(timezone.utc)
+            submitted_at = datetime.datetime.now(UTC)
             insert_stmt = insert(QueuedJob).values({
                 QueuedJob.id.key: queued_job_uuid,
                 QueuedJob.job_ref_name.key: job_ref_name,

uncountable/integration/scheduler.py

@@ -1,9 +1,10 @@
+import datetime
 import multiprocessing
 import subprocess
 import sys
 import time
 from dataclasses import dataclass
-from datetime import datetime, timezone
+from datetime import UTC
 
 from opentelemetry.trace import get_current_span
 
@@ -96,7 +97,7 @@ def _wait_queue_runner_online() -> None:
     QUEUE_RUNNER_HEALTH_CHECK_DELAY_SECS = 1
 
     num_attempts = 0
-    before = datetime.now(timezone.utc)
+    before = datetime.datetime.now(UTC)
     while num_attempts < MAX_QUEUE_RUNNER_HEALTH_CHECKS:
         try:
             if check_health(port=get_local_admin_server_port()):
@@ -105,7 +106,7 @@ def _wait_queue_runner_online() -> None:
             pass
         num_attempts += 1
         time.sleep(QUEUE_RUNNER_HEALTH_CHECK_DELAY_SECS)
-    after = datetime.now(timezone.utc)
+    after = datetime.datetime.now(UTC)
     duration_secs = (after - before).seconds
     raise Exception(f"queue runner failed to come online after {duration_secs} seconds")
 

uncountable/types/__init__.py

@@ -16,6 +16,7 @@ from . import calculations_t as calculations_t
 from . import chemical_structure_t as chemical_structure_t
 from .api.recipes import clear_recipe_outputs as clear_recipe_outputs_t
 from . import client_config_t as client_config_t
+from .api.runsheet import complete_async_upload as complete_async_upload_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 @@ from . import units_t as units_t
 from .api.entity import unlock_entity as unlock_entity_t
 from .api.recipes import unlock_recipes as unlock_recipes_t
 from .api.material_families import update_entity_material_families as update_entity_material_families_t
+from .api.condition_parameters import upsert_condition_match as upsert_condition_match_t
 from .api.field_options import upsert_field_options as upsert_field_options_t
 from . import users_t as users_t
 from . import webhook_job_t as webhook_job_t
@@ -128,6 +130,7 @@ __all__: list[str] = [
     "chemical_structure_t",
     "clear_recipe_outputs_t",
     "client_config_t",
+    "complete_async_upload_t",
     "convert_chemical_formats_t",
     "create_entities_t",
     "create_entity_t",
@@ -219,6 +222,7 @@ __all__: list[str] = [
     "unlock_entity_t",
     "unlock_recipes_t",
     "update_entity_material_families_t",
+    "upsert_condition_match_t",
     "upsert_field_options_t",
     "users_t",
     "webhook_job_t",

uncountable/types/api/batch/execute_batch.py

@@ -43,7 +43,7 @@ class RequestMethod(StrEnum):
     named_type_path="sdk.api.batch.execute_batch.BatchRequest",
     unconverted_values={"data"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class BatchRequest:
     path: str
     method: RequestMethod
@@ -54,7 +54,7 @@ class BatchRequest:
 @serial_class(
     named_type_path="sdk.api.batch.execute_batch.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     requests: list[BatchRequest]
 
@@ -64,7 +64,7 @@ class Arguments:
     named_type_path="sdk.api.batch.execute_batch.BatchResponse",
     unconverted_values={"response"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class BatchResponse:
     path: str
     method: RequestMethod
@@ -76,7 +76,7 @@ class BatchResponse:
 @serial_class(
     named_type_path="sdk.api.batch.execute_batch.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     responses: list[BatchResponse]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/batch/execute_batch_load_async.py

@@ -26,7 +26,7 @@ ENDPOINT_PATH = "api/external/batch/execute_batch_load_async"
 @serial_class(
     named_type_path="sdk.api.batch.execute_batch_load_async.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     requests: list[async_batch_t.AsyncBatchRequest]
 
@@ -35,7 +35,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.batch.execute_batch_load_async.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     job_id: base_t.ObjectId
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/chemical/convert_chemical_formats.py

@@ -28,7 +28,7 @@ ENDPOINT_PATH = "api/external/chemical/convert_chemical_formats"
 @serial_class(
     named_type_path="sdk.api.chemical.convert_chemical_formats.ChemicalStructureFile",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class ChemicalStructureFile:
     struct_file: str
 
@@ -37,7 +37,7 @@ class ChemicalStructureFile:
 @serial_class(
     named_type_path="sdk.api.chemical.convert_chemical_formats.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     source_chemical_structures: list[ChemicalStructureFile]
 
@@ -56,7 +56,7 @@ UncountableChemicalStructure = typing.Annotated[
     named_type_path="sdk.api.chemical.convert_chemical_formats.Data",
     unconverted_values={"chemical_structures"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     chemical_structures: list[UncountableChemicalStructure]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/condition_parameters/__init__.py

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

uncountable/types/api/condition_parameters/upsert_condition_match.py

@@ -0,0 +1,72 @@
+# 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
+from enum import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import base_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "ConditionParameter",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "MatchType",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/condition_parameters/upsert_condition_match"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_match.ConditionParameter",
+    to_string_values={"max_value", "min_value"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ConditionParameter:
+    condition_parameter_key: identifier_t.IdentifierKey
+    allow_null: bool
+    min_value: Decimal | None = None
+    max_value: Decimal | None = None
+    categorical_values: list[str | None] | None = None
+    match_type: MatchType | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class MatchType(StrEnum):
+    ALL = "all"
+    EQ = "eq"
+    EQCATEGORICAL = "eqCategorical"
+    RANGE = "range"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_match.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    explicit_name: bool = False
+    name: str | None = None
+    condition_parameters: list[ConditionParameter] | None = None
+    output_conditions: list[identifier_t.IdentifierKey] | None = None
+    existing_condition_match: identifier_t.IdentifierKey | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_match.Data",
+)
+@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/api/entity/create_entities.py

@@ -29,7 +29,7 @@ ENDPOINT_PATH = "api/external/entity/external_create_entities"
 @serial_class(
     named_type_path="sdk.api.entity.create_entities.EntityToCreate",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class EntityToCreate:
     field_values: list[field_values_t.FieldRefNameValue] | None = None
     parent_entity: entity_t.Entity | None = None
@@ -39,7 +39,7 @@ class EntityToCreate:
 @serial_class(
     named_type_path="sdk.api.entity.create_entities.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_type: entity_t.LimitedEntityType
     entities_to_create: list[EntityToCreate]
@@ -51,7 +51,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.create_entities.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     entities: list[entity_t.Entity]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entity.py

@@ -30,7 +30,7 @@ ENDPOINT_PATH = "api/external/entity/external_create_entity"
     named_type_path="sdk.api.entity.create_entity.EntityFieldInitialValue",
     unconverted_values={"value"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class EntityFieldInitialValue:
     field_ref_name: str
     value: base_t.JsonValue
@@ -41,7 +41,7 @@ class EntityFieldInitialValue:
 @serial_class(
     named_type_path="sdk.api.entity.create_entity.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_type: entity_t.LimitedEntityType
     definition_id: base_t.ObjectId | None = None
@@ -54,7 +54,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.create_entity.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     entity: entity_t.Entity
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_or_update_entity.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
 from ... import async_batch_t
+from ... import base_t
 from ... import entity_t
 from ... import field_values_t
 from ... import identifier_t
@@ -28,7 +29,7 @@ ENDPOINT_PATH = "api/external/entity/create_or_update_entity"
 @serial_class(
     named_type_path="sdk.api.entity.create_or_update_entity.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_type: entity_t.EntityType
     definition_key: identifier_t.IdentifierKey
@@ -40,7 +41,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.create_or_update_entity.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/api/entity/get_entities_data.py

@@ -28,7 +28,7 @@ ENDPOINT_PATH = "api/external/entity/external_get_entities_data"
 @serial_class(
     named_type_path="sdk.api.entity.get_entities_data.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_ids: list[base_t.ObjectId]
     entity_type: entity_t.EntityType
@@ -38,7 +38,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.get_entities_data.EntityDetails",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class EntityDetails:
     entity: entity_t.Entity
     field_values: list[field_values_t.FieldRefIdNameValue]
@@ -48,7 +48,7 @@ class EntityDetails:
 @serial_class(
     named_type_path="sdk.api.entity.get_entities_data.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     entity_details: list[EntityDetails]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/grant_entity_permissions.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
 from ... import async_batch_t
+from ... import base_t
 from ... import entity_t
 from ... import identifier_t
 
@@ -27,7 +28,7 @@ ENDPOINT_PATH = "api/external/entity/grant_entity_permissions"
 @serial_class(
     named_type_path="sdk.api.entity.grant_entity_permissions.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_type: entity_t.LimitedEntityType
     entity_key: identifier_t.IdentifierKey
@@ -41,7 +42,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.grant_entity_permissions.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/api/entity/list_entities.py

@@ -30,7 +30,7 @@ ENDPOINT_PATH = "api/external/entity/external_list_entities"
     named_type_path="sdk.api.entity.list_entities.Arguments",
     unconverted_values={"attributes"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     config_reference: str
     entity_type: entity_t.EntityType | None = None
@@ -44,7 +44,7 @@ class Arguments:
     named_type_path="sdk.api.entity.list_entities.EntityResult",
     unconverted_values={"column_values"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class EntityResult:
     entity: entity_t.Entity
     column_values: list[base_t.JsonValue]
@@ -54,7 +54,7 @@ class EntityResult:
 @serial_class(
     named_type_path="sdk.api.entity.list_entities.ColumnAccess",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class ColumnAccess:
     name: str
     table_label: str | None
@@ -64,7 +64,7 @@ class ColumnAccess:
 @serial_class(
     named_type_path="sdk.api.entity.list_entities.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     columns: list[ColumnAccess]
     results: list[EntityResult]

uncountable/types/api/entity/lock_entity.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
 from ... import async_batch_t
+from ... import base_t
 from ... import entity_t
 from ... import identifier_t
 
@@ -27,7 +28,7 @@ ENDPOINT_PATH = "api/external/entity/external_lock_entity"
 @serial_class(
     named_type_path="sdk.api.entity.lock_entity.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_key: identifier_t.IdentifierKey
     entity_type: entity_t.EntityType
@@ -38,7 +39,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.lock_entity.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/api/entity/lookup_entity.py

@@ -41,7 +41,7 @@ class LookupEntityQueryType(StrEnum):
 @serial_class(
     named_type_path="sdk.api.entity.lookup_entity.LookupEntityQueryBase",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class LookupEntityQueryBase:
     type: LookupEntityQueryType
 
@@ -51,7 +51,7 @@ class LookupEntityQueryBase:
     named_type_path="sdk.api.entity.lookup_entity.LookupFieldArgumentValue",
     unconverted_values={"field_value"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class LookupFieldArgumentValue:
     field_key: identifier_t.IdentifierKey
     field_value: base_t.JsonValue
@@ -63,7 +63,7 @@ class LookupFieldArgumentValue:
     named_type_path="sdk.api.entity.lookup_entity.LookupEntityFieldValue",
     parse_require={"type"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class LookupEntityFieldValue:
     type: typing.Literal[LookupEntityQueryType.FIELD_VALUE] = LookupEntityQueryType.FIELD_VALUE
     value: LookupFieldArgumentValue
@@ -86,7 +86,7 @@ LookupEntityQuery = typing.Annotated[
 @serial_class(
     named_type_path="sdk.api.entity.lookup_entity.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_type: entity_t.EntityType
     query: LookupEntityQuery
@@ -96,7 +96,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.lookup_entity.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/api/entity/resolve_entity_ids.py

@@ -27,7 +27,7 @@ ENDPOINT_PATH = "api/external/entity/external_resolve_entity_ids"
 @serial_class(
     named_type_path="sdk.api.entity.resolve_entity_ids.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_ids: list[str | base_t.ObjectId]
     entity_type: entity_t.EntityType
@@ -37,7 +37,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.resolve_entity_ids.EntityNames",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class EntityNames:
     id: str | int
     name: str
@@ -47,7 +47,7 @@ class EntityNames:
 @serial_class(
     named_type_path="sdk.api.entity.resolve_entity_ids.Data",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Data:
     items: list[EntityNames]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/set_entity_field_values.py

@@ -9,6 +9,7 @@ from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
 from ... import async_batch_t
+from ... import base_t
 from ... import entity_t
 from ... import field_values_t
 
@@ -27,7 +28,7 @@ ENDPOINT_PATH = "api/external/entity/set_entity_field_values"
 @serial_class(
     named_type_path="sdk.api.entity.set_entity_field_values.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity_identifier: entity_t.EntityIdentifier
     field_values: list[field_values_t.FieldArgumentValue]
@@ -37,7 +38,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.set_entity_field_values.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/api/entity/set_values.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 field_values_t
 from ... import response_t
@@ -27,7 +28,7 @@ ENDPOINT_PATH = "api/external/entity/external_set_values"
 @serial_class(
     named_type_path="sdk.api.entity.set_values.Arguments",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     entity: entity_t.Entity
     values: list[field_values_t.ArgumentValueRefName]
@@ -37,7 +38,7 @@ class Arguments:
 @serial_class(
     named_type_path="sdk.api.entity.set_values.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