UncountablePythonSDK 0.0.166__py3-none-any.whl → 0.0.168__py3-none-any.whl

pkgs/argument_parser/argument_parser.py

@@ -426,9 +426,11 @@ def _build_parser_inner(
             return inner
 
         v_parser = _build_parser_inner(args[1], context, type_var_map)
-        return lambda value: origin(
-            (key_parser(k), v_parser(v)) for k, v in value.items()
-        )
+
+        def parse_dict(value: typing.Any) -> typing.Any:
+            return origin((key_parser(k), v_parser(v)) for k, v in value.items())
+
+        return parse_dict
 
     if origin == typing.Literal:
         valid_values: set[T] = set(typing.get_args(parsed_type))

pkgs/filesystem_utils/_sftp_session.py

@@ -67,6 +67,7 @@ class SFTPSession(FileSystemSession):
     def __init__(self, sftp_config: FileSystemSFTPConfig) -> None:
         super().__init__()
         self.host: str = sftp_config.ip
+        self.port: int = sftp_config.port
         self.username: str = sftp_config.username
         self.key_file: str | paramiko.PKey | None = (
             sftp_config.pem_path
@@ -81,6 +82,7 @@ class SFTPSession(FileSystemSession):
         if self.key_file is not None:
             self.connection = pysftp.Connection(
                 self.host,
+                port=self.port,
                 username=self.username,
                 private_key=self.key_file,
                 cnopts=cnopts,
@@ -88,6 +90,7 @@ class SFTPSession(FileSystemSession):
         elif self.password is not None:
             self.connection = pysftp.Connection(
                 self.host,
+                port=self.port,
                 username=self.username,
                 password=self.password,
                 cnopts=cnopts,

pkgs/filesystem_utils/file_type_utils.py

@@ -65,6 +65,7 @@ class FileSystemSFTPConfig:
     password: str | None = None
     valid_extensions: tuple[str] | None = None
     recursive: bool = True
+    port: int = 22
 
 
 @dataclass(kw_only=True)

uncountable/core/client.py

@@ -264,7 +264,7 @@ class Client(ClientMethods):
         }
         with push_scope_optional(self._cfg.logger, "api_call", attributes=attributes):
             if self._cfg.logger is not None:
-                self._cfg.logger.log_info(api_request.endpoint, attributes=attributes)
+                self._cfg.logger.log_debug(api_request.endpoint, attributes=attributes)
             timeout = (
                 api_request.request_options.timeout_secs
                 if api_request.request_options is not None

uncountable/core/file_upload.py

@@ -142,7 +142,7 @@ class FileUploader:
                 self._logger, "upload_file", attributes=attributes
             ):
                 if self._logger is not None:
-                    self._logger.log_info("Uploading file", attributes=attributes)
+                    self._logger.log_debug("Uploading file", attributes=attributes)
                 with file_upload_data(file_upload) as file_bytes:
                     if file_bytes.bytes_data.read(1) == b"":
                         raise UploadFailed(

uncountable/core/query/builder.py

@@ -1,11 +1,10 @@
 from __future__ import annotations
 
 import json
-import time
 from copy import copy
 from dataclasses import make_dataclass
 from decimal import Decimal
-from typing import Self
+from typing import Callable, Self
 
 from pkgs.argument_parser import CachedParser
 from pkgs.serialization.serial_class import serial_class
@@ -29,8 +28,7 @@ from uncountable.types import (
 from uncountable.types.api.entity import list_entities as list_entities_t
 from uncountable.types.api.files.download_file import FileDownloadQueryEntityField
 
-_ASYNC_EXPORT_TIMEOUT_SECONDS = 60 * 10
-_ASYNC_EXPORT_POLL_INTERVAL_SECONDS = 1
+_FETCH_ROW_LIMIT = 100
 
 
 class _AsyncExportJob(QueryRow):
@@ -99,7 +97,9 @@ class QueryBuilder[QueryRowT: QueryRow]:
         )
         return branch
 
-    def _fetch(self, limit: int = 100, offset: int = 0) -> list[QueryRowT]:
+    def _fetch(self, limit: int, offset: int = 0) -> list[QueryRowT]:
+        if limit > _FETCH_ROW_LIMIT:
+            raise ValueError(f"Fetch row limit cannot exceed {_FETCH_ROW_LIMIT}")
         rows = self._client.fetch_listing(
             entity_type=self._base_type,
             columns=list(self._column_identifiers),
@@ -107,7 +107,6 @@ class QueryBuilder[QueryRowT: QueryRow]:
             limit=limit,
             offset=offset,
         ).results
-
         return [self._parse_row(row.column_values) for row in rows]
 
     def _hook_after_parse_row(self, row: QueryRowT) -> None:
@@ -124,13 +123,13 @@ class QueryBuilder[QueryRowT: QueryRow]:
         return parsed_model
 
     def one(self) -> QueryRowT:
-        results = self._fetch()
+        results = self._fetch(limit=2)
         if len(results) != 1:
             raise ValueError(f"Expected one result, received {len(results)}")
         return results[0]
 
     def one_or_none(self) -> QueryRowT | None:
-        results = self._fetch()
+        results = self._fetch(limit=2)
         if len(results) > 1:
             raise ValueError(f"Expected one result, received {len(results)}")
         if len(results) == 0:
@@ -142,59 +141,74 @@ class QueryBuilder[QueryRowT: QueryRow]:
         return None if len(results) == 0 else results[0]
 
     def all(self) -> list[QueryRowT]:
-        results = self._fetch()
-        if len(results) < 100:
+        results = self._fetch(limit=_FETCH_ROW_LIMIT)
+        if len(results) < _FETCH_ROW_LIMIT:
             return results
-        return self._await_async_export()
 
-    def _await_async_export(self) -> list[QueryRowT]:
+        # Generate an async listing export if the number of results >= 100
+        poll_for_async_export_results = self._submit_async_export()
+
+        # Resolve exponentially increasing pagination batches until all rows are fetched or the async job completes
+        iteration_count: int = 0
+        while True:
+            for _ in range(2**iteration_count):
+                rows = self._fetch(limit=_FETCH_ROW_LIMIT, offset=len(results))
+                results.extend(rows)
+                if len(rows) < _FETCH_ROW_LIMIT:
+                    return results
+
+            if export_results := poll_for_async_export_results():
+                return export_results
+            iteration_count += 1
+
+    def _submit_async_export(self) -> Callable[[], list[QueryRowT] | None]:
+        """
+        Requests an async export of the query results. Returns a closure for polling the results of the export.
+        """
         async_job_entity = self._client.export_listing(
             entity_type=self._base_type,
             columns=list(self._column_identifiers),
             filters=listing_t.FilterNodeColumnAnd(filters=tuple(self._filters)),
         ).entity
         if async_job_entity is None:
-            raise ValueError("Listing export failed to create async job")
-
-        status_query = QueryBuilder(client=self._client, model=_AsyncExportJob)
-
-        start_time = time.time()
-        while time.time() - start_time < _ASYNC_EXPORT_TIMEOUT_SECONDS:
-            match (
-                status_query
+            raise ValueError("async listing export generation failed")
+        async_job_status: async_jobs_t.AsyncJobStatus | str | None = None
+
+        def poll_for_results() -> list[QueryRowT] | None:
+            nonlocal async_job_status
+            if async_job_status == async_jobs_t.AsyncJobStatus.ERROR:
+                return None
+            async_job_status = (
+                QueryBuilder(client=self._client, model=_AsyncExportJob)
                 .filter(_AsyncExportJob.id == async_job_entity.id)
                 .one()
                 .status
-            ):
-                case async_jobs_t.AsyncJobStatus.COMPLETED:
-                    export_file = self._client.download_files(
-                        file_query=FileDownloadQueryEntityField(
-                            entity=entity_t.EntityIdentifier(
-                                identifier_key=identifier_t.IdentifierKeyId(
-                                    id=async_job_entity.id
-                                ),
-                                type=entity_t.EntityType.ASYNC_JOB,
-                            ),
-                            field_key=identifier_t.IdentifierKeyRefName(
-                                ref_name="unc_async_listing_export_file"
+            )
+            if async_job_status == async_jobs_t.AsyncJobStatus.COMPLETED:
+                export_file = self._client.download_files(
+                    file_query=FileDownloadQueryEntityField(
+                        entity=entity_t.EntityIdentifier(
+                            identifier_key=identifier_t.IdentifierKeyId(
+                                id=async_job_entity.id
                             ),
+                            type=entity_t.EntityType.ASYNC_JOB,
+                        ),
+                        field_key=identifier_t.IdentifierKeyRefName(
+                            ref_name="unc_async_listing_export_file"
+                        ),
+                    )
+                )[0].data.read()
+                return [
+                    self._parse_row(row.column_values)
+                    for row in CachedParser(list_entities_t.Data)
+                    .parse_api(
+                        json.loads(
+                            export_file,
+                            parse_float=Decimal,
                         )
-                    )[0].data.read()
-
-                    return [
-                        self._parse_row(row.column_values)
-                        for row in CachedParser(list_entities_t.Data)
-                        .parse_api(
-                            json.loads(
-                                export_file,
-                                parse_float=Decimal,
-                            )
-                        )
-                        .results
-                    ]
-                case async_jobs_t.AsyncJobStatus.ERROR:
-                    raise ValueError("Listing export failed")
-            time.sleep(_ASYNC_EXPORT_POLL_INTERVAL_SECONDS)
-        raise TimeoutError(
-            f"Async listing export timed out after {_ASYNC_EXPORT_TIMEOUT_SECONDS} seconds"
-        )
+                    )
+                    .results
+                ]
+            return None
+
+        return poll_for_results

uncountable/integration/executors/executors.py

@@ -93,7 +93,7 @@ def execute_job(
         job = resolve_executor(job_definition.executor, profile_metadata)
         job_location = f"{job.__class__.__module__}.{job.__class__.__name__}"
 
-        job_logger.log_info(f"Starting job: `{job_location}`")
+        job_logger.log_debug(f"Starting job: `{job_location}`")
 
         run_entity: entity_t.Entity | None = None
         try:
@@ -115,7 +115,7 @@ def execute_job(
             result = job_definition_t.JobResult(success=False)
 
         submitted_batch_job_ids = args.batch_processor.get_submitted_job_ids()
-        job_logger.log_info(
+        job_logger.log_debug(
             f"Completed job: `{job_location}` (success={result.success})",
             attributes={
                 "submitted_batch_job_ids": submitted_batch_job_ids,

uncountable/integration/queue_runner/job_scheduler.py

@@ -136,7 +136,7 @@ async def start_scheduler(
                 job_payload=payload,
                 job_ref_name=job_ref_name,
             )
-            logger.log_info(
+            logger.log_debug(
                 "Job submitted successfully.",
                 attributes={"job.queued_job_uuid": queued_job.queued_job_uuid},
             )
@@ -168,7 +168,7 @@ async def start_scheduler(
                     )
                 )
 
-                logger.log_info(
+                logger.log_debug(
                     "Job cancelled successfully.",
                     attributes={"job.queued_job_uuid": command.queued_job_uuid},
                 )
@@ -201,10 +201,10 @@ async def start_scheduler(
         def _handle_vaccuum_queued_jobs_command(
             command: CommandVaccuumQueuedJobs,
         ) -> None:
-            logger.log_info("Vaccuuming queued jobs...")
+            logger.log_debug("Vaccuuming queued jobs...")
             deleted_uuids = datastore.vaccuum_queued_jobs()
 
-            logger.log_info(
+            logger.log_debug(
                 "Queued jobs vacuumed successfully.",
                 attributes={"job.queued_job_uuids": deleted_uuids},
             )

uncountable/integration/scheduler.py

@@ -71,7 +71,7 @@ ProcessAlarm = ProcessAlarmRestart | ProcessAlarmShutdownAll
 
 
 def handle_shutdown(logger: Logger, processes: dict[ProcessName, ProcessInfo]) -> None:
-    logger.log_info("received shutdown command, shutting down sub-processes")
+    logger.log_debug("received shutdown command, shutting down sub-processes")
     for proc_info in processes.values():
         if proc_info.is_alive:
             proc_info.process.terminate()
@@ -83,7 +83,7 @@ def handle_shutdown(logger: Logger, processes: dict[ProcessName, ProcessInfo]) -
         and len(still_living_processes) > 0
     ):
         current_loop_processes = [*still_living_processes]
-        logger.log_info(
+        logger.log_debug(
             "waiting for sub-processes to shut down",
             attributes={
                 "still_living_processes": [
@@ -94,7 +94,7 @@ def handle_shutdown(logger: Logger, processes: dict[ProcessName, ProcessInfo]) -
         still_living_processes = []
         for proc_info in current_loop_processes:
             if not proc_info.is_alive:
-                logger.log_info(f"{proc_info.name} shut down successfully")
+                logger.log_debug(f"{proc_info.name} shut down successfully")
             else:
                 still_living_processes.append(proc_info)
         time.sleep(1)
@@ -121,7 +121,7 @@ def restart_process(
             processes[ProcessName.QUEUE_RUNNER] = new_info
             try:
                 _wait_queue_runner_online()
-                logger.log_info("queue runner restarted successfully")
+                logger.log_debug("queue runner restarted successfully")
             except Exception as e:
                 logger.log_exception(e)
                 logger.log_error(
@@ -135,13 +135,13 @@ def restart_process(
             cron_proc.start()
             new_info = ProcessInfo(name=ProcessName.CRON_SERVER, process=cron_proc)
             processes[ProcessName.CRON_SERVER] = new_info
-            logger.log_info("cron server restarted successfully")
+            logger.log_debug("cron server restarted successfully")
 
         case ProcessName.UWSGI:
             uwsgi_proc: AnyProcess = subprocess.Popen(["uwsgi", "--die-on-term"])
             new_info = ProcessInfo(name=ProcessName.UWSGI, process=uwsgi_proc)
             processes[ProcessName.UWSGI] = new_info
-            logger.log_info("uwsgi restarted successfully")
+            logger.log_debug("uwsgi restarted successfully")
 
 
 def check_process_alarms(
@@ -188,7 +188,7 @@ def main() -> None:
 
     def add_process(process: ProcessInfo) -> None:
         processes[process.name] = process
-        logger.log_info(f"started process {process.name}")
+        logger.log_debug(f"started process {process.name}")
 
     def _start_queue_runner() -> None:
         runner_process = multiprocessing.Process(target=start_queue_runner)

uncountable/integration/telemetry.py

@@ -94,6 +94,7 @@ def get_otel_logger() -> OTELLogger:
 
 
 class LogSeverity(StrEnum):
+    DEBUG = "Debug"
     INFO = "Info"
     WARN = "Warn"
     ERROR = "Error"
@@ -102,6 +103,8 @@ class LogSeverity(StrEnum):
 def _get_severity_number(severity: LogSeverity) -> _logs.SeverityNumber:
     """Map LogSeverity to OpenTelemetry SeverityNumber for Datadog."""
     match severity:
+        case LogSeverity.DEBUG:
+            return _logs.SeverityNumber.DEBUG
         case LogSeverity.INFO:
             return _logs.SeverityNumber.INFO
         case LogSeverity.WARN:
@@ -172,6 +175,13 @@ class Logger:
     def bind(self, *, attributes: Attributes) -> None:
         self._bound_attributes.update(attributes)
 
+    def log_debug(self, message: str, *, attributes: Attributes | None = None) -> None:
+        self._emit_log(
+            message=message, severity=LogSeverity.DEBUG, attributes=attributes
+        )
+
+    # Use log_info for user-facing job logs (e.g. example_* and custom executor output).
+    # SDK-internal instrumentation should use log_debug instead.
     def log_info(self, message: str, *, attributes: Attributes | None = None) -> None:
         self._emit_log(
             message=message, severity=LogSeverity.INFO, attributes=attributes
@@ -268,7 +278,7 @@ class PerJobResourceTracker:
                     self.max_rss = max(self.max_rss, rss)
                     now = time.monotonic()
                     if now - last_log_time >= self.log_interval:
-                        self.logger.log_info(
+                        self.logger.log_debug(
                             "Job resource usage (periodic)",
                             attributes=self._current_stats(),
                         )
@@ -310,7 +320,7 @@ class PerJobResourceTracker:
                     ),
                 )
             return
-        self.logger.log_info("Job resource usage summary", attributes=stats)
+        self.logger.log_debug("Job resource usage summary", attributes=stats)
         return
 
     def summary(self) -> Attributes:

uncountable/integration/webhook_server/entrypoint.py

@@ -91,7 +91,7 @@ def register_route(
         methods=["POST"],
     )
 
-    server_logger.log_info(
+    server_logger.log_debug(
         f"job {job.id} webhook registered at: {route}", attributes=log_attributes
     )
 

uncountable/types/__init__.py

@@ -20,6 +20,7 @@ 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 . import condition_match_t as condition_match_t
 from .api.chemical import convert_chemical_formats as convert_chemical_formats_t
 from .api.entity import create_entities as create_entities_t
 from .api.entity import create_entity as create_entity_t
@@ -145,6 +146,7 @@ from .api.material_families import update_entity_material_families as update_ent
 from .api.outputs import update_output_condition_parameter as update_output_condition_parameter_t
 from . import uploader_t as uploader_t
 from .api.condition_parameters import upsert_condition_match as upsert_condition_match_t
+from .api.condition_parameters import upsert_condition_matches as upsert_condition_matches_t
 from .api.field_options import upsert_field_options as upsert_field_options_t
 from .api.recipes import upsert_recipe_workflow_step as upsert_recipe_workflow_step_t
 from .api.recipes import upsert_step_relationships as upsert_step_relationships_t
@@ -172,6 +174,7 @@ __all__: list[str] = [
     "client_config_t",
     "complete_async_parse_t",
     "complete_async_upload_t",
+    "condition_match_t",
     "convert_chemical_formats_t",
     "create_entities_t",
     "create_entity_t",
@@ -297,6 +300,7 @@ __all__: list[str] = [
     "update_output_condition_parameter_t",
     "uploader_t",
     "upsert_condition_match_t",
+    "upsert_condition_matches_t",
     "upsert_field_options_t",
     "upsert_recipe_workflow_step_t",
     "upsert_step_relationships_t",

uncountable/types/api/condition_parameters/upsert_condition_match.py

@@ -6,11 +6,12 @@ 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_alias_annotation
 from ... import async_batch_t
 from ... import base_t
+from ... import condition_match_t
 from ... import identifier_t
 
 __all__: list[str] = [
@@ -42,11 +43,12 @@ class ConditionParameter:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-class MatchType(StrEnum):
-    ALL = "all"
-    EQ = "eq"
-    EQCATEGORICAL = "eqCategorical"
-    RANGE = "range"
+MatchType = typing.Annotated[
+    condition_match_t.MatchType,
+    serial_alias_annotation(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_match.MatchType",
+    ),
+]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/condition_parameters/upsert_condition_matches.py

@@ -0,0 +1,161 @@
+# 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 pkgs.serialization import serial_union_annotation
+from ... import async_batch_t
+from ... import base_t
+from ... import condition_match_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "ConditionParameter",
+    "ConditionParameterAll",
+    "ConditionParameterEqCategorical",
+    "ConditionParameterEqNumeric",
+    "ConditionParameterRange",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "UpsertArgs",
+    "UpsertArgsOutputConditions",
+    "UpsertArgsParameters",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/condition_parameters/upsert_condition_matches"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.ConditionParameterAll",
+    parse_require={"match_type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ConditionParameterAll:
+    condition_parameter_key: identifier_t.IdentifierKey
+    allow_null: bool
+    match_type: typing.Literal[condition_match_t.MatchType.ALL] = condition_match_t.MatchType.ALL
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.ConditionParameterEqNumeric",
+    parse_require={"match_type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ConditionParameterEqNumeric:
+    condition_parameter_key: identifier_t.IdentifierKey
+    allow_null: bool
+    numeric_values: list[Decimal | None]
+    match_type: typing.Literal[condition_match_t.MatchType.EQ] = condition_match_t.MatchType.EQ
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.ConditionParameterEqCategorical",
+    parse_require={"match_type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ConditionParameterEqCategorical:
+    condition_parameter_key: identifier_t.IdentifierKey
+    allow_null: bool
+    categorical_values: list[str | None]
+    match_type: typing.Literal[condition_match_t.MatchType.EQCATEGORICAL] = condition_match_t.MatchType.EQCATEGORICAL
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.ConditionParameterRange",
+    to_string_values={"max_value", "min_value"},
+    parse_require={"match_type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ConditionParameterRange:
+    condition_parameter_key: identifier_t.IdentifierKey
+    allow_null: bool
+    min_value: Decimal
+    max_value: Decimal
+    match_type: typing.Literal[condition_match_t.MatchType.RANGE] = condition_match_t.MatchType.RANGE
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+ConditionParameter = typing.Annotated[
+    ConditionParameterAll | ConditionParameterEqNumeric | ConditionParameterEqCategorical | ConditionParameterRange,
+    serial_union_annotation(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.ConditionParameter",
+        discriminator="match_type",
+        discriminator_map={
+            "all": ConditionParameterAll,
+            "eq": ConditionParameterEqNumeric,
+            "eqCategorical": ConditionParameterEqCategorical,
+            "range": ConditionParameterRange,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.UpsertArgsParameters",
+    parse_require={"kind"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class UpsertArgsParameters:
+    condition_parameters: list[ConditionParameter]
+    kind: typing.Literal["condition_parameters"] = "condition_parameters"
+    name: str | None = None
+    existing_condition_match_key: identifier_t.IdentifierKey | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.UpsertArgsOutputConditions",
+    parse_require={"kind"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class UpsertArgsOutputConditions:
+    output_condition_keys: list[identifier_t.IdentifierKey]
+    kind: typing.Literal["output_conditions"] = "output_conditions"
+    name: str | None = None
+    existing_condition_match_key: identifier_t.IdentifierKey | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+UpsertArgs = typing.Annotated[
+    UpsertArgsParameters | UpsertArgsOutputConditions,
+    serial_union_annotation(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.UpsertArgs",
+        discriminator="kind",
+        discriminator_map={
+            "condition_parameters": UpsertArgsParameters,
+            "output_conditions": UpsertArgsOutputConditions,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    upserts: list[UpsertArgs]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.condition_parameters.upsert_condition_matches.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data:
+    results: list[async_batch_t.AsyncBatchActionReturn]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -120,6 +120,7 @@ import uncountable.types.api.material_families.update_entity_material_families a
 import uncountable.types.api.outputs.update_output_condition_parameter as update_output_condition_parameter_t
 from uncountable.types import uploader_t
 import uncountable.types.api.condition_parameters.upsert_condition_match as upsert_condition_match_t
+import uncountable.types.api.condition_parameters.upsert_condition_matches as upsert_condition_matches_t
 import uncountable.types.api.field_options.upsert_field_options as upsert_field_options_t
 import uncountable.types.api.recipes.upsert_recipe_workflow_step as upsert_recipe_workflow_step_t
 import uncountable.types.api.recipes.upsert_step_relationships as upsert_step_relationships_t
@@ -2457,6 +2458,26 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=upsert_condition_match_t.Data)
 
+    def upsert_condition_matches(
+        self,
+        *,
+        upserts: list[upsert_condition_matches_t.UpsertArgs],
+        _request_options: client_config_t.RequestOptions | None = None,
+    ) -> upsert_condition_matches_t.Data:
+        """Bulk variant of upsert_condition_match. Accepts a list of upsert arguments and applies them via a single batched database round-trip per bulk operation (insert, update, delete) through the shared upsert_condition_matches helper. Returns one AsyncBatchActionReturn per input in the same order.
+
+        """
+        args = upsert_condition_matches_t.Arguments(
+            upserts=upserts,
+        )
+        api_request = APIRequest(
+            method=upsert_condition_matches_t.ENDPOINT_METHOD,
+            endpoint=upsert_condition_matches_t.ENDPOINT_PATH,
+            args=args,
+            request_options=_request_options,
+        )
+        return self.do_request(api_request=api_request, return_type=upsert_condition_matches_t.Data)
+
     def upsert_field_options(
         self,
         *,

uncountable/types/condition_match.py

@@ -0,0 +1,7 @@
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .condition_match_t import MatchType as MatchType
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/condition_match_t.py

@@ -0,0 +1,22 @@
+# 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
+
+__all__: list[str] = [
+    "MatchType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class MatchType(StrEnum):
+    ALL = "all"
+    EQ = "eq"
+    EQCATEGORICAL = "eqCategorical"
+    RANGE = "range"
+# DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.166.dist-info → uncountablepythonsdk-0.0.168.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.166
+Version: 0.0.168
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.166.dist-info → uncountablepythonsdk-0.0.168.dist-info}/RECORD

@@ -46,7 +46,7 @@ pkgs/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pkgs/argument_parser/__init__.py,sha256=kbarKmvTSa9lrt5hE0PUdJe3UmBHe8KbbnOLqupRZJE,1040
 pkgs/argument_parser/_is_enum.py,sha256=Gw6jJa8nBwYGqXwwCZbSnWL8Rvr5alkg5lSVAqXtOZM,257
 pkgs/argument_parser/_is_namedtuple.py,sha256=InCP2orqKbUYc4JsmE7ccri2EQPvLZeRijYPGqVSeXY,323
-pkgs/argument_parser/argument_parser.py,sha256=3qUDflMdtgC7cDZljo3-eSpikVqhT8WVb1nhul8Q_v8,26638
+pkgs/argument_parser/argument_parser.py,sha256=pjphvyGLMA2DC5qqbdiID0SsNvlekWdWtHCfcb7bRbo,26691
 pkgs/argument_parser/case_convert.py,sha256=NuJLJUJRbyVb6_Slen4uqaStEHbcOS1d-hBBfDrrw-c,605
 pkgs/argument_parser/parser_error.py,sha256=2DuYW-vQL4F1V_NEr-WW_ZdBPmH7L_uJiOuv7KhMLXw,2732
 pkgs/filesystem_utils/__init__.py,sha256=Ik9algr3R5KJkMMe-EOBiw22rSvm68yjhYZ7WIKyCQ0,1614
@@ -55,8 +55,8 @@ pkgs/filesystem_utils/_file_share_session.py,sha256=xlxCbNtOOT1xXCiL2Y_unjWZlQ99
 pkgs/filesystem_utils/_gdrive_session.py,sha256=4P2MSsA0GNxIYJxvmf7zKR8dM8oVMCCSzDIEkGA7XqE,11089
 pkgs/filesystem_utils/_local_session.py,sha256=wbmcGqqus417lY_b6VzgEaZA0SUU8Ar5MrgMkXMp58M,2339
 pkgs/filesystem_utils/_s3_session.py,sha256=8SClYopSg4Wgn6EuUZWe8SqO3oaZGzY-Ojj356AsLYg,4503
-pkgs/filesystem_utils/_sftp_session.py,sha256=N2GoSFzXDy1kMghGPKQXd4qN4_ErAG8Wcpw8I_9Ve3c,4860
-pkgs/filesystem_utils/file_type_utils.py,sha256=OL_kS1lfNFfZdwcsaXtXpYnfXfsOrUB03vc1I4H0ZEc,2116
+pkgs/filesystem_utils/_sftp_session.py,sha256=0s9y9IyYwCXZx7hPrifgQciRbNxsfWY7feF7x553lfc,4966
+pkgs/filesystem_utils/file_type_utils.py,sha256=gSZ12rnhjjc0t13_n5nV7NIXLbDj-shF-eUpZTCZtNA,2135
 pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsres6XrOQ8RoiEpcE,1045
 pkgs/serialization/__init__.py,sha256=xkCveM7syIh9N5tZ7ojjKKuFrqPwYcH9mu3iFcQ19s4,1213
 pkgs/serialization/annotation.py,sha256=UJNMb9wMUWLWW-om17HSNmfFtAEcZJcCmvirdmmGmhg,2114
@@ -110,12 +110,12 @@ uncountable/__init__.py,sha256=8l8XWNCKsu7TG94c-xa2KHpDegvxDC2FyQISdWC763Y,89
 uncountable/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/core/__init__.py,sha256=RFv0kO6rKFf1PtBPu83hCGmxqkJamRtsgQ9_-ztw7tA,341
 uncountable/core/async_batch.py,sha256=9pYGFzVCQXt8059qFHgutweGIFPquJ5Xfq6NT5P-1K0,1206
-uncountable/core/client.py,sha256=DRb2LUacLQaxwQu7YPMK4DHQd68K3CGswa8hF3D2Lac,15159
+uncountable/core/client.py,sha256=MYalNgB9SMIq0rqhAw8w43qGrwUPTqGsxlY6x4KgkS8,15160
 uncountable/core/environment.py,sha256=Z9vu7JtnSDgQB_KKcZnjTFNyARXjRr_PDW9krwxNNAo,1132
-uncountable/core/file_upload.py,sha256=oIhtUvVfbI7mJO6DaP2aHltv1DD_XV_pOnxyj3wtoLE,5729
+uncountable/core/file_upload.py,sha256=hvxz6lq1Hv4nK51bux8W_Tl22_UNBtd8U21OWteXWjI,5730
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
 uncountable/core/query/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/core/query/builder.py,sha256=Wv7KaU5n6HmN59kg7Ut-yLOBeJpvw_csMrNVwJFBOuA,7523
+uncountable/core/query/builder.py,sha256=bnsHdeV6Pe2gf2kpbVDtY8_cQmmp51CMprLfNjymN8Q,8243
 uncountable/core/query/column.py,sha256=Xxvp9C4PkHtcSI5NGAt90NBI4-xpNwpvRJr6f1L51hA,44285
 uncountable/core/query/row.py,sha256=dy68YKvMz336nyOsj_g2oNTLKBuUnKRUz5n99sApfT0,1600
 uncountable/core/query/types.py,sha256=xZgc0_O1PIxZoJWQhxkg3qnSvHj3yt-bv9QX1UCOFGw,481
@@ -127,20 +127,20 @@ uncountable/integration/entrypoint.py,sha256=BHOYPQgKvZE6HG8Rv15MkdYl8lRkvfDgv1O
 uncountable/integration/job.py,sha256=ZVcMddHdOobFHO8MZAPNH3Ht6A8Odu6uEUsqtsKmgtc,8447
 uncountable/integration/request_context.py,sha256=N_FJJxqvfUJ0yV9h3I3vFTGNJiDfyLYObczcYa44pw8,999
 uncountable/integration/scan_profiles.py,sha256=iTpzYKBHarlhYG6CKYtyAmQ7vUhEUbj2icGVHell8AU,3059
-uncountable/integration/scheduler.py,sha256=4fLxzSwmuZ0FafM3HmiThFFxfZKG9WlcQCG4fT6yyXM,8470
+uncountable/integration/scheduler.py,sha256=BajQ4txvgEBw8S9x1P0eGm-uBkKp_oecJK65SQd2hNw,8477
 uncountable/integration/server.py,sha256=P4RRGwU9jMselHPWbU6GxhRLgVtN7Ydcxr18sFn2zI8,5778
-uncountable/integration/telemetry.py,sha256=BvbL7pM_OX02UUX4gpxk-eMEEUHIqlRx9FZ2V12USPQ,14530
+uncountable/integration/telemetry.py,sha256=rWq5DUYldx6vz-L_a3dsohVX8PSJFlBUIo5NfjAYbT4,14986
 uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/db/connect.py,sha256=mE3bdV0huclH2iT_dXCQdRL4LkjIuf_myAR64RTWXEs,498
 uncountable/integration/db/session.py,sha256=96cGQXpe6IugBTdSsjdP0S5yhJ6toSmbVB6qhc3FJzE,693
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/executors/executors.py,sha256=NWe9KjJL9C76h77sutzoqhI8HkdBkLUjLrhRUmC3bpU,5147
+uncountable/integration/executors/executors.py,sha256=E3a0_yrQesfigDzjhUAwD7-l-oc9qnvPh-A8QTK6F3k,5149
 uncountable/integration/executors/generic_upload_executor.py,sha256=BdakXkAvNvLcM96fGvN0Jw2jCvGah6Q29vlXcX1nL-A,12094
 uncountable/integration/executors/script_executor.py,sha256=BBQ9f0l7uH2hgKf60jtm-pONzwk-EeOhM2qBAbv_URo,846
 uncountable/integration/http_server/__init__.py,sha256=WY2HMcL0UCAGYv8y6Pz-j0azbDGXwubFF21EH_zNPkc,189
 uncountable/integration/http_server/types.py,sha256=3JJSulRfv784SbXnXo1Pywto7RwGxgS-iJ2-a6TOnDI,1869
 uncountable/integration/queue_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/queue_runner/job_scheduler.py,sha256=RWWDADzLFoTVVIAFqSzDedh8Kcpz9ChamEnjCZZa6Aw,9799
+uncountable/integration/queue_runner/job_scheduler.py,sha256=LuhJjomXhIy7am-JsdhrF2RxtGd3U4a6mLmIbdqfAPc,9803
 uncountable/integration/queue_runner/queue_runner.py,sha256=N4sUXmlGzVquybiJ7NQZavCJOBGrxBj6k7mb-TITaN0,1139
 uncountable/integration/queue_runner/types.py,sha256=8HS6KnYMS_vc5XHeMpg0BFAQC-5P3QLzd-aDYDMMt3E,244
 uncountable/integration/queue_runner/worker.py,sha256=YU1qRGCYF7vrEcsV4zGnNslYGc6XM9WMCpoATtXB_g0,7314
@@ -160,8 +160,8 @@ uncountable/integration/queue_runner/datastore/interface.py,sha256=zonEm2O1l5GXE
 uncountable/integration/queue_runner/datastore/model.py,sha256=SH0tpXH0G7oydUbOT1z7aFZ6SNIZ6EHc9YAEaM5l1oM,832
 uncountable/integration/secret_retrieval/__init__.py,sha256=3QXVj35w8rRMxVvmmsViFYDi3lcb3g70incfalOEm6o,87
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=LBEf18KHtXZxg-ZZ80stJ1vW39AWf0CQllP6pNu3Eq8,2994
-uncountable/integration/webhook_server/entrypoint.py,sha256=TeHkpufKsbklbHXCXz57Hvh887jNOD8LV7p8x-9LE5k,4267
-uncountable/types/__init__.py,sha256=nOt7Bnspaz_A1XILTyUt9QG8JN9Wj6K67lAD_CtzPMY,13216
+uncountable/integration/webhook_server/entrypoint.py,sha256=_1360zLIvEVQVf8ToyK-E2fnaszGKs2HcxcZjNraADc,4268
+uncountable/types/__init__.py,sha256=k3T7TUSXwR7FbNComx3pP66mbc1yxz37OIy_u7ZdOi0,13421
 uncountable/types/async_batch.py,sha256=yCCWrrLQfxXVqZp-KskxLBNkNmuELdz4PJjx8ULppgs,662
 uncountable/types/async_batch_processor.py,sha256=A6ohKoJhUW2jCE8xOUNNt9ab5l6P5jWvWCNBTJPeqQk,46460
 uncountable/types/async_batch_t.py,sha256=9MIXbtMYpHOZW9MkxXb346OfvS5CjuUrBKMd7uyAfyw,4402
@@ -175,9 +175,11 @@ uncountable/types/calculations.py,sha256=fApOFpgBemt_t7IVneVR0VdI3X5EOxiG6Xhzr6R
 uncountable/types/calculations_t.py,sha256=pl-lhjyDQuj11Sf9g1-0BsSkN7Ez8UxDp8-KMQ_3enM,709
 uncountable/types/chemical_structure.py,sha256=ujyragaD26-QG5jgKnWhO7TN3N1V9b_04T2WhqNYxxo,281
 uncountable/types/chemical_structure_t.py,sha256=VFFyits_vx4t5L2euu_qFiSpsGJjURkDPr3ISnr3nPc,855
-uncountable/types/client_base.py,sha256=yD9z1uiWnym0AKSIr5eS5EYWCR0r8eF2Csag9PhqUo8,129153
+uncountable/types/client_base.py,sha256=ed9pNkxaRBwWdrMMLqcDtqt6-I-GHYZoRzVQvE5IHdU,130243
 uncountable/types/client_config.py,sha256=M7FZ0m_lGmBsIYcMn8pm92DdoVzrLpzd8sH6DqTQLKo,456
 uncountable/types/client_config_t.py,sha256=m1sVg0zaVf8DvQKaDJH_fG1abU2-VZZH0wOIMDjlECU,1944
+uncountable/types/condition_match.py,sha256=ekDzij7-e1PtVwIslSjD1T9fuBIDr5c_7QPdaasiGJw,262
+uncountable/types/condition_match_t.py,sha256=iykByQKFtFa2Gk3sqi9xkqoLHeztRHagV4zkiXqwUh4,536
 uncountable/types/curves.py,sha256=QyEyC20jsG-LGKVx6miiF-w70vKMwNkILFBDIJ5Ok9g,345
 uncountable/types/curves_t.py,sha256=DxYepdC3QKKR7mepOOBoyarNcFZQdUa5ZYH-hwCY3BI,1469
 uncountable/types/data.py,sha256=u2isf4XEug3Eu-xSIoqGaCQmW2dFaKBHCkP_WKYwwBc,500
@@ -281,7 +283,8 @@ uncountable/types/api/batch/execute_batch_load_async.py,sha256=YLr3Ae3xruoZv3BHL
 uncountable/types/api/chemical/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/chemical/convert_chemical_formats.py,sha256=V-s2EMqjodA_zS7-fOdo4iI4yFrvOc2bAwRFCsl445Y,1961
 uncountable/types/api/condition_parameters/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
-uncountable/types/api/condition_parameters/upsert_condition_match.py,sha256=7I9lAiYfu8_abkKdmDtw11br_Ii9HCqBpGoaXBr5Vog,2319
+uncountable/types/api/condition_parameters/upsert_condition_match.py,sha256=VpC3ZNTmP9qxyuoVy6MOnJR1QMH7lnbKmaLpskvbElU,2457
+uncountable/types/api/condition_parameters/upsert_condition_matches.py,sha256=iooN1XX9jqXbS16Tj1iAJ5AglLm5q8J8PJW7SX9Fwuk,6073
 uncountable/types/api/entity/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/entity/create_entities.py,sha256=cCDEra2SHvGWvz7nIxxMDSQN6OWrHMTT0JSomWUesto,1794
 uncountable/types/api/entity/create_entity.py,sha256=urT6C7iGAa7_rCv9Wcz6GM_lKg1tP55E__rjNkj-Rjc,1879
@@ -395,7 +398,7 @@ uncountable/types/api/uploader/complete_async_parse.py,sha256=ffS3ApqCNkZb6QPuYE
 uncountable/types/api/uploader/invoke_uploader.py,sha256=Bj7Dq4A90k00suacwk3bLA_dCb2aovS1kAbVam2AQnM,1395
 uncountable/types/api/user/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/user/get_current_user_info.py,sha256=Avqi_RXtRgbefrT_dwJ9MrO6eDNSSa_Nu650FSuESlg,1109
-uncountablepythonsdk-0.0.166.dist-info/METADATA,sha256=E-holHf_W4PtrzbwUMmVB-6mJ-3xaKNhUs7sZSAzJ4I,2289
-uncountablepythonsdk-0.0.166.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-uncountablepythonsdk-0.0.166.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-uncountablepythonsdk-0.0.166.dist-info/RECORD,,
+uncountablepythonsdk-0.0.168.dist-info/METADATA,sha256=Eh2iAbMOCD1M_CLUBemNthcUHEjuZ2DMeTfZeruzGYk,2289
+uncountablepythonsdk-0.0.168.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+uncountablepythonsdk-0.0.168.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+uncountablepythonsdk-0.0.168.dist-info/RECORD,,