udb 0.3.1

"""Generated UDB SDK robustness layer — typed, retrying RPC wrappers.

Generated by `udb sdk generate` from the embedded proto descriptor set. Edit the template under sdk-templates/<lang>/, not this file.

UDB version:      0.3.1
Protocol version: 1.0.0
Services:         7
RPCs:             153

This module is a *forwarding* layer over the buf-generated gRPC service stubs
under ``udb.*`` (the ``*_pb2_grpc`` modules). For every RPC across every UDB
service it emits a wrapper that forwards to the underlying stub method of the
same name while adding:

  * a configurable per-call deadline / timeout,
  * retry with exponential backoff + jitter on transient gRPC codes
    (UNAVAILABLE, DEADLINE_EXCEEDED, RESOURCE_EXHAUSTED) — never for
    client-streaming RPCs, which are not safely replayable,
  * TLS / credential and channel wiring,
  * request metadata (authorization / api-key / x-request-id and the full
    :class:`udb_client.metadata.Metadata` header set),
  * typed error mapping that unpacks the ``udb-error-detail-bin`` trailer into
    :class:`UdbDetailedRpcError` when present, else a clean
    :class:`udb_client.exceptions.UdbRpcError`.

It COMPOSES WITH — and never overwrites — the hand-written
:mod:`udb_client.client`, :mod:`udb_client.auth`, :mod:`udb_client.metadata`,
:mod:`udb_client.negotiation`, and :mod:`udb_client.exceptions` modules; it
imports them rather than re-implementing them.
"""

from __future__ import annotations

import importlib
import pkgutil
import random
import time
import uuid
from dataclasses import dataclass, field
from typing import Any, Callable, Iterable, Iterator, Sequence

import grpc

from .exceptions import UdbConfigurationError, UdbError, UdbRpcError
from .metadata import Metadata

__all__ = [
    "RetryPolicy",
    "UdbDetailedRpcError",
    "GeneratedClient",
    "AnalyticsServiceClient",
    "ApiKeyServiceClient",
    "AuthnServiceClient",
    "AuthzServiceClient",
    "NotificationServiceClient",
    "TenantServiceClient",
    "DataBrokerClient",
]

# Default set of gRPC status codes that are safe to retry for idempotent RPCs.
_RETRYABLE_CODES: frozenset[grpc.StatusCode] = frozenset(
    {
        grpc.StatusCode.UNAVAILABLE,
        grpc.StatusCode.DEADLINE_EXCEEDED,
        grpc.StatusCode.RESOURCE_EXHAUSTED,
    }
)

# Binary trailer the broker uses to carry a structured ErrorDetail (Rule: the
# server attaches `udb-error-detail-bin` with the serialized ErrorDetail proto).
_ERROR_DETAIL_TRAILER = "udb-error-detail-bin"


@dataclass(frozen=True)
class RetryPolicy:
    """Exponential-backoff-with-jitter retry policy for transient gRPC errors."""

    max_attempts: int = 4
    initial_backoff: float = 0.1
    max_backoff: float = 5.0
    multiplier: float = 2.0
    jitter: float = 0.2
    retryable_codes: frozenset[grpc.StatusCode] = field(default_factory=lambda: _RETRYABLE_CODES)

    def should_retry(self, code: grpc.StatusCode, attempt: int) -> bool:
        return attempt < self.max_attempts and code in self.retryable_codes

    def backoff_for(self, attempt: int) -> float:
        base = min(self.initial_backoff * (self.multiplier ** (attempt - 1)), self.max_backoff)
        return base + random.uniform(0.0, base * self.jitter)


class UdbDetailedRpcError(UdbRpcError):
    """An :class:`UdbRpcError` enriched with the broker's ``ErrorDetail`` trailer.

    ``detail`` is the raw bytes of the ``udb-error-detail-bin`` trailer when the
    server provided one (callers may parse it with the appropriate generated
    ``ErrorDetail`` message); otherwise ``None``.
    """

    def __init__(self, rpc_name: str, error: grpc.RpcError, detail: bytes | None):
        super().__init__(rpc_name, error)
        self.detail = detail


def _extract_error_detail(error: grpc.RpcError) -> bytes | None:
    trailers_obj: object = None
    getter = getattr(error, "trailing_metadata", None)
    if callable(getter):
        try:
            trailers_obj = getter()
        except Exception:  # pragma: no cover - defensive
            trailers_obj = None
    if not isinstance(trailers_obj, Sequence):
        return None
    for entry in trailers_obj:
        if not isinstance(entry, tuple) or len(entry) != 2:
            continue
        key, value = entry
        if key == _ERROR_DETAIL_TRAILER:
            if isinstance(value, bytes):
                return value
            if isinstance(value, bytearray):
                return bytes(value)
            return None
    return None


def _map_error(rpc_name: str, error: grpc.RpcError) -> UdbRpcError:
    detail = _extract_error_detail(error)
    if detail is not None:
        return UdbDetailedRpcError(rpc_name, error, detail)
    return UdbRpcError(rpc_name, error)


def _discover_stub_class(service_pkg: str, service_name: str) -> type:
    """Locate the buf-generated ``<service_name>Stub`` inside ``service_pkg``.

    The gRPC stub module filename is the *proto file* basename (e.g.
    ``authn_service_pb2_grpc``), which is not derivable from the service name —
    so we import the package and scan its ``*_pb2_grpc`` submodules for the
    class. Proto remains the source of truth; nothing is hand-listed.
    """

    pkg = importlib.import_module(service_pkg)
    stub_name = f"{service_name}Stub"
    # Fast path: already-imported attribute on the package.
    found = getattr(pkg, stub_name, None)
    if isinstance(found, type):
        return found
    search_paths = getattr(pkg, "__path__", None)
    if search_paths is not None:
        for mod in pkgutil.iter_modules(search_paths):
            if not mod.name.endswith("_pb2_grpc"):
                continue
            module = importlib.import_module(f"{service_pkg}.{mod.name}")
            candidate = getattr(module, stub_name, None)
            if isinstance(candidate, type):
                return candidate
    raise UdbConfigurationError(
        f"could not locate gRPC stub '{stub_name}' under package '{service_pkg}'. "
        f"Run `buf generate` so the *_pb2_grpc modules are present."
    )


