UncountablePythonSDK 0.0.72__py3-none-any.whl → 0.0.73__py3-none-any.whl

uncountable/integration/queue_runner/worker.py

@@ -102,8 +102,12 @@ def run_queued_job(
             payload=payload,
         )
 
-        return execute_job(
-            args=args,
-            profile_metadata=job_details.profile_metadata,
-            job_definition=job_details.job_definition,
-        )
+        try:
+            return execute_job(
+                args=args,
+                profile_metadata=job_details.profile_metadata,
+                job_definition=job_details.job_definition,
+            )
+        except Exception as e:
+            job_logger.log_exception(e)
+            return job_definition_t.JobResult(success=False)

uncountable/integration/server.py

@@ -12,13 +12,11 @@ from opentelemetry.trace import get_current_span
 from sqlalchemy.engine.base import Engine
 
 from uncountable.integration.cron import CronJobArgs, cron_job_executor
+from uncountable.integration.scan_profiles import ProfileDetails
 from uncountable.integration.telemetry import Logger
 from uncountable.types import base_t
-from uncountable.types.client_config_t import ClientConfigOptions
 from uncountable.types.job_definition_t import (
-    AuthRetrieval,
     CronJobDefinition,
-    JobDefinition,
     ProfileMetadata,
     WebhookJobDefinition,
 )
@@ -40,71 +38,70 @@ class IntegrationServer:
         )
         self._server_logger = Logger(get_current_span())
 
