o
    ;i2                     @   sj   d dl Z d dlZd dlZd dlZejrd dlZd dlZd dl	Z
d dlZ
G dd de jZG dd dZdS )    Nc                   @   s   e Zd ZejdddZejddd	ZejdddZejdddZejdddZ	ejd ddZ
ejd!ddZdejeejjf fddZdS )"TaskCommandRouterBasestreamgrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecPollRequest, modal_proto.task_command_router_pb2.TaskExecPollResponse]returnNc                       d S N selfr   r   r   X/home/ubuntu/.local/lib/python3.10/site-packages/modal_proto/task_command_router_grpc.pyTaskExecPoll      z"TaskCommandRouterBase.TaskExecPollgrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecStartRequest, modal_proto.task_command_router_pb2.TaskExecStartResponse]c                    r   r   r   r	   r   r   r   TaskExecStart   r   z#TaskCommandRouterBase.TaskExecStartgrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecStdinWriteRequest, modal_proto.task_command_router_pb2.TaskExecStdinWriteResponse]c                    r   r   r   r	   r   r   r   TaskExecStdinWrite   r   z(TaskCommandRouterBase.TaskExecStdinWritegrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecStdioReadRequest, modal_proto.task_command_router_pb2.TaskExecStdioReadResponse]c                    r   r   r   r	   r   r   r   TaskExecStdioRead   r   z'TaskCommandRouterBase.TaskExecStdioReadgrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecWaitRequest, modal_proto.task_command_router_pb2.TaskExecWaitResponse]c                    r   r   r   r	   r   r   r   TaskExecWait#   r   z"TaskCommandRouterBase.TaskExecWaitugrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskMountDirectoryRequest, google.protobuf.empty_pb2.Empty]c                    r   r   r   r	   r   r   r   TaskMountDirectory'   r   z(TaskCommandRouterBase.TaskMountDirectorygrpclib.server.Stream[modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest, modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse]c                    r   r   r   r	   r   r   r   TaskSnapshotDirectory+   r   z+TaskCommandRouterBase.TaskSnapshotDirectoryc                 C   s   t j| jt jjjtjjtjj	t j| j
t jjjtjjtjjt j| jt jjjtjjtjjt j| jt jjjtjjtjjt j| jt jjjtjjtjjt j| jt jjjtjjtjjjt j| jt jjjtjjtjjdS )N)9/modal.task_command_router.TaskCommandRouter/TaskExecPoll:/modal.task_command_router.TaskCommandRouter/TaskExecStart?/modal.task_command_router.TaskCommandRouter/TaskExecStdinWrite>/modal.task_command_router.TaskCommandRouter/TaskExecStdioRead9/modal.task_command_router.TaskCommandRouter/TaskExecWait?/modal.task_command_router.TaskCommandRouter/TaskMountDirectoryB/modal.task_command_router.TaskCommandRouter/TaskSnapshotDirectory) grpclibconstHandlerr   CardinalityUNARY_UNARYmodal_prototask_command_router_pb2TaskExecPollRequestTaskExecPollResponser   TaskExecStartRequestTaskExecStartResponser   TaskExecStdinWriteRequestTaskExecStdinWriteResponser   UNARY_STREAMTaskExecStdioReadRequestTaskExecStdioReadResponser   TaskExecWaitRequestTaskExecWaitResponser   TaskMountDirectoryRequestgoogleprotobuf	empty_pb2Emptyr   TaskSnapshotDirectoryRequestTaskSnapshotDirectoryResponse)r
   r   r   r   __mapping__/   sV   z!TaskCommandRouterBase.__mapping__)r   r   r   N)r   r   r   N)r   r   r   N)r   r   r   N)r   r   r   N)r   r   r   N)r   r   r   N)__name__
__module____qualname__abcabstractmethodr   r   r   r   r   r   r   typingDictstrr!   r"   r#   r:   r   r   r   r   r      s      r   c                   @   s"   e Zd ZdejjddfddZdS )TaskCommandRouterStubchannelr   Nc                 C   s   t j|dtjjtjj| _t j|dtjjtjj	| _
t j|dtjjtjj| _t j|dtjjtjj| _t j|dtjjtjj| _t j|dtjjtjjj| _t j|dtjjtjj| _d S )Nr   r   r   r   r   r   r    )r!   clientUnaryUnaryMethodr&   r'   r(   r)   r   r*   r+   r   r,   r-   r   UnaryStreamMethodr/   r0   r   r1   r2   r   r3   r4   r5   r6   r7   r   r8   r9   r   )r
   rD   r   r   r   __init__`   sT   
zTaskCommandRouterStub.__init__)r;   r<   r=   r!   rE   ChannelrH   r   r   r   r   rC   ^   s    rC   )r>   r@   grpclib.constr!   grpclib.clientTYPE_CHECKINGgrpclib.servergoogle.protobuf.empty_pb2r4   modal_proto.api_pb2r&   #modal_proto.task_command_router_pb2ABCr   rC   r   r   r   r   <module>   s   M