ڰj{$SrSSKrSSKJrJrJr SSKrSSKJrJ r SSK J r J r J r Jr "SS5r"SS 5r"S S 5rg) z# Runpod | Python | Endpoint Runner N)AnyDictOptional) HTTPAdapterRetry)API_KEY_NOT_SET_MSG FINAL_STATESUNAUTHORIZED_MSG is_completedc \rSrSrSrSS\\4SjjrSS\S\S\\S \ 4S jjr SS\S\S \ 4S jjr SS\S \ 4S jjr S r g) RunPodClientz$A client for running endpoint calls.Napi_keyc:SSKJnJn U=(d UUlURc[[5e[ R "5Ul[SSSS/S9nURRS [US 95 S S UR3S .Ul X0lg)z Initialize a RunPodClient instance. Args: api_key: Optional API key. If not provided, uses global api_key. Raises: RuntimeError: If the API key has not been initialized. r)rendpoint_url_baseNii)totalbackoff_factorstatus_forcelistzhttp://) max_retrieszapplication/jsonzBearer )z Content-Type Authorization) runpodrr RuntimeErrorrrequestsSession rp_sessionrmountrheaders)selfrglobal_api_keyrretriess S/home/mande/repo/quber/.venv/lib/python3.13/site-packages/runpod/endpoint/runner.py__init__RunPodClient.__init__s 0. << 23 3"**,aS#JO i)IJ/&t||n5  "3methodendpointdatatimeoutcURSU3nURRXURX4S9nURS:Xa[ [ 5eUR5 UR5$)a" Make a request to the specified endpoint using the given HTTP method. Args: method: The HTTP method to use ('GET' or 'POST'). endpoint: The endpoint path to which the request will be made. data: The JSON payload to send with the request. timeout: The number of seconds to wait for the server to send data before giving up. Returns: The JSON response from the server. Raises: RuntimeError: If the response returns a 401 Unauthorized status. requests.HTTPError: If the response contains an unsuccessful status code. /)rjsonr*i) rrrequestr status_coderr raise_for_statusr-)r r'r(r)r*urlresponses r#_requestRunPodClient._request9ss(''((4??** D+    3 &/0 0!!#}}r&c(URSXU5$)zPost to the endpoint.POSTr3)r r(r)r*s r#postRunPodClient.postXs}}VXW==r&c"URSXS9$)zGet from the endpoint.GETr*r7)r r(r*s r#getRunPodClient.get\s}}UH}>>r&)rrrrN)N )r@)__name__ __module__ __qualname____firstlineno____doc__rstrr$dictintr3r8r=__static_attributes__r&r#r r s{.3 3BBF%(08>>S>>s>?C?#??r&r c\rSrSrSrS\S\S\4SjrSS\S\\\ 44S jjr S r SS \ S\ 4S jjr S\ 4S jrSS \ S\ 4SjjrSrg)Jobdz1Represents a job to be run on the Runpod service. endpoint_idjob_idclientcDXlX lX0lSUlSUlg)z Initialize a Job instance with the given endpoint ID and job ID. Args: endpoint_id: The identifier for the endpoint. job_id: The identifier for the job. client: An instance of the RunPodClient to make requests with. N)rNrO rp_client job_status job_output)r rNrOrPs r#r$ Job.__init__gs"' r&sourcereturncURSUSUR3nURRUS9n[ US5(a!USUlURSS5UlU$)zBReturns the raw json of the status, raises an exception if invalidr,)r(statusoutputN)rNrOrRr=r rSrT)r rV status_url job_states r# _fetch_jobJob._fetch_jobwsn(()6(!DKK=A NN&& &;  (+ , ,'1DO'mmHd;DOr&cZURb UR$UR5S$)z&Returns the status of the job request.rY)rSr])r s r#rY Job.statuss( ?? &?? " **r&r*cTUS:aj[UR55(dL[R"S5 US-nUS::a [ S5e[UR55(dMLUR b UR $UR 5RSS5$)z Returns the output of the job request. Args: timeout: The number of seconds to wait for the server to send data before giving up. rrzJob timed out.NrZ)r rYtimesleep TimeoutErrorrTr]r=r r*s r#rZ Job.outputs Q;"4;;=11 1 1 a<&'788 #4;;=11 ?? &?? " $$Xt44r&c## [R"S5 URSS9nUS[;d[ US5S:a!UR S/5H nUSv M OUS[;agMu7f)z>Returns a generator that yields the output of the job request.rstream)rVrYrrZN)rbrcr]r lenr=)r stream_partialchunks r#rh Job.streamsx JJqM!__H_=Nx( <~h/014+//"=E/)>)\9sA7A9cjURRURSUR3SUS9$)z Cancels the job and returns the result of the cancellation request. Args: timeout: The number of seconds to wait for the server to respond before giving up. z/cancel/Nr)r*)rRr8rNrOres r#cancel Job.cancels?~~""  6#  r&)rNrOrTrSrRN)rY)r)rArBrCrDrErFr r$rrr]rYrHrZrhrorIrJr&r#rLrLdsv;Cl  DcN +5c5#5&    c  #   r&rLc \rSrSrSrSS\S\\4SjjrS\\\ 4S\ 4S jr SS\\\ 4S \ S\\\ 44S jjr SS \ S\\\ 44S jjrSS \ S\\\ 44S jjrSrg)Endpointz6Manages an endpoint to run jobs on the Runpod service.NrNrc,Xl[US9Ulg)a Initialize an Endpoint instance with the given endpoint ID. Args: endpoint_id: The identifier for the endpoint. api_key: Optional API key for this endpoint instance. Example: >>> endpoint = runpod.Endpoint("ENDPOINT_ID") >>> run_request = endpoint.run({"your_model_input_key": "your_model_input_value"}) >>> print(run_request.status()) >>> print(run_request.output()) )rN)rNr rR)r rNrs r#r$Endpoint.__init__s'%g6r& request_inputrWcURS5(dSU0nURRURS3U5n[ URUSUR5$)z Run the endpoint with the given input. Args: request_input: The input to pass into the endpoint. Returns: A Job instance for the run request. inputz/runid)r=rRr8rNrL)r rx job_requests r#run Endpoint.runsg  ))$m4Mnn))  %  4##[%6GGr&r*c&URS5(dSU0nURRURS3UUS9nUS[;aURSS5$[ URUSUR5R US9$)z Run the endpoint with the given input synchronously. Args: request_input: The input to pass into the endpoint. timeout: Maximum time to wait for the job to complete. rzz/runsyncr<rYrZNr{)r=rRr8rNr rLrZ)r rxr*r|s r#run_syncEndpoint.run_syncs  ))$m4Mnn))  ) * x L 0??8T2 2   k$/ && ! "r&cPURRURS3US9$)z Check the health of the endpoint (number/state of workers, number/state of requests). Args: timeout: The number of seconds to wait for the server to respond before giving up. z/healthr<)rRr=rNres r#healthEndpoint.healths4~~!!  ("  r&cRURRURS3SUS9$)z Purges the endpoint's job queue and returns the result of the purge request. Args: timeout: The number of seconds to wait for the server to respond before giving up. z /purge-queueNrn)rRr8rNres r# purge_queueEndpoint.purge_queues7~~"" -#  r&)rNrRr?)iQrq)rArBrCrDrErFrr$rrrLr}rHrrrrIrJr&r#rtrts@7C7(3-7"Hc3hHCH(=B"!#s(^"69" c3h"4  c  $sCx.    3  tCH~   r&rt)rErbtypingrrrrrequests.adaptersrrrunpod.endpoint.helpersrr r r r rLrtrJr&r#rsK &&0H?H?\Q Q nX X r&