Exemple #1
0
    async def __rpc_delta_runtime(self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        await self._call_rpc_handler_server_stream(
            self.delta_runtime,
            stream,
            request_kwargs,
        )
Exemple #2
0
    async def __rpc_stream_load_stats(self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        await self._call_rpc_handler_server_stream(
            self.stream_load_stats,
            stream,
            request_kwargs,
        )
Exemple #3
0
    async def __rpc_ei_job_update(self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        await self._call_rpc_handler_server_stream(
            self.ei_job_update,
            stream,
            request_kwargs,
        )
Exemple #4
0
    async def __rpc_delta_aggregated_resources(
            self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        await self._call_rpc_handler_server_stream(
            self.delta_aggregated_resources,
            stream,
            request_kwargs,
        )
Exemple #5
0
    async def __rpc_stream_metrics(self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        response = await self.stream_metrics(**request_kwargs)
        await stream.send_message(response)
Exemple #6
0
    async def __rpc_upload_register_model(self, stream: grpclib.server.Stream) -> None:
        request_kwargs = {"request_iterator": stream.__aiter__()}

        response = await self.upload_register_model(**request_kwargs)
        await stream.send_message(response)