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

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
@@ -28,6 +29,7 @@ from uncountable.types import recipe_metadata_t
 from uncountable.types import recipe_workflow_steps_t
 import uncountable.types.api.entity.set_entity_field_values as set_entity_field_values_t
 import uncountable.types.api.recipes.set_recipe_metadata as set_recipe_metadata_t
+import uncountable.types.api.condition_parameters.upsert_condition_match as upsert_condition_match_t
 import uuid
 from abc import ABC, abstractmethod
 from pkgs.serialization_util import serialize_for_api
@@ -182,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,
         *,
@@ -489,3 +524,42 @@ class AsyncBatchProcessorBase(ABC):
             path=req.path,
             batch_reference=req.batch_reference,
         )
+
+    def upsert_condition_match(
+        self,
+        *,
+        explicit_name: bool = False,
+        name: str | None = None,
+        condition_parameters: list[upsert_condition_match_t.ConditionParameter] | None = None,
+        output_conditions: list[identifier_t.IdentifierKey] | None = None,
+        existing_condition_match: identifier_t.IdentifierKey | None = None,
+        depends_on: list[str] | None = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Creates or updates condition match
+
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = upsert_condition_match_t.Arguments(
+            name=name,
+            condition_parameters=condition_parameters,
+            output_conditions=output_conditions,
+            explicit_name=explicit_name,
+            existing_condition_match=existing_condition_match,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.UPSERT_CONDITION_MATCH,
+            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,
+        )

uncountable/types/async_batch_t.py

@@ -42,6 +42,8 @@ class AsyncBatchRequestPath(StrEnum):
     ADD_TIME_SERIES_DATA = "recipes/add_time_series_data"
     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
@@ -49,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
@@ -62,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
@@ -72,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
@@ -84,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
@@ -94,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
@@ -88,6 +89,7 @@ import uncountable.types.api.recipes.unarchive_recipes as unarchive_recipes_t
 import uncountable.types.api.entity.unlock_entity as unlock_entity_t
 import uncountable.types.api.recipes.unlock_recipes as unlock_recipes_t
 import uncountable.types.api.material_families.update_entity_material_families as update_entity_material_families_t
+import uncountable.types.api.condition_parameters.upsert_condition_match as upsert_condition_match_t
 import uncountable.types.api.field_options.upsert_field_options as upsert_field_options_t
 from abc import ABC, abstractmethod
 import dataclasses
@@ -265,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,
         *,
@@ -1673,6 +1695,32 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=update_entity_material_families_t.Data)
 
