UncountablePythonSDK 0.0.177__py3-none-any.whl → 0.0.179__py3-none-any.whl

examples/integration-server/pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
   "ruff == 0.*",
   "openpyxl == 3.*",
   "more_itertools ==11.*",
-  "types-paramiko ==4.0.0.20260518",
+  "types-paramiko ==5.0.0.20260617",
   "types-openpyxl == 3.*",
   "types-pytz ==2026.*",
   "types-requests == 2.*",

pkgs/filesystem_utils/__init__.py

@@ -1,3 +1,5 @@
+# CLOSED MODULE
+
 from ._blob_session import BlobSession as BlobSession
 from ._file_share_session import FileShareSession as FileShareSession
 from ._gdrive_session import (
@@ -32,3 +34,32 @@ from .file_type_utils import (
     RemoteObjectReference as RemoteObjectReference,
 )
 from .filesystem_session import FileSystemSession as FileSystemSession
+
+__all__ = [
+    "BlobSession",
+    "FileObjectData",
+    "FileShareSession",
+    "FileSystemBlobConfig",
+    "FileSystemFileReference",
+    "FileSystemFileShareConfig",
+    "FileSystemObject",
+    "FileSystemS3Config",
+    "FileSystemSFTPConfig",
+    "FileSystemSession",
+    "FileTransfer",
+    "GDriveSession",
+    "IncompatibleFileReference",
+    "LocalSession",
+    "RemoteObjectReference",
+    "S3Session",
+    "SFTPConnection",
+    "SFTPCredentialError",
+    "SFTPSession",
+    "delete_gdrive_file",
+    "download_gdrive_file",
+    "list_gdrive_files",
+    "list_sftp_files",
+    "move_gdrive_file",
+    "move_sftp_files",
+    "upload_file_gdrive",
+]

pkgs/filesystem_utils/_file_share_session.py

@@ -26,6 +26,7 @@ class FileShareSession(FileSystemSession):
             account_url=self.config.account_url,
             share_name=self.config.share_name,
             credential=self.config.credential,
+            token_intent="backup",
         )
 
     def __enter__(self) -> Self:

pkgs/serialization/opaque_key.py

@@ -3,5 +3,7 @@ from typing import Self
 
 # Blocks a string key value from being interpreted for case conversion
 class OpaqueKey(str):  # noqa: FURB189
+    __slots__ = ()
+
     def __new__(cls, key: str) -> Self:
         return str.__new__(cls, key)

pkgs/strenum_compat/__init__.py

@@ -1,3 +1,5 @@
+# CLOSED MODULE
+
 from .strenum_compat import StrEnum
 
 __all__ = ["StrEnum"]

uncountable/core/client.py

@@ -201,7 +201,7 @@ DownloadedFiles = list[DownloadedFile]
 
 
 class Client(ClientMethods):
-    _parser_map: dict[type, CachedParser] = {}
+    _parser_map: typing.ClassVar[dict[type, CachedParser]] = {}
     _auth_details: AuthDetailsAll
     _base_url: str
     _file_uploader: FileUploader
@@ -298,6 +298,10 @@ class Client(ClientMethods):
                 else None
             )
             response = self._send_request(request, timeout=timeout)
+            if self._cfg.logger is not None:
+                self._cfg.logger.set_span_attribute(
+                    "http.status_code", response.status_code
+                )
         response_data = self._get_response_json(response, request_id=request_id)
         cached_parser = self._get_cached_parser(return_type)
         try:

uncountable/integration/db/connect.py

@@ -1,18 +1,48 @@
 import os
 from enum import StrEnum
 
+from opentelemetry.trace import get_current_span
 from sqlalchemy import create_engine
 from sqlalchemy.engine.base import Engine
 
+from uncountable.integration.telemetry import Logger
+
 
 class IntegrationDBService(StrEnum):
     CRON = "cron"
     RUNNER = "runner"
 
 
+_DB_URI_ENV_VARS: dict[IntegrationDBService, str] = {
+    IntegrationDBService.CRON: "UNC_CRON_DB_URI",
+    IntegrationDBService.RUNNER: "UNC_RUNNER_DB_URI",
+}
+
+_DEPRECATED_DB_URI_ENV_VARS: dict[IntegrationDBService, str] = {
+    IntegrationDBService.CRON: "UNC_CRON_SQLITE_URI",
+    IntegrationDBService.RUNNER: "UNC_RUNNER_SQLITE_URI",
+}
+
+
+def _resolve_db_uri(service: IntegrationDBService) -> str:
+    logger = Logger(get_current_span())
+    env_var = _DB_URI_ENV_VARS[service]
+    uri = os.environ.get(env_var)
+    if uri is not None:
+        return uri
+
+    deprecated_env_var = _DEPRECATED_DB_URI_ENV_VARS[service]
+    deprecated_uri = os.environ.get(deprecated_env_var)
+    if deprecated_uri is not None:
+        logger.log_warning(
+            f"{deprecated_env_var} is deprecated and will be removed; set {env_var} instead.",
+        )
+        return deprecated_uri
+
+    raise KeyError(
+        f"no database URI configured for the {service} service: set {env_var}"
+    )
+
+
 def create_db_engine(service: IntegrationDBService) -> Engine:
-    match service:
-        case IntegrationDBService.CRON:
-            return create_engine(os.environ["UNC_CRON_SQLITE_URI"])
-        case IntegrationDBService.RUNNER:
-            return create_engine(os.environ["UNC_RUNNER_SQLITE_URI"])
+    return create_engine(_resolve_db_uri(service))

uncountable/integration/queue_runner/command_server/command_server.py

@@ -41,7 +41,7 @@ from uncountable.integration.queue_runner.command_server.types import (
     CommandVaccuumQueuedJobs,
     CommandVaccuumQueuedJobsResponse,
 )
-from uncountable.integration.queue_runner.datastore import DatastoreSqlite
+from uncountable.integration.queue_runner.datastore import Datastore
 from uncountable.types import queued_job_t
 
 from .constants import ListQueuedJobsConstants