-    def register_profile(
-        self,
-        *,
-        profile_name: str,
-        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,
-                client_options=client_options,
-            )
-            match job_defn:
-                case CronJobDefinition():
-                    # Add to ap scheduler
-                    job_kwargs = asdict(
-                        CronJobArgs(
-                            definition=job_defn, profile_metadata=profile_metadata
-                        )
-                    )
-                    try:
-                        existing_job = self._scheduler.get_job(job_defn.id)
-                    except ValueError as e:
-                        self._server_logger.log_warning(
-                            f"could not reconstitute job {job_defn.id}: {e}"
+    def register_jobs(self, profiles: list[ProfileDetails]) -> None:
+        valid_job_ids = []
+        for profile in profiles:
+            for job_defn in profile.definition.jobs:
+                valid_job_ids.append(job_defn.id)
+                profile_metadata = ProfileMetadata(
+                    name=profile.name,
+                    auth_retrieval=profile.definition.auth_retrieval,
+                    base_url=profile.definition.base_url,
+                    client_options=profile.definition.client_options,
+                )
+                match job_defn:
+                    case CronJobDefinition():
+                        # Add to ap scheduler
+                        job_kwargs = asdict(
+                            CronJobArgs(
+                                definition=job_defn, profile_metadata=profile_metadata
+                            )
                         )
-                        self._scheduler.remove_job(job_defn.id)
-                        existing_job = None
-                    if existing_job is not None:
-                        existing_job.modify(
-                            name=job_defn.name,
-                            kwargs=job_kwargs,
-                            misfire_grace_time=None,
-                        )
-                        existing_job.reschedule(
-                            CronTrigger.from_crontab(job_defn.cron_spec)
-                        )
-                        if not job_defn.enabled:
-                            existing_job.pause()
+                        try:
+                            existing_job = self._scheduler.get_job(job_defn.id)
+                        except ValueError as e:
+                            self._server_logger.log_warning(
+                                f"could not reconstitute job {job_defn.id}: {e}"
+                            )
+                            self._scheduler.remove_job(job_defn.id)
+                            existing_job = None
+                        if existing_job is not None:
+                            existing_job.modify(
+                                name=job_defn.name,
+                                kwargs=job_kwargs,
+                                misfire_grace_time=None,
+                            )
+                            existing_job.reschedule(
+                                CronTrigger.from_crontab(job_defn.cron_spec)
+                            )
+                            if not job_defn.enabled:
+                                existing_job.pause()
+                            else:
+                                existing_job.resume()
                         else:
-                            existing_job.resume()
-                    else:
-                        job_opts: dict[str, base_t.JsonValue] = {}
-                        if not job_defn.enabled:
-                            job_opts["next_run_time"] = None
-                        self._scheduler.add_job(
-                            cron_job_executor,
-                            # IMPROVE: reconsider these defaults
-                            max_instances=1,
-                            coalesce=True,
-                            trigger=CronTrigger.from_crontab(job_defn.cron_spec),
-                            name=job_defn.name,
-                            id=job_defn.id,
-                            kwargs=job_kwargs,
-                            misfire_grace_time=None,
-                            **job_opts,
-                        )
-                case WebhookJobDefinition():
-                    pass
-                case _:
-                    assert_never(job_defn)
+                            job_opts: dict[str, base_t.JsonValue] = {}
+                            if not job_defn.enabled:
+                                job_opts["next_run_time"] = None
+                            self._scheduler.add_job(
+                                cron_job_executor,
+                                # IMPROVE: reconsider these defaults
+                                max_instances=1,
+                                coalesce=True,
+                                trigger=CronTrigger.from_crontab(job_defn.cron_spec),
+                                name=job_defn.name,
+                                id=job_defn.id,
+                                kwargs=job_kwargs,
+                                misfire_grace_time=None,
+                                **job_opts,
+                            )
+                    case WebhookJobDefinition():
+                        pass
+                    case _:
+                        assert_never(job_defn)
+        all_jobs = self._scheduler.get_jobs()
+        for job in all_jobs:
+            if job.id not in valid_job_ids:
+                self._scheduler.remove_job(job.id)
 
     def serve_forever(self) -> None:
         signal.pause()

uncountable/types/api/batch/execute_batch.py

@@ -41,6 +41,7 @@ class RequestMethod(StrEnum):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.batch.execute_batch.BatchRequest",
     unconverted_values={"data"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -51,6 +52,9 @@ class BatchRequest:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.batch.execute_batch.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     requests: list[BatchRequest]
@@ -58,6 +62,7 @@ class Arguments:
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.batch.execute_batch.BatchResponse",
     unconverted_values={"response"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -69,6 +74,9 @@ class BatchResponse:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.batch.execute_batch.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     responses: list[BatchResponse]

uncountable/types/api/batch/execute_batch_load_async.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import async_batch_t
 from ... import base_t
 
@@ -23,12 +24,18 @@ ENDPOINT_PATH = "api/external/batch/execute_batch_load_async"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.batch.execute_batch_load_async.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     requests: list[async_batch_t.AsyncBatchRequest]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.batch.execute_batch_load_async.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     job_id: base_t.ObjectId

uncountable/types/api/chemical/convert_chemical_formats.py

@@ -25,12 +25,18 @@ ENDPOINT_PATH = "api/external/chemical/convert_chemical_formats"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.chemical.convert_chemical_formats.ChemicalStructureFile",
+)
 @dataclasses.dataclass(kw_only=True)
 class ChemicalStructureFile:
     struct_file: str
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.chemical.convert_chemical_formats.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     source_chemical_structures: list[ChemicalStructureFile]
@@ -42,6 +48,7 @@ UncountableChemicalStructure = dict[str, base_t.JsonValue]
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.chemical.convert_chemical_formats.Data",
     unconverted_values={"chemical_structures"},
 )
 @dataclasses.dataclass(kw_only=True)

uncountable/types/api/entity/create_entities.py

@@ -8,6 +8,7 @@ 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
 from ... import entity_t
 from ... import field_values_t
@@ -25,20 +26,29 @@ ENDPOINT_PATH = "api/external/entity/external_create_entities"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.create_entities.EntityToCreate",
+)
 @dataclasses.dataclass(kw_only=True)
 class EntityToCreate:
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.create_entities.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG]]
+    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.OUTPUT]]
     entities_to_create: list[EntityToCreate]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.create_entities.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     entities: list[entity_t.Entity]

uncountable/types/api/entity/create_entity.py

@@ -27,6 +27,7 @@ ENDPOINT_PATH = "api/external/entity/external_create_entity"
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.entity.create_entity.EntityFieldInitialValue",
     unconverted_values={"value"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -37,14 +38,20 @@ class EntityFieldInitialValue:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.create_entity.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG]]
+    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.OUTPUT]]
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.create_entity.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     entity: entity_t.Entity

uncountable/types/api/entity/get_entities_data.py

@@ -8,6 +8,7 @@ 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
 from ... import entity_t
 from ... import field_values_t
@@ -25,6 +26,9 @@ ENDPOINT_PATH = "api/external/entity/external_get_entities_data"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.get_entities_data.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_ids: list[base_t.ObjectId]
@@ -32,6 +36,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.get_entities_data.EntityDetails",
+)
 @dataclasses.dataclass(kw_only=True)
 class EntityDetails:
     entity: entity_t.Entity
@@ -39,6 +46,9 @@ class EntityDetails:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.get_entities_data.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     entity_details: list[EntityDetails]

uncountable/types/api/entity/list_entities.py

