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

uncountable/integration/cli.py

@@ -2,20 +2,17 @@ import argparse
 
 from opentelemetry.trace import get_current_span
 
-from uncountable.core.async_batch import AsyncBatchProcessor
-from uncountable.integration.construct_client import construct_uncountable_client
-from uncountable.integration.executors.executors import execute_job
-from uncountable.integration.job import CronJobArguments
-from uncountable.integration.scan_profiles import load_profiles
-from uncountable.integration.telemetry import JobLogger, Logger
-from uncountable.types import job_definition_t
+from uncountable.core.environment import get_local_admin_server_port
+from uncountable.integration.queue_runner.command_server.command_client import (
+    send_job_queue_message,
+)
+from uncountable.integration.telemetry import Logger
+from uncountable.types import queued_job_t
 
 
 def main() -> None:
     logger = Logger(get_current_span())
 
-    profiles = load_profiles()
-
     parser = argparse.ArgumentParser(
         description="Process a job with a given command and job ID."
     )
@@ -29,45 +26,22 @@ def main() -> None:
 
     parser.add_argument("job_id", type=str, help="The ID of the job to process")
 
+    parser.add_argument(
+        "--host", type=str, default="localhost", nargs="?", help="The host to run on"
+    )
+
     args = parser.parse_args()
 
     with logger.push_scope(args.command):
         if args.command == "run":
-            job_result: job_definition_t.JobResult | None = None
-            for profile in profiles:
-                for job in profile.definition.jobs:
-                    if job.id == args.job_id:
-                        profile_meta = job_definition_t.ProfileMetadata(
-                            name=profile.name,
-                            base_url=profile.definition.base_url,
-                            auth_retrieval=profile.definition.auth_retrieval,
-                            client_options=profile.definition.client_options,
-                        )
-                        job_logger = JobLogger(
-                            base_span=logger.current_span,
-                            profile_metadata=profile_meta,
-                            job_definition=job,
-                        )
-                        client = construct_uncountable_client(
-                            profile_meta=profile_meta, job_logger=job_logger
-                        )
-                        batch_processor = AsyncBatchProcessor(client=client)
-                        job_args = CronJobArguments(
-                            job_definition=job,
-                            profile_metadata=profile_meta,
-                            client=client,
-                            batch_processor=batch_processor,
-                            logger=job_logger,
-                            payload=None,
-                        )
-                        job_result = execute_job(
-                            job_definition=job,
-                            profile_metadata=profile_meta,
-                            args=job_args,
-                        )
-                        break
-            if job_result is None:
-                raise Exception(f"no such job id {args.job_id}")
+            send_job_queue_message(
+                job_ref_name=args.job_id,
+                payload=queued_job_t.QueuedJobPayload(
+                    invocation_context=queued_job_t.InvocationContextManual()
+                ),
+                host=args.host,
+                port=get_local_admin_server_port(),
+            )
         else:
             parser.print_usage()
 

uncountable/integration/entrypoint.py

@@ -5,15 +5,7 @@ from uncountable.integration.server import IntegrationServer
 
 def main() -> None:
     with IntegrationServer(create_db_engine(IntegrationDBService.CRON)) as server:
-        for profile_details in load_profiles():
-            server.register_profile(
-                profile_name=profile_details.name,
-                base_url=profile_details.definition.base_url,
-                auth_retrieval=profile_details.definition.auth_retrieval,
-                jobs=profile_details.definition.jobs,
-                client_options=profile_details.definition.client_options,
-            )
-
+        server.register_jobs(load_profiles())
         server.serve_forever()
 
 

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