UncountablePythonSDK 0.0.182__tar.gz → 0.0.183__tar.gz

{uncountablepythonsdk-0.0.182/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.183}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.182
+Version: 0.0.183
 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.182 → uncountablepythonsdk-0.0.183/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.182
+Version: 0.0.183
 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.182 → uncountablepythonsdk-0.0.183}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -190,13 +190,7 @@ uncountable/integration/queue_runner/datastore/model.py
 uncountable/integration/secret_retrieval/__init__.py
 uncountable/integration/secret_retrieval/basic_secret.py
 uncountable/integration/secret_retrieval/retrieve_secret.py
-uncountable/integration/webhook_server/admin_auth.py
 uncountable/integration/webhook_server/entrypoint.py
-uncountable/integration/webhook_server/admin_routes/__init__.py
-uncountable/integration/webhook_server/admin_routes/_boundary.py
-uncountable/integration/webhook_server/admin_routes/health.py
-uncountable/integration/webhook_server/admin_routes/jobs.py
-uncountable/integration/webhook_server/admin_routes/register.py
 uncountable/types/__init__.py
 uncountable/types/async_batch.py
 uncountable/types/async_batch_processor.py

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/pkgs/filesystem_utils/_blob_session.py

@@ -2,7 +2,7 @@ from io import BytesIO
 from typing import Self
 
 from azure.core.credentials import AzureSasCredential
-from azure.storage.blob import BlobServiceClient, ContainerClient
+from azure.storage.blob import BlobPrefix, BlobServiceClient, ContainerClient
 
 from pkgs.filesystem_utils.file_type_utils import (
     FileObjectData,
@@ -78,6 +78,24 @@ class BlobSession(FileSystemSession):
 
         return filesystem_file_references
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to list_directories on uninitialized blob session"
+        )
+
+        prefix = _add_slash(dir_path.filepath)
+        directories: list[FileSystemObject] = []
+        for item in self.container_client.walk_blobs(
+            name_starts_with=prefix, delimiter="/"
+        ):
+            if isinstance(item, BlobPrefix):
+                directories.append(FileSystemFileReference(item.name.rstrip("/")))
+
+        return directories
+
     def download_files(
         self,
         filepaths: list[FileSystemObject],
@@ -108,6 +126,22 @@ class BlobSession(FileSystemSession):
 
         return downloaded_files
 
+    def _move_blob(self, src_name: str, dest_name: str) -> None:
+        assert self.container_client is not None, (
+            "call to _move_blob on uninitialized blob session"
+        )
+        source_blob_client = self.container_client.get_blob_client(src_name)
+        dest_blob_client = self.container_client.get_blob_client(dest_name)
+
+        source_url = (
+            f"{source_blob_client.url}?{self.config.credential.signature}"
+            if isinstance(self.config.credential, AzureSasCredential)
+            else source_blob_client.url
+        )
+
+        dest_blob_client.start_copy_from_url(source_url)
+        source_blob_client.delete_blob()
+
     def move_files(self, file_mappings: list[FileTransfer]) -> None:
         assert self.service_client is not None and self.container_client is not None, (
             "call to move_files on uninitialized blob session"
@@ -119,19 +153,25 @@ class BlobSession(FileSystemSession):
             ):
                 raise IncompatibleFileReferenceError()
 
-            source_blob_client = self.container_client.get_blob_client(
-                src_file.filepath
-            )
-            dest_blob_client = self.container_client.get_blob_client(dest_file.filepath)
+            self._move_blob(src_file.filepath, dest_file.filepath)
 
-            source_url = (
-                f"{source_blob_client.url}?{self.config.credential.signature}"
-                if isinstance(self.config.credential, AzureSasCredential)
-                else source_blob_client.url
-            )
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to move_directory on uninitialized blob session"
+        )
 
-            dest_blob_client.start_copy_from_url(source_url)
-            source_blob_client.delete_blob()
+        src_prefix = _add_slash(src_dir.filepath)
+        dest_prefix = _add_slash(dest_dir.filepath)
+        for blob in self.container_client.list_blobs(name_starts_with=src_prefix):
+            dest_name = dest_prefix + blob.name[len(src_prefix) :]
+            self._move_blob(blob.name, dest_name)
 
     def delete_files(self, filepaths: list[FileSystemObject]) -> None:
         assert self.service_client is not None and self.container_client is not None, (

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/pkgs/filesystem_utils/_file_share_session.py

@@ -79,6 +79,49 @@ class FileShareSession(FileSystemSession):
             dir_client, recursive=recursive, valid_extensions=valid_extensions
         )
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to list_directories on uninitialized file share session"
+        )
+        dir_client = self._share_client.get_directory_client(dir_path.filepath)
+        directories: list[FileSystemObject] = []
+        for item in dir_client.list_directories_and_files():
+            if item["is_directory"]:
+                item_path = f"{dir_client.directory_path}/{item['name']}".lstrip("/")
+                directories.append(FileSystemFileReference(filepath=item_path))
+        return directories
+
+    def create_directory(self, dir_path: FileSystemObject) -> None:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to create_directory on uninitialized file share session"
+        )
+        current = ""
+        for segment in dir_path.filepath.strip("/").split("/"):
+            if not segment:
+                continue
+            current = f"{current}/{segment}".lstrip("/")
+            dir_client = self._share_client.get_directory_client(current)
+            if not dir_client.exists():
+                dir_client.create_directory()
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to move_directory on uninitialized file share session"
+        )
+        self._share_client.get_directory_client(src_dir.filepath).rename_directory(
+            dest_dir.filepath
+        )
+
     def download_files(self, filepaths: list[FileSystemObject]) -> list[FileObjectData]:
         assert self._share_client is not None, (
             "call to download_files on uninitialized file share session"

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/pkgs/filesystem_utils/_s3_session.py

@@ -86,6 +86,28 @@ class S3Session(FileSystemSession):
 
         return filesystem_references
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+
+        assert self.bucket is not None, (
+            "call to list_directories on uninitialized s3 session"
+        )
+
+        prefix = _add_slash(dir_path.filepath)
+        paginator = self.bucket.meta.client.get_paginator("list_objects_v2")
+
+        directories: list[FileSystemObject] = []
+        for page in paginator.paginate(
+            Bucket=self.bucket.name, Prefix=prefix, Delimiter="/"
+        ):
+            for common_prefix in page.get("CommonPrefixes", []):
+                directories.append(
+                    FileSystemFileReference(common_prefix["Prefix"].rstrip("/"))
+                )
+
+        return directories
+
     def download_files(
         self,
         filepaths: list[FileSystemObject],
@@ -115,6 +137,18 @@ class S3Session(FileSystemSession):
 
         return downloaded_files
 
+    def _move_object(self, src_key: str, dest_key: str) -> None:
+        assert self.bucket is not None, (
+            "call to _move_object on uninitialized s3 session"
+        )
+        self.bucket.Object(dest_key).copy_from(
+            CopySource={
+                "Bucket": self.bucket.name,
+                "Key": src_key,
+            }
+        )
+        self.bucket.Object(src_key).delete()
+
     def move_files(self, file_mappings: list[FileTransfer]) -> None:
         assert self.bucket is not None, "call to move_files on uninitialized s3 session"
 
@@ -123,10 +157,22 @@ class S3Session(FileSystemSession):
                 dest_file, FileSystemFileReference
             ):
                 raise IncompatibleFileReferenceError()
-            self.bucket.Object(dest_file.filepath).copy_from(
-                CopySource={
-                    "Bucket": self.bucket.name,
-                    "Key": src_file.filepath,
-                }
-            )
-            self.bucket.Object(src_file.filepath).delete()
+            self._move_object(src_file.filepath, dest_file.filepath)
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+
+        assert self.bucket is not None, (
+            "call to move_directory on uninitialized s3 session"
+        )
+
+        src_prefix = _add_slash(src_dir.filepath)
+        dest_prefix = _add_slash(dest_dir.filepath)
+        for obj in self.bucket.objects.filter(Prefix=src_prefix):
+            dest_key = dest_prefix + obj.key[len(src_prefix) :]
+            self._move_object(obj.key, dest_key)

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/pkgs/filesystem_utils/filesystem_session.py

@@ -32,6 +32,17 @@ class FileSystemSession(ABC):
     def delete_files(self, filepaths: list[FileSystemObject]) -> None:
         raise NotImplementedError
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        raise NotImplementedError
+
+    def create_directory(self, dir_path: FileSystemObject) -> None:
+        raise NotImplementedError
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        raise NotImplementedError
+
     @abstractmethod
     def __enter__(self) -> "FileSystemSession": ...
 

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/uncountable/core/environment.py

@@ -26,10 +26,6 @@ def get_local_admin_server_port() -> int:
     return int(os.environ.get("UNC_ADMIN_SERVER_PORT", "50051"))
 
 
-def get_integration_admin_token() -> str | None:
-    return os.environ.get("UNC_INTEGRATION_ADMIN_TOKEN")
-
-
 def get_otel_enabled() -> bool:
     return os.environ.get("UNC_OTEL_ENABLED") == "true"
 

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/uncountable/integration/webhook_server/entrypoint.py

@@ -12,7 +12,6 @@ from uncountable.integration.http_server import GenericHttpRequest, HttpExceptio
 from uncountable.integration.job import CustomHttpJob, WebhookJob
 from uncountable.integration.scan_profiles import load_profiles
 from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_routes import register_admin_routes
 from uncountable.types import job_definition_t
 
 app = flask.Flask(__name__)
@@ -99,7 +98,6 @@ def register_route(
 
 def main() -> None:
     app.add_url_rule("/health", "health", lambda: ("OK", 200))
-    register_admin_routes(app)
 
     profiles = load_profiles()
     for profile_metadata in profiles:

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/uncountable/types/api/recipes/set_recipe_inputs.py

@@ -21,7 +21,7 @@ __all__: list[str] = [
 
 ENDPOINT_METHOD = "POST"
 ENDPOINT_PATH = "api/external/recipes/external_set_recipe_inputs"
-ENDPOINT_DESCRIPTION = "Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Supports setting actual values, associating lots (creating them if needed), and removing inputs. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided."
+ENDPOINT_DESCRIPTION = "Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Each entry edits one input on one recipe with a single operation: set the value (the actual value when set_actual_value is true, otherwise the set value), associate a lot (lot_recipe_id, created if the recipe is not already a lot), or remove the input (remove). A lot association targets the set value; combining it with set_actual_value fails. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided."
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.182 → uncountablepythonsdk-0.0.183}/uncountable/types/client_base.py

@@ -2177,7 +2177,7 @@ class ClientMethods(ABC):
         input_data: list[set_recipe_inputs_t.RecipeInputValue],
         _request_options: client_config_t.RequestOptions | None = None,
     ) -> set_recipe_inputs_t.Data:
-        """Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Supports setting actual values, associating lots (creating them if needed), and removing inputs. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided.
+        """Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Each entry edits one input on one recipe with a single operation: set the value (the actual value when set_actual_value is true, otherwise the set value), associate a lot (lot_recipe_id, created if the recipe is not already a lot), or remove the input (remove). A lot association targets the set value; combining it with set_actual_value fails. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided.
 
         :param input_data: The inputs to set. Must be at most 100 entries long
         """

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_auth.py

@@ -1,38 +0,0 @@
-import functools
-import hmac
-from collections.abc import Callable
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_integration_admin_token
-from uncountable.integration.http_server import HttpException
-
-_BEARER_PREFIX = "Bearer "
-
-
-def _extract_bearer_token(authorization_header: str | None) -> str | None:
-    if authorization_header is None:
-        return None
-    if not authorization_header.startswith(_BEARER_PREFIX):
-        return None
-    return authorization_header[len(_BEARER_PREFIX) :]
-
-
-def require_admin_auth(
-    view: Callable[..., ResponseReturnValue],
-) -> Callable[..., ResponseReturnValue]:
-    @functools.wraps(view)
-    def wrapper(*args: object, **kwargs: object) -> ResponseReturnValue:
-        admin_token = get_integration_admin_token()
-        if admin_token is None:
-            raise HttpException.service_unavailable()
-        provided_token = _extract_bearer_token(
-            flask.request.headers.get("Authorization")
-        )
-        if provided_token is None or not hmac.compare_digest(
-            provided_token, admin_token
-        ):
-            raise HttpException.unauthorized()
-        return view(*args, **kwargs)
-
-    return wrapper

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_routes/__init__.py

@@ -1,3 +0,0 @@
-from uncountable.integration.webhook_server.admin_routes.register import (
-    register_admin_routes as register_admin_routes,
-)

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_routes/_boundary.py

@@ -1,39 +0,0 @@
-import functools
-from collections.abc import Callable
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.integration.http_server import HttpException
-from uncountable.integration.queue_runner.command_server.types import (
-    CommandServerException,
-)
-from uncountable.integration.telemetry import Logger
-
-
-def register_admin_route(
-    *,
-    app: flask.Flask,
-    logger: Logger,
-    route: str,
-    endpoint: str,
-    view: Callable[..., ResponseReturnValue],
-    methods: list[str],
-) -> None:
-    @functools.wraps(view)
-    def wrapped(*args: object, **kwargs: object) -> ResponseReturnValue:
-        with logger.push_scope(route):
-            try:
-                return view(*args, **kwargs)
-            except HttpException as e:
-                return e.make_error_response()
-            except CommandServerException as e:
-                logger.log_exception(
-                    e,
-                    message=f"admin route {route} could not reach the command server",
-                )
-                return HttpException.bad_gateway().make_error_response()
-            except Exception as e:
-                logger.log_exception(e, message=f"unexpected error on {route}")
-                return HttpException.unknown_error().make_error_response()
-
-    app.add_url_rule(route, endpoint=endpoint, view_func=wrapped, methods=methods)

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_routes/health.py

@@ -1,30 +0,0 @@
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_local_admin_server_port
-from uncountable.integration.queue_runner.command_server.command_client import (
-    check_health,
-)
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_auth import require_admin_auth
-from uncountable.integration.webhook_server.admin_routes._boundary import (
-    register_admin_route,
-)
-
-_ADMIN_HEALTH_ROUTE = "/admin/health"
-
-
-@require_admin_auth
-def _handle_admin_health() -> ResponseReturnValue:
-    healthy = check_health(port=get_local_admin_server_port())
-    return flask.jsonify({"healthy": healthy})
-
-
-def register_health_routes(*, app: flask.Flask, logger: Logger) -> None:
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_HEALTH_ROUTE,
-        endpoint="admin_health",
-        view=_handle_admin_health,
-        methods=["GET"],
-    )

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_routes/jobs.py

@@ -1,147 +0,0 @@
-import datetime
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_local_admin_server_port
-from uncountable.integration.http_server import HttpException
-from uncountable.integration.queue_runner.command_server.command_client import (
-    send_job_cancellation_message,
-    send_job_queue_message,
-    send_list_queued_jobs_message,
-    send_retry_job_message,
-)
-from uncountable.integration.queue_runner.command_server.protocol.command_server_pb2 import (
-    ListQueuedJobsResult,
-)
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_auth import require_admin_auth
-from uncountable.integration.webhook_server.admin_routes._boundary import (
-    register_admin_route,
-)
-from uncountable.types import base_t, queued_job_t
-
-from pkgs.serialization_util import serialize_for_api
-
-_ADMIN_JOBS_ROUTE = "/admin/jobs"
-_ADMIN_ENQUEUE_JOB_ROUTE = "/admin/jobs/enqueue"
-_ADMIN_RETRY_JOB_ROUTE = "/admin/jobs/<job_uuid>/retry"
-_ADMIN_CANCEL_JOB_ROUTE = "/admin/jobs/<job_uuid>/cancel"
-_DEFAULT_JOBS_LIMIT = 100
-_VALID_JOB_STATUSES = {status.value for status in queued_job_t.JobStatus}
-
-
-def _parse_job_status(raw_status: str | None) -> queued_job_t.JobStatus | None:
-    if raw_status is None:
-        return None
-    if raw_status not in _VALID_JOB_STATUSES:
-        raise HttpException.bad_request(f"invalid status: {raw_status}")
-    return queued_job_t.JobStatus(raw_status)
-
-
-def _parse_non_negative_int(raw_value: str | None, *, default: int) -> int:
-    if raw_value is None:
-        return default
-    if not raw_value.isdigit():
-        raise HttpException.bad_request(
-            f"expected a non-negative integer, got: {raw_value}"
-        )
-    return int(raw_value)
-
-
-def _serialize_queued_job(
-    item: ListQueuedJobsResult.ListQueuedJobsResultItem,
-) -> dict[str, base_t.JsonValue]:
-    metadata = queued_job_t.QueuedJobMetadata(
-        queued_job_uuid=item.uuid,
-        job_ref_name=item.job_ref_name,
-        num_attempts=item.num_attempts,
-        status=queued_job_t.JobStatus(item.status),
-        submitted_at=item.submitted_at.ToDatetime(datetime.timezone.utc),
-    )
-    return serialize_for_api(metadata)
-
-
-@require_admin_auth
-def _handle_list_jobs() -> ResponseReturnValue:
-    status = _parse_job_status(flask.request.args.get("status"))
-    offset = _parse_non_negative_int(flask.request.args.get("offset"), default=0)
-    limit = _parse_non_negative_int(
-        flask.request.args.get("limit"), default=_DEFAULT_JOBS_LIMIT
-    )
-    jobs = send_list_queued_jobs_message(
-        status=status, offset=offset, limit=limit, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"jobs": [_serialize_queued_job(job) for job in jobs]})
-
-
-@require_admin_auth
-def _handle_enqueue_job() -> ResponseReturnValue:
-    body = flask.request.get_json(silent=True)
-    if not isinstance(body, dict):
-        raise HttpException.bad_request("expected a JSON object body")
-    job_ref_name = body.get("job_ref_name")
-    if not isinstance(job_ref_name, str):
-        raise HttpException.bad_request("job_ref_name is required")
-    webhook_payload = body.get("webhook_payload")
-    invocation_context: queued_job_t.InvocationContext = (
-        queued_job_t.InvocationContextManual()
-        if webhook_payload is None
-        else queued_job_t.InvocationContextWebhook(webhook_payload=webhook_payload)
-    )
-    queued_job_uuid = send_job_queue_message(
-        job_ref_name=job_ref_name,
-        payload=queued_job_t.QueuedJobPayload(invocation_context=invocation_context),
-        port=get_local_admin_server_port(),
-    )
-    return flask.jsonify({"queuedJobUuid": queued_job_uuid})
-
-
-@require_admin_auth
-def _handle_retry_job(job_uuid: str) -> ResponseReturnValue:
-    queued_job_uuid = send_retry_job_message(
-        job_uuid=job_uuid, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"queuedJobUuid": queued_job_uuid})
-
-
-@require_admin_auth
-def _handle_cancel_job(job_uuid: str) -> ResponseReturnValue:
-    status = send_job_cancellation_message(
-        queued_job_uuid=job_uuid, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"status": status.value})
-
-
-def register_job_routes(*, app: flask.Flask, logger: Logger) -> None:
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_JOBS_ROUTE,
-        endpoint="admin_list_jobs",
-        view=_handle_list_jobs,
-        methods=["GET"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_ENQUEUE_JOB_ROUTE,
-        endpoint="admin_enqueue_job",
-        view=_handle_enqueue_job,
-        methods=["POST"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_RETRY_JOB_ROUTE,
-        endpoint="admin_retry_job",
-        view=_handle_retry_job,
-        methods=["POST"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_CANCEL_JOB_ROUTE,
-        endpoint="admin_cancel_job",
-        view=_handle_cancel_job,
-        methods=["POST"],
-    )

uncountablepythonsdk-0.0.182/uncountable/integration/webhook_server/admin_routes/register.py

@@ -1,13 +0,0 @@
-import flask
-from opentelemetry.trace import get_current_span
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_routes.health import (
-    register_health_routes,
-)
-from uncountable.integration.webhook_server.admin_routes.jobs import register_job_routes
-
-
-def register_admin_routes(app: flask.Flask) -> None:
-    logger = Logger(get_current_span())
-    register_health_routes(app=app, logger=logger)
-    register_job_routes(app=app, logger=logger)