o
    `۷if                     @   s4  d Z ddlZddlmZ G dd deZG dd deZd	d
 ZG dd deZ	G dd deZ
G dd deZdd ZG dd deZG dd deZG dd deZdd ZG dd deZG dd deZG dd  d eZd!d" ZG d#d$ d$eZG d%d& d&eZG d'd( d(eZd)d* ZG d+d, d,eZdS )-zEClient and server classes corresponding to protobuf-defined services.    N   )	serve_pb2c                   @      e Zd ZdZdd ZdS )RayServeAPIServiceStub8Missing associated documentation comment in .proto file.c                 C   s8   |j dtjjtjjd| _|j dtjjtjjd| _	dS )IConstructor.

        Args:
            channel: A grpc.Channel.
        ./ray.serve.RayServeAPIService/ListApplicationsrequest_serializerresponse_deserializer%/ray.serve.RayServeAPIService/HealthzN)
unary_unary'src_dot_ray_dot_protobuf_dot_serve__pb2ListApplicationsRequestSerializeToStringListApplicationsResponse
FromStringListApplicationsHealthzRequestHealthzResponseHealthzselfchannel r   X/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/serve/generated/serve_pb2_grpc.py__init__      zRayServeAPIServiceStub.__init__N__name__
__module____qualname____doc__r   r   r   r   r   r          r   c                   @       e Zd ZdZdd Zdd ZdS )RayServeAPIServiceServicerr   c                 C       | tjj |d tdr   zMethod not implemented!set_codegrpc
StatusCodeUNIMPLEMENTEDset_detailsNotImplementedErrorr   requestcontextr   r   r   r          
z+RayServeAPIServiceServicer.ListApplicationsc                 C   r&   r'   r(   r/   r   r   r   r   &   r2   z"RayServeAPIServiceServicer.HealthzN)r   r    r!   r"   r   r   r   r   r   r   r%          r%   c                 C   sR   t j| jtjjtjjdt j| jtj	jtj
jdd}t d|}||f d S )Nrequest_deserializerresponse_serializer)r   r   zray.serve.RayServeAPIService)r*   unary_unary_rpc_method_handlerr   r   r   r   r   r   r   r   r   method_handlers_generic_handleradd_generic_rpc_handlersservicerserverrpc_method_handlersgeneric_handlerr   r   r   (add_RayServeAPIServiceServicer_to_server-      r?   c                   @   L   e Zd ZdZe								d	ddZe								d	ddZdS )
RayServeAPIServicer   r   NFc
           
      C   ,   t j| |dtjjtjj||||||||	S )Nr   )r*   experimentalr   r   r   r   r   r   
r0   targetoptionschannel_credentialscall_credentialsinsecurecompressionwait_for_readytimeoutmetadatar   r   r   r   C      z#RayServeAPIService.ListApplicationsc
           
      C   rC   )Nr   )r*   rD   r   r   r   r   r   r   rE   r   r   r   r   T   rO   zRayServeAPIService.Healthzr   NNFNNNN)r   r    r!   r"   staticmethodr   r   r   r   r   r   rB   @   ,    rB   c                   @   r   )UserDefinedServiceStubr   c                 C   sl   |j dtjjtjjd| _|j dtjjtjjd| _|j dtjjtj	jd| _
|jdtjjtjjd| _dS )r   &/ray.serve.UserDefinedService/__call__r	   %/ray.serve.UserDefinedService/Method1%/ray.serve.UserDefinedService/Method2'/ray.serve.UserDefinedService/StreamingN)r   r   UserDefinedMessager   UserDefinedResponser   __call__Method1UserDefinedMessage2UserDefinedResponse2Method2unary_stream	Streamingr   r   r   r   r   i   (   zUserDefinedServiceStub.__init__Nr   r   r   r   r   rS   f   r#   rS   c                   @   0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )UserDefinedServiceServicerr   c                 C   r&   r'   r(   r/   r   r   r   rZ      r2   z#UserDefinedServiceServicer.__call__c                 C   r&   r'   r(   r/   r   r   r   r[      r2   z"UserDefinedServiceServicer.Method1c                 C   r&   r'   r(   r/   r   r   r   r^      r2   z"UserDefinedServiceServicer.Method2c                 C   r&   r'   r(   r/   r   r   r   r`      r2   z$UserDefinedServiceServicer.StreamingN)r   r    r!   r"   rZ   r[   r^   r`   r   r   r   r   rc          rc   c                 C   s   t j| jtjjtjjdt j| jtjjtjjdt j| j	tj
jtjjdt j| jtjjtjjdd}t d|}||f d S )Nr4   )rZ   r[   r^   r`   zray.serve.UserDefinedService)r*   r7   rZ   r   rX   r   rY   r   r[   r^   r\   r]   unary_stream_rpc_method_handlerr`   r8   r9   r:   r   r   r   (add_UserDefinedServiceServicer_to_server   2   rf   c                   @      e Zd ZdZe								dddZe								dddZe								dd	d
Ze								dddZdS )UserDefinedServicer   r   NFc
           
      C   rC   )NrT   r*   rD   r   r   rX   r   rY   r   rE   r   r   r   rZ      rO   zUserDefinedService.__call__c
           
      C   rC   )NrU   rj   rE   r   r   r   r[      rO   zUserDefinedService.Method1c
           
      C   rC   )NrV   )r*   rD   r   r   r\   r   r]   r   rE   r   r   r   r^      rO   zUserDefinedService.Method2c
           
      C   rC   )NrW   )r*   rD   r_   r   rX   r   rY   r   rE   r   r   r   r`      rO   zUserDefinedService.StreamingrP   )	r   r    r!   r"   rQ   rZ   r[   r^   r`   r   r   r   r   ri      T    ri   c                   @   r   )FruitServiceStubr   c                 C   s   |j dtjjtjjd| _dS )r   "/ray.serve.FruitService/FruitStandr	   N)r   r   FruitAmountsr   
