diff --git a/buf.gen.python.yaml b/buf.gen.python.yaml index 6a843ca9dd..bfaf6c8c50 100644 --- a/buf.gen.python.yaml +++ b/buf.gen.python.yaml @@ -15,3 +15,5 @@ plugins: - remote: buf.build/grpc/python:v1.58.1 out: gen/python include_imports: true + - remote: buf.build/connectrpc/python:v0.9.0 + out: gen/python diff --git a/gen/python/flyteidl2/actions/actions_service_connect.py b/gen/python/flyteidl2/actions/actions_service_connect.py new file mode 100644 index 0000000000..80cab04eab --- /dev/null +++ b/gen/python/flyteidl2/actions/actions_service_connect.py @@ -0,0 +1,383 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/actions/actions_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.actions.actions_service_pb2 as flyteidl2_dot_actions_dot_actions__service__pb2 + + +class ActionsService(Protocol): + async def enqueue(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_latest_state(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_for_updates(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def abort(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ActionsServiceASGIApplication(ConnectASGIApplication[ActionsService]): + def __init__(self, service: ActionsService | AsyncGenerator[ActionsService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.actions.ActionsService/Enqueue": Endpoint.unary( + method=MethodInfo( + name="Enqueue", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.enqueue, + ), + "/flyteidl2.actions.ActionsService/GetLatestState": Endpoint.unary( + method=MethodInfo( + name="GetLatestState", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_latest_state, + ), + "/flyteidl2.actions.ActionsService/WatchForUpdates": Endpoint.server_stream( + method=MethodInfo( + name="WatchForUpdates", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_for_updates, + ), + "/flyteidl2.actions.ActionsService/Update": Endpoint.unary( + method=MethodInfo( + name="Update", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update, + ), + "/flyteidl2.actions.ActionsService/Abort": Endpoint.unary( + method=MethodInfo( + name="Abort", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.abort, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.actions.ActionsService" + + +class ActionsServiceClient(ConnectClient): + async def enqueue( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Enqueue", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_latest_state( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetLatestState", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_for_updates( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchForUpdates", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Update", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def abort( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Abort", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class ActionsServiceSync(Protocol): + def enqueue(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_latest_state(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_for_updates(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def abort(self, request: flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, ctx: RequestContext) -> flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ActionsServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: ActionsServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.actions.ActionsService/Enqueue": EndpointSync.unary( + method=MethodInfo( + name="Enqueue", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.enqueue, + ), + "/flyteidl2.actions.ActionsService/GetLatestState": EndpointSync.unary( + method=MethodInfo( + name="GetLatestState", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_latest_state, + ), + "/flyteidl2.actions.ActionsService/WatchForUpdates": EndpointSync.server_stream( + method=MethodInfo( + name="WatchForUpdates", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_for_updates, + ), + "/flyteidl2.actions.ActionsService/Update": EndpointSync.unary( + method=MethodInfo( + name="Update", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update, + ), + "/flyteidl2.actions.ActionsService/Abort": EndpointSync.unary( + method=MethodInfo( + name="Abort", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.abort, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.actions.ActionsService" + + +class ActionsServiceClientSync(ConnectClientSync): + def enqueue( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Enqueue", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.EnqueueResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_latest_state( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetLatestState", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.GetLatestStateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_for_updates( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchForUpdates", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.WatchForUpdatesResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Update", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def abort( + self, + request: flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Abort", + service_name="flyteidl2.actions.ActionsService", + input=flyteidl2_dot_actions_dot_actions__service__pb2.AbortRequest, + output=flyteidl2_dot_actions_dot_actions__service__pb2.AbortResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/app/app_logs_service_connect.py b/gen/python/flyteidl2/app/app_logs_service_connect.py new file mode 100644 index 0000000000..7467a52b54 --- /dev/null +++ b/gen/python/flyteidl2/app/app_logs_service_connect.py @@ -0,0 +1,123 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/app/app_logs_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.app.app_logs_payload_pb2 as flyteidl2_dot_app_dot_app__logs__payload__pb2 + + +class AppLogsService(Protocol): + def tail_logs(self, request: flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AppLogsServiceASGIApplication(ConnectASGIApplication[AppLogsService]): + def __init__(self, service: AppLogsService | AsyncGenerator[AppLogsService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.app.AppLogsService/TailLogs": Endpoint.server_stream( + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.app.AppLogsService", + input=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + output=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.tail_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.app.AppLogsService" + + +class AppLogsServiceClient(ConnectClient): + def tail_logs( + self, + request: flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.app.AppLogsService", + input=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + output=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class AppLogsServiceSync(Protocol): + def tail_logs(self, request: flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AppLogsServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: AppLogsServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.app.AppLogsService/TailLogs": EndpointSync.server_stream( + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.app.AppLogsService", + input=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + output=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.tail_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.app.AppLogsService" + + +class AppLogsServiceClientSync(ConnectClientSync): + def tail_logs( + self, + request: flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.app.AppLogsService", + input=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsRequest, + output=flyteidl2_dot_app_dot_app__logs__payload__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/app/app_service_connect.py b/gen/python/flyteidl2/app/app_service_connect.py new file mode 100644 index 0000000000..442d589552 --- /dev/null +++ b/gen/python/flyteidl2/app/app_service_connect.py @@ -0,0 +1,586 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/app/app_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.app.app_payload_pb2 as flyteidl2_dot_app_dot_app__payload__pb2 + + +class AppService(Protocol): + async def create(self, request: flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get(self, request: flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.GetResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update(self, request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_status(self, request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete(self, request: flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list(self, request: flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.ListResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch(self, request: flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def lease(self, request: flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AppServiceASGIApplication(ConnectASGIApplication[AppService]): + def __init__(self, service: AppService | AsyncGenerator[AppService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.app.AppService/Create": Endpoint.unary( + method=MethodInfo( + name="Create", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create, + ), + "/flyteidl2.app.AppService/Get": Endpoint.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.GetResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get, + ), + "/flyteidl2.app.AppService/Update": Endpoint.unary( + method=MethodInfo( + name="Update", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update, + ), + "/flyteidl2.app.AppService/UpdateStatus": Endpoint.unary( + method=MethodInfo( + name="UpdateStatus", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_status, + ), + "/flyteidl2.app.AppService/Delete": Endpoint.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete, + ), + "/flyteidl2.app.AppService/List": Endpoint.unary( + method=MethodInfo( + name="List", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.ListResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list, + ), + "/flyteidl2.app.AppService/Watch": Endpoint.server_stream( + method=MethodInfo( + name="Watch", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.watch, + ), + "/flyteidl2.app.AppService/Lease": Endpoint.server_stream( + method=MethodInfo( + name="Lease", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.lease, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.app.AppService" + + +class AppServiceClient(ConnectClient): + async def create( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Create", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.GetResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.GetResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def update( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Update", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_status( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateStatus", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def list( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.ListResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="List", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.ListResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Watch", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def lease( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Lease", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class AppServiceSync(Protocol): + def create(self, request: flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get(self, request: flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.GetResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update(self, request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_status(self, request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete(self, request: flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list(self, request: flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, ctx: RequestContext) -> flyteidl2_dot_app_dot_app__payload__pb2.ListResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch(self, request: flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def lease(self, request: flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AppServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: AppServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.app.AppService/Create": EndpointSync.unary( + method=MethodInfo( + name="Create", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create, + ), + "/flyteidl2.app.AppService/Get": EndpointSync.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.GetResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get, + ), + "/flyteidl2.app.AppService/Update": EndpointSync.unary( + method=MethodInfo( + name="Update", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update, + ), + "/flyteidl2.app.AppService/UpdateStatus": EndpointSync.unary( + method=MethodInfo( + name="UpdateStatus", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_status, + ), + "/flyteidl2.app.AppService/Delete": EndpointSync.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete, + ), + "/flyteidl2.app.AppService/List": EndpointSync.unary( + method=MethodInfo( + name="List", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.ListResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list, + ), + "/flyteidl2.app.AppService/Watch": EndpointSync.server_stream( + method=MethodInfo( + name="Watch", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.watch, + ), + "/flyteidl2.app.AppService/Lease": EndpointSync.server_stream( + method=MethodInfo( + name="Lease", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.lease, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.app.AppService" + + +class AppServiceClientSync(ConnectClientSync): + def create( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Create", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.CreateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.CreateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.GetResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.GetRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.GetResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def update( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Update", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_status( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateStatus", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.UpdateStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.DeleteRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.DeleteResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def list( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_app_dot_app__payload__pb2.ListResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="List", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.ListRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.ListResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Watch", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.WatchRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def lease( + self, + request: flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Lease", + service_name="flyteidl2.app.AppService", + input=flyteidl2_dot_app_dot_app__payload__pb2.LeaseRequest, + output=flyteidl2_dot_app_dot_app__payload__pb2.LeaseResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/auth/auth_service_connect.py b/gen/python/flyteidl2/auth/auth_service_connect.py new file mode 100644 index 0000000000..ec8a096882 --- /dev/null +++ b/gen/python/flyteidl2/auth/auth_service_connect.py @@ -0,0 +1,196 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/auth/auth_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.auth.auth_service_pb2 as flyteidl2_dot_auth_dot_auth__service__pb2 + + +class AuthMetadataService(Protocol): + async def get_o_auth2_metadata(self, request: flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_public_client_config(self, request: flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AuthMetadataServiceASGIApplication(ConnectASGIApplication[AuthMetadataService]): + def __init__(self, service: AuthMetadataService | AsyncGenerator[AuthMetadataService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.auth.AuthMetadataService/GetOAuth2Metadata": Endpoint.unary( + method=MethodInfo( + name="GetOAuth2Metadata", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_o_auth2_metadata, + ), + "/flyteidl2.auth.AuthMetadataService/GetPublicClientConfig": Endpoint.unary( + method=MethodInfo( + name="GetPublicClientConfig", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_public_client_config, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.auth.AuthMetadataService" + + +class AuthMetadataServiceClient(ConnectClient): + async def get_o_auth2_metadata( + self, + request: flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetOAuth2Metadata", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def get_public_client_config( + self, + request: flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetPublicClientConfig", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + +class AuthMetadataServiceSync(Protocol): + def get_o_auth2_metadata(self, request: flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_public_client_config(self, request: flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AuthMetadataServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: AuthMetadataServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.auth.AuthMetadataService/GetOAuth2Metadata": EndpointSync.unary( + method=MethodInfo( + name="GetOAuth2Metadata", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_o_auth2_metadata, + ), + "/flyteidl2.auth.AuthMetadataService/GetPublicClientConfig": EndpointSync.unary( + method=MethodInfo( + name="GetPublicClientConfig", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_public_client_config, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.auth.AuthMetadataService" + + +class AuthMetadataServiceClientSync(ConnectClientSync): + def get_o_auth2_metadata( + self, + request: flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetOAuth2Metadata", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetOAuth2MetadataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def get_public_client_config( + self, + request: flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetPublicClientConfig", + service_name="flyteidl2.auth.AuthMetadataService", + input=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigRequest, + output=flyteidl2_dot_auth_dot_auth__service__pb2.GetPublicClientConfigResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) diff --git a/gen/python/flyteidl2/auth/identity_connect.py b/gen/python/flyteidl2/auth/identity_connect.py new file mode 100644 index 0000000000..acd114e833 --- /dev/null +++ b/gen/python/flyteidl2/auth/identity_connect.py @@ -0,0 +1,123 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/auth/identity.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.auth.identity_pb2 as flyteidl2_dot_auth_dot_identity__pb2 + + +class IdentityService(Protocol): + async def user_info(self, request: flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class IdentityServiceASGIApplication(ConnectASGIApplication[IdentityService]): + def __init__(self, service: IdentityService | AsyncGenerator[IdentityService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.auth.IdentityService/UserInfo": Endpoint.unary( + method=MethodInfo( + name="UserInfo", + service_name="flyteidl2.auth.IdentityService", + input=flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + output=flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.user_info, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.auth.IdentityService" + + +class IdentityServiceClient(ConnectClient): + async def user_info( + self, + request: flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UserInfo", + service_name="flyteidl2.auth.IdentityService", + input=flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + output=flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class IdentityServiceSync(Protocol): + def user_info(self, request: flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, ctx: RequestContext) -> flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class IdentityServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: IdentityServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.auth.IdentityService/UserInfo": EndpointSync.unary( + method=MethodInfo( + name="UserInfo", + service_name="flyteidl2.auth.IdentityService", + input=flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + output=flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.user_info, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.auth.IdentityService" + + +class IdentityServiceClientSync(ConnectClientSync): + def user_info( + self, + request: flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UserInfo", + service_name="flyteidl2.auth.IdentityService", + input=flyteidl2_dot_auth_dot_identity__pb2.UserInfoRequest, + output=flyteidl2_dot_auth_dot_identity__pb2.UserInfoResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/cacheservice/cacheservice_connect.py b/gen/python/flyteidl2/cacheservice/cacheservice_connect.py new file mode 100644 index 0000000000..7038efefcd --- /dev/null +++ b/gen/python/flyteidl2/cacheservice/cacheservice_connect.py @@ -0,0 +1,383 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/cacheservice/cacheservice.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.cacheservice.cacheservice_pb2 as flyteidl2_dot_cacheservice_dot_cacheservice__pb2 + + +class CacheService(Protocol): + async def get(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def put(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_or_extend_reservation(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def release_reservation(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class CacheServiceASGIApplication(ConnectASGIApplication[CacheService]): + def __init__(self, service: CacheService | AsyncGenerator[CacheService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.cacheservice.CacheService/Get": Endpoint.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get, + ), + "/flyteidl2.cacheservice.CacheService/Put": Endpoint.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.put, + ), + "/flyteidl2.cacheservice.CacheService/Delete": Endpoint.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete, + ), + "/flyteidl2.cacheservice.CacheService/GetOrExtendReservation": Endpoint.unary( + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_or_extend_reservation, + ), + "/flyteidl2.cacheservice.CacheService/ReleaseReservation": Endpoint.unary( + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.release_reservation, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.cacheservice.CacheService" + + +class CacheServiceClient(ConnectClient): + async def get( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def put( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_or_extend_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def release_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class CacheServiceSync(Protocol): + def get(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def put(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_or_extend_reservation(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def release_reservation(self, request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class CacheServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: CacheServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.cacheservice.CacheService/Get": EndpointSync.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get, + ), + "/flyteidl2.cacheservice.CacheService/Put": EndpointSync.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.put, + ), + "/flyteidl2.cacheservice.CacheService/Delete": EndpointSync.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete, + ), + "/flyteidl2.cacheservice.CacheService/GetOrExtendReservation": EndpointSync.unary( + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_or_extend_reservation, + ), + "/flyteidl2.cacheservice.CacheService/ReleaseReservation": EndpointSync.unary( + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.release_reservation, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.cacheservice.CacheService" + + +class CacheServiceClientSync(ConnectClientSync): + def get( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def put( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_or_extend_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def release_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.CacheService", + input=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/cacheservice/v2/cacheservice_connect.py b/gen/python/flyteidl2/cacheservice/v2/cacheservice_connect.py new file mode 100644 index 0000000000..9a1d004db8 --- /dev/null +++ b/gen/python/flyteidl2/cacheservice/v2/cacheservice_connect.py @@ -0,0 +1,384 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/cacheservice/v2/cacheservice.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.cacheservice.cacheservice_pb2 as flyteidl2_dot_cacheservice_dot_cacheservice__pb2 +import flyteidl2.cacheservice.v2.cacheservice_pb2 as flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2 + + +class CacheService(Protocol): + async def get(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def put(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_or_extend_reservation(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def release_reservation(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class CacheServiceASGIApplication(ConnectASGIApplication[CacheService]): + def __init__(self, service: CacheService | AsyncGenerator[CacheService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.cacheservice.v2.CacheService/Get": Endpoint.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get, + ), + "/flyteidl2.cacheservice.v2.CacheService/Put": Endpoint.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.put, + ), + "/flyteidl2.cacheservice.v2.CacheService/Delete": Endpoint.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete, + ), + "/flyteidl2.cacheservice.v2.CacheService/GetOrExtendReservation": Endpoint.unary( + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_or_extend_reservation, + ), + "/flyteidl2.cacheservice.v2.CacheService/ReleaseReservation": Endpoint.unary( + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.release_reservation, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.cacheservice.v2.CacheService" + + +class CacheServiceClient(ConnectClient): + async def get( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def put( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_or_extend_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def release_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class CacheServiceSync(Protocol): + def get(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def put(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_or_extend_reservation(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def release_reservation(self, request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, ctx: RequestContext) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class CacheServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: CacheServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.cacheservice.v2.CacheService/Get": EndpointSync.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get, + ), + "/flyteidl2.cacheservice.v2.CacheService/Put": EndpointSync.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.put, + ), + "/flyteidl2.cacheservice.v2.CacheService/Delete": EndpointSync.unary( + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete, + ), + "/flyteidl2.cacheservice.v2.CacheService/GetOrExtendReservation": EndpointSync.unary( + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_or_extend_reservation, + ), + "/flyteidl2.cacheservice.v2.CacheService/ReleaseReservation": EndpointSync.unary( + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.release_reservation, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.cacheservice.v2.CacheService" + + +class CacheServiceClientSync(ConnectClientSync): + def get( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def put( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.PutCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.PutCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Delete", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.DeleteCacheRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.DeleteCacheResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_or_extend_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetOrExtendReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.GetOrExtendReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.GetOrExtendReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def release_reservation( + self, + request: flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ReleaseReservation", + service_name="flyteidl2.cacheservice.v2.CacheService", + input=flyteidl2_dot_cacheservice_dot_v2_dot_cacheservice__pb2.ReleaseReservationRequest, + output=flyteidl2_dot_cacheservice_dot_cacheservice__pb2.ReleaseReservationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/connector/service_connect.py b/gen/python/flyteidl2/connector/service_connect.py new file mode 100644 index 0000000000..2fd2f78f15 --- /dev/null +++ b/gen/python/flyteidl2/connector/service_connect.py @@ -0,0 +1,555 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/connector/service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.connector.connector_pb2 as flyteidl2_dot_connector_dot_connector__pb2 + + +class AsyncConnectorService(Protocol): + async def create_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_task_metrics(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def get_task_logs(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AsyncConnectorServiceASGIApplication(ConnectASGIApplication[AsyncConnectorService]): + def __init__(self, service: AsyncConnectorService | AsyncGenerator[AsyncConnectorService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.connector.AsyncConnectorService/CreateTask": Endpoint.unary( + method=MethodInfo( + name="CreateTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_task, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTask": Endpoint.unary( + method=MethodInfo( + name="GetTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_task, + ), + "/flyteidl2.connector.AsyncConnectorService/DeleteTask": Endpoint.unary( + method=MethodInfo( + name="DeleteTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete_task, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTaskMetrics": Endpoint.unary( + method=MethodInfo( + name="GetTaskMetrics", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_task_metrics, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTaskLogs": Endpoint.server_stream( + method=MethodInfo( + name="GetTaskLogs", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_task_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.connector.AsyncConnectorService" + + +class AsyncConnectorServiceClient(ConnectClient): + async def create_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_task_metrics( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTaskMetrics", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_task_logs( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="GetTaskLogs", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + + +class ConnectorMetadataService(Protocol): + async def get_connector(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_connectors(self, request: flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ConnectorMetadataServiceASGIApplication(ConnectASGIApplication[ConnectorMetadataService]): + def __init__(self, service: ConnectorMetadataService | AsyncGenerator[ConnectorMetadataService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.connector.ConnectorMetadataService/GetConnector": Endpoint.unary( + method=MethodInfo( + name="GetConnector", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_connector, + ), + "/flyteidl2.connector.ConnectorMetadataService/ListConnectors": Endpoint.unary( + method=MethodInfo( + name="ListConnectors", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.list_connectors, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.connector.ConnectorMetadataService" + + +class ConnectorMetadataServiceClient(ConnectClient): + async def get_connector( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetConnector", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def list_connectors( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListConnectors", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class AsyncConnectorServiceSync(Protocol): + def create_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete_task(self, request: flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_task_metrics(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_task_logs(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class AsyncConnectorServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: AsyncConnectorServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.connector.AsyncConnectorService/CreateTask": EndpointSync.unary( + method=MethodInfo( + name="CreateTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_task, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTask": EndpointSync.unary( + method=MethodInfo( + name="GetTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_task, + ), + "/flyteidl2.connector.AsyncConnectorService/DeleteTask": EndpointSync.unary( + method=MethodInfo( + name="DeleteTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete_task, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTaskMetrics": EndpointSync.unary( + method=MethodInfo( + name="GetTaskMetrics", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_task_metrics, + ), + "/flyteidl2.connector.AsyncConnectorService/GetTaskLogs": EndpointSync.server_stream( + method=MethodInfo( + name="GetTaskLogs", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_task_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.connector.AsyncConnectorService" + + +class AsyncConnectorServiceClientSync(ConnectClientSync): + def create_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.CreateTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete_task( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteTask", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.DeleteTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_task_metrics( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTaskMetrics", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskMetricsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_task_logs( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="GetTaskLogs", + service_name="flyteidl2.connector.AsyncConnectorService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetTaskLogsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + +class ConnectorMetadataServiceSync(Protocol): + def get_connector(self, request: flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_connectors(self, request: flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, ctx: RequestContext) -> flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ConnectorMetadataServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: ConnectorMetadataServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.connector.ConnectorMetadataService/GetConnector": EndpointSync.unary( + method=MethodInfo( + name="GetConnector", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_connector, + ), + "/flyteidl2.connector.ConnectorMetadataService/ListConnectors": EndpointSync.unary( + method=MethodInfo( + name="ListConnectors", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.list_connectors, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.connector.ConnectorMetadataService" + + +class ConnectorMetadataServiceClientSync(ConnectClientSync): + def get_connector( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetConnector", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.GetConnectorResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def list_connectors( + self, + request: flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListConnectors", + service_name="flyteidl2.connector.ConnectorMetadataService", + input=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsRequest, + output=flyteidl2_dot_connector_dot_connector__pb2.ListConnectorsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/dataproxy/dataproxy_service_connect.py b/gen/python/flyteidl2/dataproxy/dataproxy_service_connect.py new file mode 100644 index 0000000000..29962a49c5 --- /dev/null +++ b/gen/python/flyteidl2/dataproxy/dataproxy_service_connect.py @@ -0,0 +1,123 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/dataproxy/dataproxy_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.dataproxy.dataproxy_service_pb2 as flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2 + + +class DataProxyService(Protocol): + async def create_upload_location(self, request: flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, ctx: RequestContext) -> flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class DataProxyServiceASGIApplication(ConnectASGIApplication[DataProxyService]): + def __init__(self, service: DataProxyService | AsyncGenerator[DataProxyService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.dataproxy.DataProxyService/CreateUploadLocation": Endpoint.unary( + method=MethodInfo( + name="CreateUploadLocation", + service_name="flyteidl2.dataproxy.DataProxyService", + input=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + output=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_upload_location, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.dataproxy.DataProxyService" + + +class DataProxyServiceClient(ConnectClient): + async def create_upload_location( + self, + request: flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateUploadLocation", + service_name="flyteidl2.dataproxy.DataProxyService", + input=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + output=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class DataProxyServiceSync(Protocol): + def create_upload_location(self, request: flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, ctx: RequestContext) -> flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class DataProxyServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: DataProxyServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.dataproxy.DataProxyService/CreateUploadLocation": EndpointSync.unary( + method=MethodInfo( + name="CreateUploadLocation", + service_name="flyteidl2.dataproxy.DataProxyService", + input=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + output=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_upload_location, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.dataproxy.DataProxyService" + + +class DataProxyServiceClientSync(ConnectClientSync): + def create_upload_location( + self, + request: flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateUploadLocation", + service_name="flyteidl2.dataproxy.DataProxyService", + input=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationRequest, + output=flyteidl2_dot_dataproxy_dot_dataproxy__service__pb2.CreateUploadLocationResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/imagebuilder/service_connect.py b/gen/python/flyteidl2/imagebuilder/service_connect.py new file mode 100644 index 0000000000..ea3ec4e913 --- /dev/null +++ b/gen/python/flyteidl2/imagebuilder/service_connect.py @@ -0,0 +1,127 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/imagebuilder/service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.imagebuilder.payload_pb2 as flyteidl2_dot_imagebuilder_dot_payload__pb2 + + +class ImageService(Protocol): + async def get_image(self, request: flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, ctx: RequestContext) -> flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ImageServiceASGIApplication(ConnectASGIApplication[ImageService]): + def __init__(self, service: ImageService | AsyncGenerator[ImageService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.imagebuilder.ImageService/GetImage": Endpoint.unary( + method=MethodInfo( + name="GetImage", + service_name="flyteidl2.imagebuilder.ImageService", + input=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + output=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_image, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.imagebuilder.ImageService" + + +class ImageServiceClient(ConnectClient): + async def get_image( + self, + request: flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetImage", + service_name="flyteidl2.imagebuilder.ImageService", + input=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + output=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + +class ImageServiceSync(Protocol): + def get_image(self, request: flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, ctx: RequestContext) -> flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ImageServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: ImageServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.imagebuilder.ImageService/GetImage": EndpointSync.unary( + method=MethodInfo( + name="GetImage", + service_name="flyteidl2.imagebuilder.ImageService", + input=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + output=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_image, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.imagebuilder.ImageService" + + +class ImageServiceClientSync(ConnectClientSync): + def get_image( + self, + request: flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetImage", + service_name="flyteidl2.imagebuilder.ImageService", + input=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageRequest, + output=flyteidl2_dot_imagebuilder_dot_payload__pb2.GetImageResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) diff --git a/gen/python/flyteidl2/org/domain_service_connect.py b/gen/python/flyteidl2/org/domain_service_connect.py new file mode 100644 index 0000000000..7103e82066 --- /dev/null +++ b/gen/python/flyteidl2/org/domain_service_connect.py @@ -0,0 +1,391 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/org/domain_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.org.domain_service_pb2 as flyteidl2_dot_org_dot_domain__service__pb2 + + +class DomainService(Protocol): + async def list_domains(self, request: flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def create_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class DomainServiceASGIApplication(ConnectASGIApplication[DomainService]): + def __init__(self, service: DomainService | AsyncGenerator[DomainService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.org.DomainService/ListDomains": Endpoint.unary( + method=MethodInfo( + name="ListDomains", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_domains, + ), + "/flyteidl2.org.DomainService/CreateDomain": Endpoint.unary( + method=MethodInfo( + name="CreateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_domain, + ), + "/flyteidl2.org.DomainService/GetDomain": Endpoint.unary( + method=MethodInfo( + name="GetDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_domain, + ), + "/flyteidl2.org.DomainService/UpdateDomain": Endpoint.unary( + method=MethodInfo( + name="UpdateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_domain, + ), + "/flyteidl2.org.DomainService/DeleteDomain": Endpoint.unary( + method=MethodInfo( + name="DeleteDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete_domain, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.org.DomainService" + + +class DomainServiceClient(ConnectClient): + async def list_domains( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListDomains", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def create_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def update_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class DomainServiceSync(Protocol): + def list_domains(self, request: flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def create_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete_domain(self, request: flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class DomainServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: DomainServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.org.DomainService/ListDomains": EndpointSync.unary( + method=MethodInfo( + name="ListDomains", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_domains, + ), + "/flyteidl2.org.DomainService/CreateDomain": EndpointSync.unary( + method=MethodInfo( + name="CreateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_domain, + ), + "/flyteidl2.org.DomainService/GetDomain": EndpointSync.unary( + method=MethodInfo( + name="GetDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_domain, + ), + "/flyteidl2.org.DomainService/UpdateDomain": EndpointSync.unary( + method=MethodInfo( + name="UpdateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_domain, + ), + "/flyteidl2.org.DomainService/DeleteDomain": EndpointSync.unary( + method=MethodInfo( + name="DeleteDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete_domain, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.org.DomainService" + + +class DomainServiceClientSync(ConnectClientSync): + def list_domains( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListDomains", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.ListDomainsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def create_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.CreateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.GetDomainResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def update_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.UpdateDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete_domain( + self, + request: flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteDomain", + service_name="flyteidl2.org.DomainService", + input=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainRequest, + output=flyteidl2_dot_org_dot_domain__service__pb2.DeleteDomainResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/org/settings_service_connect.py b/gen/python/flyteidl2/org/settings_service_connect.py new file mode 100644 index 0000000000..00558bb4d7 --- /dev/null +++ b/gen/python/flyteidl2/org/settings_service_connect.py @@ -0,0 +1,460 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/org/settings_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.org.settings_service_pb2 as flyteidl2_dot_org_dot_settings__service__pb2 + + +class SettingsService(Protocol): + async def get_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_settings_for_edit(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_settings_for_edit_raw(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def create_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_settings_raw(self, request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class SettingsServiceASGIApplication(ConnectASGIApplication[SettingsService]): + def __init__(self, service: SettingsService | AsyncGenerator[SettingsService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.org.SettingsService/GetSettings": Endpoint.unary( + method=MethodInfo( + name="GetSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_settings, + ), + "/flyteidl2.org.SettingsService/GetSettingsForEdit": Endpoint.unary( + method=MethodInfo( + name="GetSettingsForEdit", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_settings_for_edit, + ), + "/flyteidl2.org.SettingsService/GetSettingsForEditRaw": Endpoint.unary( + method=MethodInfo( + name="GetSettingsForEditRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_settings_for_edit_raw, + ), + "/flyteidl2.org.SettingsService/CreateSettings": Endpoint.unary( + method=MethodInfo( + name="CreateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_settings, + ), + "/flyteidl2.org.SettingsService/UpdateSettings": Endpoint.unary( + method=MethodInfo( + name="UpdateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_settings, + ), + "/flyteidl2.org.SettingsService/UpdateSettingsRaw": Endpoint.unary( + method=MethodInfo( + name="UpdateSettingsRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_settings_raw, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.org.SettingsService" + + +class SettingsServiceClient(ConnectClient): + async def get_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def get_settings_for_edit( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettingsForEdit", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def get_settings_for_edit_raw( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettingsForEditRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def create_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_settings_raw( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSettingsRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class SettingsServiceSync(Protocol): + def get_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_settings_for_edit(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_settings_for_edit_raw(self, request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def create_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_settings(self, request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_settings_raw(self, request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, ctx: RequestContext) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class SettingsServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: SettingsServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.org.SettingsService/GetSettings": EndpointSync.unary( + method=MethodInfo( + name="GetSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_settings, + ), + "/flyteidl2.org.SettingsService/GetSettingsForEdit": EndpointSync.unary( + method=MethodInfo( + name="GetSettingsForEdit", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_settings_for_edit, + ), + "/flyteidl2.org.SettingsService/GetSettingsForEditRaw": EndpointSync.unary( + method=MethodInfo( + name="GetSettingsForEditRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_settings_for_edit_raw, + ), + "/flyteidl2.org.SettingsService/CreateSettings": EndpointSync.unary( + method=MethodInfo( + name="CreateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_settings, + ), + "/flyteidl2.org.SettingsService/UpdateSettings": EndpointSync.unary( + method=MethodInfo( + name="UpdateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_settings, + ), + "/flyteidl2.org.SettingsService/UpdateSettingsRaw": EndpointSync.unary( + method=MethodInfo( + name="UpdateSettingsRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_settings_raw, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.org.SettingsService" + + +class SettingsServiceClientSync(ConnectClientSync): + def get_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def get_settings_for_edit( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettingsForEdit", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def get_settings_for_edit_raw( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetSettingsForEditRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.GetSettingsForEditRawResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def create_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.CreateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_settings( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSettings", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_settings_raw( + self, + request: flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSettingsRaw", + service_name="flyteidl2.org.SettingsService", + input=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawRequest, + output=flyteidl2_dot_org_dot_settings__service__pb2.UpdateSettingsRawResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/project/project_service_connect.py b/gen/python/flyteidl2/project/project_service_connect.py new file mode 100644 index 0000000000..2c60204480 --- /dev/null +++ b/gen/python/flyteidl2/project/project_service_connect.py @@ -0,0 +1,322 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/project/project_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.project.project_service_pb2 as flyteidl2_dot_project_dot_project__service__pb2 + + +class ProjectService(Protocol): + async def create_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_projects(self, request: flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ProjectServiceASGIApplication(ConnectASGIApplication[ProjectService]): + def __init__(self, service: ProjectService | AsyncGenerator[ProjectService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.project.ProjectService/CreateProject": Endpoint.unary( + method=MethodInfo( + name="CreateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_project, + ), + "/flyteidl2.project.ProjectService/UpdateProject": Endpoint.unary( + method=MethodInfo( + name="UpdateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_project, + ), + "/flyteidl2.project.ProjectService/GetProject": Endpoint.unary( + method=MethodInfo( + name="GetProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_project, + ), + "/flyteidl2.project.ProjectService/ListProjects": Endpoint.unary( + method=MethodInfo( + name="ListProjects", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.list_projects, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.project.ProjectService" + + +class ProjectServiceClient(ConnectClient): + async def create_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def list_projects( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListProjects", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class ProjectServiceSync(Protocol): + def create_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_project(self, request: flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_projects(self, request: flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, ctx: RequestContext) -> flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class ProjectServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: ProjectServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.project.ProjectService/CreateProject": EndpointSync.unary( + method=MethodInfo( + name="CreateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_project, + ), + "/flyteidl2.project.ProjectService/UpdateProject": EndpointSync.unary( + method=MethodInfo( + name="UpdateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_project, + ), + "/flyteidl2.project.ProjectService/GetProject": EndpointSync.unary( + method=MethodInfo( + name="GetProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_project, + ), + "/flyteidl2.project.ProjectService/ListProjects": EndpointSync.unary( + method=MethodInfo( + name="ListProjects", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.list_projects, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.project.ProjectService" + + +class ProjectServiceClientSync(ConnectClientSync): + def create_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.CreateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.UpdateProjectResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_project( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetProject", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.GetProjectRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.GetProjectResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def list_projects( + self, + request: flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListProjects", + service_name="flyteidl2.project.ProjectService", + input=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsRequest, + output=flyteidl2_dot_project_dot_project__service__pb2.ListProjectsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/secret/secret_connect.py b/gen/python/flyteidl2/secret/secret_connect.py new file mode 100644 index 0000000000..ed9f944235 --- /dev/null +++ b/gen/python/flyteidl2/secret/secret_connect.py @@ -0,0 +1,383 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/secret/secret.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.secret.payload_pb2 as flyteidl2_dot_secret_dot_payload__pb2 + + +class SecretService(Protocol): + async def create_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_secrets(self, request: flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class SecretServiceASGIApplication(ConnectASGIApplication[SecretService]): + def __init__(self, service: SecretService | AsyncGenerator[SecretService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.secret.SecretService/CreateSecret": Endpoint.unary( + method=MethodInfo( + name="CreateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_secret, + ), + "/flyteidl2.secret.SecretService/UpdateSecret": Endpoint.unary( + method=MethodInfo( + name="UpdateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_secret, + ), + "/flyteidl2.secret.SecretService/GetSecret": Endpoint.unary( + method=MethodInfo( + name="GetSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get_secret, + ), + "/flyteidl2.secret.SecretService/DeleteSecret": Endpoint.unary( + method=MethodInfo( + name="DeleteSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete_secret, + ), + "/flyteidl2.secret.SecretService/ListSecrets": Endpoint.unary( + method=MethodInfo( + name="ListSecrets", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.list_secrets, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.secret.SecretService" + + +class SecretServiceClient(ConnectClient): + async def create_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def list_secrets( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListSecrets", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class SecretServiceSync(Protocol): + def create_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete_secret(self, request: flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_secrets(self, request: flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, ctx: RequestContext) -> flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class SecretServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: SecretServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.secret.SecretService/CreateSecret": EndpointSync.unary( + method=MethodInfo( + name="CreateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_secret, + ), + "/flyteidl2.secret.SecretService/UpdateSecret": EndpointSync.unary( + method=MethodInfo( + name="UpdateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_secret, + ), + "/flyteidl2.secret.SecretService/GetSecret": EndpointSync.unary( + method=MethodInfo( + name="GetSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get_secret, + ), + "/flyteidl2.secret.SecretService/DeleteSecret": EndpointSync.unary( + method=MethodInfo( + name="DeleteSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete_secret, + ), + "/flyteidl2.secret.SecretService/ListSecrets": EndpointSync.unary( + method=MethodInfo( + name="ListSecrets", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.list_secrets, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.secret.SecretService" + + +class SecretServiceClientSync(ConnectClientSync): + def create_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.CreateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.UpdateSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.GetSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.GetSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete_secret( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteSecret", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.DeleteSecretResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def list_secrets( + self, + request: flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListSecrets", + service_name="flyteidl2.secret.SecretService", + input=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsRequest, + output=flyteidl2_dot_secret_dot_payload__pb2.ListSecretsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/task/task_service_connect.py b/gen/python/flyteidl2/task/task_service_connect.py new file mode 100644 index 0000000000..ca088cef00 --- /dev/null +++ b/gen/python/flyteidl2/task/task_service_connect.py @@ -0,0 +1,330 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/task/task_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.task.task_service_pb2 as flyteidl2_dot_task_dot_task__service__pb2 + + +class TaskService(Protocol): + async def deploy_task(self, request: flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_task_details(self, request: flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_tasks(self, request: flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_versions(self, request: flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TaskServiceASGIApplication(ConnectASGIApplication[TaskService]): + def __init__(self, service: TaskService | AsyncGenerator[TaskService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.task.TaskService/DeployTask": Endpoint.unary( + method=MethodInfo( + name="DeployTask", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.deploy_task, + ), + "/flyteidl2.task.TaskService/GetTaskDetails": Endpoint.unary( + method=MethodInfo( + name="GetTaskDetails", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_task_details, + ), + "/flyteidl2.task.TaskService/ListTasks": Endpoint.unary( + method=MethodInfo( + name="ListTasks", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_tasks, + ), + "/flyteidl2.task.TaskService/ListVersions": Endpoint.unary( + method=MethodInfo( + name="ListVersions", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_versions, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.task.TaskService" + + +class TaskServiceClient(ConnectClient): + async def deploy_task( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeployTask", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_task_details( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTaskDetails", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def list_tasks( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListTasks", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def list_versions( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListVersions", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + +class TaskServiceSync(Protocol): + def deploy_task(self, request: flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_task_details(self, request: flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_tasks(self, request: flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_versions(self, request: flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, ctx: RequestContext) -> flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TaskServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: TaskServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.task.TaskService/DeployTask": EndpointSync.unary( + method=MethodInfo( + name="DeployTask", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.deploy_task, + ), + "/flyteidl2.task.TaskService/GetTaskDetails": EndpointSync.unary( + method=MethodInfo( + name="GetTaskDetails", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_task_details, + ), + "/flyteidl2.task.TaskService/ListTasks": EndpointSync.unary( + method=MethodInfo( + name="ListTasks", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_tasks, + ), + "/flyteidl2.task.TaskService/ListVersions": EndpointSync.unary( + method=MethodInfo( + name="ListVersions", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_versions, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.task.TaskService" + + +class TaskServiceClientSync(ConnectClientSync): + def deploy_task( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeployTask", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.DeployTaskResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_task_details( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTaskDetails", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.GetTaskDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def list_tasks( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListTasks", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListTasksRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListTasksResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def list_versions( + self, + request: flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListVersions", + service_name="flyteidl2.task.TaskService", + input=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsRequest, + output=flyteidl2_dot_task_dot_task__service__pb2.ListVersionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) diff --git a/gen/python/flyteidl2/trigger/trigger_service_connect.py b/gen/python/flyteidl2/trigger/trigger_service_connect.py new file mode 100644 index 0000000000..22d570eb22 --- /dev/null +++ b/gen/python/flyteidl2/trigger/trigger_service_connect.py @@ -0,0 +1,529 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/trigger/trigger_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.trigger.trigger_service_pb2 as flyteidl2_dot_trigger_dot_trigger__service__pb2 + + +class TriggerService(Protocol): + async def deploy_trigger(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_trigger_details(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_trigger_revision_details(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_trigger_revision_history(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def delete_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TriggerServiceASGIApplication(ConnectASGIApplication[TriggerService]): + def __init__(self, service: TriggerService | AsyncGenerator[TriggerService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.trigger.TriggerService/DeployTrigger": Endpoint.unary( + method=MethodInfo( + name="DeployTrigger", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.deploy_trigger, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerDetails": Endpoint.unary( + method=MethodInfo( + name="GetTriggerDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_trigger_details, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerRevisionDetails": Endpoint.unary( + method=MethodInfo( + name="GetTriggerRevisionDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_trigger_revision_details, + ), + "/flyteidl2.trigger.TriggerService/ListTriggers": Endpoint.unary( + method=MethodInfo( + name="ListTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_triggers, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerRevisionHistory": Endpoint.unary( + method=MethodInfo( + name="GetTriggerRevisionHistory", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_trigger_revision_history, + ), + "/flyteidl2.trigger.TriggerService/UpdateTriggers": Endpoint.unary( + method=MethodInfo( + name="UpdateTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_triggers, + ), + "/flyteidl2.trigger.TriggerService/DeleteTriggers": Endpoint.unary( + method=MethodInfo( + name="DeleteTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.delete_triggers, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.trigger.TriggerService" + + +class TriggerServiceClient(ConnectClient): + async def deploy_trigger( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeployTrigger", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_trigger_details( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def get_trigger_revision_details( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerRevisionDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def list_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def get_trigger_revision_history( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerRevisionHistory", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def update_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def delete_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class TriggerServiceSync(Protocol): + def deploy_trigger(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_trigger_details(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_trigger_revision_details(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_trigger_revision_history(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def delete_triggers(self, request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, ctx: RequestContext) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TriggerServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: TriggerServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.trigger.TriggerService/DeployTrigger": EndpointSync.unary( + method=MethodInfo( + name="DeployTrigger", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.deploy_trigger, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerDetails": EndpointSync.unary( + method=MethodInfo( + name="GetTriggerDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_trigger_details, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerRevisionDetails": EndpointSync.unary( + method=MethodInfo( + name="GetTriggerRevisionDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_trigger_revision_details, + ), + "/flyteidl2.trigger.TriggerService/ListTriggers": EndpointSync.unary( + method=MethodInfo( + name="ListTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_triggers, + ), + "/flyteidl2.trigger.TriggerService/GetTriggerRevisionHistory": EndpointSync.unary( + method=MethodInfo( + name="GetTriggerRevisionHistory", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_trigger_revision_history, + ), + "/flyteidl2.trigger.TriggerService/UpdateTriggers": EndpointSync.unary( + method=MethodInfo( + name="UpdateTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_triggers, + ), + "/flyteidl2.trigger.TriggerService/DeleteTriggers": EndpointSync.unary( + method=MethodInfo( + name="DeleteTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.delete_triggers, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.trigger.TriggerService" + + +class TriggerServiceClientSync(ConnectClientSync): + def deploy_trigger( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeployTrigger", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeployTriggerResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_trigger_details( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def get_trigger_revision_details( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerRevisionDetails", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def list_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.ListTriggersResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def get_trigger_revision_history( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetTriggerRevisionHistory", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.GetTriggerRevisionHistoryResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def update_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.UpdateTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def delete_triggers( + self, + request: flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="DeleteTriggers", + service_name="flyteidl2.trigger.TriggerService", + input=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersRequest, + output=flyteidl2_dot_trigger_dot_trigger__service__pb2.DeleteTriggersResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/events_proxy_service_connect.py b/gen/python/flyteidl2/workflow/events_proxy_service_connect.py new file mode 100644 index 0000000000..dabc6ed0c9 --- /dev/null +++ b/gen/python/flyteidl2/workflow/events_proxy_service_connect.py @@ -0,0 +1,123 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/events_proxy_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.events_proxy_service_pb2 as flyteidl2_dot_workflow_dot_events__proxy__service__pb2 + + +class EventsProxyService(Protocol): + async def record(self, request: flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class EventsProxyServiceASGIApplication(ConnectASGIApplication[EventsProxyService]): + def __init__(self, service: EventsProxyService | AsyncGenerator[EventsProxyService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.EventsProxyService/Record": Endpoint.unary( + method=MethodInfo( + name="Record", + service_name="flyteidl2.workflow.EventsProxyService", + input=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + output=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.record, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.EventsProxyService" + + +class EventsProxyServiceClient(ConnectClient): + async def record( + self, + request: flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Record", + service_name="flyteidl2.workflow.EventsProxyService", + input=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + output=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class EventsProxyServiceSync(Protocol): + def record(self, request: flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class EventsProxyServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: EventsProxyServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.EventsProxyService/Record": EndpointSync.unary( + method=MethodInfo( + name="Record", + service_name="flyteidl2.workflow.EventsProxyService", + input=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + output=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.record, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.EventsProxyService" + + +class EventsProxyServiceClientSync(ConnectClientSync): + def record( + self, + request: flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Record", + service_name="flyteidl2.workflow.EventsProxyService", + input=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordRequest, + output=flyteidl2_dot_workflow_dot_events__proxy__service__pb2.RecordResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/internal_run_service_connect.py b/gen/python/flyteidl2/workflow/internal_run_service_connect.py new file mode 100644 index 0000000000..d63518d70a --- /dev/null +++ b/gen/python/flyteidl2/workflow/internal_run_service_connect.py @@ -0,0 +1,448 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/internal_run_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.internal_run_service_pb2 as flyteidl2_dot_workflow_dot_internal__run__service__pb2 + + +class InternalRunService(Protocol): + async def record_action(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def record_action_stream(self, request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest], ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def update_action_status(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def update_action_status_stream(self, request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest], ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def record_action_events(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def record_action_event_stream(self, request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest], ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class InternalRunServiceASGIApplication(ConnectASGIApplication[InternalRunService]): + def __init__(self, service: InternalRunService | AsyncGenerator[InternalRunService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.InternalRunService/RecordAction": Endpoint.unary( + method=MethodInfo( + name="RecordAction", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.record_action, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionStream": Endpoint.bidi_stream( + method=MethodInfo( + name="RecordActionStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.record_action_stream, + ), + "/flyteidl2.workflow.InternalRunService/UpdateActionStatus": Endpoint.unary( + method=MethodInfo( + name="UpdateActionStatus", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_action_status, + ), + "/flyteidl2.workflow.InternalRunService/UpdateActionStatusStream": Endpoint.bidi_stream( + method=MethodInfo( + name="UpdateActionStatusStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.update_action_status_stream, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionEvents": Endpoint.unary( + method=MethodInfo( + name="RecordActionEvents", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.record_action_events, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionEventStream": Endpoint.bidi_stream( + method=MethodInfo( + name="RecordActionEventStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.record_action_event_stream, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.InternalRunService" + + +class InternalRunServiceClient(ConnectClient): + async def record_action( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="RecordAction", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def record_action_stream( + self, + request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="RecordActionStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def update_action_status( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateActionStatus", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_action_status_stream( + self, + request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="UpdateActionStatusStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def record_action_events( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="RecordActionEvents", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def record_action_event_stream( + self, + request: AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="RecordActionEventStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class InternalRunServiceSync(Protocol): + def record_action(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def record_action_stream(self, request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest], ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_action_status(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def update_action_status_stream(self, request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest], ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def record_action_events(self, request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def record_action_event_stream(self, request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest], ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class InternalRunServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: InternalRunServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.InternalRunService/RecordAction": EndpointSync.unary( + method=MethodInfo( + name="RecordAction", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.record_action, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionStream": EndpointSync.bidi_stream( + method=MethodInfo( + name="RecordActionStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.record_action_stream, + ), + "/flyteidl2.workflow.InternalRunService/UpdateActionStatus": EndpointSync.unary( + method=MethodInfo( + name="UpdateActionStatus", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_action_status, + ), + "/flyteidl2.workflow.InternalRunService/UpdateActionStatusStream": EndpointSync.bidi_stream( + method=MethodInfo( + name="UpdateActionStatusStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.update_action_status_stream, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionEvents": EndpointSync.unary( + method=MethodInfo( + name="RecordActionEvents", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.record_action_events, + ), + "/flyteidl2.workflow.InternalRunService/RecordActionEventStream": EndpointSync.bidi_stream( + method=MethodInfo( + name="RecordActionEventStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.record_action_event_stream, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.InternalRunService" + + +class InternalRunServiceClientSync(ConnectClientSync): + def record_action( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="RecordAction", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def record_action_stream( + self, + request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="RecordActionStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_action_status( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="UpdateActionStatus", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def update_action_status_stream( + self, + request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="UpdateActionStatusStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.UpdateActionStatusStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def record_action_events( + self, + request: flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="RecordActionEvents", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def record_action_event_stream( + self, + request: Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest], + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse]: + return self.execute_bidi_stream( + request=request, + method=MethodInfo( + name="RecordActionEventStream", + service_name="flyteidl2.workflow.InternalRunService", + input=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamRequest, + output=flyteidl2_dot_workflow_dot_internal__run__service__pb2.RecordActionEventStreamResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/queue_service_connect.py b/gen/python/flyteidl2/workflow/queue_service_connect.py new file mode 100644 index 0000000000..7189d727e0 --- /dev/null +++ b/gen/python/flyteidl2/workflow/queue_service_connect.py @@ -0,0 +1,253 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/queue_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.queue_service_pb2 as flyteidl2_dot_workflow_dot_queue__service__pb2 + + +class QueueService(Protocol): + async def enqueue_action(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def abort_queued_run(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def abort_queued_action(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class QueueServiceASGIApplication(ConnectASGIApplication[QueueService]): + def __init__(self, service: QueueService | AsyncGenerator[QueueService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.QueueService/EnqueueAction": Endpoint.unary( + method=MethodInfo( + name="EnqueueAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.enqueue_action, + ), + "/flyteidl2.workflow.QueueService/AbortQueuedRun": Endpoint.unary( + method=MethodInfo( + name="AbortQueuedRun", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.abort_queued_run, + ), + "/flyteidl2.workflow.QueueService/AbortQueuedAction": Endpoint.unary( + method=MethodInfo( + name="AbortQueuedAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.abort_queued_action, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.QueueService" + + +class QueueServiceClient(ConnectClient): + async def enqueue_action( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="EnqueueAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def abort_queued_run( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="AbortQueuedRun", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def abort_queued_action( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="AbortQueuedAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class QueueServiceSync(Protocol): + def enqueue_action(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def abort_queued_run(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def abort_queued_action(self, request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class QueueServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: QueueServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.QueueService/EnqueueAction": EndpointSync.unary( + method=MethodInfo( + name="EnqueueAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.enqueue_action, + ), + "/flyteidl2.workflow.QueueService/AbortQueuedRun": EndpointSync.unary( + method=MethodInfo( + name="AbortQueuedRun", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.abort_queued_run, + ), + "/flyteidl2.workflow.QueueService/AbortQueuedAction": EndpointSync.unary( + method=MethodInfo( + name="AbortQueuedAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.abort_queued_action, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.QueueService" + + +class QueueServiceClientSync(ConnectClientSync): + def enqueue_action( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="EnqueueAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.EnqueueActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def abort_queued_run( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="AbortQueuedRun", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def abort_queued_action( + self, + request: flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="AbortQueuedAction", + service_name="flyteidl2.workflow.QueueService", + input=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionRequest, + output=flyteidl2_dot_workflow_dot_queue__service__pb2.AbortQueuedActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/run_logs_service_connect.py b/gen/python/flyteidl2/workflow/run_logs_service_connect.py new file mode 100644 index 0000000000..78523294e1 --- /dev/null +++ b/gen/python/flyteidl2/workflow/run_logs_service_connect.py @@ -0,0 +1,123 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/run_logs_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.run_logs_service_pb2 as flyteidl2_dot_workflow_dot_run__logs__service__pb2 + + +class RunLogsService(Protocol): + def tail_logs(self, request: flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class RunLogsServiceASGIApplication(ConnectASGIApplication[RunLogsService]): + def __init__(self, service: RunLogsService | AsyncGenerator[RunLogsService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.RunLogsService/TailLogs": Endpoint.server_stream( + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.workflow.RunLogsService", + input=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + output=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.tail_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.RunLogsService" + + +class RunLogsServiceClient(ConnectClient): + def tail_logs( + self, + request: flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.workflow.RunLogsService", + input=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + output=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class RunLogsServiceSync(Protocol): + def tail_logs(self, request: flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class RunLogsServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: RunLogsServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.RunLogsService/TailLogs": EndpointSync.server_stream( + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.workflow.RunLogsService", + input=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + output=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.tail_logs, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.RunLogsService" + + +class RunLogsServiceClientSync(ConnectClientSync): + def tail_logs( + self, + request: flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="TailLogs", + service_name="flyteidl2.workflow.RunLogsService", + input=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsRequest, + output=flyteidl2_dot_workflow_dot_run__logs__service__pb2.TailLogsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/run_service_connect.py b/gen/python/flyteidl2/workflow/run_service_connect.py new file mode 100644 index 0000000000..b2a7256bbc --- /dev/null +++ b/gen/python/flyteidl2/workflow/run_service_connect.py @@ -0,0 +1,988 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/run_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.run_service_pb2 as flyteidl2_dot_workflow_dot_run__service__pb2 + + +class RunService(Protocol): + async def create_run(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def abort_run(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_run_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_run_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_action_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_action_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get_action_data(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_runs(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_runs(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def list_actions(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_actions(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_cluster_events(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def abort_action(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch_groups(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class RunServiceASGIApplication(ConnectASGIApplication[RunService]): + def __init__(self, service: RunService | AsyncGenerator[RunService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.RunService/CreateRun": Endpoint.unary( + method=MethodInfo( + name="CreateRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.create_run, + ), + "/flyteidl2.workflow.RunService/AbortRun": Endpoint.unary( + method=MethodInfo( + name="AbortRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.abort_run, + ), + "/flyteidl2.workflow.RunService/GetRunDetails": Endpoint.unary( + method=MethodInfo( + name="GetRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_run_details, + ), + "/flyteidl2.workflow.RunService/WatchRunDetails": Endpoint.server_stream( + method=MethodInfo( + name="WatchRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_run_details, + ), + "/flyteidl2.workflow.RunService/GetActionDetails": Endpoint.unary( + method=MethodInfo( + name="GetActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_action_details, + ), + "/flyteidl2.workflow.RunService/WatchActionDetails": Endpoint.server_stream( + method=MethodInfo( + name="WatchActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_action_details, + ), + "/flyteidl2.workflow.RunService/GetActionData": Endpoint.unary( + method=MethodInfo( + name="GetActionData", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.get_action_data, + ), + "/flyteidl2.workflow.RunService/ListRuns": Endpoint.unary( + method=MethodInfo( + name="ListRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_runs, + ), + "/flyteidl2.workflow.RunService/WatchRuns": Endpoint.server_stream( + method=MethodInfo( + name="WatchRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_runs, + ), + "/flyteidl2.workflow.RunService/ListActions": Endpoint.unary( + method=MethodInfo( + name="ListActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.list_actions, + ), + "/flyteidl2.workflow.RunService/WatchActions": Endpoint.server_stream( + method=MethodInfo( + name="WatchActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_actions, + ), + "/flyteidl2.workflow.RunService/WatchClusterEvents": Endpoint.server_stream( + method=MethodInfo( + name="WatchClusterEvents", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_cluster_events, + ), + "/flyteidl2.workflow.RunService/AbortAction": Endpoint.unary( + method=MethodInfo( + name="AbortAction", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.abort_action, + ), + "/flyteidl2.workflow.RunService/WatchGroups": Endpoint.server_stream( + method=MethodInfo( + name="WatchGroups", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch_groups, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.RunService" + + +class RunServiceClient(ConnectClient): + async def create_run( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="CreateRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def abort_run( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="AbortRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_run_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_run_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_action_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_action_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get_action_data( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="GetActionData", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def list_runs( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_runs( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def list_actions( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="ListActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_actions( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_cluster_events( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchClusterEvents", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def abort_action( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="AbortAction", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_groups( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchGroups", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class RunServiceSync(Protocol): + def create_run(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def abort_run(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_run_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_run_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_action_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_action_details(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get_action_data(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_runs(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_runs(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def list_actions(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_actions(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_cluster_events(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def abort_action(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch_groups(self, request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class RunServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: RunServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.RunService/CreateRun": EndpointSync.unary( + method=MethodInfo( + name="CreateRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.create_run, + ), + "/flyteidl2.workflow.RunService/AbortRun": EndpointSync.unary( + method=MethodInfo( + name="AbortRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.abort_run, + ), + "/flyteidl2.workflow.RunService/GetRunDetails": EndpointSync.unary( + method=MethodInfo( + name="GetRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_run_details, + ), + "/flyteidl2.workflow.RunService/WatchRunDetails": EndpointSync.server_stream( + method=MethodInfo( + name="WatchRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_run_details, + ), + "/flyteidl2.workflow.RunService/GetActionDetails": EndpointSync.unary( + method=MethodInfo( + name="GetActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_action_details, + ), + "/flyteidl2.workflow.RunService/WatchActionDetails": EndpointSync.server_stream( + method=MethodInfo( + name="WatchActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_action_details, + ), + "/flyteidl2.workflow.RunService/GetActionData": EndpointSync.unary( + method=MethodInfo( + name="GetActionData", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.get_action_data, + ), + "/flyteidl2.workflow.RunService/ListRuns": EndpointSync.unary( + method=MethodInfo( + name="ListRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_runs, + ), + "/flyteidl2.workflow.RunService/WatchRuns": EndpointSync.server_stream( + method=MethodInfo( + name="WatchRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_runs, + ), + "/flyteidl2.workflow.RunService/ListActions": EndpointSync.unary( + method=MethodInfo( + name="ListActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.list_actions, + ), + "/flyteidl2.workflow.RunService/WatchActions": EndpointSync.server_stream( + method=MethodInfo( + name="WatchActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_actions, + ), + "/flyteidl2.workflow.RunService/WatchClusterEvents": EndpointSync.server_stream( + method=MethodInfo( + name="WatchClusterEvents", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_cluster_events, + ), + "/flyteidl2.workflow.RunService/AbortAction": EndpointSync.unary( + method=MethodInfo( + name="AbortAction", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.abort_action, + ), + "/flyteidl2.workflow.RunService/WatchGroups": EndpointSync.server_stream( + method=MethodInfo( + name="WatchGroups", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch_groups, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.RunService" + + +class RunServiceClientSync(ConnectClientSync): + def create_run( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="CreateRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.CreateRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def abort_run( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="AbortRun", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortRunResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_run_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetRunDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_run_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchRunDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_action_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDetailsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_action_details( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchActionDetails", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionDetailsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get_action_data( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="GetActionData", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.GetActionDataResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def list_runs( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListRunsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_runs( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchRuns", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchRunsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def list_actions( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="ListActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.ListActionsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def watch_actions( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchActions", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchActionsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_cluster_events( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchClusterEvents", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchClusterEventsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def abort_action( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="AbortAction", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.AbortActionResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch_groups( + self, + request: flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="WatchGroups", + service_name="flyteidl2.workflow.RunService", + input=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsRequest, + output=flyteidl2_dot_workflow_dot_run__service__pb2.WatchGroupsResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/state_service_connect.py b/gen/python/flyteidl2/workflow/state_service_connect.py new file mode 100644 index 0000000000..2fe6a44229 --- /dev/null +++ b/gen/python/flyteidl2/workflow/state_service_connect.py @@ -0,0 +1,253 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/state_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.state_service_pb2 as flyteidl2_dot_workflow_dot_state__service__pb2 + + +class StateService(Protocol): + async def put(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def get(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + def watch(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, ctx: RequestContext) -> AsyncIterator[flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class StateServiceASGIApplication(ConnectASGIApplication[StateService]): + def __init__(self, service: StateService | AsyncGenerator[StateService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.StateService/Put": Endpoint.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.put, + ), + "/flyteidl2.workflow.StateService/Get": Endpoint.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.get, + ), + "/flyteidl2.workflow.StateService/Watch": Endpoint.server_stream( + method=MethodInfo( + name="Watch", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=svc.watch, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.StateService" + + +class StateServiceClient(ConnectClient): + async def put( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + async def get( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> AsyncIterator[flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Watch", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + +class StateServiceSync(Protocol): + def put(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def get(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def watch(self, request: flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, ctx: RequestContext) -> Iterator[flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse]: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class StateServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: StateServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.StateService/Put": EndpointSync.unary( + method=MethodInfo( + name="Put", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.put, + ), + "/flyteidl2.workflow.StateService/Get": EndpointSync.unary( + method=MethodInfo( + name="Get", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.get, + ), + "/flyteidl2.workflow.StateService/Watch": EndpointSync.server_stream( + method=MethodInfo( + name="Watch", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + function=service.watch, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.StateService" + + +class StateServiceClientSync(ConnectClientSync): + def put( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Put", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.PutRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.PutResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def get( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="Get", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.GetRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.GetResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) + + def watch( + self, + request: flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + ) -> Iterator[flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse]: + return self.execute_server_stream( + request=request, + method=MethodInfo( + name="Watch", + service_name="flyteidl2.workflow.StateService", + input=flyteidl2_dot_workflow_dot_state__service__pb2.WatchRequest, + output=flyteidl2_dot_workflow_dot_state__service__pb2.WatchResponse, + idempotency_level=IdempotencyLevel.UNKNOWN, + ), + headers=headers, + timeout_ms=timeout_ms, + ) diff --git a/gen/python/flyteidl2/workflow/translator_service_connect.py b/gen/python/flyteidl2/workflow/translator_service_connect.py new file mode 100644 index 0000000000..6b8bdc6f86 --- /dev/null +++ b/gen/python/flyteidl2/workflow/translator_service_connect.py @@ -0,0 +1,334 @@ +# -*- coding: utf-8 -*- +# Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT! +# source: flyteidl2/workflow/translator_service.proto + +from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping +from typing import Protocol + +from connectrpc.client import ConnectClient, ConnectClientSync +from connectrpc.code import Code +from connectrpc.compression import Compression +from connectrpc.errors import ConnectError +from connectrpc.interceptor import Interceptor, InterceptorSync +from connectrpc.method import IdempotencyLevel, MethodInfo +from connectrpc.request import Headers, RequestContext +from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync +import flyteidl2.workflow.translator_service_pb2 as flyteidl2_dot_workflow_dot_translator__service__pb2 + + +class TranslatorService(Protocol): + async def literals_to_launch_form_json(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def launch_form_json_to_literals(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def task_spec_to_launch_form_json(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + async def json_values_to_literals(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TranslatorServiceASGIApplication(ConnectASGIApplication[TranslatorService]): + def __init__(self, service: TranslatorService | AsyncGenerator[TranslatorService], *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + service=service, + endpoints=lambda svc: { + "/flyteidl2.workflow.TranslatorService/LiteralsToLaunchFormJson": Endpoint.unary( + method=MethodInfo( + name="LiteralsToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.literals_to_launch_form_json, + ), + "/flyteidl2.workflow.TranslatorService/LaunchFormJsonToLiterals": Endpoint.unary( + method=MethodInfo( + name="LaunchFormJsonToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.launch_form_json_to_literals, + ), + "/flyteidl2.workflow.TranslatorService/TaskSpecToLaunchFormJson": Endpoint.unary( + method=MethodInfo( + name="TaskSpecToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.task_spec_to_launch_form_json, + ), + "/flyteidl2.workflow.TranslatorService/JsonValuesToLiterals": Endpoint.unary( + method=MethodInfo( + name="JsonValuesToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=svc.json_values_to_literals, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.TranslatorService" + + +class TranslatorServiceClient(ConnectClient): + async def literals_to_launch_form_json( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="LiteralsToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def launch_form_json_to_literals( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="LaunchFormJsonToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def task_spec_to_launch_form_json( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="TaskSpecToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + async def json_values_to_literals( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse: + return await self.execute_unary( + request=request, + method=MethodInfo( + name="JsonValuesToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + +class TranslatorServiceSync(Protocol): + def literals_to_launch_form_json(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def launch_form_json_to_literals(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def task_spec_to_launch_form_json(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + def json_values_to_literals(self, request: flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, ctx: RequestContext) -> flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse: + raise ConnectError(Code.UNIMPLEMENTED, "Not implemented") + + +class TranslatorServiceWSGIApplication(ConnectWSGIApplication): + def __init__(self, service: TranslatorServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None, compressions: Iterable[Compression] | None = None) -> None: + super().__init__( + endpoints={ + "/flyteidl2.workflow.TranslatorService/LiteralsToLaunchFormJson": EndpointSync.unary( + method=MethodInfo( + name="LiteralsToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.literals_to_launch_form_json, + ), + "/flyteidl2.workflow.TranslatorService/LaunchFormJsonToLiterals": EndpointSync.unary( + method=MethodInfo( + name="LaunchFormJsonToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.launch_form_json_to_literals, + ), + "/flyteidl2.workflow.TranslatorService/TaskSpecToLaunchFormJson": EndpointSync.unary( + method=MethodInfo( + name="TaskSpecToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.task_spec_to_launch_form_json, + ), + "/flyteidl2.workflow.TranslatorService/JsonValuesToLiterals": EndpointSync.unary( + method=MethodInfo( + name="JsonValuesToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + function=service.json_values_to_literals, + ), + }, + interceptors=interceptors, + read_max_bytes=read_max_bytes, + compressions=compressions, + ) + + @property + def path(self) -> str: + """Returns the URL path to mount the application to when serving multiple applications.""" + return "/flyteidl2.workflow.TranslatorService" + + +class TranslatorServiceClientSync(ConnectClientSync): + def literals_to_launch_form_json( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="LiteralsToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LiteralsToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def launch_form_json_to_literals( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="LaunchFormJsonToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.LaunchFormJsonToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def task_spec_to_launch_form_json( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="TaskSpecToLaunchFormJson", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.TaskSpecToLaunchFormJsonResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + ) + + def json_values_to_literals( + self, + request: flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + *, + headers: Headers | Mapping[str, str] | None = None, + timeout_ms: int | None = None, + use_get: bool = False, + ) -> flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse: + return self.execute_unary( + request=request, + method=MethodInfo( + name="JsonValuesToLiterals", + service_name="flyteidl2.workflow.TranslatorService", + input=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsRequest, + output=flyteidl2_dot_workflow_dot_translator__service__pb2.JsonValuesToLiteralsResponse, + idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, + ), + headers=headers, + timeout_ms=timeout_ms, + use_get=use_get, + )