sphinx_learn 0.1.0__py3-none-any.whl

sphinx_ai/gen/ra/jobs_connect.py

@@ -0,0 +1,148 @@
+# Generated from ra/jobs.proto. DO NOT EDIT.
+# Generated by protoc-gen-connectrpc-py v0.11.1 with parameter "".
+# ruff: noqa: PGH004
+# ruff: noqa
+# fmt: off
+
+from __future__ import annotations
+
+from typing import Protocol, TYPE_CHECKING
+
+from connectrpc.client import ConnectClient, ConnectClientSync
+from connectrpc.code import Code
+from connectrpc.errors import ConnectError
+from connectrpc.method import IdempotencyLevel, MethodInfo
+from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync
+
+from .jobs_pb import SubmitJobRequest, SubmitJobResponse
+
+if TYPE_CHECKING:
+    from collections.abc import AsyncGenerator, Iterable, Mapping
+
+    from connectrpc.codec import Codec
+    from connectrpc.compression import Compression
+    from connectrpc.interceptor import Interceptor, InterceptorSync
+    from connectrpc.request import Headers, RequestContext
+
+
+class RaJobService(Protocol):
+    async def submit_job(self, request: SubmitJobRequest, ctx: RequestContext[SubmitJobRequest, SubmitJobResponse]) -> SubmitJobResponse:
+        raise ConnectError(Code.UNIMPLEMENTED, 'Not implemented')
+
+
+class RaJobServiceASGIApplication(ConnectASGIApplication[RaJobService]):
+    def __init__(
+        self,
+        service: RaJobService | AsyncGenerator[RaJobService],
+        *,
+        interceptors: Iterable[Interceptor] = (),
+        read_max_bytes: int | None = None,
+        compressions: Iterable[Compression] | None = None,
+        codecs: Iterable[Codec] | None = None,
+    ) -> None:
+        super().__init__(
+            service=service,
+            endpoints=lambda svc: {
+                "/ra.RaJobService/SubmitJob": Endpoint.unary(
+                    method=MethodInfo(
+                        name="SubmitJob",
+                        service_name="ra.RaJobService",
+                        input=SubmitJobRequest,
+                        output=SubmitJobResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=svc.submit_job,
+                ),
+            },
+            interceptors=interceptors,
+            read_max_bytes=read_max_bytes,
+            compressions=compressions,
+            codecs=codecs,
+        )
+
+    @property
+    def path(self) -> str:
+        """Returns the URL path to mount the application to when serving multiple applications."""
+        return "/ra.RaJobService"
+
+
+class RaJobServiceClient(ConnectClient):
+    async def submit_job(
+        self,
+        request: SubmitJobRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None, 
+        timeout_ms: int | None = None,
+    ) -> SubmitJobResponse:
+        return await self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="SubmitJob",
+                service_name="ra.RaJobService",
+                input=SubmitJobRequest,
+                output=SubmitJobResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+class RaJobServiceSync(Protocol):
+    def submit_job(self, request: SubmitJobRequest, ctx: RequestContext[SubmitJobRequest, SubmitJobResponse]) -> SubmitJobResponse:
+        raise ConnectError(Code.UNIMPLEMENTED, 'Not implemented')
+
+
+class RaJobServiceWSGIApplication(ConnectWSGIApplication):
+    def __init__(
+        self,
+        service: RaJobServiceSync,
+        interceptors: Iterable[InterceptorSync] = (),
+        read_max_bytes: int | None = None,
+        compressions: Iterable[Compression] | None = None,
+        codecs: Iterable[Codec] | None = None,
+    ) -> None:
+        super().__init__(
+            endpoints={
+                "/ra.RaJobService/SubmitJob": EndpointSync.unary(
+                    method=MethodInfo(
+                        name="SubmitJob",
+                        service_name="ra.RaJobService",
+                        input=SubmitJobRequest,
+                        output=SubmitJobResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.submit_job,
+                ),
+            },
+            interceptors=interceptors,
+            read_max_bytes=read_max_bytes,
+            compressions=compressions,
+            codecs=codecs,
+        )
+
+    @property
+    def path(self) -> str:
+        """Returns the URL path to mount the application to when serving multiple applications."""
+        return "/ra.RaJobService"
+
+
+class RaJobServiceClientSync(ConnectClientSync):
+    def submit_job(
+        self,
+        request: SubmitJobRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None, 
+        timeout_ms: int | None = None,
+    ) -> SubmitJobResponse:
+        return self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="SubmitJob",
+                service_name="ra.RaJobService",
+                input=SubmitJobRequest,
+                output=SubmitJobResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )