UncountablePythonSDK 0.0.117__tar.gz → 0.0.118__tar.gz

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/.github/workflows/documentation.yml

@@ -17,7 +17,7 @@ jobs:
       - name: Set up UV
         uses: astral-sh/setup-uv@f0ec1fc3b38f5e7cd731bb6ce540c5af426746bb
         with:
-          version: "0.7.1"
+          version: "0.8.4"
           activate-environment: true
           enable-cache: true
       - name: Install dependencies

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/.github/workflows/publish.yml

@@ -25,7 +25,7 @@ jobs:
       - name: Set up UV
         uses: astral-sh/setup-uv@f0ec1fc3b38f5e7cd731bb6ce540c5af426746bb
         with:
-          version: "0.7.1"
+          version: "0.8.4"
           activate-environment: true
           enable-cache: true
       - name: Install build dependencies

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/PKG-INFO

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

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

@@ -138,6 +138,7 @@ uncountable/integration/queue_runner/worker.py
 uncountable/integration/queue_runner/command_server/__init__.py
 uncountable/integration/queue_runner/command_server/command_client.py
 uncountable/integration/queue_runner/command_server/command_server.py
+uncountable/integration/queue_runner/command_server/constants.py
 uncountable/integration/queue_runner/command_server/types.py
 uncountable/integration/queue_runner/command_server/protocol/__init__.py
 uncountable/integration/queue_runner/command_server/protocol/command_server.proto

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/docs/justfile

@@ -1,5 +1,5 @@
 docs-setup-python:
-  curl -LsSf https://astral.sh/uv/0.7.1/install.sh | sh
+  curl -LsSf https://astral.sh/uv/0.8.4/install.sh | sh
   uv pip install -r requirements.txt
 
 docs-clean:

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/examples/integration-server/jobs/materials_auto/example_instrument.py

@@ -1,7 +1,17 @@
+import time
 from dataclasses import dataclass
+from decimal import Decimal
 
 from uncountable.integration.job import JobArguments, WebhookJob, register_job
-from uncountable.types import base_t, entity_t, job_definition_t
+from uncountable.types import (
+    base_t,
+    entity_t,
+    identifier_t,
+    integrations_t,
+    job_definition_t,
+    publish_realtime_data_t,
+)
+from uncountable.types.client_base import APIRequest
 
 
 @dataclass(kw_only=True)
@@ -31,6 +41,28 @@ class InstrumentExample(WebhookJob[InstrumentPayload]):
 
         args.logger.log_info(f"Instrument ID: {instrument_id}")
 
+        for i in range(10):
+            req_args = publish_realtime_data_t.Arguments(
+                data_package=integrations_t.DataPackageNumericReading(
+                    value=Decimal(i * 15),
+                    target_entity=entity_t.EntityIdentifier(
+                        identifier_key=identifier_t.IdentifierKeyId(
+                            id=payload.equipment_id
+                        ),
+                        type=entity_t.EntityType.EQUIPMENT,
+                    ),
+                ),
+            )
+            api_request = APIRequest(
+                method=publish_realtime_data_t.ENDPOINT_METHOD,
+                endpoint=publish_realtime_data_t.ENDPOINT_PATH,
+                args=req_args,
+            )
+            args.client.do_request(
+                api_request=api_request, return_type=publish_realtime_data_t.Data
+            )
+            time.sleep(0.75)
+
         return job_definition_t.JobResult(success=True)
 
     @property

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/examples/integration-server/pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
   "ruff == 0.*",
   "openpyxl == 3.*",
   "more_itertools == 10.*",
-  "types-paramiko ==3.5.0.20240918",
+  "types-paramiko ==3.5.0.20250708",
   "types-openpyxl == 3.*",
   "types-pysftp == 0.*",
   "types-pytz ==2025.*",

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/pkgs/argument_parser/__init__.py

@@ -7,6 +7,7 @@ from .argument_parser import ParserFunction as ParserFunction
 from .argument_parser import ParserOptions as ParserOptions
 from .argument_parser import SourceEncoding as SourceEncoding
 from .argument_parser import build_parser as build_parser
+from .argument_parser import is_missing as is_missing
 from .argument_parser import is_optional as is_optional
 from .argument_parser import is_union as is_union
 from .case_convert import camel_to_snake_case as camel_to_snake_case

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/pkgs/argument_parser/argument_parser.py

@@ -223,7 +223,7 @@ def _build_parser_inner(
             )
 
     if dataclasses.is_dataclass(parsed_type):