class _ServiceClientBase:
    """Shared transport/robustness machinery for every generated service client.

    Builds (or adopts) a gRPC channel, instantiates the discovered buf stub, and
    provides the retrying ``_invoke_*`` helpers the per-RPC wrappers forward to.
    """

    _SERVICE_PKG: str = ""
    _SERVICE_NAME: str = ""
    _SERVICE_FULL: str = ""

    def __init__(
        self,
        target: str = "",
        metadata: Metadata | None = None,
        *,
        secure: bool = False,
        root_certificates: bytes | None = None,
        call_credentials: grpc.CallCredentials | None = None,
        channel_credentials: grpc.ChannelCredentials | None = None,
        channel_options: Sequence[tuple[str, Any]] | None = None,
        timeout: float | None = 30.0,
        retry: RetryPolicy | None = None,
        api_key: str = "",
        bearer_token: str = "",
        channel: grpc.Channel | None = None,
    ) -> None:
        if not target and channel is None:
            raise UdbConfigurationError("UDB target is required, e.g. '127.0.0.1:50051'")
        self._metadata = metadata
        self._timeout = timeout
        self._retry = retry or RetryPolicy()
        self._api_key = api_key
        self._bearer_token = bearer_token
        self._owns_channel = channel is None
        if channel is None:
            options = list(channel_options or ())
            if channel_credentials is not None:
                channel = grpc.secure_channel(target, channel_credentials, options=options)
            elif secure:
                creds = grpc.ssl_channel_credentials(root_certificates)
                if call_credentials is not None:
                    creds = grpc.composite_channel_credentials(creds, call_credentials)
                channel = grpc.secure_channel(target, creds, options=options)
            else:
                channel = grpc.insecure_channel(target, options=options)
        self._channel = channel
        stub_cls = _discover_stub_class(self._SERVICE_PKG, self._SERVICE_NAME)
        self._stub = stub_cls(channel)

    # ── lifecycle ───────────────────────────────────────────────────────────
    @property
    def stub(self) -> Any:
        """The raw buf-generated stub, for RPCs or knobs not surfaced here."""

        return self._stub

    def bind_metadata(self, metadata: Metadata) -> None:
        self._metadata = metadata

    def close(self) -> None:
        if self._owns_channel:
            self._channel.close()

    def __enter__(self) -> "_ServiceClientBase":
        return self

    def __exit__(self, *_: object) -> None:
        self.close()

    # ── metadata assembly ────────────────────────────────────────────────────
    def _effective_metadata(self, metadata: Metadata | None) -> Metadata | None:
        return metadata or self._metadata

    def _call_metadata(
        self, metadata: Metadata | None, request_id: str | None
    ) -> tuple[tuple[str, str], ...]:
        headers: list[tuple[str, str]] = []
        effective = self._effective_metadata(metadata)
        if effective is not None:
            headers.extend(effective.to_grpc_metadata())
        if self._bearer_token:
            headers.append(("authorization", f"Bearer {self._bearer_token}"))
        if self._api_key:
            headers.append(("x-udb-api-key", self._api_key))
        headers.append(("x-request-id", request_id or uuid.uuid4().hex))
        return tuple(headers)

    # ── retrying invocation primitives ───────────────────────────────────────
    def _invoke_unary(
        self,
        method: Callable[..., Any],
        rpc_name: str,
        request: Any,
        *,
        metadata: Metadata | None,
        timeout: float | None,
        retryable: bool,
    ) -> Any:
        attempt = 0
        while True:
            attempt += 1
            request_id = uuid.uuid4().hex
            try:
                return method(
                    request,
                    metadata=self._call_metadata(metadata, request_id),
                    timeout=self._timeout if timeout is None else timeout,
                )
            except grpc.RpcError as error:
                code = error.code()
                if retryable and self._retry.should_retry(code, attempt):
                    time.sleep(self._retry.backoff_for(attempt))
                    continue
                raise _map_error(rpc_name, error) from error

    def _invoke_server_streaming(
        self,
        method: Callable[..., Any],
        rpc_name: str,
        request: Any,
        *,
        metadata: Metadata | None,
        timeout: float | None,
        retryable: bool,
    ) -> Iterator[Any]:
        # Retry only applies before the first item is yielded; once streaming has
        # begun a partial replay is not safe, so we surface errors as they occur.
        attempt = 0
        while True:
            attempt += 1
            request_id = uuid.uuid4().hex
            stream = method(
                request,
                metadata=self._call_metadata(metadata, request_id),
                timeout=self._timeout if timeout is None else timeout,
            )
            iterator = iter(stream)
            try:
                first = next(iterator)
            except StopIteration:
                return iter(())
            except grpc.RpcError as error:
                code = error.code()
                if retryable and self._retry.should_retry(code, attempt):
                    time.sleep(self._retry.backoff_for(attempt))
                    continue
                raise _map_error(rpc_name, error) from error

            def _drain() -> Iterator[Any]:
                yield first
                try:
                    for item in iterator:
                        yield item
                except grpc.RpcError as error:
                    raise _map_error(rpc_name, error) from error

            return _drain()

    def _invoke_client_streaming(
        self,
        method: Callable[..., Any],
        rpc_name: str,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None,
        timeout: float | None,
    ) -> Any:
        # Client-streaming / bidi requests are never auto-retried: the request
        # iterator is generally not replayable.
        request_id = uuid.uuid4().hex
        try:
            return method(
                request_iterator,
                metadata=self._call_metadata(metadata, request_id),
                timeout=self._timeout if timeout is None else timeout,
            )
        except grpc.RpcError as error:
            raise _map_error(rpc_name, error) from error


class AnalyticsServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.analytics.services.v1.AnalyticsService`` service (7 RPC(s)).

    Forwards to the buf-generated ``AnalyticsServiceStub`` under
    ``udb.core.analytics.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.analytics.services.v1"
    _SERVICE_NAME = "AnalyticsService"
    _SERVICE_FULL = "udb.core.analytics.services.v1.AnalyticsService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def get_executor_performance(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetExecutorPerformance`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/GetExecutorPerformance``.

        Forwards ``GetExecutorPerformanceRequest`` (udb.core.analytics.services.v1) and returns
        ``GetExecutorPerformanceResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetExecutorPerformance,
            "GetExecutorPerformance",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_pipeline_summary(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetPipelineSummary`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/GetPipelineSummary``.

        Forwards ``GetPipelineSummaryRequest`` (udb.core.analytics.services.v1) and returns
        ``GetPipelineSummaryResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetPipelineSummary,
            "GetPipelineSummary",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_reconciliation_analytics(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetReconciliationAnalytics`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/GetReconciliationAnalytics``.

        Forwards ``GetReconciliationAnalyticsRequest`` (udb.core.analytics.services.v1) and returns
        ``GetReconciliationAnalyticsResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetReconciliationAnalytics,
            "GetReconciliationAnalytics",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_sla_compliance(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetSlaCompliance`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/GetSlaCompliance``.

        Forwards ``GetSlaComplianceRequest`` (udb.core.analytics.services.v1) and returns
        ``GetSlaComplianceResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetSlaCompliance,
            "GetSlaCompliance",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_throughput(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetThroughput`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/GetThroughput``.

        Forwards ``GetThroughputRequest`` (udb.core.analytics.services.v1) and returns
        ``GetThroughputResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetThroughput,
            "GetThroughput",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def record_pipeline_metric(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RecordPipelineMetric`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/RecordPipelineMetric``.

        Forwards ``RecordPipelineMetricRequest`` (udb.core.analytics.services.v1) and returns
        ``RecordPipelineMetricResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.RecordPipelineMetric,
            "RecordPipelineMetric",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def trigger_snapshot(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``TriggerSnapshot`` — unary call to ``/udb.core.analytics.services.v1.AnalyticsService/TriggerSnapshot``.

        Forwards ``TriggerSnapshotRequest`` (udb.core.analytics.services.v1) and returns
        ``TriggerSnapshotResponse`` (udb.core.analytics.services.v1).
        """

        return self._invoke_unary(
            self._stub.TriggerSnapshot,
            "TriggerSnapshot",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class ApiKeyServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.apikey.services.v1.ApiKeyService`` service (7 RPC(s)).

    Forwards to the buf-generated ``ApiKeyServiceStub`` under
    ``udb.core.apikey.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.apikey.services.v1"
    _SERVICE_NAME = "ApiKeyService"
    _SERVICE_FULL = "udb.core.apikey.services.v1.ApiKeyService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def create_api_key(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateApiKey`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/CreateApiKey``.

        Forwards ``CreateApiKeyRequest`` (udb.core.apikey.services.v1) and returns
        ``CreateApiKeyResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreateApiKey,
            "CreateApiKey",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_api_key(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetApiKey`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/GetApiKey``.

        Forwards ``GetApiKeyRequest`` (udb.core.apikey.services.v1) and returns
        ``GetApiKeyResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetApiKey,
            "GetApiKey",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_api_key_usage_stats(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetApiKeyUsageStats`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/GetApiKeyUsageStats``.

        Forwards ``GetApiKeyUsageStatsRequest`` (udb.core.apikey.services.v1) and returns
        ``GetApiKeyUsageStatsResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetApiKeyUsageStats,
            "GetApiKeyUsageStats",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_api_keys(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListApiKeys`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/ListApiKeys``.

        Forwards ``ListApiKeysRequest`` (udb.core.apikey.services.v1) and returns
        ``ListApiKeysResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListApiKeys,
            "ListApiKeys",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def revoke_api_key(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RevokeApiKey`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/RevokeApiKey``.

        Forwards ``RevokeApiKeyRequest`` (udb.core.apikey.services.v1) and returns
        ``RevokeApiKeyResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.RevokeApiKey,
            "RevokeApiKey",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def update_api_key(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpdateApiKey`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/UpdateApiKey``.

        Forwards ``UpdateApiKeyRequest`` (udb.core.apikey.services.v1) and returns
        ``UpdateApiKeyResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpdateApiKey,
            "UpdateApiKey",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def validate_api_key(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ValidateApiKey`` — unary call to ``/udb.core.apikey.services.v1.ApiKeyService/ValidateApiKey``.

        Forwards ``ValidateApiKeyRequest`` (udb.core.apikey.services.v1) and returns
        ``ValidateApiKeyResponse`` (udb.core.apikey.services.v1).
        """

        return self._invoke_unary(
            self._stub.ValidateApiKey,
            "ValidateApiKey",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class AuthnServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.authn.services.v1.AuthnService`` service (23 RPC(s)).

    Forwards to the buf-generated ``AuthnServiceStub`` under
    ``udb.core.authn.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.authn.services.v1"
    _SERVICE_NAME = "AuthnService"
    _SERVICE_FULL = "udb.core.authn.services.v1.AuthnService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def admin_reset_password(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``AdminResetPassword`` — unary call to ``/udb.core.authn.services.v1.AuthnService/AdminResetPassword``.

        Forwards ``AdminResetPasswordRequest`` (udb.core.authn.services.v1) and returns
        ``AdminResetPasswordResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.AdminResetPassword,
            "AdminResetPassword",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def authenticate(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Authenticate`` — unary call to ``/udb.core.authn.services.v1.AuthnService/Authenticate``.

        Forwards ``AuthnRequest`` (udb.core.authn.services.v1) and returns
        ``AuthnResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.Authenticate,
            "Authenticate",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def change_password(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ChangePassword`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ChangePassword``.

        Forwards ``ChangePasswordRequest`` (udb.core.authn.services.v1) and returns
        ``ChangePasswordResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ChangePassword,
            "ChangePassword",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def change_user_status(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ChangeUserStatus`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ChangeUserStatus``.

        Forwards ``ChangeUserStatusRequest`` (udb.core.authn.services.v1) and returns
        ``ChangeUserStatusResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ChangeUserStatus,
            "ChangeUserStatus",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def confirm_m_f_a_enrollment(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ConfirmMFAEnrollment`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ConfirmMFAEnrollment``.

        Forwards ``ConfirmMFAEnrollmentRequest`` (udb.core.authn.services.v1) and returns
        ``ConfirmMFAEnrollmentResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ConfirmMFAEnrollment,
            "ConfirmMFAEnrollment",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def create_session(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateSession`` — unary call to ``/udb.core.authn.services.v1.AuthnService/CreateSession``.

        Forwards ``CreateSessionRequest`` (udb.core.authn.services.v1) and returns
        ``CreateSessionResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreateSession,
            "CreateSession",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def create_user(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateUser`` — unary call to ``/udb.core.authn.services.v1.AuthnService/CreateUser``.

        Forwards ``CreateUserRequest`` (udb.core.authn.services.v1) and returns
        ``CreateUserResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreateUser,
            "CreateUser",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def enroll_m_f_a(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``EnrollMFA`` — unary call to ``/udb.core.authn.services.v1.AuthnService/EnrollMFA``.

        Forwards ``EnrollMFARequest`` (udb.core.authn.services.v1) and returns
        ``EnrollMFAResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.EnrollMFA,
            "EnrollMFA",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_session(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetSession`` — unary call to ``/udb.core.authn.services.v1.AuthnService/GetSession``.

        Forwards ``GetSessionRequest`` (udb.core.authn.services.v1) and returns
        ``GetSessionResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetSession,
            "GetSession",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_user(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetUser`` — unary call to ``/udb.core.authn.services.v1.AuthnService/GetUser``.

        Forwards ``GetUserRequest`` (udb.core.authn.services.v1) and returns
        ``GetUserResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetUser,
            "GetUser",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_sessions(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListSessions`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ListSessions``.

        Forwards ``ListSessionsRequest`` (udb.core.authn.services.v1) and returns
        ``ListSessionsResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListSessions,
            "ListSessions",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_users(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListUsers`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ListUsers``.

        Forwards ``ListUsersRequest`` (udb.core.authn.services.v1) and returns
        ``ListUsersResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListUsers,
            "ListUsers",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def login(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Login`` — unary call to ``/udb.core.authn.services.v1.AuthnService/Login``.

        Forwards ``LoginRequest`` (udb.core.authn.services.v1) and returns
        ``LoginResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.Login,
            "Login",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def logout(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Logout`` — unary call to ``/udb.core.authn.services.v1.AuthnService/Logout``.

        Forwards ``LogoutRequest`` (udb.core.authn.services.v1) and returns
        ``LogoutResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.Logout,
            "Logout",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def refresh_session(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RefreshSession`` — unary call to ``/udb.core.authn.services.v1.AuthnService/RefreshSession``.

        Forwards ``RefreshSessionRequest`` (udb.core.authn.services.v1) and returns
        ``RefreshSessionResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.RefreshSession,
            "RefreshSession",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def refresh_token(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RefreshToken`` — unary call to ``/udb.core.authn.services.v1.AuthnService/RefreshToken``.

        Forwards ``RefreshTokenRequest`` (udb.core.authn.services.v1) and returns
        ``RefreshTokenResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.RefreshToken,
            "RefreshToken",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def resend_o_t_p(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ResendOTP`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ResendOTP``.

        Forwards ``ResendOTPRequest`` (udb.core.authn.services.v1) and returns
        ``ResendOTPResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ResendOTP,
            "ResendOTP",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def revoke_session(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RevokeSession`` — unary call to ``/udb.core.authn.services.v1.AuthnService/RevokeSession``.

        Forwards ``RevokeSessionRequest`` (udb.core.authn.services.v1) and returns
        ``RevokeSessionResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.RevokeSession,
            "RevokeSession",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def send_o_t_p(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``SendOTP`` — unary call to ``/udb.core.authn.services.v1.AuthnService/SendOTP``.

        Forwards ``SendOTPRequest`` (udb.core.authn.services.v1) and returns
        ``SendOTPResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.SendOTP,
            "SendOTP",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def update_user(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpdateUser`` — unary call to ``/udb.core.authn.services.v1.AuthnService/UpdateUser``.

        Forwards ``UpdateUserRequest`` (udb.core.authn.services.v1) and returns
        ``UpdateUserResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpdateUser,
            "UpdateUser",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def validate_c_s_r_f(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ValidateCSRF`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ValidateCSRF``.

        Forwards ``ValidateCSRFRequest`` (udb.core.authn.services.v1) and returns
        ``ValidateCSRFResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ValidateCSRF,
            "ValidateCSRF",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def validate_token(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ValidateToken`` — unary call to ``/udb.core.authn.services.v1.AuthnService/ValidateToken``.

        Forwards ``ValidateTokenRequest`` (udb.core.authn.services.v1) and returns
        ``ValidateTokenResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.ValidateToken,
            "ValidateToken",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def verify_o_t_p(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``VerifyOTP`` — unary call to ``/udb.core.authn.services.v1.AuthnService/VerifyOTP``.

        Forwards ``VerifyOTPRequest`` (udb.core.authn.services.v1) and returns
        ``VerifyOTPResponse`` (udb.core.authn.services.v1).
        """

        return self._invoke_unary(
            self._stub.VerifyOTP,
            "VerifyOTP",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class AuthzServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.authz.services.v1.AuthzService`` service (23 RPC(s)).

    Forwards to the buf-generated ``AuthzServiceStub`` under
    ``udb.core.authz.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.authz.services.v1"
    _SERVICE_NAME = "AuthzService"
    _SERVICE_FULL = "udb.core.authz.services.v1.AuthzService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def assign_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``AssignRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/AssignRole``.

        Forwards ``AssignRoleRequest`` (udb.core.authz.services.v1) and returns
        ``AssignRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.AssignRole,
            "AssignRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def authorize(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Authorize`` — unary call to ``/udb.core.authz.services.v1.AuthzService/Authorize``.

        Forwards ``AuthzRequest`` (udb.core.authz.services.v1) and returns
        ``AuthzResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.Authorize,
            "Authorize",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def batch_check_permissions(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``BatchCheckPermissions`` — unary call to ``/udb.core.authz.services.v1.AuthzService/BatchCheckPermissions``.

        Forwards ``BatchCheckPermissionsRequest`` (udb.core.authz.services.v1) and returns
        ``BatchCheckPermissionsResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.BatchCheckPermissions,
            "BatchCheckPermissions",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def check_access(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CheckAccess`` — unary call to ``/udb.core.authz.services.v1.AuthzService/CheckAccess``.

        Forwards ``CheckAccessRequest`` (udb.core.authz.services.v1) and returns
        ``CheckAccessResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.CheckAccess,
            "CheckAccess",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def create_policy_rule(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreatePolicyRule`` — unary call to ``/udb.core.authz.services.v1.AuthzService/CreatePolicyRule``.

        Forwards ``CreatePolicyRuleRequest`` (udb.core.authz.services.v1) and returns
        ``CreatePolicyRuleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreatePolicyRule,
            "CreatePolicyRule",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def create_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/CreateRole``.

        Forwards ``CreateRoleRequest`` (udb.core.authz.services.v1) and returns
        ``CreateRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreateRole,
            "CreateRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def delete_policy_rule(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DeletePolicyRule`` — unary call to ``/udb.core.authz.services.v1.AuthzService/DeletePolicyRule``.

        Forwards ``DeletePolicyRuleRequest`` (udb.core.authz.services.v1) and returns
        ``DeletePolicyRuleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.DeletePolicyRule,
            "DeletePolicyRule",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def delete_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DeleteRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/DeleteRole``.

        Forwards ``DeleteRoleRequest`` (udb.core.authz.services.v1) and returns
        ``DeleteRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.DeleteRole,
            "DeleteRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_native_access(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetNativeAccess`` — unary call to ``/udb.core.authz.services.v1.AuthzService/GetNativeAccess``.

        Forwards ``NativeAccessRequest`` (udb.core.authz.services.v1) and returns
        ``NativeAccessResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetNativeAccess,
            "GetNativeAccess",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_policy_bundle(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetPolicyBundle`` — unary call to ``/udb.core.authz.services.v1.AuthzService/GetPolicyBundle``.

        Forwards ``PolicyBundleRequest`` (udb.core.authz.services.v1) and returns
        ``PolicyBundleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetPolicyBundle,
            "GetPolicyBundle",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_policy_rule(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetPolicyRule`` — unary call to ``/udb.core.authz.services.v1.AuthzService/GetPolicyRule``.

        Forwards ``GetPolicyRuleRequest`` (udb.core.authz.services.v1) and returns
        ``GetPolicyRuleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetPolicyRule,
            "GetPolicyRule",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/GetRole``.

        Forwards ``GetRoleRequest`` (udb.core.authz.services.v1) and returns
        ``GetRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetRole,
            "GetRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def lint_authz_policies(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``LintAuthzPolicies`` — unary call to ``/udb.core.authz.services.v1.AuthzService/LintAuthzPolicies``.

        Forwards ``LintAuthzPoliciesRequest`` (udb.core.authz.services.v1) and returns
        ``LintAuthzPoliciesResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.LintAuthzPolicies,
            "LintAuthzPolicies",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_access_decision_audits(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListAccessDecisionAudits`` — unary call to ``/udb.core.authz.services.v1.AuthzService/ListAccessDecisionAudits``.

        Forwards ``ListAccessDecisionAuditsRequest`` (udb.core.authz.services.v1) and returns
        ``ListAccessDecisionAuditsResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListAccessDecisionAudits,
            "ListAccessDecisionAudits",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_policy_rules(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListPolicyRules`` — unary call to ``/udb.core.authz.services.v1.AuthzService/ListPolicyRules``.

        Forwards ``ListPolicyRulesRequest`` (udb.core.authz.services.v1) and returns
        ``ListPolicyRulesResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListPolicyRules,
            "ListPolicyRules",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_roles(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListRoles`` — unary call to ``/udb.core.authz.services.v1.AuthzService/ListRoles``.

        Forwards ``ListRolesRequest`` (udb.core.authz.services.v1) and returns
        ``ListRolesResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListRoles,
            "ListRoles",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_user_permissions(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListUserPermissions`` — unary call to ``/udb.core.authz.services.v1.AuthzService/ListUserPermissions``.

        Forwards ``ListUserPermissionsRequest`` (udb.core.authz.services.v1) and returns
        ``ListUserPermissionsResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListUserPermissions,
            "ListUserPermissions",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_user_roles(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListUserRoles`` — unary call to ``/udb.core.authz.services.v1.AuthzService/ListUserRoles``.

        Forwards ``ListUserRolesRequest`` (udb.core.authz.services.v1) and returns
        ``ListUserRolesResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListUserRoles,
            "ListUserRoles",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def put_authz_policy(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PutAuthzPolicy`` — unary call to ``/udb.core.authz.services.v1.AuthzService/PutAuthzPolicy``.

        Forwards ``PutAuthzPolicyRequest`` (udb.core.authz.services.v1) and returns
        ``AuthMutationResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.PutAuthzPolicy,
            "PutAuthzPolicy",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def put_relationship(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PutRelationship`` — unary call to ``/udb.core.authz.services.v1.AuthzService/PutRelationship``.

        Forwards ``PutRelationshipRequest`` (udb.core.authz.services.v1) and returns
        ``AuthMutationResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.PutRelationship,
            "PutRelationship",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def put_role_binding(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PutRoleBinding`` — unary call to ``/udb.core.authz.services.v1.AuthzService/PutRoleBinding``.

        Forwards ``PutRoleBindingRequest`` (udb.core.authz.services.v1) and returns
        ``AuthMutationResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.PutRoleBinding,
            "PutRoleBinding",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def revoke_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RevokeRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/RevokeRole``.

        Forwards ``RevokeRoleRequest`` (udb.core.authz.services.v1) and returns
        ``RevokeRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.RevokeRole,
            "RevokeRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def update_role(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpdateRole`` — unary call to ``/udb.core.authz.services.v1.AuthzService/UpdateRole``.

        Forwards ``UpdateRoleRequest`` (udb.core.authz.services.v1) and returns
        ``UpdateRoleResponse`` (udb.core.authz.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpdateRole,
            "UpdateRole",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class NotificationServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.notification.services.v1.NotificationService`` service (11 RPC(s)).

    Forwards to the buf-generated ``NotificationServiceStub`` under
    ``udb.core.notification.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.notification.services.v1"
    _SERVICE_NAME = "NotificationService"
    _SERVICE_FULL = "udb.core.notification.services.v1.NotificationService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def get_delivery_stats(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetDeliveryStats`` — unary call to ``/udb.core.notification.services.v1.NotificationService/GetDeliveryStats``.

        Forwards ``GetDeliveryStatsRequest`` (udb.core.notification.services.v1) and returns
        ``GetDeliveryStatsResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetDeliveryStats,
            "GetDeliveryStats",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_notification(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetNotification`` — unary call to ``/udb.core.notification.services.v1.NotificationService/GetNotification``.

        Forwards ``GetNotificationRequest`` (udb.core.notification.services.v1) and returns
        ``GetNotificationResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetNotification,
            "GetNotification",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_preference(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetPreference`` — unary call to ``/udb.core.notification.services.v1.NotificationService/GetPreference``.

        Forwards ``GetPreferenceRequest`` (udb.core.notification.services.v1) and returns
        ``GetPreferenceResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetPreference,
            "GetPreference",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_template(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetTemplate`` — unary call to ``/udb.core.notification.services.v1.NotificationService/GetTemplate``.

        Forwards ``GetTemplateRequest`` (udb.core.notification.services.v1) and returns
        ``GetTemplateResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetTemplate,
            "GetTemplate",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_notifications(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListNotifications`` — unary call to ``/udb.core.notification.services.v1.NotificationService/ListNotifications``.

        Forwards ``ListNotificationsRequest`` (udb.core.notification.services.v1) and returns
        ``ListNotificationsResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListNotifications,
            "ListNotifications",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_preferences(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListPreferences`` — unary call to ``/udb.core.notification.services.v1.NotificationService/ListPreferences``.

        Forwards ``ListPreferencesRequest`` (udb.core.notification.services.v1) and returns
        ``ListPreferencesResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListPreferences,
            "ListPreferences",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_templates(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListTemplates`` — unary call to ``/udb.core.notification.services.v1.NotificationService/ListTemplates``.

        Forwards ``ListTemplatesRequest`` (udb.core.notification.services.v1) and returns
        ``ListTemplatesResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListTemplates,
            "ListTemplates",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def retry_notification(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RetryNotification`` — unary call to ``/udb.core.notification.services.v1.NotificationService/RetryNotification``.

        Forwards ``RetryNotificationRequest`` (udb.core.notification.services.v1) and returns
        ``RetryNotificationResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.RetryNotification,
            "RetryNotification",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def send_notification(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``SendNotification`` — unary call to ``/udb.core.notification.services.v1.NotificationService/SendNotification``.

        Forwards ``SendNotificationRequest`` (udb.core.notification.services.v1) and returns
        ``SendNotificationResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.SendNotification,
            "SendNotification",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def set_preference(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``SetPreference`` — unary call to ``/udb.core.notification.services.v1.NotificationService/SetPreference``.

        Forwards ``SetPreferenceRequest`` (udb.core.notification.services.v1) and returns
        ``SetPreferenceResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.SetPreference,
            "SetPreference",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def upsert_template(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpsertTemplate`` — unary call to ``/udb.core.notification.services.v1.NotificationService/UpsertTemplate``.

        Forwards ``UpsertTemplateRequest`` (udb.core.notification.services.v1) and returns
        ``UpsertTemplateResponse`` (udb.core.notification.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpsertTemplate,
            "UpsertTemplate",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class TenantServiceClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.core.tenant.services.v1.TenantService`` service (6 RPC(s)).

    Forwards to the buf-generated ``TenantServiceStub`` under
    ``udb.core.tenant.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.core.tenant.services.v1"
    _SERVICE_NAME = "TenantService"
    _SERVICE_FULL = "udb.core.tenant.services.v1.TenantService"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def create_tenant(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateTenant`` — unary call to ``/udb.core.tenant.services.v1.TenantService/CreateTenant``.

        Forwards ``CreateTenantRequest`` (udb.core.tenant.services.v1) and returns
        ``CreateTenantResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.CreateTenant,
            "CreateTenant",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_tenant(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetTenant`` — unary call to ``/udb.core.tenant.services.v1.TenantService/GetTenant``.

        Forwards ``GetTenantRequest`` (udb.core.tenant.services.v1) and returns
        ``GetTenantResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetTenant,
            "GetTenant",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_tenant_config(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetTenantConfig`` — unary call to ``/udb.core.tenant.services.v1.TenantService/GetTenantConfig``.

        Forwards ``GetTenantConfigRequest`` (udb.core.tenant.services.v1) and returns
        ``GetTenantConfigResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.GetTenantConfig,
            "GetTenantConfig",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_tenants(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListTenants`` — unary call to ``/udb.core.tenant.services.v1.TenantService/ListTenants``.

        Forwards ``ListTenantsRequest`` (udb.core.tenant.services.v1) and returns
        ``ListTenantsResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.ListTenants,
            "ListTenants",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def update_tenant(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpdateTenant`` — unary call to ``/udb.core.tenant.services.v1.TenantService/UpdateTenant``.

        Forwards ``UpdateTenantRequest`` (udb.core.tenant.services.v1) and returns
        ``UpdateTenantResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpdateTenant,
            "UpdateTenant",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def update_tenant_config(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``UpdateTenantConfig`` — unary call to ``/udb.core.tenant.services.v1.TenantService/UpdateTenantConfig``.

        Forwards ``UpdateTenantConfigRequest`` (udb.core.tenant.services.v1) and returns
        ``UpdateTenantConfigResponse`` (udb.core.tenant.services.v1).
        """

        return self._invoke_unary(
            self._stub.UpdateTenantConfig,
            "UpdateTenantConfig",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────

    # ── client-streaming RPCs ─────────────────────────────────────────────────

    # ── bidirectional-streaming RPCs ──────────────────────────────────────────

class DataBrokerClient(_ServiceClientBase):
    """Robustness wrapper for the ``udb.services.v1.DataBroker`` service (76 RPC(s)).

    Forwards to the buf-generated ``DataBrokerStub`` under
    ``udb.services.v1`` with retry, deadlines, metadata, and typed errors.
    """

    _SERVICE_PKG = "udb.services.v1"
    _SERVICE_NAME = "DataBroker"
    _SERVICE_FULL = "udb.services.v1.DataBroker"

    # ── unary RPCs ────────────────────────────────────────────────────────────
    def activate_catalog(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ActivateCatalog`` — unary call to ``/udb.services.v1.DataBroker/ActivateCatalog``.

        Forwards ``CatalogVersionRequest`` (udb.entity.v1) and returns
        ``CatalogVersionResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ActivateCatalog,
            "ActivateCatalog",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def analytical_query(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``AnalyticalQuery`` — unary call to ``/udb.services.v1.DataBroker/AnalyticalQuery``.

        Forwards ``AnalyticalQueryRequest`` (udb.entity.v1) and returns
        ``AnalyticalQueryResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.AnalyticalQuery,
            "AnalyticalQuery",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def apply_migration(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ApplyMigration`` — unary call to ``/udb.services.v1.DataBroker/ApplyMigration``.

        Forwards ``MigrationApplyRequest`` (udb.entity.v1) and returns
        ``MigrationStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ApplyMigration,
            "ApplyMigration",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def approve_migration_plan(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ApproveMigrationPlan`` — unary call to ``/udb.services.v1.DataBroker/ApproveMigrationPlan``.

        Forwards ``MigrationRunRequest`` (udb.entity.v1) and returns
        ``MigrationStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ApproveMigrationPlan,
            "ApproveMigrationPlan",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def cache_delete(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CacheDelete`` — unary call to ``/udb.services.v1.DataBroker/CacheDelete``.

        Forwards ``CacheDeleteRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.CacheDelete,
            "CacheDelete",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def cache_get(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CacheGet`` — unary call to ``/udb.services.v1.DataBroker/CacheGet``.

        Forwards ``CacheGetRequest`` (udb.entity.v1) and returns
        ``CacheGetResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.CacheGet,
            "CacheGet",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def cache_scan(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CacheScan`` — unary call to ``/udb.services.v1.DataBroker/CacheScan``.

        Forwards ``CacheScanRequest`` (udb.entity.v1) and returns
        ``CacheScanResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.CacheScan,
            "CacheScan",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def cache_set(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CacheSet`` — unary call to ``/udb.services.v1.DataBroker/CacheSet``.

        Forwards ``CacheSetRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.CacheSet,
            "CacheSet",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def create_materialized_view(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``CreateMaterializedView`` — unary call to ``/udb.services.v1.DataBroker/CreateMaterializedView``.

        Forwards ``ViewDefinition`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.CreateMaterializedView,
            "CreateMaterializedView",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def delete(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Delete`` — unary call to ``/udb.services.v1.DataBroker/Delete``.

        Forwards ``DeleteRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.Delete,
            "Delete",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def delete_policy(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DeletePolicy`` — unary call to ``/udb.services.v1.DataBroker/DeletePolicy``.

        Forwards ``PolicyRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DeletePolicy,
            "DeletePolicy",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def dismiss_dlq_event(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DismissDlqEvent`` — unary call to ``/udb.services.v1.DataBroker/DismissDlqEvent``.

        Forwards ``DlqActionRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DismissDlqEvent,
            "DismissDlqEvent",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def document_delete(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DocumentDelete`` — unary call to ``/udb.services.v1.DataBroker/DocumentDelete``.

        Forwards ``DocumentDeleteRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DocumentDelete,
            "DocumentDelete",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def document_find(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DocumentFind`` — unary call to ``/udb.services.v1.DataBroker/DocumentFind``.

        Forwards ``DocumentFindRequest`` (udb.entity.v1) and returns
        ``DocumentSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DocumentFind,
            "DocumentFind",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def document_get(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DocumentGet`` — unary call to ``/udb.services.v1.DataBroker/DocumentGet``.

        Forwards ``DocumentGetRequest`` (udb.entity.v1) and returns
        ``DocumentSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DocumentGet,
            "DocumentGet",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def document_upsert(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DocumentUpsert`` — unary call to ``/udb.services.v1.DataBroker/DocumentUpsert``.

        Forwards ``DocumentUpsertRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DocumentUpsert,
            "DocumentUpsert",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def drop_resource(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``DropResource`` — unary call to ``/udb.services.v1.DataBroker/DropResource``.

        Forwards ``ResourceAdminRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.DropResource,
            "DropResource",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def enqueue_outbox_event(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``EnqueueOutboxEvent`` — unary call to ``/udb.services.v1.DataBroker/EnqueueOutboxEvent``.

        Forwards ``EnqueueOutboxEventRequest`` (udb.entity.v1) and returns
        ``EnqueueOutboxEventResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.EnqueueOutboxEvent,
            "EnqueueOutboxEvent",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def ensure_project(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``EnsureProject`` — unary call to ``/udb.services.v1.DataBroker/EnsureProject``.

        Forwards ``EnsureProjectRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.EnsureProject,
            "EnsureProject",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def ensure_resource(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``EnsureResource`` — unary call to ``/udb.services.v1.DataBroker/EnsureResource``.

        Forwards ``ResourceAdminRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.EnsureResource,
            "EnsureResource",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def generate_presigned_url(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GeneratePresignedUrl`` — unary call to ``/udb.services.v1.DataBroker/GeneratePresignedUrl``.

        Forwards ``UrlRequest`` (udb.entity.v1) and returns
        ``UrlResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GeneratePresignedUrl,
            "GeneratePresignedUrl",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def generic_dispatch(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GenericDispatch`` — unary call to ``/udb.services.v1.DataBroker/GenericDispatch``.

        Forwards ``GenericDispatchRequest`` (udb.entity.v1) and returns
        ``GenericDispatchResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GenericDispatch,
            "GenericDispatch",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_admin_summary(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetAdminSummary`` — unary call to ``/udb.services.v1.DataBroker/GetAdminSummary``.

        Forwards ``AdminSummaryRequest`` (udb.entity.v1) and returns
        ``AdminSummaryResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetAdminSummary,
            "GetAdminSummary",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_capabilities(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetCapabilities`` — unary call to ``/udb.services.v1.DataBroker/GetCapabilities``.

        Forwards ``CapabilitiesRequest`` (udb.entity.v1) and returns
        ``CapabilitiesResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetCapabilities,
            "GetCapabilities",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_catalog_manifest(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetCatalogManifest`` — unary call to ``/udb.services.v1.DataBroker/GetCatalogManifest``.

        Forwards ``CatalogManifestRequest`` (udb.entity.v1) and returns
        ``CatalogManifestResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetCatalogManifest,
            "GetCatalogManifest",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_catalog_version(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetCatalogVersion`` — unary call to ``/udb.services.v1.DataBroker/GetCatalogVersion``.

        Forwards ``CatalogVersionRequest`` (udb.entity.v1) and returns
        ``CatalogVersionResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetCatalogVersion,
            "GetCatalogVersion",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_catalog_versions(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetCatalogVersions`` — unary call to ``/udb.services.v1.DataBroker/GetCatalogVersions``.

        Forwards ``CatalogManifestRequest`` (udb.entity.v1) and returns
        ``CatalogVersionListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetCatalogVersions,
            "GetCatalogVersions",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_cdc_status(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetCdcStatus`` — unary call to ``/udb.services.v1.DataBroker/GetCdcStatus``.

        Forwards ``CdcControlRequest`` (udb.entity.v1) and returns
        ``CdcStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetCdcStatus,
            "GetCdcStatus",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_dlq_event(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetDlqEvent`` — unary call to ``/udb.services.v1.DataBroker/GetDlqEvent``.

        Forwards ``DlqEventRequest`` (udb.entity.v1) and returns
        ``DlqEventResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetDlqEvent,
            "GetDlqEvent",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_health_report(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetHealthReport`` — unary call to ``/udb.services.v1.DataBroker/GetHealthReport``.

        Forwards ``HealthReportRequest`` (udb.entity.v1) and returns
        ``HealthReportResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetHealthReport,
            "GetHealthReport",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_migration_status(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetMigrationStatus`` — unary call to ``/udb.services.v1.DataBroker/GetMigrationStatus``.

        Forwards ``MigrationRunRequest`` (udb.entity.v1) and returns
        ``MigrationStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetMigrationStatus,
            "GetMigrationStatus",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def get_saga(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GetSaga`` — unary call to ``/udb.services.v1.DataBroker/GetSaga``.

        Forwards ``SagaRequest`` (udb.entity.v1) and returns
        ``SagaResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GetSaga,
            "GetSaga",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def graph_mutate(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GraphMutate`` — unary call to ``/udb.services.v1.DataBroker/GraphMutate``.

        Forwards ``GraphMutationRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GraphMutate,
            "GraphMutate",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def graph_query(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``GraphQuery`` — unary call to ``/udb.services.v1.DataBroker/GraphQuery``.

        Forwards ``GraphQueryRequest`` (udb.entity.v1) and returns
        ``GraphResultSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.GraphQuery,
            "GraphQuery",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def initiate_multipart_upload(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``InitiateMultipartUpload`` — unary call to ``/udb.services.v1.DataBroker/InitiateMultipartUpload``.

        Forwards ``MultipartUploadRequest`` (udb.entity.v1) and returns
        ``MultipartUploadResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.InitiateMultipartUpload,
            "InitiateMultipartUpload",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def lint_policies(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``LintPolicies`` — unary call to ``/udb.services.v1.DataBroker/LintPolicies``.

        Forwards ``CapabilitiesRequest`` (udb.entity.v1) and returns
        ``PolicyLintResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.LintPolicies,
            "LintPolicies",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_admin_audit_logs(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListAdminAuditLogs`` — unary call to ``/udb.services.v1.DataBroker/ListAdminAuditLogs``.

        Forwards ``AdminAuditLogRequest`` (udb.entity.v1) and returns
        ``AdminAuditLogResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListAdminAuditLogs,
            "ListAdminAuditLogs",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_dlq_events(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListDlqEvents`` — unary call to ``/udb.services.v1.DataBroker/ListDlqEvents``.

        Forwards ``DlqListRequest`` (udb.entity.v1) and returns
        ``DlqListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListDlqEvents,
            "ListDlqEvents",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_message_schemas(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListMessageSchemas`` — unary call to ``/udb.services.v1.DataBroker/ListMessageSchemas``.

        Forwards ``MessageSchemaListRequest`` (udb.entity.v1) and returns
        ``MessageSchemaListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListMessageSchemas,
            "ListMessageSchemas",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_migration_runs(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListMigrationRuns`` — unary call to ``/udb.services.v1.DataBroker/ListMigrationRuns``.

        Forwards ``MigrationRunListRequest`` (udb.entity.v1) and returns
        ``MigrationRunListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListMigrationRuns,
            "ListMigrationRuns",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_policies(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListPolicies`` — unary call to ``/udb.services.v1.DataBroker/ListPolicies``.

        Forwards ``PolicyListRequest`` (udb.entity.v1) and returns
        ``PolicyListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListPolicies,
            "ListPolicies",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_projects(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListProjects`` — unary call to ``/udb.services.v1.DataBroker/ListProjects``.

        Forwards ``ProjectListRequest`` (udb.entity.v1) and returns
        ``ProjectListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListProjects,
            "ListProjects",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_resources(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListResources`` — unary call to ``/udb.services.v1.DataBroker/ListResources``.

        Forwards ``ResourceAdminRequest`` (udb.entity.v1) and returns
        ``ResourceListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListResources,
            "ListResources",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def list_sagas(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ListSagas`` — unary call to ``/udb.services.v1.DataBroker/ListSagas``.

        Forwards ``SagaListRequest`` (udb.entity.v1) and returns
        ``SagaListResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ListSagas,
            "ListSagas",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def lookup_message_schema(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``LookupMessageSchema`` — unary call to ``/udb.services.v1.DataBroker/LookupMessageSchema``.

        Forwards ``MessageSchemaLookupRequest`` (udb.entity.v1) and returns
        ``MessageSchemaLookupResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.LookupMessageSchema,
            "LookupMessageSchema",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def mark_saga_reviewed(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``MarkSagaReviewed`` — unary call to ``/udb.services.v1.DataBroker/MarkSagaReviewed``.

        Forwards ``SagaRequest`` (udb.entity.v1) and returns
        ``SagaResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.MarkSagaReviewed,
            "MarkSagaReviewed",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def pause_cdc(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PauseCdc`` — unary call to ``/udb.services.v1.DataBroker/PauseCdc``.

        Forwards ``CdcControlRequest`` (udb.entity.v1) and returns
        ``CdcStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.PauseCdc,
            "PauseCdc",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def plan_migration(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PlanMigration`` — unary call to ``/udb.services.v1.DataBroker/PlanMigration``.

        Forwards ``MigrationPlanRequest`` (udb.entity.v1) and returns
        ``MigrationPlanResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.PlanMigration,
            "PlanMigration",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def preview_cdc_redaction(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PreviewCdcRedaction`` — unary call to ``/udb.services.v1.DataBroker/PreviewCdcRedaction``.

        Forwards ``CdcRedactionPreviewRequest`` (udb.entity.v1) and returns
        ``CdcRedactionPreviewResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.PreviewCdcRedaction,
            "PreviewCdcRedaction",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def put_policy(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``PutPolicy`` — unary call to ``/udb.services.v1.DataBroker/PutPolicy``.

        Forwards ``PutPolicyRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.PutPolicy,
            "PutPolicy",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def quarantine_dlq_event(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``QuarantineDlqEvent`` — unary call to ``/udb.services.v1.DataBroker/QuarantineDlqEvent``.

        Forwards ``DlqActionRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.QuarantineDlqEvent,
            "QuarantineDlqEvent",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def reload_policies(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ReloadPolicies`` — unary call to ``/udb.services.v1.DataBroker/ReloadPolicies``.

        Forwards ``CapabilitiesRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ReloadPolicies,
            "ReloadPolicies",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def replay_dlq_event(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ReplayDlqEvent`` — unary call to ``/udb.services.v1.DataBroker/ReplayDlqEvent``.

        Forwards ``DlqActionRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ReplayDlqEvent,
            "ReplayDlqEvent",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def resume_cdc(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ResumeCdc`` — unary call to ``/udb.services.v1.DataBroker/ResumeCdc``.

        Forwards ``CdcControlRequest`` (udb.entity.v1) and returns
        ``CdcStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ResumeCdc,
            "ResumeCdc",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def retry_saga_compensation(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RetrySagaCompensation`` — unary call to ``/udb.services.v1.DataBroker/RetrySagaCompensation``.

        Forwards ``SagaRequest`` (udb.entity.v1) and returns
        ``SagaResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.RetrySagaCompensation,
            "RetrySagaCompensation",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def rollback_catalog(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``RollbackCatalog`` — unary call to ``/udb.services.v1.DataBroker/RollbackCatalog``.

        Forwards ``CatalogVersionRequest`` (udb.entity.v1) and returns
        ``CatalogVersionResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.RollbackCatalog,
            "RollbackCatalog",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def scan_projection_drift(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ScanProjectionDrift`` — unary call to ``/udb.services.v1.DataBroker/ScanProjectionDrift``.

        Forwards ``ProjectionDriftScanRequest`` (udb.entity.v1) and returns
        ``ProjectionDriftScanResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ScanProjectionDrift,
            "ScanProjectionDrift",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def select(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Select`` — unary call to ``/udb.services.v1.DataBroker/Select``.

        Forwards ``SelectRequest`` (udb.entity.v1) and returns
        ``RecordSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.Select,
            "Select",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def stage_catalog(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``StageCatalog`` — unary call to ``/udb.services.v1.DataBroker/StageCatalog``.

        Forwards ``StageCatalogRequest`` (udb.entity.v1) and returns
        ``CatalogVersionResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.StageCatalog,
            "StageCatalog",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def step_down_cdc_leader(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``StepDownCdcLeader`` — unary call to ``/udb.services.v1.DataBroker/StepDownCdcLeader``.

        Forwards ``CdcControlRequest`` (udb.entity.v1) and returns
        ``CdcStatusResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.StepDownCdcLeader,
            "StepDownCdcLeader",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def time_series_query(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``TimeSeriesQuery`` — unary call to ``/udb.services.v1.DataBroker/TimeSeriesQuery``.

        Forwards ``TimeSeriesQueryRequest`` (udb.entity.v1) and returns
        ``TimeSeriesQueryResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.TimeSeriesQuery,
            "TimeSeriesQuery",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def time_series_write(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``TimeSeriesWrite`` — unary call to ``/udb.services.v1.DataBroker/TimeSeriesWrite``.

        Forwards ``TimeSeriesWriteRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.TimeSeriesWrite,
            "TimeSeriesWrite",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def upsert(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``Upsert`` — unary call to ``/udb.services.v1.DataBroker/Upsert``.

        Forwards ``UpsertRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.Upsert,
            "Upsert",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def validate_catalog(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``ValidateCatalog`` — unary call to ``/udb.services.v1.DataBroker/ValidateCatalog``.

        Forwards ``StageCatalogRequest`` (udb.entity.v1) and returns
        ``CatalogValidationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.ValidateCatalog,
            "ValidateCatalog",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def vector_hybrid_search(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``VectorHybridSearch`` — unary call to ``/udb.services.v1.DataBroker/VectorHybridSearch``.

        Forwards ``VectorHybridSearchRequest`` (udb.entity.v1) and returns
        ``VectorSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.VectorHybridSearch,
            "VectorHybridSearch",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def vector_search(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``VectorSearch`` — unary call to ``/udb.services.v1.DataBroker/VectorSearch``.

        Forwards ``VectorSearchRequest`` (udb.entity.v1) and returns
        ``VectorSet`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.VectorSearch,
            "VectorSearch",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def vector_upsert(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``VectorUpsert`` — unary call to ``/udb.services.v1.DataBroker/VectorUpsert``.

        Forwards ``VectorUpsertRequest`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.VectorUpsert,
            "VectorUpsert",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def verify_admin_audit_log(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Any:
        """``VerifyAdminAuditLog`` — unary call to ``/udb.services.v1.DataBroker/VerifyAdminAuditLog``.

        Forwards ``AdminAuditVerifyRequest`` (udb.entity.v1) and returns
        ``AdminAuditVerifyResponse`` (udb.entity.v1).
        """

        return self._invoke_unary(
            self._stub.VerifyAdminAuditLog,
            "VerifyAdminAuditLog",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── server-streaming RPCs ─────────────────────────────────────────────────
    def get_object(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Iterator[Any]:
        """``GetObject`` — server-streaming call to ``/udb.services.v1.DataBroker/GetObject``.

        Forwards ``ObjectRequest`` (udb.entity.v1) and yields
        ``Chunk`` (udb.entity.v1) messages. Retry applies only
        before the first message is received.
        """

        return self._invoke_server_streaming(
            self._stub.GetObject,
            "GetObject",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def publish_c_d_c(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Iterator[Any]:
        """``PublishCDC`` — server-streaming call to ``/udb.services.v1.DataBroker/PublishCDC``.

        Forwards ``CDCSubscriptionRequest`` (udb.entity.v1) and yields
        ``CDCEnvelope`` (udb.events.v1) messages. Retry applies only
        before the first message is received.
        """

        return self._invoke_server_streaming(
            self._stub.PublishCDC,
            "PublishCDC",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )

    def select_v2(
        self,
        request: Any,
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
        retry: bool = True,
    ) -> Iterator[Any]:
        """``SelectV2`` — server-streaming call to ``/udb.services.v1.DataBroker/SelectV2``.

        Forwards ``SelectRequest`` (udb.entity.v1) and yields
        ``RecordBatchV2`` (udb.entity.v1) messages. Retry applies only
        before the first message is received.
        """

        return self._invoke_server_streaming(
            self._stub.SelectV2,
            "SelectV2",
            request,
            metadata=metadata,
            timeout=timeout,
            retryable=retry,
        )


    # ── client-streaming RPCs ─────────────────────────────────────────────────
    def put_object(
        self,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
    ) -> Any:
        """``PutObject`` — client-streaming call to ``/udb.services.v1.DataBroker/PutObject``.

        Consumes an iterator of ``Chunk`` (udb.entity.v1) and returns
        ``MutationResponse`` (udb.entity.v1). Not auto-retried (request stream
        is not replayable).
        """

        return self._invoke_client_streaming(
            self._stub.PutObject,
            "PutObject",
            request_iterator,
            metadata=metadata,
            timeout=timeout,
        )


    # ── bidirectional-streaming RPCs ──────────────────────────────────────────
    def batch_select(
        self,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
    ) -> Iterator[Any]:
        """``BatchSelect`` — bidirectional-streaming call to ``/udb.services.v1.DataBroker/BatchSelect``.

        Consumes an iterator of ``SelectRequest`` (udb.entity.v1) and yields
        ``RecordSet`` (udb.entity.v1). Not auto-retried.
        """

        stream = self._invoke_client_streaming(
            self._stub.BatchSelect,
            "BatchSelect",
            request_iterator,
            metadata=metadata,
            timeout=timeout,
        )
        try:
            for item in stream:
                yield item
        except grpc.RpcError as error:
            raise _map_error("BatchSelect", error) from error

    def batch_upsert(
        self,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
    ) -> Iterator[Any]:
        """``BatchUpsert`` — bidirectional-streaming call to ``/udb.services.v1.DataBroker/BatchUpsert``.

        Consumes an iterator of ``UpsertRequest`` (udb.entity.v1) and yields
        ``MutationResponse`` (udb.entity.v1). Not auto-retried.
        """

        stream = self._invoke_client_streaming(
            self._stub.BatchUpsert,
            "BatchUpsert",
            request_iterator,
            metadata=metadata,
            timeout=timeout,
        )
        try:
            for item in stream:
                yield item
        except grpc.RpcError as error:
            raise _map_error("BatchUpsert", error) from error

    def begin_tx(
        self,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
    ) -> Iterator[Any]:
        """``BeginTx`` — bidirectional-streaming call to ``/udb.services.v1.DataBroker/BeginTx``.

        Consumes an iterator of ``Mutation`` (udb.entity.v1) and yields
        ``TxStatus`` (udb.entity.v1). Not auto-retried.
        """

        stream = self._invoke_client_streaming(
            self._stub.BeginTx,
            "BeginTx",
            request_iterator,
            metadata=metadata,
            timeout=timeout,
        )
        try:
            for item in stream:
                yield item
        except grpc.RpcError as error:
            raise _map_error("BeginTx", error) from error

    def vector_batch_upsert(
        self,
        request_iterator: Iterable[Any],
        *,
        metadata: Metadata | None = None,
        timeout: float | None = None,
    ) -> Iterator[Any]:
        """``VectorBatchUpsert`` — bidirectional-streaming call to ``/udb.services.v1.DataBroker/VectorBatchUpsert``.

        Consumes an iterator of ``VectorUpsertRequest`` (udb.entity.v1) and yields
        ``MutationResponse`` (udb.entity.v1). Not auto-retried.
        """

        stream = self._invoke_client_streaming(
            self._stub.VectorBatchUpsert,
            "VectorBatchUpsert",
            request_iterator,
            metadata=metadata,
            timeout=timeout,
        )
        try:
            for item in stream:
                yield item
        except grpc.RpcError as error:
            raise _map_error("VectorBatchUpsert", error) from error




class GeneratedClient:
    """Aggregate facade exposing one robustness client per UDB service.

    All sub-clients share the same channel/metadata/retry configuration, so a
    single :class:`GeneratedClient` covers every RPC across every service.
    """

    def __init__(self, target: str = "", metadata: Metadata | None = None, **kwargs: Any) -> None:
        self._kwargs = kwargs
        self._target = target
        self._metadata = metadata
        self.AnalyticsService: AnalyticsServiceClient = AnalyticsServiceClient(
            target, metadata, **kwargs
        )
        self.ApiKeyService: ApiKeyServiceClient = ApiKeyServiceClient(
            target, metadata, **kwargs
        )
        self.AuthnService: AuthnServiceClient = AuthnServiceClient(
            target, metadata, **kwargs
        )
        self.AuthzService: AuthzServiceClient = AuthzServiceClient(
            target, metadata, **kwargs
        )
        self.NotificationService: NotificationServiceClient = NotificationServiceClient(
            target, metadata, **kwargs
        )
        self.TenantService: TenantServiceClient = TenantServiceClient(
            target, metadata, **kwargs
        )
        self.DataBroker: DataBrokerClient = DataBrokerClient(
            target, metadata, **kwargs
        )

    def bind_metadata(self, metadata: Metadata) -> None:
        self._metadata = metadata
        self.AnalyticsService.bind_metadata(metadata)
        self.ApiKeyService.bind_metadata(metadata)
        self.AuthnService.bind_metadata(metadata)
        self.AuthzService.bind_metadata(metadata)
        self.NotificationService.bind_metadata(metadata)
        self.TenantService.bind_metadata(metadata)
        self.DataBroker.bind_metadata(metadata)

    def close(self) -> None:
        self.AnalyticsService.close()
        self.ApiKeyService.close()
        self.AuthnService.close()
        self.AuthzService.close()
        self.NotificationService.close()
        self.TenantService.close()
        self.DataBroker.close()

    def __enter__(self) -> "GeneratedClient":
        return self

    def __exit__(self, *_: object) -> None:
        self.close()