UncountablePythonSDK 0.0.153__tar.gz → 0.0.155__tar.gz

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.153
+Version: 0.0.155
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -31,7 +31,7 @@ Requires-Dist: opentelemetry-api==1.*
 Requires-Dist: opentelemetry-exporter-otlp-proto-common==1.*
 Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.*
 Requires-Dist: opentelemetry-sdk>=1.39.1
-Requires-Dist: paramiko==3.*
+Requires-Dist: paramiko==4.*
 Requires-Dist: boto3==1.*
 Requires-Dist: flask==3.*
 Requires-Dist: simplejson==3.*

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/UncountablePythonSDK.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.153
+Version: 0.0.155
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -31,7 +31,7 @@ Requires-Dist: opentelemetry-api==1.*
 Requires-Dist: opentelemetry-exporter-otlp-proto-common==1.*
 Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.*
 Requires-Dist: opentelemetry-sdk>=1.39.1
-Requires-Dist: paramiko==3.*
+Requires-Dist: paramiko==4.*
 Requires-Dist: boto3==1.*
 Requires-Dist: flask==3.*
 Requires-Dist: simplejson==3.*

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ docs/static/favicons/manifest.json
 docs/static/favicons/mstile-150x150.png
 docs/static/favicons/safari-pinned-tab.svg
 examples/async_batch.py
+examples/async_batch_large_parse.py
 examples/basic_auth.py
 examples/create_entity.py
 examples/create_ingredient_sdk.py
@@ -328,6 +329,7 @@ uncountable/types/api/material_families/__init__.py
 uncountable/types/api/material_families/update_entity_material_families.py
 uncountable/types/api/notebooks/__init__.py
 uncountable/types/api/notebooks/add_notebook_content.py
+uncountable/types/api/notebooks/get_notebook_content.py
 uncountable/types/api/outputs/__init__.py
 uncountable/types/api/outputs/get_output_data.py
 uncountable/types/api/outputs/get_output_names.py

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/UncountablePythonSDK.egg-info/requires.txt

@@ -13,7 +13,7 @@ opentelemetry-api==1.*
 opentelemetry-exporter-otlp-proto-common==1.*
 opentelemetry-exporter-otlp-proto-http==1.*
 opentelemetry-sdk>=1.39.1
-paramiko==3.*
+paramiko==4.*
 boto3==1.*
 flask==3.*
 simplejson==3.*

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/docs/requirements.txt

@@ -1,9 +1,9 @@
 furo==2025.12.19
 myst-parser==5.0.0
-sphinx-autoapi==3.6.1
+sphinx-autoapi==3.7.0
 sphinx-copybutton==0.5.2
-Sphinx==8.2.3
+Sphinx==9.1.0
 sphinx_design==0.7.0
-sphinx-favicon==1.0.1
+sphinx-favicon==1.1.0
 astroid==4.0.3
-docutils==0.21.2
+docutils==0.22.4

uncountablepythonsdk-0.0.155/examples/async_batch_large_parse.py

@@ -0,0 +1,47 @@
+import os
+from decimal import Decimal
+
+from uncountable.core import AsyncBatchProcessor, AuthDetailsApiKey, Client
+from uncountable.types.generic_upload_t import UploadDestinationRecipe
+from uncountable.types.identifier import IdentifierKeyId
+from uncountable.types.uploader_t import (
+    DataChannel,
+    DecimalValue,
+    NumericChannelData,
+    ParsedFileData,
+)
+
+client = Client(
+    base_url=os.environ["UNC_BASE_URL"],
+    auth_details=AuthDetailsApiKey(
+        api_id=os.environ["UNC_API_ID"], api_secret_key=os.environ["UNC_API_SECRET_KEY"]
+    ),
+)
+
+ROWS = 1_000_000
+CHANNELS = 2
+
+parsed_file_data = [
+    ParsedFileData(
+        file_name="large_timeseries.csv",
+        file_structures=[
+            DataChannel(
+                channel=NumericChannelData(
+                    name=f"channel_{i}",
+                    data=[DecimalValue(value=Decimal("1.0"))] * ROWS,
+                ),
+            )
+            for i in range(CHANNELS)
+        ],
+    )
+]
+
+batch_loader = AsyncBatchProcessor(client=client)
+batch_loader.complete_async_parse(
+    parsed_file_data=parsed_file_data,
+    async_job_key=IdentifierKeyId(id=1),
+    upload_destination=UploadDestinationRecipe(
+        recipe_key=IdentifierKeyId(id=1),
+    ),
+)
+job_id = batch_loader.send()

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/pyproject.toml

