ڰjObSrSSKrSSKrSSKrSSKrSSKJrJrJrJ r SSK J r \ "5r \R"5r\RR!S\"\R$"555rSr\RR!SS5SLr"S S 5r"S S \ \5r"S S5rg)z] Handles getting stuff from environment variables and updating the global state like job id. N)AnyDictOptionalSet) RunPodLogger RUNPOD_POD_IDiRUNPOD_WEBHOOK_GET_JOBc ~\rSrSrSrSS\S\\\\4S\\SS4Sjjr S \ S\ 4S jr S\ 4S jrS\4S jrS rg)Jobz Represents a job object. Args: job_id: The id of the job, a unique string. job_input: The input to the job. webhook: The webhook to send the job output to. Nidinputwebhookreturnc rXlX lX0lUR5HupV[ XU5 M gN)rrritemssetattr)selfrrrkwargskeyvalues c/home/mande/repo/quber/.venv/lib/python3.13/site-packages/runpod/serverless/modules/worker_state.py__init__ Job.__init__&s/   ,,.JC Du %)otherc`[U[5(aURUR:H$g)NF) isinstancer r)rrs r__eq__ Job.__eq__4s% eS ! !77ehh& &rc,[UR5$r)hashrrs r__hash__ Job.__hash__9sDGG}rcUR$rrr%s r__str__ Job.__str__<s wwr)rrr)NN)__name__ __module__ __qualname____firstlineno____doc__strrrrrobjectboolr!intr&r*__static_attributes__rrr r sv+/!% &  &S#X' &# &  &Ft #rr c^\rSrSrSrSrSrSrS\4Sjr SSjr SS jr SU4S jjr S \ 4U4S jjrS \ 4U4S jjrS \ S\\4SjrS\\4SjrS\4SjrSrU=r$) JobsProgressCzETrack the state of current jobs in progress (in-memory, per process).Nc[RcV[RU5[l[R [R5 S[Rl[R$r)r8 _instanceset__new__r_mirror)clss rr=JobsProgress.__new__HsM  ! ! )%([[%5L " LL// 0.2L " " *%%%rcgrr6r%s rrJobsProgress.__init__Qs rrcVSURRSUR53$)N: ) __class__r, get_job_listr%s r__repr__JobsProgress.__repr__Us*4>>**+3t/@/@/B.CDDrc0XlUR5 g)zAttach a PingJobMirror that mirrors the in-progress job ids to the ping process. Every add/remove/clear then pushes the snapshot to it.N)r>_notify_mirror)rmirrors r set_mirrorJobsProgress.set_mirrorXs  rcrURb*URRUR55 gg)z@Push the current job-id snapshot to the attached mirror, if any.N)r>r<rFr%s rrJJobsProgress._notify_mirror^s, << # LL  T..0 1 $rcB>[TU]5 UR5 gr)superclearrJ)rrEs rrRJobsProgress.clearcs   relementc>[U[5(a [US9n[U[5(a [S0UD6n[U[5(d [ S5e[ TU]U5nUR5 U$)z Adds a Job object to the set. If the added element is a string, then `Job(id=element)` is added. If the added element is a dict, then `Job(**element)` is added. r)z.Only Job objects can be added to JobsProgress.r6)r r1r dict TypeErrorrQaddrJrrTresultrEs rrXJobsProgress.addgsk gs # #WoG gt $ $nGnG'3''LM MW%  rc>[U[5(a [US9n[U[5(a [S0UD6n[U[5(d [ S5e[ TU]U5nUR5 U$)z Removes a Job object from the set. If the element is a string, then `Job(id=element)` is removed. If the element is a dict, then `Job(**element)` is removed. r)z2Only Job objects can be removed from JobsProgress.r6)r r1r rVrWrQdiscardrJrYs rremoveJobsProgress.remove{sk gs # #WoG gt $ $nGnG'3''PQ Q)  rc[U[5(a [US9n[U[5(d [S5eUH nX!:XdM Us $ g)Nr)z4Only Job objects can be retrieved from JobsProgress.)r r1r rW)rrTjobs rgetJobsProgress.getsJ gs # #WoG'3''RS SC~ rcT[U5(dgSRSU55$)zL Returns the list of job IDs as a comma-separated string, or None if empty. N,c38# UHn[U5v M g7fr)r1).0ras r ,JobsProgress.get_job_list..s1DSCDs)lenjoinr%s rrFJobsProgress.get_job_lists%4yyxx1D111rc[U5$)z Returns the number of jobs. )rjr%s r get_job_countJobsProgress.get_job_counts4yr)r>)rN)r,r-r.r/r0r;r=rr1rGrLrJrRrrXr^rr rbrFr4rnr5 __classcell__)rEs@rr8r8CsOI& E#E 2 3(c( 3 8C= 2hsm2srr8cZ\rSrSrSr\S4S\4SjjrS\\ SS4Sjr S\\ 4S jr S r g) PingJobMirrora One-way snapshot of in-progress job ids from the worker (main) process to the separate ping process. Backed by a fixed-size shared-memory buffer created in the main process and passed to the ping process via ``Process(args=...)``. It lives only in this worker's own process tree, so it cannot be shared across workers and never touches the filesystem. All operations are best-effort and never raise into the caller (a failure here must not break job processing or kill the ping). Ncapacityc\U=(d [nXlURSU5Ulg)Nc)multiprocessing _capacityArray_buffer)rrtctxs rrPingJobMirror.__init__s#$_!yyh/ rjob_idsrcU=(d SRS5nURS- n[U5U:a:USUnURS5nUS:waUSUn[R SUS35 UR R5 X R lSSS5 g!,(df  g=f![a"n[RS U35 SnAgSnAff=f) z=Write the current job-id snapshot. Best-effort; never raises.utf-8rN,z(PingJobMirror: job-id snapshot exceeded z bytes; truncatedzPingJobMirror.set failed: ) encoderxrjrfindlogwarnrzget_lockr Exceptionerror)rr}datalimitcuterrs rr<PingJobMirror.sets :Mr))'2DNNQ&E4y5 FU|jj&"9:D>ugEVW&&(%) ")(( : II23%8 9 9 :s6B B9B(B9( B62B96B99 C%C  C%c.URR5 URRnSSS5 WRS5nU=(d S$!,(df  N*=f![a"n[ R SU35 SnAgSnAff=f)zrs ++#n""$ JJNN?C ,= >  7>$F !!Ne3s8eV,,r