Add connectrpc Python stubs for flyteidl2

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

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,
+        )