UncountablePythonSDK 0.0.115__py3-none-any.whl → 0.0.142.dev0__py3-none-any.whl

uncountable/integration/telemetry.py

@@ -1,17 +1,24 @@
 import functools
+import json
 import os
+import threading
 import time
 import traceback
+import types
 import typing
 from contextlib import contextmanager
 from enum import StrEnum
 from typing import Generator, assert_never, cast
 
+import psutil
 from opentelemetry import _logs, trace
 from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.sdk._logs import Logger as OTELLogger
-from opentelemetry.sdk._logs import LoggerProvider, LogRecord
+from opentelemetry.sdk._logs import (
+    LoggerProvider,
+    LogRecord,
+)
 from opentelemetry.sdk._logs.export import BatchLogRecordProcessor, ConsoleLogExporter
 from opentelemetry.sdk.resources import Attributes, Resource
 from opentelemetry.sdk.trace import TracerProvider
@@ -32,6 +39,11 @@ def _cast_attributes(attributes: dict[str, base_t.JsonValue]) -> Attributes:
     return cast(Attributes, attributes)
 
 
+def one_line_formatter(record: LogRecord) -> str:
+    json_data = record.to_json()
+    return json.dumps(json.loads(json_data), separators=(",", ":")) + "\n"
+
+
 @functools.cache
 def get_otel_resource() -> Resource:
     attributes: dict[str, base_t.JsonValue] = {
@@ -60,7 +72,9 @@ def get_otel_tracer() -> Tracer:
 @functools.cache
 def get_otel_logger() -> OTELLogger:
     provider = LoggerProvider(resource=get_otel_resource())
-    provider.add_log_record_processor(BatchLogRecordProcessor(ConsoleLogExporter()))
+    provider.add_log_record_processor(
+        BatchLogRecordProcessor(ConsoleLogExporter(formatter=one_line_formatter))
+    )
     if get_otel_enabled():
         provider.add_log_record_processor(BatchLogRecordProcessor(OTLPLogExporter()))
     _logs.set_logger_provider(provider)
@@ -87,8 +101,27 @@ class Logger:
     def current_trace_id(self) -> int | None:
         return self.current_span.get_span_context().trace_id
 
-    def _patch_attributes(self, attributes: Attributes | None) -> Attributes:
-        return attributes or {}
+    def _patch_attributes(
+        self,
+        attributes: Attributes | None,
+        *,
+        message: str | None = None,
+        severity: LogSeverity | None = None,
+    ) -> Attributes:
+        patched_attributes = {**(attributes if attributes is not None else {})}
+        if message is not None:
+            patched_attributes["message"] = message
+        elif "body" in patched_attributes:
+            patched_attributes["message"] = patched_attributes["body"]
+
+        if severity is not None:
+            patched_attributes["status"] = severity.lower()
+        elif "severity_text" in patched_attributes and isinstance(
+            patched_attributes["severity_text"], str
+        ):
+            patched_attributes["status"] = patched_attributes["severity_text"].lower()
+
+        return patched_attributes
 
     def _emit_log(
         self, message: str, *, severity: LogSeverity, attributes: Attributes | None
@@ -98,7 +131,9 @@ class Logger:
             body=message,
             severity_text=severity,
             timestamp=time.time_ns(),
-            attributes=self._patch_attributes(attributes),
+            attributes=self._patch_attributes(
+                message=message, severity=severity, attributes=attributes
+            ),
             span_id=self.current_span_id,
             trace_id=self.current_trace_id,
             trace_flags=DEFAULT_TRACE_OPTIONS,
@@ -132,7 +167,9 @@ class Logger:
         attributes: Attributes | None = None,
     ) -> None:
         traceback_str = "".join(traceback.format_exception(exception))
-        patched_attributes = self._patch_attributes(attributes)
+        patched_attributes = self._patch_attributes(
+            message=message, severity=LogSeverity.ERROR, attributes=attributes
+        )
         self.current_span.record_exception(
             exception=exception, attributes=patched_attributes
         )
@@ -151,6 +188,76 @@ class Logger:
             yield self
 
 
+class PerJobResourceTracker:
+    def __init__(self, logger: "JobLogger", sample_interval: float = 0.5) -> None:
+        self.logger = logger
+        self.sample_interval = sample_interval
+        self._process = psutil.Process(os.getpid())
+        self._stop_event = threading.Event()
+        self._thread: threading.Thread | None = None
+
+        self.max_rss: int = 0
+        self.start_cpu_times: psutil._common.pcputimes | None = None
+        self.end_cpu_times: psutil._common.pcputimes | None = None
+        self.start_wall_time: float | None = None
+        self.end_wall_time: float | None = None
+
+    def start(self) -> None:
+        self.start_cpu_times = self._process.cpu_times()
+        self.start_wall_time = time.monotonic()
+
+        def _monitor() -> None:
+            try:
+                while not self._stop_event.is_set():
+                    rss = self._process.memory_info().rss
+                    self.max_rss = max(self.max_rss, rss)
+                    time.sleep(self.sample_interval)
+            except Exception:
+                self._stop_event.set()
+
+        self._thread = threading.Thread(target=_monitor, daemon=True)
+        self._thread.start()
+
+    def stop(self) -> None:
+        self._stop_event.set()
+        if self._thread is not None:
+            self._thread.join()
+        self.end_cpu_times = self._process.cpu_times()
+        self.end_wall_time = time.monotonic()
+
+    def __enter__(self) -> typing.Self:
+        self.start()
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: types.TracebackType | None,
+    ) -> None:
+        self.stop()
+        stats = dict(self.summary())
+        self.logger.log_info("Job resource usage summary", attributes=stats)
+
+    def summary(self) -> Attributes:
+        assert self.start_cpu_times is not None
+        assert self.end_cpu_times is not None
+        assert self.start_wall_time is not None
+        assert self.end_wall_time is not None
+
+        cpu_user = self.end_cpu_times.user - self.start_cpu_times.user
+        cpu_sys = self.end_cpu_times.system - self.start_cpu_times.system
+        cpu_total = cpu_user + cpu_sys
+        elapsed = self.end_wall_time - self.start_wall_time
+        return {
+            "cpu_user_s": round(cpu_user, 3),
+            "cpu_system_s": round(cpu_sys, 3),
+            "cpu_total_s": round(cpu_total, 3),
+            "wall_time_s": round(elapsed, 3),
+            "peak_rss_mb": round(self.max_rss / (1024 * 1024), 2),
+        }
+
+
 class JobLogger(Logger):
     def __init__(
         self,
@@ -158,20 +265,31 @@ class JobLogger(Logger):
         base_span: Span,
         profile_metadata: job_definition_t.ProfileMetadata,
         job_definition: job_definition_t.JobDefinition,
+        queued_job_uuid: str,
     ) -> None:
         self.profile_metadata = profile_metadata
         self.job_definition = job_definition
