&`iQdZddlZddlZddlZddlZddlZddlmZddlZddl m cm cm Z ddl m cm cmZddlmZerddlmZejeZde_GddZdS) zaThis file implements a threaded stream controller to return logs back from the ray clientserver. N) TYPE_CHECKING)log_once)WorkerFceZdZdddefdZdejfdZddZd e j de fd Z d e d efd Zd e d efdZd e fdZddZddZdS)LogstreamClient client_workerrmetadatac||_||_tj|_||_|jd|_dS)zInitializes a thread-safe log stream over a Ray Client gRPC channel. Args: client_worker: The Ray Client worker that manages this client metadata: metadata to pass to gRPC requests N) r _metadataqueueQueue request_queue_start_logthread log_threadstartlast_req)selfrr s n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/client/logsclient.py__init__zLogstreamClient.__init__sS+!"[]]//11  returnc:tj|jddS)NT)targetargsdaemon) threadingThread _log_mainrs rrz LogstreamClient._start_logthread*st~BtLLLLrNcd}|jjsU|r>tj|_|jr|j|jtj|jj } | t|jj d|j }n$#t$rtjdYwxYw |D]O}|jdkr!||j|j||j|jPdS#t*j$r'}||}|sYd}~dSYd}~nd}~wwxYw|jjSdSdS)NF)r ?rlevelmsg)r _in_shutdownr r rrputray_client_pb2_grpcRayletLogStreamerStubchannel Logstreamitergetr ValueErrortimesleepr$ stdstreamr%loggrpcRpcError_process_rpc_error)r reconnectingstub log_streamrecordes rrzLogstreamClient._log_main-s $1  :%*[]]"=:&**4=999&t||dS)zLog the message from the log stream. By default, calls logger.log but this can be overridden. Args: level: The loglevel of the received log message msg: The content of the message r#N)r>r2)rr$r%s rr2zLogstreamClient.logas   C (((((rcf|dkr tjn tj}t||ddS)zLog the stdout/stderr entry from the log stream. By default, calls print but this can be overridden. Args: level: The loglevel of the received log message msg: The content of the message )fileendN)sysstderrstdoutprint)rr$r% print_files rr1zLogstreamClient.stdstreamks4$)B;;SZZCJ  c ++++++rct|tj}d|_||_|j|||_dS)NT) r>setLevelray_client_pb2LogSettingsRequestenabledloglevelrr'r)rr$reqs rset_logstream_levelz#LogstreamClient.set_logstream_levelvsR/11   s### rc~|jd|j|jdSdS)N)rr'rjoinr s rclosezLogstreamClient.close~sC t$$$ ? & O " " " " " ' &rc|tj}d|_|j|||_dS)NF)rOrPrQrr'r)rrSs r disable_logszLogstreamClient.disable_logss9/11  s### r)rN)__name__ __module__ __qualname__listrrrrrr3r4boolr5intstrr2r1rTrWrYrrrrrs h $    M)"2MMMM>DMd*))3)))) ,s , , , , ,#### rr)__doc__loggingr rHrr/typingrr3!ray.core.generated.ray_client_pb2core generatedrO&ray.core.generated.ray_client_pb2_grpcr(ray.util.debugrray.util.client.workerr getLoggerrZr> propagaterrrrrls(  ::::::::::::DDDDDDDDDDDD######.------  8 $ $llllllllllr