%`i( ddlZddlmZddlmZmZmZmZddlZddlm Z ddl Z ddl m Z ddlmZ dded eeeeefd ed eejfd Z ddeeded eeeeefd efdZdS)N)futures)AnyOptionalSequenceTuple)aio)authentication_utils)load_certs_from_envFaddressoptions asynchronous credentialscN|rtnt}|pg}t|}|dtj|d<|dtj|d<|}g}tj rEddl m }m }|r||n||| |j} ||f} nst"jdddvr5t)\} } } tj| | | }|j} ||f} n |j} |f} |r | | ||d }n| | d |i}|rtj|g|R}|S) a#Create a gRPC channel with authentication interceptors if token auth is enabled. This function handles: - TLS configuration via RAY_USE_TLS environment variable or custom credentials - Authentication interceptors when token auth is enabled - Keepalive settings from Ray config - Both synchronous and asynchronous channels Args: address: The gRPC server address (host:port) options: Optional gRPC channel options as sequence of (key, value) tuples asynchronous: If True, create async channel; otherwise sync credentials: Optional custom gRPC credentials for TLS. If provided, takes precedence over RAY_USE_TLS environment variable. Returns: grpc.Channel or grpc.aio.Channel: Configured gRPC channel with interceptors zgrpc.keepalive_time_mszgrpc.keepalive_timeout_msr)+SyncAuthenticationMetadataClientInterceptorget_async_auth_interceptorsN RAY_USE_TLS0)1true)certificate_chain private_keyroot_certificates)r interceptorsr )aiogrpcgrpcdictgetray_configgrpc_client_keepalive_time_ms grpc_client_keepalive_timeout_msitemsr is_token_auth_enabledBray._private.authentication.grpc_authentication_client_interceptorrrextendappendsecure_channelosenvironlowerr ssl_channel_credentialsinsecure_channelintercept_channel)r r r r grpc_module options_dictrrrchannel_creator base_argsserver_cert_chainrca_certtls_credentialschannels k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/grpc_utils.pyinit_grpc_channelr7 s%0*3''tKmG==L-9-=-= #+"K"K"M"M..L)*1=0@0@#S[%Q%Q%S%S11L,-  ""GL133 O         O    ; ; = = > > > >    K K M M N N N%4k*  s + + 1 1 3 3} D D2E2G2G/;6/#%    &4o. &6J E!/ l   "/9>g>>  E,WD|DDDG N grpc_server max_workersthread_name_prefixc|rtnt}g}tjrI|r$ddlm}||n#ddlm}|||r||r|nd|}nC|tdtj ||} || |r|nd|}|S)aCreate a gRPC server with authentication interceptors if token auth is enabled. This function handles: - Authentication interceptors when token auth is enabled - Both synchronous and asynchronous servers - Thread pool configuration for sync servers Args: max_workers: Max thread pool workers (required for sync, ignored for async) thread_name_prefix: Thread name prefix for sync thread pool options: Optional gRPC server options as sequence of (key, value) tuples asynchronous: If True, create async server; otherwise sync Returns: grpc.Server or grpc.aio.Server: Configured gRPC server with interceptors r)$AsyncAuthenticationServerInterceptor)#SyncAuthenticationServerInterceptorN)rr z4max_workers is required for synchronous gRPC servers)r:r;) rrr r#Bray._private.authentication.grpc_authentication_server_interceptorr=r&r>server ValueErrorrThreadPoolExecutor) r:r;r r r.rr=r>r@executors r6$create_grpc_server_with_interceptorsrD_sQ,*3''tKL133 G  G          D D F F G G G G          C C E E F F F ##)5?4$    STT T-#1   ## )5?4$   Mr8)NFN)Nr9NF)r( concurrentrtypingrrrrrrrrray._private.authenticationr ray._private.tls_utilsr strboolChannelCredentialsr7intrDr8r6rNsc 111111111111  <<<<<<666666 4859 OO O huS#X/ 0OO$12 OOOOf"&+37 <<#<<huS#X/ 0< <<<<<<r8