@@ -53,7 +53,7 @@ from .protocol.command_server_pb2_grpc import (
 queued_job_payload_parser = CachedParser(queued_job_t.QueuedJobPayload)
 
 
-async def serve(command_queue: CommandQueue, datastore: DatastoreSqlite) -> None:
+async def serve(command_queue: CommandQueue, datastore: Datastore) -> None:
     server = grpc_aio.server()
 
     class CommandServerHandler(CommandServerServicer):

uncountable/integration/queue_runner/datastore/__init__.py

@@ -1,3 +1,11 @@
+from .datastore_postgres import DatastorePostgres
 from .datastore_sqlite import DatastoreSqlite
+from .factory import create_datastore
+from .interface import Datastore
 
-__all__: list[str] = ["DatastoreSqlite"]
+__all__: list[str] = [
+    "Datastore",
+    "DatastorePostgres",
+    "DatastoreSqlite",
+    "create_datastore",
+]

uncountable/integration/queue_runner/datastore/datastore_postgres.py

@@ -0,0 +1,12 @@
+from sqlalchemy.engine import Engine
+
+from uncountable.integration.queue_runner.datastore.datastore_sql_base import (
+    DatastoreSqlBase,
+)
+from uncountable.integration.queue_runner.datastore.model import Base
+
+
+class DatastorePostgres(DatastoreSqlBase):
+    @classmethod
+    def setup(cls, engine: Engine) -> None:
+        Base.metadata.create_all(engine)

uncountable/integration/queue_runner/datastore/datastore_sql_base.py

@@ -0,0 +1,286 @@
+import datetime
+import uuid
+from datetime import UTC
+
+from sqlalchemy import delete, insert, or_, select, update
+
+from pkgs.argument_parser import CachedParser
+from pkgs.serialization_util import serialize_for_storage
+from uncountable.integration.db.session import DBSessionMaker
+from uncountable.integration.queue_runner.datastore.interface import Datastore
+from uncountable.integration.queue_runner.datastore.model import QueuedJob
+from uncountable.types import queued_job_t
+
+queued_job_payload_parser = CachedParser(queued_job_t.QueuedJobPayload)
+
+MAX_QUEUE_WINDOW_DAYS = 30
+
+
+class DatastoreSqlBase(Datastore):
+    def __init__(self, session_maker: DBSessionMaker) -> None:
+        self.session_maker = session_maker
+        super().__init__()
+
+    def add_job_to_queue(
+        self, job_payload: queued_job_t.QueuedJobPayload, job_ref_name: str
+    ) -> queued_job_t.QueuedJob:
+        with self.session_maker() as session:
+            serialized_payload = serialize_for_storage(job_payload)
+            queued_job_uuid = str(uuid.uuid4())
+            num_attempts = 0
+            submitted_at = datetime.datetime.now(UTC)
+            insert_stmt = insert(QueuedJob).values({
+                QueuedJob.id.key: queued_job_uuid,
+                QueuedJob.job_ref_name.key: job_ref_name,
+                QueuedJob.payload.key: serialized_payload,
+                QueuedJob.status.key: queued_job_t.JobStatus.QUEUED,
+                QueuedJob.num_attempts: num_attempts,
+                QueuedJob.submitted_at: submitted_at,
+            })
+            session.execute(insert_stmt)
+            return queued_job_t.QueuedJob(
+                queued_job_uuid=queued_job_uuid,
+                job_ref_name=job_ref_name,
+                payload=job_payload,
+                status=queued_job_t.JobStatus.QUEUED,
+                submitted_at=submitted_at,
+                num_attempts=num_attempts,
+            )
+
+    def retry_job(
+        self,
+        queued_job_uuid: str,
+    ) -> queued_job_t.QueuedJob | None:
+        with self.session_maker() as session:
+            select_stmt = select(
+                QueuedJob.id,
+                QueuedJob.payload,
+                QueuedJob.num_attempts,
+                QueuedJob.job_ref_name,
+                QueuedJob.status,
+                QueuedJob.submitted_at,
+            ).filter(QueuedJob.id == queued_job_uuid)
+            existing_job = session.execute(select_stmt).one_or_none()
+
+            if (
+                existing_job is None
+                or existing_job.status != queued_job_t.JobStatus.FAILED
+            ):
+                return None
+
+            update_stmt = (
+                update(QueuedJob)
+                .values({QueuedJob.status.key: queued_job_t.JobStatus.QUEUED})
+                .filter(QueuedJob.id == queued_job_uuid)
+            )
+            session.execute(update_stmt)
+
+            return queued_job_t.QueuedJob(
+                queued_job_uuid=existing_job.id,
+                job_ref_name=existing_job.job_ref_name,
+                num_attempts=existing_job.num_attempts,
+                status=queued_job_t.JobStatus.QUEUED,
+                submitted_at=existing_job.submitted_at,
+                payload=queued_job_payload_parser.parse_storage(existing_job.payload),
+            )
+
+    def increment_num_attempts(self, queued_job_uuid: str) -> int:
+        with self.session_maker() as session:
+            update_stmt = (
+                update(QueuedJob)
+                .values({QueuedJob.num_attempts.key: QueuedJob.num_attempts + 1})
+                .filter(QueuedJob.id == queued_job_uuid)
+            )
+            session.execute(update_stmt)
+            session.flush()
+            # IMPROVE: python3's sqlite does not support the RETURNING clause
+            select_stmt = select(QueuedJob.num_attempts).filter(
+                QueuedJob.id == queued_job_uuid
+            )
+            return int(session.execute(select_stmt).one().num_attempts)
+
+    def remove_job_from_queue(self, queued_job_uuid: str) -> None:
+        with self.session_maker() as session:
+            delete_stmt = delete(QueuedJob).filter(QueuedJob.id == queued_job_uuid)
+            session.execute(delete_stmt)
+
+    def update_job_status(
+        self, queued_job_uuid: str, status: queued_job_t.JobStatus
+    ) -> None:
+        with self.session_maker() as session:
+            update_stmt = (
+                update(QueuedJob)
+                .values({QueuedJob.status.key: status})
+                .filter(QueuedJob.id == queued_job_uuid)
+            )
+            session.execute(update_stmt)
+
+    def list_queued_job_metadata(
+        self,
+        *,
+        offset: int = 0,
+        limit: int | None = 100,
+        status: queued_job_t.JobStatus | None = None,
+        job_ref_name: str | None = None,
+        submitted_from: datetime.datetime | None = None,
+        submitted_to: datetime.datetime | None = None,
+    ) -> list[queued_job_t.QueuedJobMetadata]:
+        with self.session_maker() as session:
+            select_statement = (
+                select(
+                    QueuedJob.id,
+                    QueuedJob.job_ref_name,
+                    QueuedJob.num_attempts,
+                    QueuedJob.status,
+                    QueuedJob.submitted_at,
+                )
+                .order_by(QueuedJob.submitted_at.desc())
+                .offset(offset)
+                .limit(limit)
+            )
+
+            if status is not None:
+                select_statement = select_statement.filter(QueuedJob.status == status)
+            if job_ref_name is not None:
+                select_statement = select_statement.filter(
+                    QueuedJob.job_ref_name == job_ref_name
+                )
+            if submitted_from is not None:
+                select_statement = select_statement.filter(
+                    QueuedJob.submitted_at >= submitted_from
+                )
+            if submitted_to is not None:
+                select_statement = select_statement.filter(
+                    QueuedJob.submitted_at < submitted_to
+                )
+
+            queued_job_metadata: list[queued_job_t.QueuedJobMetadata] = [
+                queued_job_t.QueuedJobMetadata(
+                    queued_job_uuid=row.id,
+                    job_ref_name=row.job_ref_name,
+                    num_attempts=row.num_attempts,
+                    status=row.status or queued_job_t.JobStatus.QUEUED,
+                    submitted_at=row.submitted_at,
+                )
+                for row in session.execute(select_statement)
+            ]
+
+            return queued_job_metadata
+
+    def get_next_queued_job_for_ref_name(
+        self, job_ref_name: str
+    ) -> queued_job_t.QueuedJob | None:
+        with self.session_maker() as session:
+            select_stmt = (
+                select(
+                    QueuedJob.id,
+                    QueuedJob.payload,
+                    QueuedJob.num_attempts,
+                    QueuedJob.job_ref_name,
+                    QueuedJob.status,
+                    QueuedJob.submitted_at,
+                )
+                .filter(QueuedJob.job_ref_name == job_ref_name)
+                .filter(
+                    or_(
+                        QueuedJob.status == queued_job_t.JobStatus.QUEUED,
+                        QueuedJob.status.is_(None),
+                    )
+                )
+                .limit(1)
+                .order_by(QueuedJob.submitted_at)
+            )
+
+            for row in session.execute(select_stmt):
+                parsed_payload = queued_job_payload_parser.parse_storage(row.payload)
+                return queued_job_t.QueuedJob(
+                    queued_job_uuid=row.id,
+                    job_ref_name=row.job_ref_name,
+                    num_attempts=row.num_attempts,
+                    status=row.status or queued_job_t.JobStatus.QUEUED,
+                    submitted_at=row.submitted_at,
+                    payload=parsed_payload,
+                )
+
+            return None
+
+    def load_job_queue(self) -> list[queued_job_t.QueuedJob]:
+        with self.session_maker() as session:
+            select_stmt = (
+                select(
+                    QueuedJob.id,
+                    QueuedJob.payload,
+                    QueuedJob.num_attempts,
+                    QueuedJob.job_ref_name,
+                    QueuedJob.status,
+                    QueuedJob.submitted_at,
+                )
+                .filter(
+                    or_(
+                        QueuedJob.status == queued_job_t.JobStatus.QUEUED,
+                        QueuedJob.status.is_(None),
+                    )
+                )
+                .order_by(QueuedJob.submitted_at)
+            )
+
+            queued_jobs: list[queued_job_t.QueuedJob] = []
+            for row in session.execute(select_stmt):
+                parsed_payload = queued_job_payload_parser.parse_storage(row.payload)
+                queued_jobs.append(
+                    queued_job_t.QueuedJob(
+                        queued_job_uuid=row.id,
+                        job_ref_name=row.job_ref_name,
+                        num_attempts=row.num_attempts,
+                        status=row.status or queued_job_t.JobStatus.QUEUED,
+                        submitted_at=row.submitted_at,
+                        payload=parsed_payload,
+                    )
+                )
+
+            return queued_jobs
+
+    def get_queued_job(self, *, uuid: str) -> queued_job_t.QueuedJob | None:
+        with self.session_maker() as session:
+            select_stmt = select(
+                QueuedJob.id,
+                QueuedJob.payload,
+                QueuedJob.num_attempts,
+                QueuedJob.job_ref_name,
+                QueuedJob.status,
+                QueuedJob.submitted_at,
+            ).filter(QueuedJob.id == uuid)
+
+            row = session.execute(select_stmt).one_or_none()
+            return (
+                queued_job_t.QueuedJob(
+                    queued_job_uuid=row.id,
+                    job_ref_name=row.job_ref_name,
+                    num_attempts=row.num_attempts,
+                    status=row.status or queued_job_t.JobStatus.QUEUED,
+                    submitted_at=row.submitted_at,
+                    payload=queued_job_payload_parser.parse_storage(row.payload),
+                )
+                if row is not None
+                else None
+            )
+
+    def vaccuum_queued_jobs(self) -> list[str]:
+        with self.session_maker() as session:
+            select_stmt = (
+                select(QueuedJob.id)
+                .filter(QueuedJob.status == queued_job_t.JobStatus.QUEUED)
+                .filter(
+                    QueuedJob.submitted_at
+                    <= (
+                        datetime.datetime.now(UTC)
+                        - datetime.timedelta(days=MAX_QUEUE_WINDOW_DAYS)
+                    )
+                )
+            )
+            deleted_uuids: list[str] = [row.id for row in session.execute(select_stmt)]
+
+            delete_stmt = delete(QueuedJob).filter(QueuedJob.id.in_(deleted_uuids))
+            session.execute(delete_stmt)
+
+        return deleted_uuids

uncountable/integration/queue_runner/datastore/datastore_sqlite.py

@@ -1,27 +1,13 @@
-import datetime
-import uuid
-from datetime import UTC
-
-from sqlalchemy import delete, insert, or_, select, text, update
+from sqlalchemy import text
 from sqlalchemy.engine import Engine
 
-from pkgs.argument_parser import CachedParser
-from pkgs.serialization_util import serialize_for_storage
-from uncountable.integration.db.session import DBSessionMaker
-from uncountable.integration.queue_runner.datastore.interface import Datastore
-from uncountable.integration.queue_runner.datastore.model import Base, QueuedJob
-from uncountable.types import queued_job_t
-
-queued_job_payload_parser = CachedParser(queued_job_t.QueuedJobPayload)
-
-MAX_QUEUE_WINDOW_DAYS = 30
-
+from uncountable.integration.queue_runner.datastore.datastore_sql_base import (
+    DatastoreSqlBase,
+)
+from uncountable.integration.queue_runner.datastore.model import Base
 
-class DatastoreSqlite(Datastore):
-    def __init__(self, session_maker: DBSessionMaker) -> None:
-        self.session_maker = session_maker
-        super().__init__()
 
+class DatastoreSqlite(DatastoreSqlBase):
     @classmethod
     def setup(cls, engine: Engine) -> None:
         Base.metadata.create_all(engine)
@@ -46,267 +32,3 @@ class DatastoreSqlite(Datastore):
                     "create index if not exists ix_queued_jobs_submitted_at on queued_jobs (submitted_at)"
                 )
             )