+    def upsert_condition_match(
+        self,
+        *,
+        explicit_name: bool = False,
+        name: str | None = None,
+        condition_parameters: list[upsert_condition_match_t.ConditionParameter] | None = None,
+        output_conditions: list[identifier_t.IdentifierKey] | None = None,
+        existing_condition_match: identifier_t.IdentifierKey | None = None,
+    ) -> upsert_condition_match_t.Data:
+        """Creates or updates condition match
+
+        """
+        args = upsert_condition_match_t.Arguments(
+            name=name,
+            condition_parameters=condition_parameters,
+            output_conditions=output_conditions,
+            explicit_name=explicit_name,
+            existing_condition_match=existing_condition_match,
+        )
+        api_request = APIRequest(
+            method=upsert_condition_match_t.ENDPOINT_METHOD,
+            endpoint=upsert_condition_match_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=upsert_condition_match_t.Data)
+
     def upsert_field_options(
         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

uncountable/types/curves_t.py

@@ -28,7 +28,7 @@ class CurveAxisType(StrEnum):
 @serial_class(
     named_type_path="sdk.curves.Curve",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Curve:
     id: base_t.ObjectId
     x_type: typing.Literal[CurveAxisType.NUMERIC] | typing.Literal[CurveAxisType.TIMESTAMP]
@@ -42,7 +42,7 @@ class Curve:
     named_type_path="sdk.curves.CurveValue",
     to_string_values={"quantity"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class CurveValue:
     index: int
     quantity: Decimal | None = None

uncountable/types/data_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] = [
     "CielabColor",
@@ -24,11 +25,11 @@ __all__: list[str] = [
     named_type_path="sdk.data.RgbColor",
     unconverted_keys={"B", "G", "R"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RgbColor:
-    R: int
-    G: int
-    B: int
+    R: Decimal
+    G: Decimal
+    B: Decimal
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -36,11 +37,11 @@ class RgbColor:
     named_type_path="sdk.data.CielabColor",
     unconverted_keys={"L", "a", "b"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class CielabColor:
-    L: int
-    a: int
-    b: int
+    L: Decimal
+    a: Decimal
+    b: Decimal
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -48,11 +49,11 @@ class CielabColor:
     named_type_path="sdk.data.XyzColor",
     unconverted_keys={"X", "Y", "Z"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class XyzColor:
-    X: int
-    Y: int
-    Z: int
+    X: Decimal
+    Y: Decimal
+    Z: Decimal
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -60,11 +61,11 @@ class XyzColor:
     named_type_path="sdk.data.LChColor",
     unconverted_keys={"C", "L", "h"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class LChColor:
-    L: int
-    C: int
-    h: int
+    L: Decimal
+    C: Decimal
+    h: Decimal
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -72,11 +73,11 @@ class LChColor:
     named_type_path="sdk.data.HslColor",
     unconverted_keys={"L", "h", "s"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class HslColor:
-    h: int
-    s: int
-    L: int
+    h: Decimal
+    s: Decimal
+    L: Decimal
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -84,7 +85,7 @@ class HslColor:
     named_type_path="sdk.data.RecipeOutputColor",
     unconverted_keys={"CIELAB", "LCH", "RGB", "XYZ"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeOutputColor:
     RGB: RgbColor
     CIELAB: CielabColor

uncountable/types/entity_t.py

@@ -39,8 +39,10 @@ __all__: list[str] = [
         "characterization": "Characterization",
         "comment": "Comment",
         "comment_thread": "Comment Thread",
+        "condition_match": "Condition Match",
         "condition_parameter": "Condition Parameter",
         "condition_parameter_mat_family": "Condition Parameter Material Family",
+        "condition_parameter_rule": "Condition Parameter Rule",
         "condition_parameter_value": "Condition Parameter Value",
         "constraint": "Constraint",
         "constraint_set": "Constraint Set",
@@ -113,6 +115,7 @@ __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",
+        "output_group_member": "Output Group Member",
         "output_preferences": "Output Preferences",
         "permission": "Permission",
         "permissions_recipe_link_inheritance": "Permissions Recipe Link Inheritance",
@@ -217,8 +220,10 @@ class EntityType(StrEnum):
     CHARACTERIZATION = "characterization"
     COMMENT = "comment"
     COMMENT_THREAD = "comment_thread"
+    CONDITION_MATCH = "condition_match"
     CONDITION_PARAMETER = "condition_parameter"
     CONDITION_PARAMETER_MAT_FAMILY = "condition_parameter_mat_family"
+    CONDITION_PARAMETER_RULE = "condition_parameter_rule"
     CONDITION_PARAMETER_VALUE = "condition_parameter_value"
     CONSTRAINT = "constraint"
     CONSTRAINT_SET = "constraint_set"
@@ -291,6 +296,7 @@ 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"
+    OUTPUT_GROUP_MEMBER = "output_group_member"
     OUTPUT_PREFERENCES = "output_preferences"
     PERMISSION = "permission"
     PERMISSIONS_RECIPE_LINK_INHERITANCE = "permissions_recipe_link_inheritance"
@@ -384,7 +390,7 @@ class EntityType(StrEnum):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 LimitedEntityType = typing.Annotated[
-    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INGREDIENT_ATTRIBUTE] | 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] | typing.Literal[EntityType.CONSTRAINT] | typing.Literal[EntityType.INGREDIENT_CATEGORY_ALL] | typing.Literal[EntityType.TIME_SERIES_SEGMENT] | typing.Literal[EntityType.EQUIPMENT_MAINTENANCE] | typing.Literal[EntityType.MAINTENANCE_SCHEDULE],
+    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INGREDIENT_ATTRIBUTE] | 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] | typing.Literal[EntityType.CONSTRAINT] | typing.Literal[EntityType.INGREDIENT_CATEGORY_ALL] | typing.Literal[EntityType.TIME_SERIES_SEGMENT] | typing.Literal[EntityType.EQUIPMENT_MAINTENANCE] | typing.Literal[EntityType.MAINTENANCE_SCHEDULE] | typing.Literal[EntityType.CONDITION_PARAMETER_RULE] | typing.Literal[EntityType.INGREDIENT],
     serial_alias_annotation(
     named_type_path="sdk.entity.LimitedEntityType",
     ),
@@ -413,7 +419,7 @@ GrantableEntityPermissionType = typing.Annotated[
 @serial_class(
     named_type_path="sdk.entity.Entity",
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Entity:
     id: base_t.ObjectId
     type: EntityType
@@ -423,7 +429,7 @@ class Entity:
 @serial_class(
     named_type_path="sdk.entity.EntityIdentifier",
 )
-@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 EntityIdentifier:
     identifier_key: identifier_t.IdentifierKey
     type: EntityType

uncountable/types/experiment_groups_t.py

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