+        self.queued_job_uuid = queued_job_uuid
         super().__init__(base_span)
 
-    def _patch_attributes(self, attributes: Attributes | None) -> Attributes:
+    def _patch_attributes(
+        self,
+        attributes: Attributes | None,
+        *,
+        message: str | None = None,
+        severity: LogSeverity | None = None,
+    ) -> Attributes:
         patched_attributes: dict[str, base_t.JsonValue] = {
-            **(attributes if attributes is not None else {})
+            **super()._patch_attributes(
+                attributes=attributes, message=message, severity=severity
+            )
         }
         patched_attributes["profile.name"] = self.profile_metadata.name
         patched_attributes["profile.base_url"] = self.profile_metadata.base_url
         patched_attributes["job.name"] = self.job_definition.name
         patched_attributes["job.id"] = self.job_definition.id
         patched_attributes["job.definition_type"] = self.job_definition.type
+        patched_attributes["job.queued_job_uuid"] = self.queued_job_uuid
         match self.job_definition:
             case job_definition_t.CronJobDefinition():
                 patched_attributes["job.definition.cron_spec"] = (
@@ -197,6 +315,9 @@ class JobLogger(Logger):
                 assert_never(self.job_definition.executor)
         return _cast_attributes(patched_attributes)
 
+    def resource_tracking(self) -> PerJobResourceTracker:
+        return PerJobResourceTracker(self)
+
 
 @contextmanager
 def push_scope_optional(

uncountable/integration/webhook_server/entrypoint.py

@@ -73,6 +73,8 @@ def register_route(
 
 
 def main() -> None:
+    app.add_url_rule("/health", "health", lambda: ("OK", 200))
+
     profiles = load_profiles()
     for profile_metadata in profiles:
         server_logger = Logger(get_current_span())

uncountable/types/__init__.py

@@ -2,6 +2,7 @@
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
+from .api.notebooks import add_notebook_content as add_notebook_content_t
 from .api.recipes import add_recipe_to_project as add_recipe_to_project_t
 from .api.recipes import add_time_series_data as add_time_series_data_t
 from .api.recipes import archive_recipes as archive_recipes_t
@@ -16,6 +17,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.uploader import complete_async_parse as complete_async_parse_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
@@ -35,12 +37,15 @@ from . import entity_t as entity_t
 from .api.batch import execute_batch as execute_batch_t
 from .api.batch import execute_batch_load_async as execute_batch_load_async_t
 from . import experiment_groups_t as experiment_groups_t
+from .api.runsheet import export_default_runsheet as export_default_runsheet_t
 from .api.entity import export_entities as export_entities_t
 from . import exports_t as exports_t
+from .api.listing import fetch_listing as fetch_listing_t
 from . import field_values_t as field_values_t
 from . import fields_t as fields_t
 from . import generic_upload_t as generic_upload_t
 from .api.recipes import get_column_calculation_values as get_column_calculation_values_t
+from .api.user import get_current_user_info as get_current_user_info_t
 from .api.recipes import get_curve as get_curve_t
 from .api.entity import get_entities_data as get_entities_data_t
 from .api.inputs import get_input_data as get_input_data_t
@@ -49,6 +54,7 @@ from .api.inputs import get_input_names as get_input_names_t
 from .api.inputs import get_inputs_data as get_inputs_data_t
 from .api.outputs import get_output_data as get_output_data_t
 from .api.outputs import get_output_names as get_output_names_t
+from .api.outputs import get_output_organization as get_output_organization_t
 from .api.project import get_projects as get_projects_t
 from .api.project import get_projects_data as get_projects_data_t
 from .api.recipes import get_recipe_calculations as get_recipe_calculations_t
@@ -63,19 +69,27 @@ from . import identifier_t as identifier_t
 from . import input_attributes_t as input_attributes_t
 from . import inputs_t as inputs_t
 from . import integration_server_t as integration_server_t
+from . import integration_session_t as integration_session_t
+from . import integrations_t as integrations_t
 from .api.uploader import invoke_uploader as invoke_uploader_t
 from . import job_definition_t as job_definition_t
+from .api.entity import list_aggregate as list_aggregate_t
 from .api.entity import list_entities as list_entities_t
 from .api.id_source import list_id_source as list_id_source_t
+from . import listing_t as listing_t
 from .api.entity import lock_entity as lock_entity_t
 from .api.recipes import lock_recipes as lock_recipes_t
 from .api.entity import lookup_entity as lookup_entity_t
 from .api.id_source import match_id_source as match_id_source_t
+from . import notices_t as notices_t
+from . import notifications_t as notifications_t
 from . import outputs_t as outputs_t
 from . import overrides_t as overrides_t
 from . import permissions_t as permissions_t
 from . import phases_t as phases_t
 from . import post_base_t as post_base_t
+from .api.integrations import publish_realtime_data as publish_realtime_data_t
+from .api.integrations import push_notification as push_notification_t
 from . import queued_job_t as queued_job_t
 from . import recipe_identifiers_t as recipe_identifiers_t
 from . import recipe_inputs_t as recipe_inputs_t
@@ -85,6 +99,7 @@ from . import recipe_output_metadata_t as recipe_output_metadata_t
 from . import recipe_tags_t as recipe_tags_t
 from . import recipe_workflow_steps_t as recipe_workflow_steps_t
 from . import recipes_t as recipes_t
+from .api.integrations import register_sockets_token as register_sockets_token_t
 from .api.recipes import remove_recipe_from_project as remove_recipe_from_project_t
 from .api.recipe_links import remove_recipe_link as remove_recipe_link_t
 from .api.entity import resolve_entity_ids as resolve_entity_ids_t
@@ -92,6 +107,7 @@ from .api.outputs import resolve_output_conditions as resolve_output_conditions_
 from . import response_t as response_t
 from .api.triggers import run_trigger as run_trigger_t
 from . import secret_retrieval_t as secret_retrieval_t
+from .api.entity import set_barcode as set_barcode_t
 from .api.permissions import set_core_permissions as set_core_permissions_t
 from .api.entity import set_entity_field_values as set_entity_field_values_t
 from .api.inputs import set_input_attribute_values as set_input_attribute_values_t
@@ -104,13 +120,16 @@ from .api.recipes import set_recipe_output_annotations as set_recipe_output_anno
 from .api.recipes import set_recipe_output_file as set_recipe_output_file_t
 from .api.recipes import set_recipe_outputs as set_recipe_outputs_t
 from .api.recipes import set_recipe_tags as set_recipe_tags_t
+from .api.recipes import set_recipe_total as set_recipe_total_t
 from .api.entity import set_values as set_values_t
+from . import sockets_t as sockets_t
 from .api.entity import transition_entity_phase as transition_entity_phase_t
 from .api.recipes import unarchive_recipes as unarchive_recipes_t
 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 . import uploader_t as uploader_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
@@ -119,6 +138,7 @@ from . import workflows_t as workflows_t
 
 
 __all__: list[str] = [
+    "add_notebook_content_t",
     "add_recipe_to_project_t",
     "add_time_series_data_t",
     "archive_recipes_t",
@@ -133,6 +153,7 @@ __all__: list[str] = [
     "chemical_structure_t",
     "clear_recipe_outputs_t",
     "client_config_t",
+    "complete_async_parse_t",
     "complete_async_upload_t",
     "convert_chemical_formats_t",
     "create_entities_t",
@@ -152,12 +173,15 @@ __all__: list[str] = [
     "execute_batch_t",
     "execute_batch_load_async_t",
     "experiment_groups_t",
+    "export_default_runsheet_t",
     "export_entities_t",
     "exports_t",
+    "fetch_listing_t",
     "field_values_t",
     "fields_t",
     "generic_upload_t",
     "get_column_calculation_values_t",
+    "get_current_user_info_t",
     "get_curve_t",
     "get_entities_data_t",
     "get_input_data_t",
@@ -166,6 +190,7 @@ __all__: list[str] = [
     "get_inputs_data_t",
     "get_output_data_t",
     "get_output_names_t",
+    "get_output_organization_t",
     "get_projects_t",
     "get_projects_data_t",
     "get_recipe_calculations_t",
@@ -180,19 +205,27 @@ __all__: list[str] = [
     "input_attributes_t",
     "inputs_t",
     "integration_server_t",
+    "integration_session_t",
+    "integrations_t",
     "invoke_uploader_t",
     "job_definition_t",
+    "list_aggregate_t",
     "list_entities_t",
     "list_id_source_t",
+    "listing_t",
     "lock_entity_t",
     "lock_recipes_t",
     "lookup_entity_t",
     "match_id_source_t",
+    "notices_t",
+    "notifications_t",
     "outputs_t",
     "overrides_t",
     "permissions_t",
     "phases_t",
     "post_base_t",
+    "publish_realtime_data_t",
+    "push_notification_t",
     "queued_job_t",
     "recipe_identifiers_t",
     "recipe_inputs_t",
@@ -202,6 +235,7 @@ __all__: list[str] = [
     "recipe_tags_t",
     "recipe_workflow_steps_t",
     "recipes_t",
+    "register_sockets_token_t",
     "remove_recipe_from_project_t",
     "remove_recipe_link_t",
     "resolve_entity_ids_t",
@@ -209,6 +243,7 @@ __all__: list[str] = [
     "response_t",
     "run_trigger_t",
     "secret_retrieval_t",
+    "set_barcode_t",
     "set_core_permissions_t",
     "set_entity_field_values_t",
     "set_input_attribute_values_t",
@@ -221,13 +256,16 @@ __all__: list[str] = [
     "set_recipe_output_file_t",
     "set_recipe_outputs_t",
     "set_recipe_tags_t",
+    "set_recipe_total_t",
     "set_values_t",
+    "sockets_t",
     "transition_entity_phase_t",
     "unarchive_recipes_t",
     "units_t",
     "unlock_entity_t",
     "unlock_recipes_t",
     "update_entity_material_families_t",
+    "uploader_t",
     "upsert_condition_match_t",
     "upsert_field_options_t",
     "users_t",

uncountable/types/api/entity/create_or_update_entity.py

@@ -35,6 +35,7 @@ class Arguments:
     definition_key: identifier_t.IdentifierKey
     field_values: list[field_values_t.FieldArgumentValue]
     entity_key: identifier_t.IdentifierKey | None = None
+    on_create_init_field_values: list[field_values_t.FieldArgumentValue] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/export_entities.py

@@ -18,12 +18,24 @@ __all__: list[str] = [
     "Data",
     "ENDPOINT_METHOD",
     "ENDPOINT_PATH",
+    "ListingAttribute",
 ]
 
 ENDPOINT_METHOD = "POST"
 ENDPOINT_PATH = "api/external/entity/export_entities"
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.export_entities.ListingAttribute",
+    unconverted_values={"value"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ListingAttribute:
+    key: str
+    value: base_t.JsonValue
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.api.entity.export_entities.Arguments",
@@ -34,6 +46,7 @@ class Arguments:
     type: exports_t.ExportType = exports_t.ExportType.EXCEL
     client_timezone: exports_t.ListingExportUserTimezone | None = None
     limit: int | None = None
+    attributes: list[ListingAttribute] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/list_aggregate.py

@@ -0,0 +1,79 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import base_t
+
+__all__: list[str] = [
+    "Arguments",
+    "AttributeValue",
+    "ColumnAccess",
+    "ColumnResults",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "GET"
+ENDPOINT_PATH = "api/external/entity/list_aggregate"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_aggregate.AttributeValue",
+    unconverted_values={"value"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AttributeValue:
+    name: str
+    value: base_t.JsonValue
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_aggregate.Arguments",
+    unconverted_values={"attribute_values"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    config_reference: str
+    attribute_values: list[AttributeValue] | None = None
+    offset: int | None = None
+    limit: int | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_aggregate.ColumnResults",
+    unconverted_values={"column_values"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ColumnResults:
+    column_values: list[base_t.JsonValue]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_aggregate.ColumnAccess",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ColumnAccess:
+    name: str
+    table_label: str | None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_aggregate.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data:
+    columns: list[ColumnAccess]
+    results: list[ColumnResults]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/list_entities.py

@@ -11,9 +11,12 @@ from pkgs.serialization import serial_class
 from pkgs.serialization import OpaqueKey
 from ... import base_t
 from ... import entity_t
+from ... import identifier_t
 
 __all__: list[str] = [
+    "AdditionalFilterConfig",
     "Arguments",
+    "AttributeValue",
     "ColumnAccess",
     "Data",
     "ENDPOINT_METHOD",
@@ -25,6 +28,27 @@ ENDPOINT_METHOD = "GET"
 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.AttributeValue",
+    unconverted_values={"value"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AttributeValue:
+    name: str
+    value: base_t.JsonValue
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.list_entities.AdditionalFilterConfig",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AdditionalFilterConfig:
+    config_key: identifier_t.IdentifierKey
+    attribute_values: list[AttributeValue] | None = None
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.api.entity.list_entities.Arguments",
@@ -37,6 +61,7 @@ class Arguments:
     attributes: dict[OpaqueKey, base_t.JsonValue] | None = None
     offset: int | None = None
     limit: int | None = None
+    additional_filter_configs: list[AdditionalFilterConfig] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/set_barcode.py

@@ -0,0 +1,43 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import base_t
+from ... import entity_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/entity/set_barcode"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_barcode.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    entity_key: entity_t.EntityIdentifier
+    barcode_value: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.set_barcode.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/transition_entity_phase.py

@@ -66,8 +66,9 @@ TransitionIdentifier = typing.Annotated[
 )
 @dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
-    entity: entity_t.Entity
     transition: TransitionIdentifier
+    entity: entity_t.Entity | None = None
+    entity_identifier: entity_t.EntityIdentifier | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec