o 6Wj+@s<ddlmZddlZddlZddlZddlmZddlmZm Z ddl Z ddl m Z m Z mZmZddlmZmZdd lmZdd lmZdd lmZmZgd Zd ZdZeeZdAddZ dBddZ!dCddZ"dDdd Z#dEd"d#Z$dedd$d%dd&dFd4d5Z%dd6dGd9d:Z&dedd$d%dd&dHd=d>Z'dd6dId?d@Z(dS)J) annotationsN)uuid4)Iterator AsyncIterator)TRANSIENT_ERRORSjitterbackoffis_fatal_status_error)Headersomit)APIStatusError)BetaSelfHostedWork)Work AsyncWork) iter_work aiter_work POLL_BLOCK_MSigN@attemptintreturnfloatcCs t|tdS)N)cap)r _POLL_BACKOFF_CAP)rrq/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/anthropic/lib/environments/_poller.py_backoffs rlowhighcCs t||SN)r)rrrrr_jitter"s r!err ExceptionboolcCst|Sr )r )r"rrr _is_fatal_4xx&sr%codecCst|to |j|kSr ) isinstancer status_code)r"r&rrr _is_status*sr)strcCstdtjddS)N- )socket gethostnamerhexrrrr_default_worker_id.sr0FT) worker_idblock_msreclaim_older_than_msdrain auto_stop extra_headersworkrenvironment_idr1 str | Noner2 int | Noner3r4r5r6Headers | NoneIterator[BetaSelfHostedWork]c cs|pt}td|||d}d} z|j||dur|nt|dur$|nt||d} Wn7tyb} z+t| r>td| |d7}t|t dd } t d || | t | WYd} ~ qd} ~ wwd}| dur||rstd |dSt t d d qtd | j t| jddz |j| j ||dWnIty} z=t| rtd| j | t|| j ||dWYd} ~ q| d7} t| t dd } t d| j | | | t | WYd} ~ qd} ~ wwd} |s| Vqz1| VWz |j| j ||dWnTty} zt| ds t d| j | WYd} ~ n8d} ~ wwz |j| j ||dWwtyJ} zt| ds@t d| j | WYd} ~ wWYd} ~ wd} ~ wwq)u Iterate work items claimed from a self-hosted environment. Each yielded :class:`BetaSelfHostedWork` has already been ack'd. The ``work`` resource must be bound to a client authenticated for the environment — the poller itself does not handle credentials. Use ``client.beta.environments.work.poller(...)`` for the user-facing entry point that constructs a scoped sub-client for you. Two consumption shapes are supported: - **Long-running runner** (``drain=False, auto_stop=True``, the default): loops forever, sleeps with jitter on empty polls, and calls ``work.stop`` when the consuming for-loop body returns or raises. The poller owns the whole work-item lifecycle. - **Drain-and-dispatch** (``drain=True, auto_stop=False``): returns as soon as the queue is empty and never calls ``work.stop`` — use this when each yielded item is handed off to another process (e.g. a webhook handler that spawns a sandbox per work item) and that process owns ``stop``. Args: block_ms: How long the server holds an empty poll open (long-poll). Pass ``None`` to omit the param for a non-blocking poll — the server rejects ``0``. Drain callers usually want ``None`` so the final empty poll returns immediately. drain: When True, return after the first empty poll instead of sleeping and re-polling. Lets a webhook-driven dispatcher drain the queue and respond. auto_stop: When True (default), call ``work.stop`` after the consumer's loop body completes. Set False when the work item is handed off to another process that owns the stop call — otherwise the lease is terminated out from under it. reclaim_older_than_ms: Forwarded to ``work.poll``. Reclaim un-ack'd work older than this many ms. Useful in drain mode so a dead runner's work re-surfaces on the next webhook delivery. extra_headers: Optional headers passed through per request on every poll / ack / stop call (including the force-stop of an unprocessable item). They are threaded into each call's ``extra_headers=`` and are never assigned onto the client, so client state is not mutated. Credentials and ``x-stainless-helper`` come from the bound client, not this argument; a header given here overrides the bound client's same-named default for that one request, so use it for caller passthrough (e.g. trace ids), not to set auth. 7poller starting environment_id=%s drain=%s auto_stop=%srTNr2r3anthropic_worker_idr6 poll failed permanently error=%s?-poll failed attempt=%d backoff=%.1fs error=%squeue drained environment_id=%s@$claimed work work_id=%s work_type=%styper8r6:ack failed permanently; force-stopping work_id=%s error=%sDack failed, backing off work_id=%s attempt=%d backoff=%.1fs error=%sstop failed work_id=%s error=%s)r0loginfopollr rr%errorrr!warningtimesleepidgetattrdataack_force_stop_quietlystopr#r) r7r8r1r2r3r4r5r6 poll_attempt ack_attemptitemewaitrrrr5s 6                 r)r6work_idNonec Csfz |j||d|dWdSty2}zt|ds'td||WYd}~dSWYd}~dSd}~ww)zBest-effort ``work.stop(force=True)`` for an item that can't be processed. A 409 just means the work already stopped; anything else is logged but not raised, since the poll loop must keep going regardless. Tr8forcer6rL;force-stop of unprocessable work failed work_id=%s error=%sNrZr#r)rNrQr7rar8r6r_rrrrYs rYr!AsyncIterator[BetaSelfHostedWork]c Cs|pt}td|||d}d} z|j||dur|nt|dur$|nt||dIdH} Wn:tyh} z.t| rAtd| |d7}t|t dd } t d || | t | IdHWYd} ~ qd} ~ wwd}| dur|rytd |dSt t d d IdHqtd | j t| jddz|j| j ||dIdHWnOty} zCt| rtd| j | t|| j ||dIdHWYd} ~ q| d7} t| t dd } t d| j | | | t | IdHWYd} ~ qd} ~ wwd} |s| Vqz4| VWz|j| j ||dIdHWnWty,} zt| ds"t d| j | WYd} ~ n;d} ~ wwz|j| j ||dIdHWwtyb} zt| dsXt d| j | WYd} ~ wWYd} ~ wd} ~ wwq)zAsync version of :func:`iter_work`. See its docstring for semantics, including how ``extra_headers`` is passed through per request. r=rTNr>r@rArBrCrDrErFrGrHrIrJrKrLrM)r0rNrOrPr rr%rQrr!rRanyiorTrUrVrWrX_aforce_stop_quietlyrZr#r)r[rrrrs            rc snz|j||d|dIdHWdSty6}zt|ds+td||WYd}~dSWYd}~dSd}~ww)z-Async version of :func:`_force_stop_quietly`.TrcNrLrerfrgrrrrj s rj)rrrr)rrrrrr)r"r#rr$)r"r#r&rrr$)rr*)r7rr8r*r1r9r2r:r3r:r4r$r5r$r6r;rr<) r7rrar*r8r*r6r;rrb)r7rr8r*r1r9r2r:r3r:r4r$r5r$r6r;rrh) r7rrar*r8r*r6r;rrb)) __future__rrSr-logginguuidrcollections.abcrrri_retryrrr r _typesr r _exceptionsrtypes.beta.environmentsr resources.beta.environments.workrr__all__rr getLogger__name__rNrr!r%r)r0rrYrrjrrrrsL           {Q