tni dZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z eejd d Zed e Zeejd d Zed e Ze Zde fdZ dde fdZddZdZdS)z2 This module is used to handle HTTP requests. N) ClientError)FibonacciRetry RetryClient) ClientSession) RunPodLogger) WORKER_IDRUNPOD_WEBHOOK_POST_OUTPUT JOB_DONE_URLz$RUNPOD_POD_IDRUNPOD_WEBHOOK_POST_STREAMJOB_STREAM_URLclient_sessioncKtd}t||}|ddddd}|j|fi|4d {V}|d {Vd d d d {Vd S#1d {VswxYwYd S) z4 Wrapper for transmitting results via POST. )attempts)r retry_optionszutf-8z!application/x-www-form-urlencoded)charsetz Content-TypeT)dataheadersraise_for_statusN)rrposttext)rurljob_datar retry_clientkwargsclient_responses ^/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/runpod/serverless/modules/rp_http.py _transmitrsS#A...M%]L ?  ! F!| ////%%%%%%%?""$$$$$$$$$%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%sA// A9<A9Fsessionc:K |d|jd<tj|d}|rdnd}|d|dd|z}t |||d {Vt ||dnx#t$r.}t d ||dYd }~nEd }~wttf$r.}t d ||dYd }~nd }~wwxYw|tkr=| d d d kr%t d|dd Sd Sd S#|tkr<| d d d kr#t d|dwwwxYw)zR A helper function to handle the result, either for sending or streaming. idz X-Request-IDF) ensure_asciitruefalsez$IDz &isStream=Nz Failed to return job results. | z$Error while returning job result. | status IN_PROGRESSz Finished.)rjsondumpsreplacerlogdebugrerror TypeError RuntimeErrorr getinfo) r rjob url_template log_message is_streamserialized_job_datarerrs r_handle_resultr83s -*-d)'"jFFF'4FFW ""5#d)447OI7O7OO&9::::::::: ["CI.... GGG :S::CIFFFFFFFF | $KKK >>>D JJJJJJJJK L ( ( Xt,, == HH[#d) , , , , , ) (== L ( ( Xt,, == HH[#d) , , , , )=sCBB E D$B?:E?D$C<7E<DEA FcJKt|||td|d{VdS)z! Return the job results. z Results sent.)r5N)r8r )r rr2r5s r send_resultr:SsT 3 o   cFKt|||tdd{VdS)z( Return the stream job results. zIntermediate results sent.N)r8r )r rr2s r stream_resultr=\sN 30L  r;)F)__doc__r(osaiohttpr aiohttp_retryrrrunpod.http_clientr#runpod.serverless.modules.rp_loggerr worker_stater strenvironr0JOB_DONE_URL_TEMPLATEr*r JOB_STREAM_URL_TEMPLATEr r+rr8r:r=r;rrJs| 55555555,,,,,,<<<<<<######JNN/@@%,,-=yII #JNN/1ABB)001A9MMlnn%M%%%%.QV-- ----@r;