-        return _build_parser_dataclass(parsed_type, context)  # type: ignore[arg-type]
+        return _build_parser_dataclass(parsed_type, context)
 
     # namedtuple support
     if is_namedtuple_type(parsed_type):
@@ -244,7 +244,7 @@ def _build_parser_inner(
         })
 
     if parsed_type == type(None):  # noqa: E721
-        return lambda value: _invoke_membership_parser({None}, value)  # type: ignore
+        return lambda value: _invoke_membership_parser({None}, value)
 
     origin = typing.get_origin(parsed_type)
     if origin is tuple:
@@ -271,7 +271,7 @@ def _build_parser_inner(
         arg_parsers = [_build_parser_inner(arg, context) for arg in sorted_args]
         return lambda value: _invoke_fallback_parsers(parsed_type, arg_parsers, value)
 
-    if parsed_type is typing.Any:  # type: ignore[comparison-overlap]
+    if parsed_type is typing.Any:
         return lambda value: value
 
     if origin in (list, set):

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/pkgs/type_spec/builder.py

@@ -745,6 +745,8 @@ class SpecTypeDefnStringEnum(SpecTypeDefn):
                     builder.ensure(
                         isinstance(enum_value, str), "enum value should be string"
                     )
+                    assert isinstance(enum_value, str)
+
                     deprecated = value.get("deprecated", False)
                     builder.ensure(
                         isinstance(deprecated, bool),

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/pkgs/type_spec/emit_open_api.py

@@ -279,7 +279,12 @@ def _emit_stability_level(
         case EmitOpenAPIStabilityLevel.draft:
             stability_info["x-beta"] = True
         case EmitOpenAPIStabilityLevel.beta:
-            stability_info["x-badges"] = [{"name": "Beta", "color": "DarkOrange"}]
+            stability_info["x-badges"] = [
+                {
+                    "name": "Beta",
+                    "color": "DarkOrange",
+                }
+            ]
         case EmitOpenAPIStabilityLevel.stable:
             pass
         case _:
@@ -334,18 +339,57 @@ def _emit_endpoint_response_examples(
     return {"examples": response_examples}
 
 
+def _create_warning_banner(api_type: str, message: str) -> str:
+    return (
+        f'<div style="background-color: #fff3cd; border: 1px solid #ffeaa7; '
+        f'border-radius: 4px; padding: 12px; margin-bottom: 16px;">'
+        f"<strong>⚠️ {api_type} API:</strong> {message}"
+        f"</div>"
+    )
+
+
+def _get_stability_warning(
+    stability_level: EmitOpenAPIStabilityLevel | None,
+) -> str:
+    resolved_stability_level = (
+        stability_level
+        if stability_level is not None
+        else EmitOpenAPIStabilityLevel.stable
+    )
+
+    match resolved_stability_level:
+        case EmitOpenAPIStabilityLevel.draft:
+            return _create_warning_banner(
+                "Draft",
+                "This endpoint is in draft status and may change significantly. Not recommended for production use.",
+            )
+        case EmitOpenAPIStabilityLevel.beta:
+            return _create_warning_banner(
+                "Beta",
+                "This endpoint is in beta and its required parameters may change. Use with caution in production environments.",
+            )
+        case EmitOpenAPIStabilityLevel.stable:
+            return ""
+
+
 def _emit_endpoint_description(
-    description: str, guides: list[EmitOpenAPIGuide]
+    description: str,
+    guides: list[EmitOpenAPIGuide],
+    stability_level: EmitOpenAPIStabilityLevel | None = None,
 ) -> dict[str, str]:
+    stability_warning = _get_stability_warning(stability_level)
+
     full_guides = "<br/>".join([
         _write_guide_as_html(guide, is_open=False)
         for guide in sorted(guides, key=lambda g: g.ref_name)
     ])
-    return {
-        "description": description
-        if len(guides) == 0
-        else f"{description}<br/>{full_guides}"
-    }
+
+    full_description_parts = [
+        part for part in [stability_warning, description, full_guides] if part
+    ]
+    full_description = "<br/>".join(full_description_parts)
+
+    return {"description": full_description}
 
 
 def _emit_namespace(
@@ -381,7 +425,9 @@ def _emit_namespace(
                 "summary": endpoint.summary,
             }
             | _emit_endpoint_deprecated(endpoint.deprecated)
-            | _emit_endpoint_description(endpoint.description, ctx.endpoint.guides)
+            | _emit_endpoint_description(
+                endpoint.description, ctx.endpoint.guides, endpoint.stability_level
+            )
             | _emit_stability_level(endpoint.stability_level)
             | _emit_endpoint_parameters(endpoint, argument_type, ctx.endpoint.examples)
             | _emit_endpoint_request_body(

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/pkgs/type_spec/value_spec/emit_python.py

@@ -75,9 +75,14 @@ def _emit_function_wrapper(function: value_spec_t.Function) -> str:
             ):
                 python_type += " | None"
                 any_pass_null = True
+
+            if python_type.startswith("base_t.ExtJsonValue"):
+                return_statement = f"self._extract({index})"
+            else:
+                return_statement = f"cast({python_type}, self._extract({index}))"
             out.write(
                 f"""{INDENT}def get_{argument.ref_name}(self) -> {python_type}:
-{INDENT}{INDENT}return cast({python_type}, self._extract({index}))
+{INDENT}{INDENT}return {return_statement}
 """
             )
         out.write("\n")

uncountablepythonsdk-0.0.118/uncountable/integration/cli.py

@@ -0,0 +1,106 @@
+import argparse
+
+from dateutil import tz
+from opentelemetry.trace import get_current_span
+from tabulate import tabulate
+
+from uncountable.core.environment import get_local_admin_server_port
+from uncountable.integration.queue_runner.command_server.command_client import (
+    send_job_queue_message,
+    send_list_queued_jobs_message,
+)
+from uncountable.integration.telemetry import Logger
+from uncountable.types import queued_job_t
+
+
+def register_enqueue_job_parser(
+    sub_parser_manager: argparse._SubParsersAction,
+) -> None:
+    run_parser = sub_parser_manager.add_parser(
+        "run",
+        description="Process a job with a given command and job ID.",
+    )
+    run_parser.add_argument("job_id", type=str, help="The ID of the job to process")
+
+    def _handle_enqueue_job(args: argparse.Namespace) -> None:
+        send_job_queue_message(
+            job_ref_name=args.job_id,
+            payload=queued_job_t.QueuedJobPayload(
+                invocation_context=queued_job_t.InvocationContextManual()
+            ),
+            host=args.host,
+            port=get_local_admin_server_port(),
+        )
+
+    run_parser.set_defaults(func=_handle_enqueue_job)
+
+
+def register_list_queued_jobs(
+    sub_parser_manager: argparse._SubParsersAction,
+) -> None:
+    list_queued_jobs_parser = sub_parser_manager.add_parser(
+        "list-queued-jobs", description="List jobs queued on the integration server."
+    )
+
+    list_queued_jobs_parser.add_argument(
+        "--offset",
+        type=int,
+        default=0,
+        help="Number of jobs to skip. Should be non-negative.",
+    )
+    list_queued_jobs_parser.add_argument(
+        "--limit",
+        type=int,
+        default=100,
+        help="A number between 1 and 100 specifying the number of jobs to return in the result set.",
+    )
+
+    def _handle_list_queued_jobs(args: argparse.Namespace) -> None:
+        queued_jobs = send_list_queued_jobs_message(
+            offset=args.offset,
+            limit=args.limit,
+            host=args.host,
+            port=get_local_admin_server_port(),
+        )
+
+        headers = ["UUID", "Job Ref Name", "Attempts", "Submitted At"]
+        rows = [
+            [
+                job.uuid,
+                job.job_ref_name,
+                job.num_attempts,
+                job.submitted_at.ToDatetime(tz.UTC).astimezone(tz.tzlocal()),
+            ]
+            for job in queued_jobs
+        ]
+        print(tabulate(rows, headers=headers, tablefmt="grid"))
+
+    list_queued_jobs_parser.set_defaults(func=_handle_list_queued_jobs)
+
+
+def main() -> None:
+    logger = Logger(get_current_span())
+
+    parser = argparse.ArgumentParser(
+        description="Execute a given integrations server command."
+    )
+
+    parser.add_argument(
+        "--host", type=str, default="localhost", nargs="?", help="The host to run on"
+    )
+
+    subparser_action = parser.add_subparsers(
+        dest="command",
+        required=True,
+        help="The command to execute (e.g., 'run')",
+    )
+
+    register_enqueue_job_parser(subparser_action)
+    register_list_queued_jobs(subparser_action)
+
+    args = parser.parse_args()
+    with logger.push_scope(args.command):
+        args.func(args)
+
+
+main()

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/uncountable/integration/queue_runner/command_server/command_client.py

@@ -10,6 +10,8 @@ from uncountable.integration.queue_runner.command_server.protocol.command_server
     CheckHealthResult,
     EnqueueJobRequest,
     EnqueueJobResult,
+    ListQueuedJobsRequest,
+    ListQueuedJobsResult,
 )
 from uncountable.integration.queue_runner.command_server.types import (
     CommandServerBadResponse,
@@ -66,3 +68,27 @@ def check_health(*, host: str = _LOCAL_RPC_HOST, port: int) -> bool:
         assert isinstance(response, CheckHealthResult)
 
         return response.success
+
+
+def send_list_queued_jobs_message(
+    *,
+    offset: int,
+    limit: int,
+    host: str = "localhost",
+    port: int,
+) -> list[ListQueuedJobsResult.ListQueuedJobsResultItem]:
+    with command_server_connection(host=host, port=port) as stub:
+        request = ListQueuedJobsRequest(
+            offset=offset,
+            limit=limit,
+        )
+
+        try:
+            response = stub.ListQueuedJobs(
+                request, timeout=_DEFAULT_MESSAGE_TIMEOUT_SECS
+            )
+        except grpc.RpcError as e:
+            raise ValueError(e.details())  # type: ignore
+
+        assert isinstance(response, ListQueuedJobsResult)
+        return list(response.queued_jobs)

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/uncountable/integration/queue_runner/command_server/command_server.py

@@ -1,7 +1,8 @@
 import asyncio
 
 import simplejson as json
-from grpc import aio
+from google.protobuf.timestamp_pb2 import Timestamp
+from grpc import StatusCode, aio
 
 from pkgs.argument_parser import CachedParser
 from uncountable.core.environment import get_local_admin_server_port
@@ -10,14 +11,18 @@ from uncountable.integration.queue_runner.command_server.protocol.command_server
     CheckHealthResult,
     EnqueueJobRequest,
     EnqueueJobResult,
+    ListQueuedJobsRequest,
+    ListQueuedJobsResult,
 )
 from uncountable.integration.queue_runner.command_server.types import (
     CommandEnqueueJob,
     CommandEnqueueJobResponse,
     CommandQueue,
 )
+from uncountable.integration.queue_runner.datastore import DatastoreSqlite
 from uncountable.types import queued_job_t
 
+from .constants import ListQueuedJobsConstants
 from .protocol.command_server_pb2_grpc import (
     CommandServerServicer,
     add_CommandServerServicer_to_server,
@@ -26,7 +31,7 @@ from .protocol.command_server_pb2_grpc import (
 queued_job_payload_parser = CachedParser(queued_job_t.QueuedJobPayload)
 
 
-async def serve(command_queue: CommandQueue) -> None:
+async def serve(command_queue: CommandQueue, datastore: DatastoreSqlite) -> None:
     server = aio.server()
 
     class CommandServerHandler(CommandServerServicer):
@@ -54,6 +59,41 @@ async def serve(command_queue: CommandQueue) -> None:
         ) -> CheckHealthResult:
             return CheckHealthResult(success=True)
 
+        async def ListQueuedJobs(
+            self, request: ListQueuedJobsRequest, context: aio.ServicerContext
+        ) -> ListQueuedJobsResult:
+            if (
+                request.limit < ListQueuedJobsConstants.LIMIT_MIN
+                or request.limit > ListQueuedJobsConstants.LIMIT_MAX
+            ):
+                await context.abort(
+                    StatusCode.INVALID_ARGUMENT, "Limit must be between 1 and 100."
+                )
+
+            if request.offset < ListQueuedJobsConstants.OFFSET_MIN:
+                await context.abort(
+                    StatusCode.INVALID_ARGUMENT, "Offset cannot be negative."
+                )
+
+            queued_job_metadata = datastore.list_queued_job_metadata(
+                offset=request.offset, limit=request.limit
+            )
+
+            response_list: list[ListQueuedJobsResult.ListQueuedJobsResultItem] = []
+            for item in queued_job_metadata:
+                proto_timestamp = Timestamp()
+                proto_timestamp.FromDatetime(item.submitted_at)
+
+                response_list.append(
+                    ListQueuedJobsResult.ListQueuedJobsResultItem(
+                        uuid=item.queued_job_uuid,
+                        job_ref_name=item.job_ref_name,
+                        num_attempts=item.num_attempts,
+                        submitted_at=proto_timestamp,
+                    )
+                )
+            return ListQueuedJobsResult(queued_jobs=response_list)
+
     add_CommandServerServicer_to_server(CommandServerHandler(), server)
 
     listen_addr = f"[::]:{get_local_admin_server_port()}"

uncountablepythonsdk-0.0.118/uncountable/integration/queue_runner/command_server/constants.py

@@ -0,0 +1,4 @@
+class ListQueuedJobsConstants:
+    LIMIT_MIN = 1
+    LIMIT_MAX = 100
+    OFFSET_MIN = 0

{uncountablepythonsdk-0.0.117 → uncountablepythonsdk-0.0.118}/uncountable/integration/queue_runner/command_server/protocol/command_server.proto

@@ -1,8 +1,10 @@
 syntax = "proto3";
+import "google/protobuf/timestamp.proto";
 
 service CommandServer {
 	rpc EnqueueJob(EnqueueJobRequest) returns (EnqueueJobResult) {}
 	rpc CheckHealth(CheckHealthRequest) returns (CheckHealthResult) {}
+	rpc ListQueuedJobs(ListQueuedJobsRequest) returns (ListQueuedJobsResult) {}
 }
 
 message EnqueueJobRequest {
@@ -20,3 +22,19 @@ message CheckHealthRequest {}
 message CheckHealthResult {
 	bool success = 1;
 }
+
+message ListQueuedJobsRequest {
+	uint32 offset = 1;
+	uint32 limit = 2;
+}
+
+message ListQueuedJobsResult {
+	message ListQueuedJobsResultItem {
+	string uuid = 1;
+	string job_ref_name = 2;
+	int64 num_attempts = 3;
+	google.protobuf.Timestamp submitted_at = 4;
+	}
+
+	repeated ListQueuedJobsResultItem queued_jobs  = 1;
+}

uncountablepythonsdk-0.0.118/uncountable/integration/queue_runner/command_server/protocol/command_server_pb2.py

@@ -0,0 +1,49 @@
+# ruff: noqa
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: uncountable/integration/queue_runner/command_server/protocol/command_server.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\nQuncountable/integration/queue_runner/command_server/protocol/command_server.proto\x1a\x1fgoogle/protobuf/timestamp.proto"E\n\x11\x45nqueueJobRequest\x12\x14\n\x0cjob_ref_name\x18\x01 \x01(\t\x12\x1a\n\x12serialized_payload\x18\x02 \x01(\t"H\n\x10\x45nqueueJobResult\x12\x1b\n\x13successfully_queued\x18\x01 \x01(\x08\x12\x17\n\x0fqueued_job_uuid\x18\x02 \x01(\t"\x14\n\x12\x43heckHealthRequest"$\n\x11\x43heckHealthResult\x12\x0f\n\x07success\x18\x01 \x01(\x08"6\n\x15ListQueuedJobsRequest\x12\x0e\n\x06offset\x18\x01 \x01(\r\x12\r\n\x05limit\x18\x02 \x01(\r"\xe4\x01\n\x14ListQueuedJobsResult\x12\x43\n\x0bqueued_jobs\x18\x01 \x03(\x0b\x32..ListQueuedJobsResult.ListQueuedJobsResultItem\x1a\x86\x01\n\x18ListQueuedJobsResultItem\x12\x0c\n\x04uuid\x18\x01 \x01(\t\x12\x14\n\x0cjob_ref_name\x18\x02 \x01(\t\x12\x14\n\x0cnum_attempts\x18\x03 \x01(\x03\x12\x30\n\x0csubmitted_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp2\xc3\x01\n\rCommandServer\x12\x35\n\nEnqueueJob\x12\x12.EnqueueJobRequest\x1a\x11.EnqueueJobResult"\x00\x12\x38\n\x0b\x43heckHealth\x12\x13.CheckHealthRequest\x1a\x12.CheckHealthResult"\x00\x12\x41\n\x0eListQueuedJobs\x12\x16.ListQueuedJobsRequest\x1a\x15.ListQueuedJobsResult"\x00\x62\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(
+    DESCRIPTOR,
+    "uncountable.integration.queue_runner.command_server.protocol.command_server_pb2",
+    _globals,
+)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    DESCRIPTOR._options = None
+    _globals["_ENQUEUEJOBREQUEST"]._serialized_start = 118
+    _globals["_ENQUEUEJOBREQUEST"]._serialized_end = 187
+    _globals["_ENQUEUEJOBRESULT"]._serialized_start = 189
+    _globals["_ENQUEUEJOBRESULT"]._serialized_end = 261
+    _globals["_CHECKHEALTHREQUEST"]._serialized_start = 263
+    _globals["_CHECKHEALTHREQUEST"]._serialized_end = 283
+    _globals["_CHECKHEALTHRESULT"]._serialized_start = 285
+    _globals["_CHECKHEALTHRESULT"]._serialized_end = 321
+    _globals["_LISTQUEUEDJOBSREQUEST"]._serialized_start = 323
+    _globals["_LISTQUEUEDJOBSREQUEST"]._serialized_end = 377
+    _globals["_LISTQUEUEDJOBSRESULT"]._serialized_start = 380
+    _globals["_LISTQUEUEDJOBSRESULT"]._serialized_end = 608
+    _globals["_LISTQUEUEDJOBSRESULT_LISTQUEUEDJOBSRESULTITEM"]._serialized_start = 474
+    _globals["_LISTQUEUEDJOBSRESULT_LISTQUEUEDJOBSRESULTITEM"]._serialized_end = 608
+    _globals["_COMMANDSERVER"]._serialized_start = 611
+    _globals["_COMMANDSERVER"]._serialized_end = 806
+# @@protoc_insertion_point(module_scope)

uncountablepythonsdk-0.0.118/uncountable/integration/queue_runner/command_server/protocol/command_server_pb2.pyi

@@ -0,0 +1,87 @@
+# ruff: noqa
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class EnqueueJobRequest(_message.Message):
+    __slots__ = ("job_ref_name", "serialized_payload")
+    JOB_REF_NAME_FIELD_NUMBER: _ClassVar[int]
+    SERIALIZED_PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    job_ref_name: str
+    serialized_payload: str
+    def __init__(
+        self,
+        job_ref_name: _Optional[str] = ...,
+        serialized_payload: _Optional[str] = ...,
+    ) -> None: ...
+
+class EnqueueJobResult(_message.Message):
+    __slots__ = ("successfully_queued", "queued_job_uuid")
+    SUCCESSFULLY_QUEUED_FIELD_NUMBER: _ClassVar[int]
+    QUEUED_JOB_UUID_FIELD_NUMBER: _ClassVar[int]
+    successfully_queued: bool
+    queued_job_uuid: str
+    def __init__(
+        self, successfully_queued: bool = ..., queued_job_uuid: _Optional[str] = ...
+    ) -> None: ...
+
+class CheckHealthRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class CheckHealthResult(_message.Message):
+    __slots__ = ("success",)
+    SUCCESS_FIELD_NUMBER: _ClassVar[int]
+    success: bool
+    def __init__(self, success: bool = ...) -> None: ...
+
+class ListQueuedJobsRequest(_message.Message):
+    __slots__ = ("offset", "limit")
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    offset: int
+    limit: int
+    def __init__(
+        self, offset: _Optional[int] = ..., limit: _Optional[int] = ...
+    ) -> None: ...
+
+class ListQueuedJobsResult(_message.Message):
+    __slots__ = ("queued_jobs",)
+    class ListQueuedJobsResultItem(_message.Message):
+        __slots__ = ("uuid", "job_ref_name", "num_attempts", "submitted_at")
+        UUID_FIELD_NUMBER: _ClassVar[int]
+        JOB_REF_NAME_FIELD_NUMBER: _ClassVar[int]
+        NUM_ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
+        SUBMITTED_AT_FIELD_NUMBER: _ClassVar[int]
+        uuid: str
+        job_ref_name: str
+        num_attempts: int
+        submitted_at: _timestamp_pb2.Timestamp
+        def __init__(
+            self,
+            uuid: _Optional[str] = ...,
+            job_ref_name: _Optional[str] = ...,
+            num_attempts: _Optional[int] = ...,
+            submitted_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        ) -> None: ...
+
+    QUEUED_JOBS_FIELD_NUMBER: _ClassVar[int]
+    queued_jobs: _containers.RepeatedCompositeFieldContainer[
+        ListQueuedJobsResult.ListQueuedJobsResultItem
+    ]
+    def __init__(
+        self,
+        queued_jobs: _Optional[
+            _Iterable[_Union[ListQueuedJobsResult.ListQueuedJobsResultItem, _Mapping]]
+        ] = ...,
+    ) -> None: ...