-
-    def add_job_to_queue(
-        self, job_payload: queued_job_t.QueuedJobPayload, job_ref_name: str
-    ) -> queued_job_t.QueuedJob:
-        with self.session_maker() as session:
-            serialized_payload = serialize_for_storage(job_payload)
-            queued_job_uuid = str(uuid.uuid4())
-            num_attempts = 0
-            submitted_at = datetime.datetime.now(UTC)
-            insert_stmt = insert(QueuedJob).values({
-                QueuedJob.id.key: queued_job_uuid,
-                QueuedJob.job_ref_name.key: job_ref_name,
-                QueuedJob.payload.key: serialized_payload,
-                QueuedJob.status.key: queued_job_t.JobStatus.QUEUED,
-                QueuedJob.num_attempts: num_attempts,
-                QueuedJob.submitted_at: submitted_at,
-            })
-            session.execute(insert_stmt)
-            return queued_job_t.QueuedJob(
-                queued_job_uuid=queued_job_uuid,
-                job_ref_name=job_ref_name,
-                payload=job_payload,
-                status=queued_job_t.JobStatus.QUEUED,
-                submitted_at=submitted_at,
-                num_attempts=num_attempts,
-            )
-
-    def retry_job(
-        self,
-        queued_job_uuid: str,
-    ) -> queued_job_t.QueuedJob | None:
-        with self.session_maker() as session:
-            select_stmt = select(
-                QueuedJob.id,
-                QueuedJob.payload,
-                QueuedJob.num_attempts,
-                QueuedJob.job_ref_name,
-                QueuedJob.status,
-                QueuedJob.submitted_at,
-            ).filter(QueuedJob.id == queued_job_uuid)
-            existing_job = session.execute(select_stmt).one_or_none()
-
-            if (
-                existing_job is None
-                or existing_job.status != queued_job_t.JobStatus.FAILED
-            ):
-                return None
-
-            update_stmt = (
-                update(QueuedJob)
-                .values({QueuedJob.status.key: queued_job_t.JobStatus.QUEUED})
-                .filter(QueuedJob.id == queued_job_uuid)
-            )
-            session.execute(update_stmt)
-
-            return queued_job_t.QueuedJob(
-                queued_job_uuid=existing_job.id,
-                job_ref_name=existing_job.job_ref_name,
-                num_attempts=existing_job.num_attempts,
-                status=queued_job_t.JobStatus.QUEUED,
-                submitted_at=existing_job.submitted_at,
-                payload=queued_job_payload_parser.parse_storage(existing_job.payload),
-            )
-
-    def increment_num_attempts(self, queued_job_uuid: str) -> int:
-        with self.session_maker() as session:
-            update_stmt = (
-                update(QueuedJob)
-                .values({QueuedJob.num_attempts.key: QueuedJob.num_attempts + 1})
-                .filter(QueuedJob.id == queued_job_uuid)
-            )
-            session.execute(update_stmt)
-            session.flush()
-            # IMPROVE: python3's sqlite does not support the RETURNING clause
-            select_stmt = select(QueuedJob.num_attempts).filter(
-                QueuedJob.id == queued_job_uuid
-            )
-            return int(session.execute(select_stmt).one().num_attempts)
-
-    def remove_job_from_queue(self, queued_job_uuid: str) -> None:
-        with self.session_maker() as session:
-            delete_stmt = delete(QueuedJob).filter(QueuedJob.id == queued_job_uuid)
-            session.execute(delete_stmt)
-
-    def update_job_status(
-        self, queued_job_uuid: str, status: queued_job_t.JobStatus
-    ) -> None:
-        with self.session_maker() as session:
-            update_stmt = (
-                update(QueuedJob)
-                .values({QueuedJob.status.key: status})
-                .filter(QueuedJob.id == queued_job_uuid)
-            )
-            session.execute(update_stmt)
-
-    def list_queued_job_metadata(
-        self,
-        *,
-        offset: int = 0,
-        limit: int | None = 100,
-        status: queued_job_t.JobStatus | None = None,
-        job_ref_name: str | None = None,
-        submitted_from: datetime.datetime | None = None,
-        submitted_to: datetime.datetime | None = None,
-    ) -> list[queued_job_t.QueuedJobMetadata]:
-        with self.session_maker() as session:
-            select_statement = (
-                select(
-                    QueuedJob.id,
-                    QueuedJob.job_ref_name,
-                    QueuedJob.num_attempts,
-                    QueuedJob.status,
-                    QueuedJob.submitted_at,
-                )
-                .order_by(QueuedJob.submitted_at.desc())
-                .offset(offset)
-                .limit(limit)
-            )
-
-            if status is not None:
-                select_statement = select_statement.filter(QueuedJob.status == status)
-            if job_ref_name is not None:
-                select_statement = select_statement.filter(
-                    QueuedJob.job_ref_name == job_ref_name
-                )
-            if submitted_from is not None:
-                select_statement = select_statement.filter(
-                    QueuedJob.submitted_at >= submitted_from
-                )
-            if submitted_to is not None:
-                select_statement = select_statement.filter(
-                    QueuedJob.submitted_at < submitted_to
-                )
-
-            queued_job_metadata: list[queued_job_t.QueuedJobMetadata] = [
-                queued_job_t.QueuedJobMetadata(
-                    queued_job_uuid=row.id,
-                    job_ref_name=row.job_ref_name,
-                    num_attempts=row.num_attempts,
-                    status=row.status or queued_job_t.JobStatus.QUEUED,
-                    submitted_at=row.submitted_at,
-                )
-                for row in session.execute(select_statement)
-            ]
-
-            return queued_job_metadata
-
-    def get_next_queued_job_for_ref_name(
-        self, job_ref_name: str
-    ) -> queued_job_t.QueuedJob | None:
-        with self.session_maker() as session:
-            select_stmt = (
-                select(
-                    QueuedJob.id,
-                    QueuedJob.payload,
-                    QueuedJob.num_attempts,
-                    QueuedJob.job_ref_name,
-                    QueuedJob.status,
-                    QueuedJob.submitted_at,
-                )
-                .filter(QueuedJob.job_ref_name == job_ref_name)
-                .filter(
-                    or_(
-                        QueuedJob.status == queued_job_t.JobStatus.QUEUED,
-                        QueuedJob.status.is_(None),
-                    )
-                )
-                .limit(1)
-                .order_by(QueuedJob.submitted_at)
-            )
-
-            for row in session.execute(select_stmt):
-                parsed_payload = queued_job_payload_parser.parse_storage(row.payload)
-                return queued_job_t.QueuedJob(
-                    queued_job_uuid=row.id,
-                    job_ref_name=row.job_ref_name,
-                    num_attempts=row.num_attempts,
-                    status=row.status or queued_job_t.JobStatus.QUEUED,
-                    submitted_at=row.submitted_at,
-                    payload=parsed_payload,
-                )
-
-            return None
-
-    def load_job_queue(self) -> list[queued_job_t.QueuedJob]:
-        with self.session_maker() as session:
-            select_stmt = (
-                select(
-                    QueuedJob.id,
-                    QueuedJob.payload,
-                    QueuedJob.num_attempts,
-                    QueuedJob.job_ref_name,
-                    QueuedJob.status,
-                    QueuedJob.submitted_at,
-                )
-                .filter(
-                    or_(
-                        QueuedJob.status == queued_job_t.JobStatus.QUEUED,
-                        QueuedJob.status.is_(None),
-                    )
-                )
-                .order_by(QueuedJob.submitted_at)
-            )
-
-            queued_jobs: list[queued_job_t.QueuedJob] = []
-            for row in session.execute(select_stmt):
-                parsed_payload = queued_job_payload_parser.parse_storage(row.payload)
-                queued_jobs.append(
-                    queued_job_t.QueuedJob(
-                        queued_job_uuid=row.id,
-                        job_ref_name=row.job_ref_name,
-                        num_attempts=row.num_attempts,
-                        status=row.status or queued_job_t.JobStatus.QUEUED,
-                        submitted_at=row.submitted_at,
-                        payload=parsed_payload,
-                    )
-                )
-
-            return queued_jobs
-
-    def get_queued_job(self, *, uuid: str) -> queued_job_t.QueuedJob | None:
-        with self.session_maker() as session:
-            select_stmt = select(
-                QueuedJob.id,
-                QueuedJob.payload,
-                QueuedJob.num_attempts,
-                QueuedJob.job_ref_name,
-                QueuedJob.status,
-                QueuedJob.submitted_at,
-            ).filter(QueuedJob.id == uuid)
-
-            row = session.execute(select_stmt).one_or_none()
-            return (
-                queued_job_t.QueuedJob(
-                    queued_job_uuid=row.id,
-                    job_ref_name=row.job_ref_name,
-                    num_attempts=row.num_attempts,
-                    status=row.status or queued_job_t.JobStatus.QUEUED,
-                    submitted_at=row.submitted_at,
-                    payload=queued_job_payload_parser.parse_storage(row.payload),
-                )
-                if row is not None
-                else None
-            )
-
-    def vaccuum_queued_jobs(self) -> list[str]:
-        with self.session_maker() as session:
-            select_stmt = (
-                select(QueuedJob.id)
-                .filter(QueuedJob.status == queued_job_t.JobStatus.QUEUED)
-                .filter(
-                    QueuedJob.submitted_at
-                    <= (
-                        datetime.datetime.now(UTC)
-                        - datetime.timedelta(days=MAX_QUEUE_WINDOW_DAYS)
-                    )
-                )
-            )
-            deleted_uuids: list[str] = [row.id for row in session.execute(select_stmt)]
-
-            delete_stmt = delete(QueuedJob).filter(QueuedJob.id.in_(deleted_uuids))
-            session.execute(delete_stmt)
-
-        return deleted_uuids

