ڰj%hSrSSKrSSKJrJrJr SSKJrJr SSK J r "SS5r "SS 5r g) z,Module for running endpoints asynchronously.N)AnyDictOptional) FINAL_STATES is_completed) ClientSessionc\rSrSrSrS\S\S\S\4SjrSS\S \ \\ 44S jjr S \4S jr S r SS \S \ 4SjjrS \ 4SjrS \4SjrSrg)Job z5Class representing a job for an asynchronous endpoint endpoint_idjob_idsessionheaderschSSKJn XlX lX0lXPlX@lSUlSUlg)z Initialize a Job instance. Args: endpoint_id: The identifier for the endpoint. job_id: The identifier for the job. session: The aiohttp ClientSession. headers: Headers to use for requests. r)endpoint_url_baseN)runpodrr r rr job_status job_output)selfr r rrrs c/home/mande/repo/quber/.venv/lib/python3.13/site-packages/runpod/endpoint/asyncio/asyncio_runner.py__init__ Job.__init__s2 '  !2 sourcereturncZ# URSURSUSUR3nURR X R S9IShvN nUR 5IShvN n[US5(a!USUlUR SS5Ul U$NRN<7f)z~Returns the raw json of the status, raises an exception if invalid. Args: source: The URL source path of the job status. /rNstatusoutput) rr r rgetrjsonrrr)rr status_url job_states r _fetch_jobJob._fetch_job&s%%&a(8(8'96(!DKK= Q ,,**:||*LL #..**  (+ , ,'1DO'mmHd;DOM*s$AB+B'B+,B)-;B+)B+cz# URb UR$UR5IShvN nUS$N 7f)zAGets jobs' status Returns: COMPLETED, FAILED or IN_PROGRESS Nr)rr%)rr$s rr Job.status8s: ?? &?? "//++ "",s -;9 ;c# [UR5IShvN 5(dG[R"S5IShvN [UR5IShvN 5(dMFggNUN0N7f)N)rrasynciosleep)rs r_wait_for_completionJob._wait_for_completionDsLT[[]233--" " "T[[]2332 " 3s9A7A1&A7A3A7A5 A7/A73A75A7timeoutcD# URb UR$[R"UR5U5IShvN UR 5IShvN nUR SS5$N.![Ran[ S5UeSnAff=fN@7f)zpWaits for serverless API job to complete or fail Returns: Output of job Raises: KeyError if job Failed NzJob timed out.r )rr+wait_forr- TimeoutErrorr%r!)rr/excjob_datas rr Job.outputHs ?? &?? " :""4#<#<#>H H H**||Hd++ I## :/0c 9 :+sFB (A6A4A6 B BB 4A66B BBB c:# [R"S5IShvN URSS9IShvN nUS[;d[ UR S/55S:a"UR S/5H nUS7v M OUS[;agMNyNd7f)z>Returns a generator that yields the output of the job request.r*Nstream)rrrr )r+r,r%rlenr!)rstream_partialchunks rr7 Job.stream[s--" " "#'??(?#CCNx( <~))(B781<+//"=E/)>)\9 "Cs BBBBA#BBcD# URSURSUR3nURR XR S9IShvN nUR 5IShvN sSSS5IShvN $N-NN !,IShvN (df  g=f7f)z=Cancels current job Returns: Output of cancel operation rz/cancel/rN)rr r rpostrr")r cancel_urlresps rcancel Job.cancelisw ../q1A1A0B(4;;-X <<$$Z$FF$$GFF$GFFFsZAB BB B*B+B. B :B;B BB B B BB )r rrr rrrN)r)r)__name__ __module__ __qualname____firstlineno____doc__strrdictrrrr%rr-intr r7r@__static_attributes__rrr r s~?C}W[.s$sCx.$ #c ##,C,,& c %d%rr ch\rSrSrSrSS\S\S\\4SjjrS\ S \ 4S jr S \ 4S jr S \ 4S jr S rg)EndpointtzClass for running endpointNr rapi_keycSSKJnJn XlX lXPlUSURS3UlU=(d UUlURc [ S5eSSUR3S .Ulg) z Initialize an async Endpoint instance. Args: endpoint_id: The identifier for the endpoint. session: The aiohttp ClientSession. api_key: Optional API key for this endpoint instance. r)rOrrz/runNz(API key must be provided or set globallyzapplication/jsonzBearer )z Content-Type Authorization)rrOrr r endpoint_url RuntimeErrorr)rr rrOglobal_api_keyrs rrEndpoint.__init__wsw ' !20143C3C2DDI0. << IJ J/&t||n5  rendpoint_inputrc# URRURURSU0S9IShvN nUR 5IShvN nSSS5IShvN URR 5nWSUS'[ URUSURU5$NsN]NO!,IShvN (df  Nd=f7f)zz Runs endpoint with specified input. Args: endpoint_input: any dictionary with input Returns: Newly created job input)rr"Nidz X-Request-ID)rr=rRrr"copyr r )rrVr? json_resp job_headerss rrun Endpoint.runs<<$$   t||7N:S%  "iik)I  ll'') &/o N#4##Yt_dllKPP *    sW6C B,C B2B.B2 C  B0!A C .B20C 2C 8B; 9 C C c,# URSURS3nURRXRS9IShvN nUR 5IShvN sSSS5IShvN $N-NN !,IShvN (df  g=f7f)z< Checks health of endpoint Returns: Health of endpoint rz/healthrN)rr rr!rr")r health_urlr?s rhealthEndpoint.healthso../q1A1A0B'J <<##J #EE$FEE$FEEEZABA4B A:A6A:" B.A8/B6A:8B:BB  B Bc,# URSURS3nURRXRS9IShvN nUR 5IShvN sSSS5IShvN $N-NN !,IShvN (df  g=f7f)z5 Purges queue of endpoint Returns: Purge status rz /purge-queuerN)rr rr=rr")r purge_urlr?s r purge_queueEndpoint.purge_queueso--.a0@0@/AN <<$$Y $EE$FEE$FEEErc)rOr rRrrr)N)rBrCrDrErFrGrrrrHr r]rarfrJrKrrrMrMtsW$+/ C - "3- >QQQ( %d % %4 %rrM) rFr+typingrrrrunpod.endpoint.helpersrrrunpod.http_clientrr rMrKrrrks22&&>,e%e%PL%L%r