@@ -66,7 +66,7 @@ dependencies = [
     "opentelemetry-exporter-otlp-proto-common==1.*",
     "opentelemetry-exporter-otlp-proto-http==1.*",
     "opentelemetry-sdk >= 1.39.1",
-    "paramiko ==3.*",
+    "paramiko ==4.*",
     "boto3 == 1.*",
     "flask == 3.*",
     "simplejson == 3.*",

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/core/client.py

@@ -1,7 +1,9 @@
 import base64
 import datetime
+import pathlib
 import re
 import typing
+import zipfile
 from dataclasses import dataclass
 from datetime import UTC, timedelta
 from enum import StrEnum
@@ -257,6 +259,7 @@ class Client(ClientMethods):
         attributes: Attributes = {
             "method": http_request.method,
             "endpoint": api_request.endpoint,
+            "request_id": request_id,
         }
         with push_scope_optional(self._cfg.logger, "api_call", attributes=attributes):
             if self._cfg.logger is not None:
@@ -379,10 +382,34 @@ class Client(ClientMethods):
         assert isinstance(http_request, HTTPGetRequest)
         request.params = http_request.query_params
         response = self._send_request(request)
+
+        if response.status_code == 404:
+            return []
+
         self._validate_response_status(response, request_id)
 
         content = response.content
         content_disposition = response.headers.get("Content-Disposition", None)
+        file_download_type = response.headers.get(
+            download_file_t.UNC_FILE_DOWNLOAD_TYPE_HEADER, None
+        )
+        if file_download_type == download_file_t.FileDownloadType.ARCHIVE:
+            archive = zipfile.ZipFile(BytesIO(content))
+            files: list[DownloadedFile] = []
+
+            for filepath in archive.namelist():
+                with archive.open(filepath, "r") as f:
+                    filename = pathlib.Path(filepath).name
+                    read_file = f.read()
+                    files.append(
+                        DownloadedFile(
+                            name=filename,
+                            size=len(read_file),
+                            data=BytesIO(read_file),
+                        )
+                    )
+            return files
+
         return [
             DownloadedFile(
                 name=self._get_downloaded_filename(cd=content_disposition),

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/integration/construct_client.py

@@ -30,10 +30,10 @@ def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
 
 
 def _construct_client_config(
-    profile_meta: ProfileMetadata, job_logger: JobLogger
-) -> ClientConfig | None:
+    *, profile_meta: ProfileMetadata, job_logger: JobLogger
+) -> ClientConfig:
     if profile_meta.client_options is None:
-        return None
+        return ClientConfig(logger=job_logger)
     return ClientConfig(
         allow_insecure_tls=profile_meta.client_options.allow_insecure_tls,
         extra_headers=profile_meta.client_options.extra_headers,
@@ -47,6 +47,6 @@ def construct_uncountable_client(
     return Client(
         base_url=profile_meta.base_url,
         auth_details=_construct_auth_details(profile_meta),
-        config=_construct_client_config(profile_meta, logger),
+        config=_construct_client_config(profile_meta=profile_meta, job_logger=logger),
         app_base_url=profile_meta.app_base_url,
     )

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/integration/queue_runner/worker.py

@@ -1,9 +1,18 @@
 import asyncio
-import resource
+import sys
+import typing
 from concurrent.futures import ProcessPoolExecutor
 from dataclasses import dataclass
 
 import psutil
+
+if typing.TYPE_CHECKING:
+    import resource as resource
+elif sys.platform != "win32":
+    import resource as resource
+else:
+    resource = None
+
 from opentelemetry.trace import get_current_span
 
 from uncountable.core.async_batch import AsyncBatchProcessor
@@ -84,9 +93,10 @@ def run_queued_job(
     queued_job: queued_job_t.QueuedJob,
 ) -> job_definition_t.JobResult:
     with get_otel_tracer().start_as_current_span(name="run_queued_job") as span:
-        total_mem = psutil.virtual_memory().total
-        limit_bytes = int(total_mem * 0.9)
-        resource.setrlimit(resource.RLIMIT_AS, (limit_bytes, limit_bytes))
+        if resource is not None:
+            total_mem = psutil.virtual_memory().total
+            limit_bytes = int(total_mem * 0.9)
+            resource.setrlimit(resource.RLIMIT_AS, (limit_bytes, limit_bytes))
 
         job_details = get_registered_job_details(queued_job.job_ref_name)
         job_logger = JobLogger(

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/types/__init__.py

@@ -54,6 +54,7 @@ from .api.inputs import get_input_data as get_input_data_t
 from .api.input_groups import get_input_group_names as get_input_group_names_t
 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.notebooks import get_notebook_content as get_notebook_content_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
@@ -199,6 +200,7 @@ __all__: list[str] = [
     "get_input_group_names_t",
     "get_input_names_t",
     "get_inputs_data_t",
+    "get_notebook_content_t",
     "get_output_data_t",
     "get_output_names_t",
     "get_output_organization_t",

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/types/api/notebooks/add_notebook_content.py

@@ -160,6 +160,8 @@ class InputContentImage:
         "ketcher": "Ketcher",
         "mol_2000": "MOL 2000",
         "mol_3000": "MOL 3000",
+        "rxn": "RXN",
+        "sdf": "SDF",
     },
 )
 class ChemicalStructureSourceType(StrEnum):
@@ -167,6 +169,8 @@ class ChemicalStructureSourceType(StrEnum):
     KETCHER = "ketcher"
     MOL_2000 = "mol_2000"
     MOL_3000 = "mol_3000"
+    RXN = "rxn"
+    SDF = "sdf"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -179,6 +183,7 @@ class InputContentChemicalStructure:
     type: typing.Literal[InputContentType.CHEMICAL_STRUCTURE] = InputContentType.CHEMICAL_STRUCTURE
     structure_format: ChemicalStructureSourceType
     structure_data: str
+    split_sdf_into_separate_cells: bool | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.155/uncountable/types/api/notebooks/get_notebook_content.py

@@ -0,0 +1,276 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from enum import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from pkgs.serialization import serial_string_enum
+from ... import base_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "ChemicalStructureOutputFormat",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "OutputContent",
+    "OutputContentChemicalStructure",
+    "OutputContentChemicalStructureError",
+    "OutputContentEquation",
+    "OutputContentFiles",
+    "OutputContentImage",
+    "OutputContentReactionTable",
+    "OutputContentTable",
+    "OutputContentTextHtml",
+    "OutputContentType",
+    "OutputTableCell",
+    "OutputTableCellContent",
+    "OutputTableCellContentImage",
+    "OutputTableCellContentText",
+    "OutputTableCellContentType",
+    "OutputTableCellStyle",
+]
+
+ENDPOINT_METHOD = "GET"
+ENDPOINT_PATH = "api/external/notebooks/get_notebook_content"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "text_html": "Text HTML",
+        "table": "Table",
+        "chemical_structure": "Chemical Structure",
+        "chemical_structure_error": "Chemical Structure Error",
+        "equation": "Equation",
+        "files": "Files",
+        "image": "Image",
+        "reaction_table": "Reaction Table",
+    },
+)
+class OutputContentType(StrEnum):
+    TEXT_HTML = "text_html"
+    TABLE = "table"
+    CHEMICAL_STRUCTURE = "chemical_structure"
+    CHEMICAL_STRUCTURE_ERROR = "chemical_structure_error"
+    EQUATION = "equation"
+    FILES = "files"
+    IMAGE = "image"
+    REACTION_TABLE = "reaction_table"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "mol_3000": "MOL 3000",
+        "rxn_3000": "RXN 3000",
+    },
+)
+class ChemicalStructureOutputFormat(StrEnum):
+    MOL_3000 = "mol_3000"
+    RXN_3000 = "rxn_3000"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentTextHtml",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentTextHtml:
+    type: typing.Literal[OutputContentType.TEXT_HTML] = OutputContentType.TEXT_HTML
+    text_html: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "text": "Text",
+        "image": "Image",
+    },
+)
+class OutputTableCellContentType(StrEnum):
+    TEXT = "text"
+    IMAGE = "image"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputTableCellContentText",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputTableCellContentText:
+    type: typing.Literal[OutputTableCellContentType.TEXT] = OutputTableCellContentType.TEXT
+    text: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputTableCellContentImage",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputTableCellContentImage:
+    type: typing.Literal[OutputTableCellContentType.IMAGE] = OutputTableCellContentType.IMAGE
+    file_name: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+OutputTableCellContent = typing.Annotated[
+    OutputTableCellContentText | OutputTableCellContentImage,
+    serial_union_annotation(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputTableCellContent",
+        discriminator="type",
+        discriminator_map={
+            "text": OutputTableCellContentText,
+            "image": OutputTableCellContentImage,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputTableCellStyle",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputTableCellStyle:
+    background_color: str | None = None
+    bold: bool | None = None
+    italic: bool | None = None
+    underline: bool | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputTableCell",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputTableCell:
+    content: OutputTableCellContent
+    style: OutputTableCellStyle | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentTable",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentTable:
+    type: typing.Literal[OutputContentType.TABLE] = OutputContentType.TABLE
+    rows: list[list[OutputTableCell]]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentChemicalStructure",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentChemicalStructure:
+    type: typing.Literal[OutputContentType.CHEMICAL_STRUCTURE] = OutputContentType.CHEMICAL_STRUCTURE
+    structure_format: ChemicalStructureOutputFormat
+    structure_data: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentEquation",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentEquation:
+    type: typing.Literal[OutputContentType.EQUATION] = OutputContentType.EQUATION
+    equation: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentFiles",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentFiles:
+    type: typing.Literal[OutputContentType.FILES] = OutputContentType.FILES
+    file_names: list[str]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentImage",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentImage:
+    type: typing.Literal[OutputContentType.IMAGE] = OutputContentType.IMAGE
+    file_names: list[str]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentChemicalStructureError",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentChemicalStructureError:
+    type: typing.Literal[OutputContentType.CHEMICAL_STRUCTURE_ERROR] = OutputContentType.CHEMICAL_STRUCTURE_ERROR
+    message: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContentReactionTable",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class OutputContentReactionTable:
+    type: typing.Literal[OutputContentType.REACTION_TABLE] = OutputContentType.REACTION_TABLE
+    reaction_data: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+OutputContent = typing.Annotated[
+    OutputContentTextHtml | OutputContentTable | OutputContentChemicalStructure | OutputContentEquation | OutputContentChemicalStructureError | OutputContentFiles | OutputContentImage | OutputContentReactionTable,
+    serial_union_annotation(
+    named_type_path="sdk.api.notebooks.get_notebook_content.OutputContent",
+        discriminator="type",
+        discriminator_map={
+            "text_html": OutputContentTextHtml,
+            "table": OutputContentTable,
+            "chemical_structure": OutputContentChemicalStructure,
+            "equation": OutputContentEquation,
+            "chemical_structure_error": OutputContentChemicalStructureError,
+            "files": OutputContentFiles,
+            "image": OutputContentImage,
+            "reaction_table": OutputContentReactionTable,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    notebook_key: identifier_t.IdentifierKey
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.get_notebook_content.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data:
+    contents: list[OutputContent]
+# DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/types/api/triggers/run_trigger.py

@@ -30,6 +30,7 @@ ENDPOINT_PATH = "api/external/triggers/run_trigger"
 @dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     trigger_ref_name: str
+    entity_identifier: entity_t.EntityIdentifier | None = None
     entity: entity_t.Entity | None = None
 
 

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/types/async_batch_processor.py

@@ -35,6 +35,7 @@ import uncountable.types.api.integrations.push_notification as push_notification
 from uncountable.types import recipe_identifiers_t
 from uncountable.types import recipe_metadata_t
 from uncountable.types import recipe_workflow_steps_t
+import uncountable.types.api.triggers.run_trigger as run_trigger_t
 import uncountable.types.api.entity.set_barcode as set_barcode_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
@@ -680,6 +681,44 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def run_trigger(
+        self,
+        *,
+        trigger_ref_name: str,
+        entity_identifier: entity_t.EntityIdentifier | None = None,
+        entity: entity_t.Entity | None = None,
+        depends_on: list[str] | None = None,
+        _request_options: client_config_t.RequestOptions | None = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Runs a trigger.  Requires admin access
+
+        :param entity_identifier: Identifier of the entity to run the trigger on.
+        :param entity: [Deprecated: use entity_identifier] Entity to run the trigger on.
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = run_trigger_t.Arguments(
+            entity_identifier=entity_identifier,
+            entity=entity,
+            trigger_ref_name=trigger_ref_name,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.RUN_TRIGGER,
+            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 set_barcode(
         self,
         *,

{uncountablepythonsdk-0.0.153 → uncountablepythonsdk-0.0.155}/uncountable/types/async_batch_t.py

@@ -52,6 +52,7 @@ class AsyncBatchRequestPath(StrEnum):
     SET_BARCODE = "entity/set_barcode"
     LOCK_ENTITY = "entity/lock_entity"
     UNLOCK_ENTITY = "entity/unlock_entity"
+    RUN_TRIGGER = "triggers/run_trigger"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec