UncountablePythonSDK 0.0.175__tar.gz → 0.0.177__tar.gz

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/.github/workflows/publish.yml

@@ -28,7 +28,7 @@ jobs:
         with:
           version: "0.11.0"
           activate-environment: true
-          enable-cache: true
+          enable-cache: false
       - name: Install build dependencies
         run: |
           uv pip install --upgrade build

{uncountablepythonsdk-0.0.175/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.177}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.175
+Version: 0.0.177
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -17,7 +17,7 @@ Classifier: Typing :: Typed
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 Requires-Dist: aiotus==1.*
-Requires-Dist: aiohttp==3.*
+Requires-Dist: aiohttp>=3.14.1
 Requires-Dist: requests==2.*
 Requires-Dist: SQLAlchemy>=1.4.0
 Requires-Dist: APScheduler==3.*

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.175
+Version: 0.0.177
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -17,7 +17,7 @@ Classifier: Typing :: Typed
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 Requires-Dist: aiotus==1.*
-Requires-Dist: aiohttp==3.*
+Requires-Dist: aiohttp>=3.14.1
 Requires-Dist: requests==2.*
 Requires-Dist: SQLAlchemy>=1.4.0
 Requires-Dist: APScheduler==3.*

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -148,6 +148,7 @@ uncountable/integration/scan_profiles.py
 uncountable/integration/scheduler.py
 uncountable/integration/server.py
 uncountable/integration/telemetry.py
+uncountable/integration/webhook_signature_key.py
 uncountable/integration/db/__init__.py
 uncountable/integration/db/connect.py
 uncountable/integration/db/session.py
@@ -166,6 +167,7 @@ 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/provision_webhook.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
@@ -177,6 +179,7 @@ uncountable/integration/queue_runner/datastore/datastore_sqlite.py
 uncountable/integration/queue_runner/datastore/interface.py
 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/entrypoint.py
 uncountable/types/__init__.py
@@ -290,8 +293,10 @@ uncountable/types/uploader.py
 uncountable/types/uploader_t.py
 uncountable/types/users.py
 uncountable/types/users_t.py
+uncountable/types/webhook.py
 uncountable/types/webhook_job.py
 uncountable/types/webhook_job_t.py
+uncountable/types/webhook_t.py
 uncountable/types/workflows.py
 uncountable/types/workflows_t.py
 uncountable/types/api/__init__.py

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/UncountablePythonSDK.egg-info/requires.txt

@@ -1,5 +1,5 @@
 aiotus==1.*
-aiohttp==3.*
+aiohttp>=3.14.1
 requests==2.*
 SQLAlchemy>=1.4.0
 APScheduler==3.*

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/docs/conf.py

@@ -7,6 +7,8 @@
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 
 import datetime
+import importlib
+import inspect
 
 from docutils import nodes  # type: ignore[import-untyped]
 from sphinx.addnodes import pending_xref  # type: ignore[import-not-found]
@@ -63,6 +65,20 @@ favicons = [
 ]
 
 
+def _resolve_alias_to_real_module(target_module: str) -> str:
+    if "." not in target_module:
+        return target_module
+    parent_name, attr_name = target_module.rsplit(".", 1)
+    try:
+        parent = importlib.import_module(parent_name)
+    except ImportError:
+        return target_module
+    aliased = getattr(parent, attr_name, None)
+    if not inspect.ismodule(aliased) or aliased.__name__ == target_module:
+        return target_module
+    return aliased.__name__
+
+
 def _hook_missing_reference(
     _app: Sphinx, _env: BuildEnvironment, node: pending_xref, contnode: nodes.Text
 ) -> nodes.reference | None:
@@ -80,6 +96,10 @@ def _hook_missing_reference(
         return None
 
     target_module, target_name = target.rsplit(".", 1)
+    real_module = _resolve_alias_to_real_module(target_module)
+    if real_module != target_module:
+        target_module = real_module
+        target = f"{target_module}.{target_name}"
 
     # construct relative path from current doc page to target page
     relative_segments_to_root = [".." for _ in current_doc.split("/")]

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pkgs/type_spec/actions_registry/__main__.py

@@ -92,6 +92,7 @@ def main() -> None:
                 ]
                 if definition.visibility_scope is not None
                 else None,