uncountable/integration/queue_runner/datastore/factory.py

@@ -0,0 +1,24 @@
+from sqlalchemy.engine import Engine
+
+from uncountable.integration.db.session import get_session_maker
+from uncountable.integration.queue_runner.datastore.datastore_postgres import (
+    DatastorePostgres,
+)
+from uncountable.integration.queue_runner.datastore.datastore_sqlite import (
+    DatastoreSqlite,
+)
+from uncountable.integration.queue_runner.datastore.interface import Datastore
+
+
+def create_datastore(engine: Engine) -> Datastore:
+    session_maker = get_session_maker(engine)
+    dialect_name = engine.dialect.name
+    match dialect_name:
+        case "sqlite":
+            datastore: Datastore = DatastoreSqlite(session_maker)
+        case "postgresql":
+            datastore = DatastorePostgres(session_maker)
+        case _:
+            raise ValueError(f"unsupported database dialect: {dialect_name!r}")
+    datastore.setup(engine)
+    return datastore

uncountable/integration/queue_runner/datastore/interface.py

@@ -1,10 +1,16 @@
 from abc import ABC, abstractmethod
 from datetime import datetime
 
+from sqlalchemy.engine import Engine
+
 from uncountable.types import queued_job_t
 
 
 class Datastore(ABC):
