AxjdZddlZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZe ZGd d ZdS) zx This module defines the Heartbeat class. The heartbeat is responsible for sending periodic pings to the Runpod server. N)Process)Retry)SyncClientSession) RunPodLogger) WORKER_ID) __version__cNeZdZdZdZd d dZeddZdd Zdd Z d Z dS) Heartbeatz&Sends heartbeats to the Runpod server.F returnNctjdd|_|jdt |_t tjdddz|_d|_t|_ |j j dtjd d it|gd d gd }tj|||}|j d||j d|dS)z2 Initializes the Heartbeat class. RUNPOD_WEBHOOK_PING PING_NOT_SETz$RUNPOD_POD_IDRUNPOD_PING_INTERVALi'iN AuthorizationRUNPOD_AI_API_KEY)iiiiiGET)totalstatus_forcelistallowed_methodsbackoff_factor)pool_connections pool_maxsize max_retrieszhttp://zhttps://)osenvirongetPING_URLreplacerint PING_INTERVAL_mirrorr_sessionheadersupdaterrequestsadapters HTTPAdaptermount)selfrretriesretry_strategyadapters RC:\Projects\Historical\venv\Lib\site-packages\runpod/serverless/modules/rp_ping.py__init__zHeartbeat.__init__s5 ' JJ  0F!N!NOOSWW *++  $$ bjnn-@"EE F   666"G    #//-)&0   Iw/// J00000cZt}||_||dS)z Static helper to run the ping loop in a separate process. Creates a new Heartbeat instance to avoid pickling issues. N)r r% ping_loop)mirrortesthbs r1 process_loopzHeartbeat.process_loop9s, [[  Tr3ctjdstddStjdstddS|jr |jdkrtddStj sFttj ||f}d |_ | d t_ dSdS) zS Sends heartbeat pings to the Runpod server in a separate process. rz:Not deployed on Runpod serverless, pings will not be sent.N RUNPOD_POD_IDz.Not running on Runpod, pings will not be sent.rz$Ping URL not set, cannot start ping.)targetargsT)rrr logdebuginfor!errorr _process_startedrr9daemonstart)r-r6r7processs r1 start_pingzHeartbeat.start_pingCsz~~122  IIR S S S Fz~~o..  HHE F F F F  $->"A"A II< = = = F) .Y%;64.QQQG!GN MMOOO)-I & & &  . .r3ch |tj|j|rdS2)zO Sends heartbeat pings to the Runpod server until interrupted. TN) _send_pingtimesleepr$)r-r7s r1r5zHeartbeat.ping_loopYs<  OO    Jt) * * *   r3c~|j|jnd}|td} |j|j||jdz}t d|jd|j dS#tj $r)}t d|dYd}~dSd}~wwxYw) z9 Sends a heartbeat to the Runpod server. N)job_idrunpod_version)paramstimeoutzHeartbeat Sent | URL: z | Status: zPing Request Error: z, attempting to restart ping.) r%r rMr&r!r$r>r?url status_coder)RequestExceptionrA)r-job_ids ping_paramsresulterrs r1rHzHeartbeat._send_pingcs)- (@$,""$$$d!(NKK  Q]&& k4;MPQ;Q'F IITTT@RTT     ( Q Q Q IIOSOOO P P P P P P P P P QsABB<B77B<)r r )r N)NF)F) __name__ __module__ __qualname____doc__rBr2 staticmethodr9rFr5rHr3r1r r s0011111@\....,QQQQQr3r )r[rrImultiprocessingrr)urllib3.util.retryrrunpod.http_clientr#runpod.serverless.modules.rp_loggerr&runpod.serverless.modules.worker_staterrunpod.versionrrMr>r r]r3r1rds  ######$$$$$$000000<<<<<<<<<<<<888888lnn_Q_Q_Q_Q_Q_Q_Q_Q_Q_Qr3