+                kind=definition.kind,
                 arguments=definition.arguments,
             )
             action_definitions_with_arguments_list.append(

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pkgs/type_spec/actions_registry/emit_python.py

@@ -54,6 +54,10 @@ def _emit_action_definition(
         out.write(
             f"{sub_indent}visibility_scope={_emit_visibility_scope(action.visibility_scope)},\n"
         )
+    if action.kind is not None:
+        out.write(
+            f"{sub_indent}kind=actions_registry_t.ActionDefinitionKind.{action.kind.name},\n"
+        )
     if action.arguments is not None:
         out.write(f"{sub_indent}arguments={_py_str(action.arguments)},\n")
     out.write(f"{indent}),\n")

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pkgs/type_spec/actions_registry/emit_typescript.py

@@ -147,6 +147,10 @@ def _emit_action_definition(
         out.write(
             f"{sub_indent}visibilityScope: {_emit_visibility_scope(action_definition.visibility_scope)},\n"
         )
+    if action_definition.kind is not None:
+        out.write(
+            f"{sub_indent}kind: ActionsRegistryT.ActionDefinitionKind.{ts_name(action_definition.kind, name_case=builder.NameCase.convert)},\n"
+        )
     out.write(f"{indent}}} as ActionsRegistryT.ActionDefinition{argument_type},\n")
     return out.getvalue()
 

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pkgs/type_spec/value_spec/__main__.py

@@ -173,6 +173,7 @@ key_description = "description"
 key_brief = "brief"
 key_name = "name"
 key_draft = "draft"
+key_deferred_value_spec = "deferred_value_spec"
 
 
 TypeT = TypeVar("TypeT")
@@ -236,6 +237,14 @@ def main() -> None:
                     arg_meta = get(args_meta, in_argument.ref_name)
                     arg_description = get_as(arg_meta, key_description, str)
                     arg_name = get_as(arg_meta, key_name, str)
+                    deferred_value_spec = None
+                    if (
+                        isinstance(arg_meta, dict)
+                        and arg_meta.get(key_deferred_value_spec) is not None
+                    ):
+                        deferred_value_spec = get_as(
+                            arg_meta, key_deferred_value_spec, bool
+                        )
 
                     arguments.append(
                         value_spec_t.FunctionArgument(
@@ -245,6 +254,7 @@ def main() -> None:
                             type=convert_to_value_spec_type(in_argument.type_path),
                             on_null=in_argument.on_null,
                             extant=in_argument.extant,
+                            deferred_value_spec=deferred_value_spec,
                         )
                     )
 

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pkgs/type_spec/value_spec/emit_python.py

@@ -29,6 +29,7 @@ from decimal import Decimal
 
 from main.base.types import value_spec_t, base_t
 
+from . import expression
 from .function_wrapper import ExtractorArgs, WrapArgs
 """
     )
@@ -70,6 +71,12 @@ def _emit_function_wrapper(function: value_spec_t.Function) -> str:
 
 {INDENT}def get_{argument.ref_name}_at(self, at: int) -> base_t.ExtJsonValue:
 {INDENT}{INDENT}return self._extract_at({index}, at)
+"""
+            )
+        elif argument.deferred_value_spec:
+            out.write(
+                f"""{INDENT}def get_{argument.ref_name}(self) -> expression.ParseNode | None:
+{INDENT}{INDENT}return self.extract_node({index})
 """
             )
         else:
@@ -213,6 +220,8 @@ def _emit_argument(argument: value_spec_t.FunctionArgument, indent: str) -> str:
     out.write(
         f"{sub_indent}extant=value_spec_t.ArgumentExtant.{argument.extant.name},\n"
     )
+    if argument.deferred_value_spec:
+        out.write(f"{sub_indent}deferred_value_spec={argument.deferred_value_spec},\n")
     out.write(f"{sub_indent}type={_emit_type(argument.type, sub_indent)},\n")
     out.write(f"{indent})")
 

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/pyproject.toml

@@ -52,7 +52,7 @@ requires-python = ">=3.12"
 # May be compatible with older versions
 dependencies = [
     "aiotus ==1.*",
-    "aiohttp == 3.*",
+    "aiohttp >= 3.14.1",
     "requests == 2.*",
     "SQLAlchemy >= 1.4.0",
     "APScheduler == 3.*",

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/core/file_upload.py

@@ -17,6 +17,7 @@ from uncountable.types import uploader_t
 from .types import AuthDetailsAll, AuthDetailsApiKey
 
 _CHUNK_SIZE = 5 * 1024 * 1024  # s3 requires 5MiB minimum
+_PARSED_FILE_DATA_FILENAME = "parsed_file_data.json"
 
 
 class FileUploadType(StrEnum):
@@ -51,17 +52,27 @@ class UploaderFileUpload:
 FileUpload = MediaFileUpload | DataFileUpload | UploaderFileUpload
 
 
-@dataclass(kw_only=True)
-class _ParsedFileDataWrapper:
-    parsed_file_data: list[uploader_t.ParsedFileData]
-
-
 @dataclass(kw_only=True)
 class FileBytes:
     name: str
     bytes_data: BytesIO
 
 
+def _build_parsed_file_data_bytes(
+    *, parsed_file_data: list[uploader_t.ParsedFileData]
+) -> FileBytes:
+    buffer = BytesIO()
+    buffer.write(b'{"parsed_file_data": [')
+    for index, file_data in enumerate(parsed_file_data):
+        if index > 0:
+            buffer.write(b", ")
+        serialized_file_data = serialize_for_storage(file_data)
+        buffer.write(json.dumps(serialized_file_data).encode("utf-8"))
+    buffer.write(b"]}")
+    buffer.seek(0)
+    return FileBytes(name=_PARSED_FILE_DATA_FILENAME, bytes_data=buffer)
+
+
 @contextmanager
 def file_upload_data(file_upload: FileUpload) -> Generator[FileBytes, None, None]:
     match file_upload:
@@ -73,15 +84,9 @@ def file_upload_data(file_upload: FileUpload) -> Generator[FileBytes, None, None
         case DataFileUpload():
             yield FileBytes(name=file_upload.name, bytes_data=file_upload.data)
         case UploaderFileUpload():
-            wrapper = _ParsedFileDataWrapper(
+            yield _build_parsed_file_data_bytes(
                 parsed_file_data=file_upload.parsed_file_data
             )
-            serialized = serialize_for_storage(wrapper)
-            json_bytes = json.dumps(serialized).encode("utf-8")
-            yield FileBytes(
-                name="parsed_file_data.json",
-                bytes_data=BytesIO(json_bytes),
-            )
 
 
 @dataclass(kw_only=True)
@@ -135,7 +140,7 @@ class FileUploader:
                 case DataFileUpload():
                     attributes["file_name"] = file_upload.name
                 case UploaderFileUpload():
-                    attributes["file_name"] = "parsed_file_data.json"
+                    attributes["file_name"] = _PARSED_FILE_DATA_FILENAME
                 case _:
                     assert_never(file_upload)
             with push_scope_optional(

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/core/query/builder.py

@@ -13,6 +13,7 @@ from uncountable.core.query.column import (
     Column,
     ColumnFilter,
     EnumColumn,
+    JsonColumn,
     NullableColumn,
     PrimaryKeyColumn,
     get_column_base,
@@ -72,8 +73,16 @@ class QueryBuilder[QueryRowT: QueryRow]:
             column.identifier for column in column_definitions.values()
         )
         self._property_names: tuple[str, ...] = tuple(column_definitions.keys())
+        json_property_names = {
+            name
+            for name, column in column_definitions.items()
+            if isinstance(column, JsonColumn)
+        }
         self._parser: CachedParser = CachedParser(
-            serial_class(unconverted_keys={*self._property_names})(
+            serial_class(
+                unconverted_keys={*self._property_names},
+                unconverted_values=json_property_names,
+            )(
                 make_dataclass(
                     cls_name=self._model.__name__,
                     fields=[

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/core/query/column.py

@@ -465,6 +465,27 @@ class NullableTextsColumn[
     # IMPROVE: Implement "Not Exists" filter (when supported in SDK)
 
 
+class JsonColumn(Column[base_t.JsonValue, NullableColumn]):
+    __hash__ = Column.__hash__
+
+    def __init__(self, identifier: listing_t.ColumnIdentifier) -> None:
+        super().__init__(
+            identifier=identifier,
+            nullable=NullableColumn,
+            value_type=(
+                dict[str, base_t.JsonValue] | list[base_t.JsonValue] | base_t.JsonScalar
+            ),
+        )
+
+    def __ne__(self, other: None) -> ColumnFilter:  # type: ignore[override]
+        return ColumnFilter(
+            identifier=self.identifier,
+            filter=lambda identifier: listing_t.FilterSpecExists(column=identifier),
+        )
+
+    # IMPROVE: Implement "Not Exists" filter (when supported in SDK)
+
+
 class PrimaryKeyColumn(
     Column[base_t.ObjectId, NonNullableColumn],
 ):
@@ -684,6 +705,10 @@ class IdColumn[
         column: NullableTextsColumn[BaseNullableT],
     ) -> NullableTextsColumn[BaseNullableT]: ...
 
+    # Json
+    @overload
+    def __rshift__(self, column: JsonColumn) -> JsonColumn: ...
+
     # Ids
     @overload
     def __rshift__[
@@ -734,6 +759,7 @@ class IdColumn[
         | NumericColumn[BaseNullableT]
         | TextsColumn[BaseNullableT]
         | NullableTextsColumn[BaseNullableT]
+        | JsonColumn
         | IdsColumn[BaseNullableT]
         | IdColumn[BaseNullableT],
     ) -> (
@@ -768,6 +794,7 @@ class IdColumn[
         | NullableTextsColumn[BaseNullableT]
         | NullableTextsColumn[NonNullableColumn | NullableColumn]
         | NullableTextsColumn[NullableColumn]
+        | JsonColumn
         | IdsColumn[BaseNullableT]
         | IdsColumn[NonNullableColumn | NullableColumn]
         | IdsColumn[NullableColumn]
@@ -869,6 +896,8 @@ class IdColumn[
                     if self.nullable is NonNullableColumn
                     else NullableColumn,
                 )
+            case JsonColumn():
+                return JsonColumn(identifier=transitive_identifier)
             case IdsColumn():
                 return IdsColumn(
                     id_type=column.id_type,

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/cli.py

@@ -12,6 +12,7 @@ from uncountable.integration.queue_runner.command_server.command_client import (
     send_job_cancellation_message,
     send_job_queue_message,
     send_list_queued_jobs_message,
+    send_provision_webhook_message,
     send_retry_job_message,
 )
 from uncountable.integration.queue_runner.command_server.types import (
@@ -220,6 +221,33 @@ def register_retry_job_parser(
     retry_failed_jobs_parser.set_defaults(func=_handle_retry_job)
 
 
+def register_provision_webhook_parser(
+    sub_parser_manager: argparse._SubParsersAction,
+    parents: list[argparse.ArgumentParser],
+) -> None:
+    provision_parser = sub_parser_manager.add_parser(
+        "provision-webhook",
+        parents=parents,
+        help="Provision an Uncountable webhook job by creating its webhook entity and signature key",
+        description="Provision an Uncountable webhook job by creating its webhook entity and signature key",
+    )
+    provision_parser.add_argument(
+        "job_id",
+        type=str,
+        help="The id of the Uncountable webhook job to provision",
+    )
+
+    def _handle_provision_webhook(args: argparse.Namespace) -> None:
+        profile_name = send_provision_webhook_message(
+            job_id=args.job_id,
+            host=args.host,
+            port=get_local_admin_server_port(),
+        )
+        print(f"provisioned webhook {profile_name}/{args.job_id}")
+
+    provision_parser.set_defaults(func=_handle_provision_webhook)
+
+
 def main() -> None:
     logger = Logger(get_current_span())
 
@@ -242,6 +270,7 @@ def main() -> None:
     register_retry_job_parser(subparser_action, parents=[base_parser])
     register_list_queued_jobs(subparser_action, parents=[base_parser])
     register_cancel_queued_job_parser(subparser_action, parents=[base_parser])
+    register_provision_webhook_parser(subparser_action, parents=[base_parser])
 
     args = main_parser.parse_args()
     with logger.push_scope(args.command):

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/construct_client.py

@@ -1,8 +1,8 @@
 from uncountable.core import AuthDetailsApiKey, Client
 from uncountable.core.client import ClientConfig
 from uncountable.core.types import AuthDetailsAll, AuthDetailsOAuth
-from uncountable.integration.secret_retrieval.retrieve_secret import retrieve_secret
-from uncountable.integration.telemetry import JobLogger
+from uncountable.integration.secret_retrieval.basic_secret import retrieve_basic_secret
+from uncountable.integration.telemetry import Logger
 from uncountable.types import auth_retrieval_t
 from uncountable.types.client_config_t import RequestContext
 from uncountable.types.job_definition_t import (
@@ -13,16 +13,16 @@ from uncountable.types.job_definition_t import (
 def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
     match profile_meta.auth_retrieval:
         case auth_retrieval_t.AuthRetrievalOAuth():
-            refresh_token = retrieve_secret(
+            refresh_token = retrieve_basic_secret(
                 profile_meta.auth_retrieval.refresh_token_secret,
                 profile_metadata=profile_meta,
             )
             return AuthDetailsOAuth(refresh_token=refresh_token)
         case auth_retrieval_t.AuthRetrievalBasic():
-            api_id = retrieve_secret(
+            api_id = retrieve_basic_secret(
                 profile_meta.auth_retrieval.api_id_secret, profile_metadata=profile_meta
             )
-            api_key = retrieve_secret(
+            api_key = retrieve_basic_secret(
                 profile_meta.auth_retrieval.api_key_secret,
                 profile_metadata=profile_meta,
             )
@@ -32,7 +32,7 @@ def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
 
 def construct_uncountable_client(
     profile_meta: ProfileMetadata,
-    logger: JobLogger,
+    logger: Logger,
     *,
     request_context: RequestContext | None = None,
     headers: dict[str, str] | None = None,

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/job.py

@@ -25,8 +25,11 @@ from uncountable.integration.queue_runner.command_server.types import (
     CommandServerException,
 )
 from uncountable.integration.request_context import extract_request_context
-from uncountable.integration.secret_retrieval.retrieve_secret import retrieve_secret
+from uncountable.integration.secret_retrieval import retrieve_secret
 from uncountable.integration.telemetry import JobLogger
+from uncountable.integration.webhook_signature_key import (
+    webhook_signature_key_retrieval,
+)
 from uncountable.types import (
     base_t,
     job_definition_t,
@@ -166,6 +169,26 @@ class CustomHttpJob(Job[GenericHttpRequest]):
         return JobResult(success=False)
 
 
+def _resolve_webhook_signature_key(
+    *,
+    job_definition: job_definition_t.HttpJobDefinitionBase,
+    profile_meta: ProfileMetadata,
+) -> str:
+    if isinstance(job_definition, job_definition_t.WebhookJobDefinition):
+        return retrieve_secret(job_definition.signature_key_secret, profile_meta)
+    if isinstance(job_definition, job_definition_t.UncountableWebhookJobDefinition):
+        return retrieve_secret(
+            webhook_signature_key_retrieval(job_id=job_definition.id),
+            profile_meta,
+        )
+    raise HttpException.configuration_error(
+        message=(
+            "[internal] non-webhook job definition passed to webhook validation: "
+            f"{profile_meta.name}/{job_definition.id}"
+        )
+    )
+
+
 class WebhookJob[WPT](Job[webhook_job_t.WebhookEventPayload]):
     @property
     def payload_type(self) -> type[webhook_job_t.WebhookEventPayload]:
@@ -182,10 +205,8 @@ class WebhookJob[WPT](Job[webhook_job_t.WebhookEventPayload]):
         job_definition: job_definition_t.HttpJobDefinitionBase,
         profile_meta: ProfileMetadata,
     ) -> None:
-        assert isinstance(job_definition, job_definition_t.WebhookJobDefinition)
-        signature_key = retrieve_secret(
-            profile_metadata=profile_meta,
-            secret_retrieval=job_definition.signature_key_secret,
+        signature_key = _resolve_webhook_signature_key(
+            job_definition=job_definition, profile_meta=profile_meta
         )
         passed_signature = request.headers.get("Uncountable-Webhook-Signature")
         if passed_signature is None:
@@ -195,7 +216,7 @@ class WebhookJob[WPT](Job[webhook_job_t.WebhookEventPayload]):
             signature_key.encode("utf-8"), msg=request.body_bytes, digestmod="sha256"
         ).hexdigest()
 
-        if request_body_signature != passed_signature:
+        if not hmac.compare_digest(request_body_signature, passed_signature):
             raise HttpException.payload_failed_signature()
 
     @staticmethod

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/queue_runner/command_server/command_client.py

@@ -17,6 +17,8 @@ from uncountable.integration.queue_runner.command_server.protocol.command_server
     EnqueueJobResult,
     ListQueuedJobsRequest,
     ListQueuedJobsResult,
+    ProvisionWebhookRequest,
+    ProvisionWebhookResult,
     RetryJobRequest,
     RetryJobResult,
     VaccuumQueuedJobsRequest,
@@ -33,6 +35,7 @@ from .protocol.command_server_pb2_grpc import CommandServerStub
 
 _LOCAL_RPC_HOST = "localhost"
 _DEFAULT_MESSAGE_TIMEOUT_SECS = 2
+_PROVISION_WEBHOOK_TIMEOUT_SECS = 30
 
 
 @contextmanager
@@ -171,3 +174,20 @@ def send_vaccuum_queued_jobs_message(*, host: str = "localhost", port: int) -> N
 
         assert isinstance(response, VaccuumQueuedJobsResult)
         return None
+
+
+def send_provision_webhook_message(
+    *, job_id: str, host: str = "localhost", port: int
+) -> str:
+    with command_server_connection(host=host, port=port) as stub:
+        request = ProvisionWebhookRequest(job_id=job_id)
+
+        try:
+            response = stub.ProvisionWebhook(
+                request, timeout=_PROVISION_WEBHOOK_TIMEOUT_SECS
+            )
+        except grpc.RpcError as e:
+            raise ValueError(e.details())
+
+        assert isinstance(response, ProvisionWebhookResult)
+        return response.profile_name

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/queue_runner/command_server/command_server.py

@@ -19,11 +19,16 @@ from uncountable.integration.queue_runner.command_server.protocol.command_server
     EnqueueJobResult,
     ListQueuedJobsRequest,
     ListQueuedJobsResult,
+    ProvisionWebhookRequest,
+    ProvisionWebhookResult,
     RetryJobRequest,
     RetryJobResult,
     VaccuumQueuedJobsRequest,
     VaccuumQueuedJobsResult,
 )
+from uncountable.integration.queue_runner.command_server.provision_webhook import (
+    provision_webhook,
+)
 from uncountable.integration.queue_runner.command_server.types import (
     CommandCancelJob,
     CommandCancelJobResponse,
@@ -199,6 +204,20 @@ async def serve(command_queue: CommandQueue, datastore: DatastoreSqlite) -> None
             )
             return VaccuumQueuedJobsResult()
 
+        async def ProvisionWebhook(
+            self, request: ProvisionWebhookRequest, context: grpc_aio.ServicerContext
+        ) -> ProvisionWebhookResult:
+            loop = asyncio.get_running_loop()
+            profile_name = await loop.run_in_executor(
+                None, lambda: provision_webhook(job_id=request.job_id)
+            )
+            if profile_name is None:
+                await context.abort(
+                    StatusCode.INVALID_ARGUMENT,
+                    f"no Uncountable webhook job found with id {request.job_id}",
+                )
+            return ProvisionWebhookResult(profile_name=profile_name)
+
     add_CommandServerServicer_to_server(CommandServerHandler(), server)
 
     listen_addr = f"[::]:{get_local_admin_server_port()}"

{uncountablepythonsdk-0.0.175 → uncountablepythonsdk-0.0.177}/uncountable/integration/queue_runner/command_server/protocol/command_server.proto

@@ -8,6 +8,7 @@ service CommandServer {
 	rpc ListQueuedJobs(ListQueuedJobsRequest) returns (ListQueuedJobsResult) {}
 	rpc VaccuumQueuedJobs(VaccuumQueuedJobsRequest) returns (VaccuumQueuedJobsResult) {}
 	rpc CancelJob(CancelJobRequest) returns (CancelJobResult) {}
+	rpc ProvisionWebhook(ProvisionWebhookRequest) returns (ProvisionWebhookResult) {}
 }
 
 message EnqueueJobRequest {
@@ -75,3 +76,11 @@ enum CancelJobStatus {
 message CancelJobResult {
 	CancelJobStatus status = 1;
 }
+
+message ProvisionWebhookRequest {
+	string job_id = 1;
+}
+
+message ProvisionWebhookResult {
+	string profile_name = 1;
+}