FruitCostsr   
FruitStandr   r   r   r   r   	  s
   zFruitServiceStub.__init__Nr   r   r   r   r   rl     r#   rl   c                   @   r   )FruitServiceServicerr   c                 C   r&   r'   r(   r/   r   r   r   rp     r2   zFruitServiceServicer.FruitStandN)r   r    r!   r"   rp   r   r   r   r   rq     r#   rq   c                 C   s:   dt j| jtjjtjjdi}t d|}|	|f d S )Nrp   r4   zray.serve.FruitService)
r*   r7   rp   r   rn   r   ro   r   r8   r9   r:   r   r   r   "add_FruitServiceServicer_to_server   s   rr   c                   @   s.   e Zd ZdZe								dddZdS )FruitServicer   r   NFc
           
      C   rC   )Nrm   )r*   rD   r   r   rn   r   ro   r   rE   r   r   r   rp   1  rO   zFruitService.FruitStandrP   )r   r    r!   r"   rQ   rp   r   r   r   r   rs   .  s    rs   c                   @   r   )RayServeBenchmarkServiceStubr   c                 C   s8   |j dtjjtjjd| _|j dtjjtjjd| _dS )r   -/ray.serve.RayServeBenchmarkService/grpc_callr	   4/ray.serve.RayServeBenchmarkService/call_with_stringN)	r   r   	ArrayDatar   ModelOutputr   	grpc_call
