kQjt/dZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddlmZddlmZddlmZmZmZe d Zgd ZGd d eeZd ZdedefdZdededefdZGddeeZdddeegefdeefdZy)zIHandles to asynchronous registry jobs, SQL queries, and local operations.N) timedelta)AnyCallableGenericOptionalTypeVarcast)LOOP)_lancedb)JobDescriptionJobFailureInfoJobInfoT)AsyncJobJobr rrceZdZdZ ddeedeeegeffdZe dee fdZ de fdZ dd ee defd Zd Zdd Ze dee fd Ze dee fdZe deefdZe deefdZe deefdZe deefdZe dee fdZe dee fdZde deefdZddddeedee ddfdZde fdZy)rzA handle to an operation that may still be running. The operation may already be complete when the handle is created. ``T`` is the endpoint's terminal result type; unit-result jobs resolve to ``None``. Ninnerresult_decoderc ||_||_yN)_inner_result_decoder)selfrrs P/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/lancedb/job.py__init__zAsyncJob.__init__%s  -returncJ|j|jjSdS)aOThe operation's opaque ID, interpreted according to :attr:`job_type`. Registry jobs expose a registry job ID; SQL-backed jobs expose a connection-scoped query ID as a string. Local operations return ``None``. Query IDs cannot be used with registry job APIs or to resume after a client restart. Nridrs rr!z AsyncJob.id-s "&!8t{{~~BdBrclK|jy|jjd{S7w)a4The operation's current lifecycle state: "running", "finished", "failed", or "cancelled". A point snapshot; unlike `wait` it does not block or raise on a terminal failure state. States a newer server reports that this client version does not know pass through as-is. Nfinished)rstatusr"s rr%zAsyncJob.status8s. ;; [[''))))s +424timeoutcvK|jttdS|#|jjd{}nDt j |jj|j d{}|j|j|Stt|S7v73w)a.Wait until the operation reaches a terminal state. Returns the endpoint's typed result, or ``None`` for a unit-result job. Raises `JobFailedError` if the operation failed, `JobCancelledError` if it was cancelled, and `TimeoutError` if `timeout` elapses first. N)rr rwaitasynciowait_for total_secondsr)rr&results rr(z AsyncJob.waitDs ;; 4= ?;;++--F"++DKK,<,<,>@U@U@WXXF    +''/ /Av .Xs"t{{##HDHrc$|jdS)z:When the job was created, in milliseconds since the epoch. creation_ms_fieldr"s rr:zAsyncJob.creation_ms{{=))rc$|jdS)z:The job-type-specific specification it was submitted with.specr;r"s rr?z AsyncJob.spec{{6""rc$|jdS)aThe job-type-specific terminal result, as reported data rather than the typed model :meth:`AsyncJob.wait` returns. `None` until the job succeeds, so a job that never terminates reports its progress through :meth:`AsyncJob.events` instead. r,r;r"s rr,zAsyncJob.results{{8$$rc$|jdS)zCWhy the job failed, when it failed and the server reports a reason.failurer;r"s rrCzAsyncJob.failure{{9%%rc$|jdSN _spec_jsonr;r"s rrGzAsyncJob._spec_json{{<((rc$|jdSN _result_jsonr;r"s rrKzAsyncJob._result_json{{>**rnamecj|j|jjnd}| t||SdSr)r _descriptiongetattr)rrM descriptions rr<zAsyncJob._fields326++2Idkk..t -8-Dw{D)N$NrlimitfilterrSrTpa.TablecK|j td|jj||d{S7w)aThis job's recorded lifecycle events. Where the properties above report a terminal result only once the job reaches one, events are written as the job runs and outlive the workers that produced them. A distributed job records a `claim`/`claim_complete` pair per unit of work, each carrying `rows_processed`, so a job that never finishes still accounts for what it did. Parameters ---------- limit: int, optional Maximum event rows to return. The server caps results at 1000 by default and 10,000 at most, and truncates without saying so, so pass this for a job that emits an event per fragment. filter: str, optional SQL-like expression over the `state`, `updated_by`, `emitted_from`, `emitted_by`, and `claim_entity` columns, such as ``state = 'claim_complete'``. N8job event history is only available for server-side jobsrR)rNotImplementedErroreventsrrSrTs rrYzAsyncJob.eventssD2 ;; %J [[''eF'CCCCs8A?Actd|S)Nr _job_reprr"s r__repr__zAsyncJob.__repr__sT**rrrN)__name__ __module__ __qualname____doc__rrrrrpropertystrr!r%rr(r/r2r5r8intr:r?r,rrCrGrKr<rYr^rrrrs8<.}.!3%(!34.CHSMCC *c *(9"5&# $"x}""I(3-II*Xc]**#hsm##% %%&.1&&)HSM))+hsm++O3O8C=O $ $ D}D D  D>+#+rrz valuerc tj|d}|j ddt zS#t$rt|cYSwxYw)z?Render a job payload as indented JSON, aligned under its field.)indent )jsondumps TypeErrorreprreplace _REPR_INDENT)rhrendereds r _repr_payloadrtsL::eA.   D$"5 66 E{s2A A kindjobc@|j}|&|jd|jdnd}|d|dSg}|j|jd|j|jd|dD]'}t||}||j|d|)d D]0}t||}||j|dt |2|j |jd |j dj d |D}|d|d S) zRender every field the handle currently knows, omitting the rest. One field per line, with the JSON payloads indented, because a refresh job's spec and result are the point of printing it. zid=z, (znot refreshed)zstate=)r8r:=)r?r,zfailure=c34K|]}dt|dyw)rl,N)rr).0fields r z_job_repr..sC5R ~eWA.Csz ))r5r!appendrPrtrCjoin)rurvr5knownfieldsrMrhbodys rr]r]s= IIE }&)ff&8#cffZr"bq~.. F vv CFF:&' MMF5)$%+/T"   MMTF!E9- ./#<T"   MMTF!M%$8#9: ;< {{ 01 77CFC CDV1TF# rceZdZdZdeeefdZedee fdZ de fdZ ddee defd Z d Zdd Zedee fd Zedee fd ZedeefdZedeefdZedeefdZedeefdZedee fdZedee fdZde deefdZddddeedee ddfdZde fdZy)rz@Synchronous counterpart of `AsyncJob` with the same result type.rc||_yr)r)rrs rrz Job.__init__s  rrcJ|j|jjSdS)zeIdentifies the operation on the server that is running it. See :attr:`AsyncJob.id`. Nr r"s rr!zJob.ids "&!8t{{~~BdBrcv|jytj|jjS)zThe operation's current lifecycle state: "running", "finished", "failed", or "cancelled". See :meth:`AsyncJob.status`. r$)rr runr%r"s rr%z Job.statuss- ;; xx **,--rNr&c|jttdStj|jj |S)a/Block until the operation reaches a terminal state. Returns the endpoint's typed result, or ``None`` for a unit-result job. Raises `JobFailedError` if the operation failed, `JobCancelledError` if it was cancelled, and `TimeoutError` if `timeout` elapses first. N)rr rr rr()rr&s rr(zJob.wait s8 ;; 4= xx ((122rcx|jytj|jjyr.)rr rr/r"s rr/z Job.cancels( ;;   ##%&rcx|jytj|jjy)zhAsk the backend for this job's current state and record. See :meth:`AsyncJob.refresh`. N)rr rr2r"s rr2z Job.refreshs* ;;   $$&'rcJ|j|jjSdS)z>The last observed lifecycle state. See :attr:`AsyncJob.state`.r$)rr5r"s rr5z Job.state%s"%)KK$;t{{  KKrcJ|j|jjSdS)z.The job's type. See :attr:`AsyncJob.job_type`.Nr7r"s rr8z Job.job_type*s"(,{{'>t{{##HDHrc$|jdS)z;When the job was created. See :attr:`AsyncJob.creation_ms`.r:r;r"s rr:zJob.creation_ms/r=rc$|jdS)z3The job's specification. See :attr:`AsyncJob.spec`.r?r;r"s rr?zJob.spec4r@rc$|jdS)z7The job's terminal result. See :attr:`AsyncJob.result`.r,r;r"s rr,z Job.result9s{{8$$rc$|jdS)z1Why the job failed. See :attr:`AsyncJob.failure`.rCr;r"s rrCz Job.failure>rDrc$|jdSrFr;r"s rrGzJob._spec_jsonCrHrc$|jdSrJr;r"s rrKzJob._result_jsonGrLrrMcJ|jt|j|SdSr)rrP)rrMs rr<z Job._fieldKs!-1[[-Dwt{{D)N$NrrRrSrTrUc|j tdtj|jj ||S)zTThis job's recorded lifecycle events. See :meth:`AsyncJob.events`. rWrR)rrXr rrYrZs rrYz Job.eventsNsC ;; %J xx **v*FGGrctd|S)Nrr\r"s rr^z Job.__repr__^s%%rrr_)r`rarbrcrrrrrdrer!r%rr(r/r2r5r8rfr:rr?r,rrCrGrKr<rYr^rgrrrrsJhx{3CHSMCC.. 3HY/ 31 3' (Lx}LLI(3-II*Xc]**#hsm##% %%&.1&&)HSM))+hsm++O3O8C=O $ $ H}H H  H &#&rrrz _lancedb.Jobrct||S)z?Bind an internal JSON-producing job to its public result model.)r)rrs r _typed_jobrbs E> **r)rcr)rmdatetimertypingrrrrrr pyarrowpalancedb.background_loopr rxr r rrr__all__rrrrertr]rrrgrrrsP BB(== CL g+wqzg+T 777Ccc