&`i3tddlmZmZmZmZmZddlZddlmZedGddZ dS))AnyDictListOptionalTupleN) PublicAPIbeta) stabilityceZdZdZdejjjfdZde e e ffdZ dej fdZde fdZdeee e ffdZde fd Zdeefd Zdee fd Zdeee e ffd Zd ej fdZdejfdZde fdZdeee e ffdZdeee e ffdZdejjjfdZdS)RayServegRPCContextaContext manager to set and get gRPC context. This class implements most of the methods from ServicerContext (see: https://grpc.github.io/grpc/python/grpc.html#grpc.ServicerContext). It's serializable and can be passed with the request to be used on the deployment. grpc_contextc||_||_||_d|D|_||_ | |_ | |_ d|D|_d|_dS)Ncg|] \}}||f Sr.0keyvalues g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/serve/grpc_util.py z0RayServegRPCContext.__init__..s-% % % 'S%S%L% % % cg|] \}}||f Srrrs rrz0RayServegRPCContext.__init__..s-# # # 'S%S%L# # # r) auth_context _auth_contextcode_codedetails_detailsinvocation_metadata_invocation_metadatapeer_peerpeer_identities_peer_identitiespeer_identity_key_peer_identity_keytrailing_metadata_trailing_metadata _compressionselfr s r__init__zRayServegRPCContext.__init__s)6688!&&(( $,,.. % % +7+K+K+M+M% % % !"&&(( , < < > >"."@"@"B"B# # +7+I+I+K+K# # # !rreturnc|jS)zGets the auth context for the call. Returns: A map of strings to an iterable of bytes for each auth property. )rr+s rrz RayServegRPCContext.auth_context s !!rc|jS)zAccesses the value to be used as status code upon RPC completion. Returns: The StatusCode value for the RPC. rr/s rrzRayServegRPCContext.code(s zrc|jS)zAccesses the value to be used as detail string upon RPC completion. Returns: The details string of the RPC. rr/s rrzRayServegRPCContext.details0s }rc|jS)zoAccesses the metadata sent by the client. Returns: The invocation :term:`metadata`. )r r/s rrz'RayServegRPCContext.invocation_metadata8s ((rc|jS)zIdentifies the peer that invoked the RPC being serviced. Returns: A string identifying the peer that invoked the RPC being serviced. The string format is determined by gRPC runtime. )r"r/s rr!zRayServegRPCContext.peer@s zrc|jS)a2Gets one or more peer identity(s). Equivalent to servicer_context.auth_context().get(servicer_context.peer_identity_key()) Returns: An iterable of the identities, or None if the call is not authenticated. Each identity is returned as a raw bytes type. )r$r/s rr#z#RayServegRPCContext.peer_identitiesIs $$rc|jS)a8The auth property used to identify the peer. For example, "x509_common_name" or "x509_subject_alternative_name" are used to identify an SSL peer. Returns: The auth property (string) that indicates the peer identity, or None if the call is not authenticated. )r&r/s rr%z%RayServegRPCContext.peer_identity_keyUs &&rc|jS)Nr(r/s rr'z%RayServegRPCContext.trailing_metadataas &&rrc||_dS)a$Sets the value to be used as status code upon RPC completion. This method need not be called by method implementations if they wish the gRPC runtime to determine the status code of the RPC. Args: code: A StatusCode object to be sent to the client. Nr1)r+rs rset_codezRayServegRPCContext.set_codeds r compressionc||_dS)zSet the compression algorithm to be used for the entire call. Args: compression: An element of grpc.compression, e.g. grpc.compression.Gzip. N)r))r+r<s rset_compressionz#RayServegRPCContext.set_compressionos(rrc||_dS)a4Sets the value to be used as detail string upon RPC completion. Calling this method is only needed if method implementations have details to transmit. Args: details: A UTF-8-encodable string to be sent to the client upon termination of the RPC. Nr3)r+rs r set_detailszRayServegRPCContext.set_detailsxs  rc:|jD]\}}|dkr||fgcSgS)N request_idr9)r+rrs r_request_id_metadataz(RayServegRPCContext._request_id_metadatasA1 & &JCl""e ~%%%# rr'c>||z|_dS)aSets the trailing metadata for the RPC. Sets the trailing metadata to be sent upon completion of the RPC. If this method is invoked multiple times throughout the lifetime of an RPC, the value supplied in the final invocation + request id will be the value sent over the wire. This method need not be called by implementations if they have no metadata to add to what the gRPC runtime will transmit. Args: trailing_metadata: The trailing :term:`metadata`. N)rCr()r+r's rset_trailing_metadataz)RayServegRPCContext.set_trailing_metadatas$#'";";"="=@Q"Qrc|jr||j|jr||j|jr||j|jr||jdSdS)z>Serve's internal method to set attributes on the gRPC context.N)rr;r)r>rr@r(rEr*s r_set_on_grpc_contextz(RayServegRPCContext._set_on_grpc_contexts : .  ! !$* - - -   <  ( (): ; ; ; = 4  $ $T] 3 3 3  " H  . .t/F G G G G G H HrN) __name__ __module__ __qualname____doc__grpc_cythoncygrpc_ServicerContextr,rstrrr StatusCoderrrrrr!rbytesr#r%r'r; Compressionr>r@rCrErGrrrr r s !T\%8%I ! ! ! !"d38n""""do)T%S/%:))))c %% % % % % '8C= ' ' ' ''4c3h#8'''' T_    (4+;((((  3     d5c?&;RtE#s(O7LRRRR" H1D1U H H H H H Hrr ) typingrrrrrrLray.util.annotationsrr rrrrVs33333333333333 ****** V`H`H`H`H`H`H`H`H`H`Hr