StringDatacall_with_stringr   r   r   r   r   F  r   z%RayServeBenchmarkServiceStub.__init__Nr   r   r   r   r   rt   C  r#   rt   c                   @   r$   ) RayServeBenchmarkServiceServicerr   c                 C   r&   r'   r(   r/   r   r   r   ry   [  r2   z*RayServeBenchmarkServiceServicer.grpc_callc                 C   r&   r'   r(   r/   r   r   r   r{   a  r2   z1RayServeBenchmarkServiceServicer.call_with_stringN)r   r    r!   r"   ry   r{   r   r   r   r   r|   X  r3   r|   c                 C   sR   t j| jtjjtjjdt j| jtj	jtjjdd}t 
d|}||f d S )Nr4   )ry   r{   z"ray.serve.RayServeBenchmarkService)r*   r7   ry   r   rw   r   rx   r   r{   rz   r8   r9   r:   r   r   r   .add_RayServeBenchmarkServiceServicer_to_serverh  r@   r}   c                   @   rA   )
RayServeBenchmarkServicer   r   NFc
           
      C   rC   )Nru   )r*   rD   r   r   rw   r   rx   r   rE   r   r   r   ry   ~  rO   z"RayServeBenchmarkService.grpc_callc
           
      C   rC   )Nrv   )r*   rD   r   r   rz   r   rx   r   rE   r   r   r   r{     rO   z)RayServeBenchmarkService.call_with_stringrP   )r   r    r!   r"   rQ   ry   r{   r   r   r   r   r~   {  rR   r~   c                   @   r   )ASGIServiceStubr   c                 C   sl   |j dtjjtjjd| _|jdtjjtjjd| _|j dtjjtjjd| _	|jdtjjtjjd| _
dS )r   $/ray.serve.ASGIService/HandleRequestr	   -/ray.serve.ASGIService/HandleRequestStreaming1/ray.serve.ASGIService/HandleRequestWithRejection:/ray.serve.ASGIService/HandleRequestWithRejectionStreamingN)r   r   ASGIRequestr   ASGIResponser   HandleRequestr_   HandleRequestStreamingHandleRequestWithRejection#HandleRequestWithRejectionStreamingr   r   r   r   r     ra   zASGIServiceStub.__init__Nr   r   r   r   r   r     r#   r   c                   @   rb   )ASGIServiceServicerr   c                 C   r&   r'   r(   r/   r   r   r   r     r2   z!ASGIServiceServicer.HandleRequestc                 C   r&   r'   r(   r/   r   r   r   r     r2   z*ASGIServiceServicer.HandleRequestStreamingc                 C   r&   r'   r(   r/   r   r   r   r     r2   z.ASGIServiceServicer.HandleRequestWithRejectionc                 C   r&   r'   r(   r/   r   r   r   r     r2   z7ASGIServiceServicer.HandleRequestWithRejectionStreamingN)r   r    r!   r"   r   r   r   r   r   r   r   r   r     rd   r   c                 C   s   t j| jtjjtjjdt j| j	tjjtjjdt j| j
tjjtjjdt j| jtjjtjjdd}t d|}||f d S )Nr4   )r   r   r   r   zray.serve.ASGIService)r*   r7   r   r   r   r   r   r   re   r   r   r   r8   r9   r:   r   r   r   !add_ASGIServiceServicer_to_server  rg   r   c                   @   rh   )ASGIServicer   r   NFc
           
      C   rC   )Nr   r*   rD   r   r   r   r   r   r   rE   r   r   r   r     rO   zASGIService.HandleRequestc
           
      C   rC   )Nr   r*   rD   r_   r   r   r   r   r   rE   r   r   r   r     rO   z"ASGIService.HandleRequestStreamingc
           
      C   rC   )Nr   r   rE   r   r   r   r     rO   z&ASGIService.HandleRequestWithRejectionc
           
      C   rC   )Nr   r   rE   r   r   r   r   /  rO   z/ASGIService.HandleRequestWithRejectionStreamingrP   )	r   r    r!   r"   rQ   r   r   r   r   r   r   r   r   r     rk   r   )r"   r*    r   r   objectr   r%   r?   rB   rS   rc   rf   ri   rl   rq   rr   rs   rt   r|   r}   r~   r   r   r   r   r   r   r   r   <module>   s.   &H
&