@@ -28,6 +28,7 @@ ENDPOINT_PATH = "api/external/entity/external_list_entities"
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.entity.list_entities.Arguments",
     unconverted_values={"attributes"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -41,6 +42,7 @@ class Arguments:
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.entity.list_entities.EntityResult",
     unconverted_values={"column_values"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -50,6 +52,9 @@ class EntityResult:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_entities.ColumnAccess",
+)
 @dataclasses.dataclass(kw_only=True)
 class ColumnAccess:
     name: str
@@ -57,6 +62,9 @@ class ColumnAccess:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_entities.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     columns: list[ColumnAccess]

uncountable/types/api/entity/lock_entity.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import async_batch_t
 from ... import entity_t
 from ... import identifier_t
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/entity/external_lock_entity"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.lock_entity.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_key: identifier_t.IdentifierKey
@@ -32,6 +36,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.lock_entity.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass

uncountable/types/api/entity/resolve_entity_ids.py

@@ -8,6 +8,7 @@ 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
 from ... import entity_t
 
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/entity/external_resolve_entity_ids"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.resolve_entity_ids.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_ids: list[typing.Union[str, base_t.ObjectId]]
@@ -31,6 +35,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.resolve_entity_ids.EntityNames",
+)
 @dataclasses.dataclass(kw_only=True)
 class EntityNames:
     id: typing.Union[str, int]
@@ -38,6 +45,9 @@ class EntityNames:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.resolve_entity_ids.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     items: list[EntityNames]

uncountable/types/api/entity/set_values.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import entity_t
 from ... import field_values_t
 from ... import response_t
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/entity/external_set_values"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_values.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity: entity_t.Entity
@@ -31,6 +35,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_values.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data(response_t.Response):
     pass

uncountable/types/api/entity/transition_entity_phase.py

@@ -29,6 +29,7 @@ ENDPOINT_PATH = "api/external/entity/transition_entity_phase"
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.entity.transition_entity_phase.TransitionIdentifierPhases",
     parse_require={"type"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -40,6 +41,7 @@ class TransitionIdentifierPhases:
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.api.entity.transition_entity_phase.TransitionIdentifierTransition",
     parse_require={"type"},
 )
 @dataclasses.dataclass(kw_only=True)
@@ -53,6 +55,9 @@ TransitionIdentifier = typing.Union[TransitionIdentifierPhases, TransitionIdenti
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.transition_entity_phase.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity: entity_t.Entity
@@ -60,6 +65,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.transition_entity_phase.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data(response_t.Response):
     pass

uncountable/types/api/entity/unlock_entity.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import async_batch_t
 from ... import entity_t
 from ... import identifier_t
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/entity/external_unlock_entity"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.unlock_entity.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     entity_key: identifier_t.IdentifierKey
@@ -31,6 +35,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.unlock_entity.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass

uncountable/types/api/equipment/associate_equipment_input.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import async_batch_t
 from ... import base_t
 from ... import identifier_t
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/equipment/associate_equipment_input"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.equipment.associate_equipment_input.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     equipment_key: identifier_t.IdentifierKey
@@ -31,6 +35,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.equipment.associate_equipment_input.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data(async_batch_t.AsyncBatchActionReturn):
     pass

uncountable/types/api/field_options/upsert_field_options.py

@@ -8,6 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -23,6 +24,9 @@ ENDPOINT_PATH = "api/external/field_options/upsert_field_options"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.field_options.upsert_field_options.FieldOption",
+)
 @dataclasses.dataclass(kw_only=True)
 class FieldOption:
     option_id: str
@@ -32,6 +36,9 @@ class FieldOption:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.field_options.upsert_field_options.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     option_set_key: identifier_t.IdentifierKey
@@ -39,6 +46,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.field_options.upsert_field_options.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     pass

uncountable/types/api/id_source/list_id_source.py

@@ -8,6 +8,7 @@ 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
 from ... import id_source_t
 
@@ -24,6 +25,9 @@ ENDPOINT_PATH = "api/external/id_source/list_id_source"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.id_source.list_id_source.Arguments",
+)
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
     spec: id_source_t.IdSourceSpec
@@ -33,6 +37,9 @@ class Arguments:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.id_source.list_id_source.IdName",
+)
 @dataclasses.dataclass(kw_only=True)
 class IdName:
     id: typing.Union[base_t.ObjectId, str]
@@ -40,6 +47,9 @@ class IdName:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.id_source.list_id_source.Data",
+)
 @dataclasses.dataclass(kw_only=True)
 class Data:
     results: list[IdName]