+    @classmethod
+    @abstractmethod
+    def setup(cls, engine: Engine) -> None: ...
+
     @abstractmethod
     def add_job_to_queue(
         self, job_payload: queued_job_t.QueuedJobPayload, job_ref_name: str
@@ -13,6 +19,14 @@ class Datastore(ABC):
     @abstractmethod
     def remove_job_from_queue(self, queued_job_uuid: str) -> None: ...
 
+    @abstractmethod
+    def update_job_status(
+        self, queued_job_uuid: str, status: queued_job_t.JobStatus
+    ) -> None: ...
+
+    @abstractmethod
+    def retry_job(self, queued_job_uuid: str) -> queued_job_t.QueuedJob | None: ...
+
     @abstractmethod
     def increment_num_attempts(self, queued_job_uuid: str) -> int: ...
 
@@ -38,3 +52,6 @@ class Datastore(ABC):
 
     @abstractmethod
     def get_queued_job(self, *, uuid: str) -> queued_job_t.QueuedJob | None: ...
+
+    @abstractmethod
+    def vaccuum_queued_jobs(self) -> list[str]: ...

uncountable/integration/queue_runner/datastore/model.py

@@ -21,7 +21,7 @@ class QueuedJob(Base):
     payload = Column(JSON, nullable=False)
     num_attempts = Column(BigInteger, nullable=False, default=0, server_default="0")
     status = Column(
-        Enum(queued_job_t.JobStatus, length=None),
+        Enum(queued_job_t.JobStatus, length=None, native_enum=False),
         default=queued_job_t.JobStatus.QUEUED,
         nullable=True,
         index=True,

uncountable/integration/queue_runner/job_scheduler.py

@@ -23,8 +23,7 @@ from uncountable.integration.queue_runner.command_server.types import (
     CommandCancelJobStatus,
     CommandVaccuumQueuedJobs,
 )
-from uncountable.integration.queue_runner.datastore import DatastoreSqlite
-from uncountable.integration.queue_runner.datastore.interface import Datastore
+from uncountable.integration.queue_runner.datastore import Datastore
 from uncountable.integration.queue_runner.worker import Worker
 from uncountable.integration.scan_profiles import load_profiles
 from uncountable.integration.telemetry import Logger
@@ -90,9 +89,7 @@ def _start_workers(
     return job_worker_lookup
 
 
-async def start_scheduler(
-    command_queue: CommandQueue, datastore: DatastoreSqlite
-) -> None:
+async def start_scheduler(command_queue: CommandQueue, datastore: Datastore) -> None:
     logger = Logger(get_current_span())
     result_queue: ResultQueue = asyncio.Queue()
 

uncountable/integration/queue_runner/queue_runner.py

@@ -1,20 +1,16 @@
 import asyncio
 
 from uncountable.integration.db.connect import IntegrationDBService, create_db_engine
-from uncountable.integration.db.session import get_session_maker
 from uncountable.integration.queue_runner.command_server import serve
 from uncountable.integration.queue_runner.command_server.types import CommandQueue
-from uncountable.integration.queue_runner.datastore import DatastoreSqlite
+from uncountable.integration.queue_runner.datastore import create_datastore
 from uncountable.integration.queue_runner.job_scheduler import start_scheduler
 
 
 async def queue_runner_loop() -> None:
     command_queue: CommandQueue = asyncio.Queue()
     engine = create_db_engine(IntegrationDBService.RUNNER)
-    session_maker = get_session_maker(engine)
-
-    datastore = DatastoreSqlite(session_maker)
-    datastore.setup(engine)
+    datastore = create_datastore(engine)
 
     command_server = asyncio.create_task(serve(command_queue, datastore))
 

uncountable/integration/queue_runner/worker.py

@@ -21,7 +21,7 @@ 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 JobArguments
-from uncountable.integration.queue_runner.datastore.interface import Datastore
+from uncountable.integration.queue_runner.datastore import Datastore
 from uncountable.integration.queue_runner.types import ListenQueue, ResultQueue
 from uncountable.integration.scan_profiles import load_profiles
 from uncountable.integration.telemetry import JobLogger, Logger, get_otel_tracer

uncountable/integration/telemetry.py

@@ -27,6 +27,7 @@ from opentelemetry.sdk.trace.export import (
     SimpleSpanProcessor,
 )
 from opentelemetry.trace import Span, Tracer
+from opentelemetry.util.types import AttributeValue
 
 from pkgs.serialization_util import dict_fields
 from uncountable.core.environment import (
@@ -188,6 +189,9 @@ class Logger:
         )
         otel_logger.emit(log_record)
 
+    def set_span_attribute(self, key: str, value: AttributeValue) -> None:
+        self._current_span_with_fallback().set_attribute(key, value)
+
     def bind(self, *, attributes: Attributes) -> None:
         # Bound attributes flow into log records and the X-UNC-METADATA request
         # header on outbound API calls. The header is capped at 2048 bytes; if

uncountable/types/api/integrations/publish_realtime_data.py

@@ -22,7 +22,7 @@ __all__: list[str] = [
 
 ENDPOINT_METHOD = "POST"
 ENDPOINT_PATH = "api/external/integrations/publish_realtime_data"
-ENDPOINT_DESCRIPTION = "Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. API is in beta \u2014 contact Uncountable for realtime integration guidance."
+ENDPOINT_DESCRIPTION = "Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. Contact Uncountable for realtime integration guidance."
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -1713,7 +1713,7 @@ class ClientMethods(ABC):
         data_package: integrations_t.DataPackage,
         _request_options: client_config_t.RequestOptions | None = None,
     ) -> publish_realtime_data_t.Data:
-        """Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. API is in beta — contact Uncountable for realtime integration guidance.
+        """Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. Contact Uncountable for realtime integration guidance.
 
         """
         args = publish_realtime_data_t.Arguments(

uncountable/types/entity_t.py

@@ -43,6 +43,7 @@ __all__: list[str] = [
         "comment": "Comment",
         "comment_thread": "Comment Thread",
         "condition_match": "Condition Match",
+        "condition_nickname": "Condition Nickname",
         "condition_parameter": "Condition Parameter",
         "condition_parameter_mat_family": "Condition Parameter Material Family",
         "condition_parameter_matches": "Condition Parameter Matches",
@@ -52,6 +53,7 @@ __all__: list[str] = [
         "constraint": "Constraint",
         "constraint_set": "Constraint Set",
         "control_type": "Control Type",
+        "credit_balance": "Credit Balance",
         "curve_settings": "Curve Settings",
         "custom_entity": "Custom Entity",
         "default_permissions_new_project": "Default Permissions New Project",
@@ -83,6 +85,7 @@ __all__: list[str] = [
         "external_ai_connector": "External AI Connector",
         "external_ai_model": "External AI Model",
         "external_ai_session": "External AI Session",
+        "external_ai_usage": "External AI Usage",
         "experiment_group": "Experiment Group",
         "experiment_group_member": "Experiment Group Member",
         "experiment_group_type": "Experiment Group Type",
@@ -116,6 +119,7 @@ __all__: list[str] = [
         "inventory_history": "Inventory History",
         "inventory_reconciliation_action": "Inventory Reconciliation Action",
         "inventory_reconciliation_session": "Inventory Reconciliation Session",
+        "it_admin_permission": "IT Admin Permission",
         "lab_location": "Lab Location",
         "lab_request": "Lab Request",
         "license": "License",
@@ -128,6 +132,7 @@ __all__: list[str] = [
         "measurement_timeline_type": "Measurement Timeline Type",
         "milestone": "Milestone",
         "ml_job": "Batch Processing Job",
+        "monitor": "Monitor",
         "monitor_dataset": "Monitor Dataset",
         "naming_scheme": "Naming Scheme",
         "naming_counter": "Naming Counter",
@@ -228,6 +233,7 @@ __all__: list[str] = [
         "timesheet_entry": "Timesheet Entry",
         "training_module": "Training Module",
         "training_module_answer": "Training Module Answer",
+        "training_module_attempt": "Training Module Attempt",
         "training_module_block": "Training Module Block",
         "training_module_page": "Training Module Page",
         "training_module_question": "Training Module Question",
@@ -295,6 +301,7 @@ class EntityType(StrEnum):
     COMMENT = "comment"
     COMMENT_THREAD = "comment_thread"
     CONDITION_MATCH = "condition_match"
+    CONDITION_NICKNAME = "condition_nickname"
     CONDITION_PARAMETER = "condition_parameter"
     CONDITION_PARAMETER_MAT_FAMILY = "condition_parameter_mat_family"
     CONDITION_PARAMETER_MATCHES = "condition_parameter_matches"
@@ -304,6 +311,7 @@ class EntityType(StrEnum):
     CONSTRAINT = "constraint"
     CONSTRAINT_SET = "constraint_set"
     CONTROL_TYPE = "control_type"
+    CREDIT_BALANCE = "credit_balance"
     CURVE_SETTINGS = "curve_settings"
     CUSTOM_ENTITY = "custom_entity"
     DEFAULT_PERMISSIONS_NEW_PROJECT = "default_permissions_new_project"
@@ -335,6 +343,7 @@ class EntityType(StrEnum):
     EXTERNAL_AI_CONNECTOR = "external_ai_connector"
     EXTERNAL_AI_MODEL = "external_ai_model"
     EXTERNAL_AI_SESSION = "external_ai_session"
+    EXTERNAL_AI_USAGE = "external_ai_usage"
     EXPERIMENT_GROUP = "experiment_group"
     EXPERIMENT_GROUP_MEMBER = "experiment_group_member"
     EXPERIMENT_GROUP_TYPE = "experiment_group_type"
@@ -368,6 +377,7 @@ class EntityType(StrEnum):
     INVENTORY_HISTORY = "inventory_history"
     INVENTORY_RECONCILIATION_ACTION = "inventory_reconciliation_action"
     INVENTORY_RECONCILIATION_SESSION = "inventory_reconciliation_session"
+    IT_ADMIN_PERMISSION = "it_admin_permission"
     LAB_LOCATION = "lab_location"
     LAB_REQUEST = "lab_request"
     LICENSE = "license"
@@ -380,6 +390,7 @@ class EntityType(StrEnum):
     MEASUREMENT_TIMELINE_TYPE = "measurement_timeline_type"
     MILESTONE = "milestone"
     ML_JOB = "ml_job"
+    MONITOR = "monitor"
     MONITOR_DATASET = "monitor_dataset"
     NAMING_SCHEME = "naming_scheme"
     NAMING_COUNTER = "naming_counter"
@@ -480,6 +491,7 @@ class EntityType(StrEnum):
     TIMESHEET_ENTRY = "timesheet_entry"
     TRAINING_MODULE = "training_module"
     TRAINING_MODULE_ANSWER = "training_module_answer"
+    TRAINING_MODULE_ATTEMPT = "training_module_attempt"
     TRAINING_MODULE_BLOCK = "training_module_block"
     TRAINING_MODULE_PAGE = "training_module_page"
     TRAINING_MODULE_QUESTION = "training_module_question"

{uncountablepythonsdk-0.0.177.dist-info → uncountablepythonsdk-0.0.179.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.177
+Version: 0.0.179
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -20,6 +20,7 @@ Requires-Dist: aiotus==1.*
 Requires-Dist: aiohttp>=3.14.1
 Requires-Dist: requests==2.*
 Requires-Dist: SQLAlchemy>=1.4.0
+Requires-Dist: psycopg[binary]==3.*
 Requires-Dist: APScheduler==3.*
 Requires-Dist: python-dateutil==2.*
 Requires-Dist: PyYAML==6.*

{uncountablepythonsdk-0.0.177.dist-info → uncountablepythonsdk-0.0.179.dist-info}/RECORD

@@ -32,7 +32,7 @@ examples/set_input_pdf_attribute.py,sha256=A-QqaWeeLyBXQ1zvq91cBZmvGJWEUuVF5TDj0
 examples/set_recipe_metadata_file.py,sha256=cRVXGz4UN4aqnNrNSzyBmikYHpe63lMIuzOpMwD9EDU,1036
 examples/set_recipe_output_file_sdk.py,sha256=Lz1amqppnWTX83z-C090wCJ4hcKmCD3kb-4v0uBRi0Y,782
 examples/upload_files.py,sha256=qMaSvMSdTMPOOP55y1AwEurc0SOdZAMvEydlqJPsGpg,432
-examples/integration-server/pyproject.toml,sha256=gpxFJZwimt3acaOCWaGSLl_vLLPiNF5XHlQIKyzgc3A,9069
+examples/integration-server/pyproject.toml,sha256=qntU6IFi2nKoXcS4hr6J7x75EaD4WQW6KpjgHcKZzPo,9069
 examples/integration-server/jobs/materials_auto/concurrent_cron.py,sha256=xsK3H9ZEaniedC2nJUB0rqOcFI8y-ojfl_nLSJb9AMM,312
 examples/integration-server/jobs/materials_auto/example_cron.py,sha256=y1nAtGwbPJfIrfQsrHVmJLAHmQtCEHIy1g-0PjeXx04,735
 examples/integration-server/jobs/materials_auto/example_http.py,sha256=htTR61XhG_fF2Zv2gXRGYE5Yc3mSDdigrOR89kcupZY,1380
@@ -55,9 +55,9 @@ pkgs/argument_parser/parser_function_type.py,sha256=xmsg0YVseYbUx52qC9vP-_m9F_He
 pkgs/argument_parser/parser_inner.py,sha256=yteyamw3ZyMSQtDujkXgQVIpbtrpFbtjahtDNVKoGEE,23887
 pkgs/argument_parser/parser_options.py,sha256=NAwNZSIH8EjyfJXJSxbn4kYmF3K7D3HuOedlSjjsF28,1332
 pkgs/argument_parser/type_predicates.py,sha256=CkQexiseN05XDz4e34K3gIOlER5LG5Ke8u0UHK4gC04,1157
-pkgs/filesystem_utils/__init__.py,sha256=6xVtJo-5OMv8UGanNc7SfZte_iSXs5lE5au47QePy60,1381
+pkgs/filesystem_utils/__init__.py,sha256=G-Nl0sxJsSgBB6yajzzY9aeSSBnkzJHGq3lj5LFn8fw,2056
 pkgs/filesystem_utils/_blob_session.py,sha256=YB_idar5qVuF9QN_A6qjpTuyeZ7JDOLri1LGLfPHLow,5158
-pkgs/filesystem_utils/_file_share_session.py,sha256=xlxCbNtOOT1xXCiL2Y_unjWZlQ9961J7mAz4D0-uea8,5095
+pkgs/filesystem_utils/_file_share_session.py,sha256=Uos2ugL4mP8SUDNK5ILtAzEYwzOeaiaBYAEkdoyvn8M,5130
 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
@@ -68,7 +68,7 @@ pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsre
 pkgs/serialization/__init__.py,sha256=5OWO4IicaVrTWgRdC6E1K4dLYy2xQ6xQujDOuglykNU,1059
 pkgs/serialization/annotation.py,sha256=JXP2caXNoj8h_Pvx_26iarPlsaPcIqe1qssTksJiKjM,3408
 pkgs/serialization/missing_sentry.py,sha256=VCXiYs7s5MkfG8kbmwDAswdq0fL4zFRj4B2uSQzmTos,819
-pkgs/serialization/opaque_key.py,sha256=GKmZgxzzEghpukcuzSGXqG9afl_KZ7MFtLAY-Xqgu9k,213
+pkgs/serialization/opaque_key.py,sha256=QepNPlAnzpOgLI0ztc9BRB7Xr_wLoVozxfHmL0C6slg,233
 pkgs/serialization/serial_alias.py,sha256=ABEGtSEapeW95wo9NT4srEhzdGaP5RhB8EyWSyWhwKg,1337
 pkgs/serialization/serial_class.py,sha256=AfWHdGlnEY2O2GbwU-SXZp7ZbTlPoigZ624QGp8kkuE,7099
 pkgs/serialization/serial_generic.py,sha256=qdG46rw5jnvckmKezxwMSJEIF-9Y4K5FdTezS1vRSgI,476
@@ -80,7 +80,7 @@ pkgs/serialization_util/_get_type_for_serialization.py,sha256=dW5_W9MFd6wgWfW5ql
 pkgs/serialization_util/convert_to_snakecase.py,sha256=H2BAo5ZdcCDN77RpLb-uP0s7-FQ5Ukwnsd3VYc1vD0M,583
 pkgs/serialization_util/dataclasses.py,sha256=uhNGXQPQLZblDFQuuwkAGmKOPiRyfDzCdg72CVtYJGA,390
 pkgs/serialization_util/serialization_helpers.py,sha256=v8omOQ2BRZZCzMeUwu29fCEqx0UNDJ_iNbqpW4oHk84,7804
-pkgs/strenum_compat/__init__.py,sha256=wXRFeNvBm8RU6dy1PFJ5sRLgUIEeH_DVR95Sv5qpGbk,59
+pkgs/strenum_compat/__init__.py,sha256=Bq1kcPIQsXRiLUGWc63QGnVa5BlhvuFWpMMvdNcHy00,76
 pkgs/strenum_compat/strenum_compat.py,sha256=uOUAgpYTjHs1MX8dG81jRlyTkt3KNbkV_25zp7xTX2s,36
 pkgs/type_spec/__init__.py,sha256=h5DmJTca4QVV10sZR1x0-MlkZfuGYDfapR3zHvXfzto,19
 pkgs/type_spec/__main__.py,sha256=5bJaX9Y_-FavP0qwzhk-z-V97UY7uaezJTa1zhO_HHQ,1048
@@ -118,7 +118,7 @@ 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=XSAPfcSU51d3MgAZOuT6FMrdrM452KQfWzNHjLOGSMY,16549
+uncountable/core/client.py,sha256=myEWbiYFMprqy7JMKMt1aBZAu7gQfGb9mmG9fLHBNgM,16743
 uncountable/core/environment.py,sha256=Z9vu7JtnSDgQB_KKcZnjTFNyARXjRr_PDW9krwxNNAo,1132
 uncountable/core/file_upload.py,sha256=IisvIBdm2f4CJTRQ9Awi5JuI16PSERjXrzZgmrUiCHY,5960
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
@@ -137,10 +137,10 @@ uncountable/integration/request_context.py,sha256=N_FJJxqvfUJ0yV9h3I3vFTGNJiDfyL
 uncountable/integration/scan_profiles.py,sha256=86CQgrQ-zXl3p41ERazzbzhTIIjIh_BprWVfYTSHHE4,3259
 uncountable/integration/scheduler.py,sha256=BajQ4txvgEBw8S9x1P0eGm-uBkKp_oecJK65SQd2hNw,8477
 uncountable/integration/server.py,sha256=P4RRGwU9jMselHPWbU6GxhRLgVtN7Ydcxr18sFn2zI8,5778
-uncountable/integration/telemetry.py,sha256=7IWhCxchtVYJwwwFYduwnMEUdxO8srx2RnocLzcQmtg,15965
+uncountable/integration/telemetry.py,sha256=zvg-K8uy49vQxWgInXAx-Sz8jHC0IWT1kdzPQdp9ROA,16162
 uncountable/integration/webhook_signature_key.py,sha256=vXlNtj0qLqZxjKDLmRvlChvLS5xoPAGmYcYcs4WIVaM,3882
 uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/db/connect.py,sha256=mE3bdV0huclH2iT_dXCQdRL4LkjIuf_myAR64RTWXEs,498
+uncountable/integration/db/connect.py,sha256=Rl1853wku0w3hEOsAA8TM4iydaLhfEE-BGEb6G6NJSw,1396
 uncountable/integration/db/session.py,sha256=96cGQXpe6IugBTdSsjdP0S5yhJ6toSmbVB6qhc3FJzE,693
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/executors/executors.py,sha256=E3a0_yrQesfigDzjhUAwD7-l-oc9qnvPh-A8QTK6F3k,5149
@@ -149,13 +149,13 @@ uncountable/integration/executors/script_executor.py,sha256=BBQ9f0l7uH2hgKf60jtm
 uncountable/integration/http_server/__init__.py,sha256=ld7hD3wOGwAbnVRBrkTUCP5zNG-kptMd4wGJyQvUWVc,170
 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=MrQLYB4UqkUXKRPDHTqc6C6ZlQ3i8Cy-qopnqZQZbWE,9897
-uncountable/integration/queue_runner/queue_runner.py,sha256=N4sUXmlGzVquybiJ7NQZavCJOBGrxBj6k7mb-TITaN0,1139
+uncountable/integration/queue_runner/job_scheduler.py,sha256=HS6cpr863bho-EPBl4cQzK-AoBv7_Ti38Lx_0WGPx8k,9800
+uncountable/integration/queue_runner/queue_runner.py,sha256=kB5W4jCbIEPMWfHxEaQ8suKMekT4vc4fmBOMFusGQMQ,994
 uncountable/integration/queue_runner/types.py,sha256=8HS6KnYMS_vc5XHeMpg0BFAQC-5P3QLzd-aDYDMMt3E,244
-uncountable/integration/queue_runner/worker.py,sha256=mAA-NAxjMH3q1pooIbCzNBN6pIPAlSOn4eO1UITayeQ,7356
+uncountable/integration/queue_runner/worker.py,sha256=6qXP2Y9_yxYfvdC3pVL-a-W6fSSiKmzetB1oekK2ZUQ,7346
 uncountable/integration/queue_runner/command_server/__init__.py,sha256=hMCDLWct8zW4B2a9BaIAsMhtaEgFlxONjeow-6nf6dg,675
 uncountable/integration/queue_runner/command_server/command_client.py,sha256=N1UBs1Z9O6iDOEqXNzjkVFMqDUhm-0-M41I1y6s_5y0,6316
-uncountable/integration/queue_runner/command_server/command_server.py,sha256=Gbz9hY6jE51_7lREIK33v1T8BTRVDIRWOuBQ8gWFFDM,8671
+uncountable/integration/queue_runner/command_server/command_server.py,sha256=Q9avIU5EPBPSeQJzZ6_zEg9PJez4BE5NNUkHsA2lHM8,8659
 uncountable/integration/queue_runner/command_server/constants.py,sha256=7J9mQIAMOfV50wnwpn7HgrPFEi3Ritj6HwrGYwxGLoU,88
 uncountable/integration/queue_runner/command_server/provision_webhook.py,sha256=yPYEqCi9lDFYOeGBJgwcLS3IdfYGcaZFEYo6_8sFiS8,1102
 uncountable/integration/queue_runner/command_server/types.py,sha256=ZNqJE6b6rfMZGdcPS-7umB_8x2a7dzTfRBsdGoCzDjY,2077
@@ -164,10 +164,13 @@ uncountable/integration/queue_runner/command_server/protocol/command_server.prot
 uncountable/integration/queue_runner/command_server/protocol/command_server_pb2.py,sha256=voigJ1ScgpWx0FL59pKinN8MLj0seuqoZuOqvCtSPss,6013
 uncountable/integration/queue_runner/command_server/protocol/command_server_pb2.pyi,sha256=CkaTBkhsGngUxD-am691wKFAARCrivqiJHxP1pLe7jw,5984
 uncountable/integration/queue_runner/command_server/protocol/command_server_pb2_grpc.py,sha256=wzPYtvuccPVW1qEXaMxU-KB3ptnZYnjHsuit6L1Pk6U,17415
-uncountable/integration/queue_runner/datastore/__init__.py,sha256=6BefApqN8D2zlVOH14QAeVzwQ8j5NIb41-njT02Za0k,88
-uncountable/integration/queue_runner/datastore/datastore_sqlite.py,sha256=odfDBOWk6AhhdI8GMsvYWgqDDjaRULkhpoanyrcP0AE,12070
-uncountable/integration/queue_runner/datastore/interface.py,sha256=zonEm2O1l5GXEBaCy_FiFRNt6IY_BRljsGECn5lu76E,1145
-uncountable/integration/queue_runner/datastore/model.py,sha256=SH0tpXH0G7oydUbOT1z7aFZ6SNIZ6EHc9YAEaM5l1oM,832
+uncountable/integration/queue_runner/datastore/__init__.py,sha256=zwMMQbZ6FwAJU3yL8iVgKf9uTjpAjcVjqR1Akmhkrzw,282
+uncountable/integration/queue_runner/datastore/datastore_postgres.py,sha256=oKRwOWlxgodJyqKr-jDcP9VbXKKTrU26lI-7m4HFBuk,359
+uncountable/integration/queue_runner/datastore/datastore_sql_base.py,sha256=eDQFDzn8y1IHoA888Hy69dUNl9MeCFQWW_VmnFqKdfo,11107
+uncountable/integration/queue_runner/datastore/datastore_sqlite.py,sha256=W9hWKKvZrp3JyaO_tmK-qZL3ACnMnCfYwvDSJHHd6qs,1197
+uncountable/integration/queue_runner/datastore/factory.py,sha256=bv_7BxQpgjrggJKz5YDllgP_BihEJ0-CdpLvaG4ooj4,868
+uncountable/integration/queue_runner/datastore/interface.py,sha256=kuAU8lrjVN3NTllo99nfJwBojt5dg_wgdivyYCB4EbY,1581
+uncountable/integration/queue_runner/datastore/model.py,sha256=06_fGS0QTALEA_381I1zz63yXYRooaAVJdfYi0ubSd0,851
 uncountable/integration/secret_retrieval/__init__.py,sha256=sisf8cxCYcJmCXCKjEPoT2xIxm1j-Ru9iTfMVxH6NM0,171
 uncountable/integration/secret_retrieval/basic_secret.py,sha256=qjDhhBxhhQdMIsC3WJjFshJkqSC0v9hg3QJQoCi46oE,3286
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=c0VkCvppWKXwgxdblQuj2WhRaEGBPa7vLDoS6mlNOAQ,3919
@@ -186,7 +189,7 @@ uncountable/types/calculations.py,sha256=fApOFpgBemt_t7IVneVR0VdI3X5EOxiG6Xhzr6R
 uncountable/types/calculations_t.py,sha256=_K4MDVoxa6AgxlxPrEDBujWhEGWEhUQuE2-ZgiLg9OU,660
 uncountable/types/chemical_structure.py,sha256=ujyragaD26-QG5jgKnWhO7TN3N1V9b_04T2WhqNYxxo,281
 uncountable/types/chemical_structure_t.py,sha256=tbA7NBO-vv7Hnk3-ETxc6Pw3wqWeenBG27Y9_bdPCu4,785
-uncountable/types/client_base.py,sha256=kwy2tJ5L1v02YFeT_wFuFh543ffMEnhNuzFxL1MdvgQ,135001
+uncountable/types/client_base.py,sha256=h3uIZLAyJR_FPo3SAKXgkk6HDTGAaLmcio3klzcIIM0,134982
 uncountable/types/client_config.py,sha256=M7FZ0m_lGmBsIYcMn8pm92DdoVzrLpzd8sH6DqTQLKo,456
 uncountable/types/client_config_t.py,sha256=-YNp-zFvk5OL6_WGwGSx30ToK2bGIOh0kv_HN7KcCys,1796
 uncountable/types/condition_match.py,sha256=ekDzij7-e1PtVwIslSjD1T9fuBIDr5c_7QPdaasiGJw,262
@@ -196,7 +199,7 @@ uncountable/types/curves_t.py,sha256=WIML06C67yC4KWdPJR2HV_Ac7eBFTDRDUL3YgeXbFOs
 uncountable/types/data.py,sha256=u2isf4XEug3Eu-xSIoqGaCQmW2dFaKBHCkP_WKYwwBc,500
 uncountable/types/data_t.py,sha256=U_dIh5GcyaJlkrHEI4Qf4TOo1uJX07PYO70sYzzKtTo,2615
 uncountable/types/entity.py,sha256=Zclk1LYcRaYrMDhqyCjMSLEg0fE6_q8LHvV22Qvscgs,566
-uncountable/types/entity_t.py,sha256=fVDQyZvpBHp6llk05entEf2ODOVcCsEGlkE1IyDfY2o,27676
+uncountable/types/entity_t.py,sha256=OxaVn2vuQRNHxVg_iU6d26c8xSYNNyt1hQ-N6XrEf1k,28224
 uncountable/types/experiment_groups.py,sha256=qUpFOx1AKgzaT_4khCOv5Xs6jwiQGbvHH-GUh3v1nv4,288
 uncountable/types/experiment_groups_t.py,sha256=QjZBdjnbsD1D-VpPSSV6hBg-Q7mujjKvYtdMewtXNVU,712
 uncountable/types/exports.py,sha256=VMmxUO2PpV1Y63hZ2AnVor4H-B6aswJ7YpSru_u89lU,334
@@ -341,7 +344,7 @@ uncountable/types/api/inputs/set_input_category.py,sha256=EWQdlp-qhgtOOf0bwdjvix
 uncountable/types/api/inputs/set_input_subcategories.py,sha256=5nJgffEe-9Fh_VUXqZkI6Uwoa36D0d4lIhADtRGr7tk,1615
 uncountable/types/api/inputs/set_intermediate_type.py,sha256=Qsrjv67hvNHIad58kFm-8BcD1Ny8wzkuLz3aIB7ybSw,1641
 uncountable/types/api/integrations/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
-uncountable/types/api/integrations/publish_realtime_data.py,sha256=HJkfczKaLOySHWgI0qQcDQzanSUPRkv0HhnHipUKjnk,1960
+uncountable/types/api/integrations/publish_realtime_data.py,sha256=R5efPtt3RMEwoazLiqblndRPd57RTF1YajgwWiieH1Y,1938
 uncountable/types/api/integrations/push_notification.py,sha256=cqJv-qXI97jg8HFdoDF-KYaxSUgV3fqD1N614WZvaBw,1701
 uncountable/types/api/integrations/register_sockets_token.py,sha256=LJ_1LSfjVjTz-uhZkjcIw80Ah8D9pgdjQGjsTSS8gtM,1393
 uncountable/types/api/listing/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
@@ -414,7 +417,7 @@ uncountable/types/api/uploader/complete_async_parse.py,sha256=Os1HAubxTlbut6Gylj
 uncountable/types/api/uploader/invoke_uploader.py,sha256=QU1KmAFgGHSWskZEKfaAww8rCj-hVNP0i4kMQE3x3Fc,1822
 uncountable/types/api/user/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/user/get_current_user_info.py,sha256=BT7bGp5SOHz3XdKWhhdZ6acnInafT2PgdJ-H4YH3W7c,1181
-uncountablepythonsdk-0.0.177.dist-info/METADATA,sha256=gvqE0J2wNYcjjbTIjMPWR8fF1tB5p0Z2DCG0Z4-tiFc,2175
-uncountablepythonsdk-0.0.177.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-uncountablepythonsdk-0.0.177.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-uncountablepythonsdk-0.0.177.dist-info/RECORD,,
+uncountablepythonsdk-0.0.179.dist-info/METADATA,sha256=uTTLUVf_leXkYxohdAYWsCN81x1Xr7GsQfzJ_JG-tpY,2211
+uncountablepythonsdk-0.0.179.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+uncountablepythonsdk-0.0.179.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+uncountablepythonsdk-0.0.179.dist-info/RECORD,,

{uncountablepythonsdk-0.0.177.dist-info → uncountablepythonsdk-0.0.179.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (82.0.1)
+Generator: setuptools (83.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any