jdZddlmZddlmZmZmZmZddlm Z ddl m Z m Z ddl m Z mZmZddlZddlmZmZmZe rdd lmZdd lmZdd lmZmZe Gd d Ze GddZe GddZdZdZ GddZ!GddZ"e GddZ#GddZ$GddZ%dS)z#Data models for the sandbox client.) annotations) AsyncIterator AwaitableCallableIterator) dataclass)datetimetimezone) TYPE_CHECKINGAnyOptionalN)SandboxConnectionErrorSandboxOperationErrorSandboxServerReloadError) AsyncSandbox)Sandbox)_AsyncWSStreamControl_WSStreamControlcJeZdZUdZded<ded<ded<ed d Zd S) ExecutionResultz+Result of executing a command in a sandbox.strstdoutstderrint exit_codereturnboolc|jdkS)z.Return True if the command exited with code 0.r)rselfs c/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langsmith/sandbox/_models.pysuccesszExecutionResult.success#s~""Nrr)__name__ __module__ __qualname____doc____annotations__propertyr"r#r!rrsU55KKKKKKNNN ###X###r#rcDeZdZUdZded<dZded<ed d ZdS) ResourceStatusaLightweight provisioning status for any async-created resource. Attributes: status: Resource lifecycle status. One of "provisioning", "ready", "failed". status_message: Human-readable details when status is "failed", None otherwise. rstatusN Optional[str]status_messagedatadict[str, Any]rcj||dd|dS)z/Create a ResourceStatus from API response dict.r. provisioningr0)r.r0getclsr1s r! from_dictzResourceStatus.from_dict5s>s88Hn5588$455    r#)r1r2rr-)r%r&r'r(r)r0 classmethodr9r+r#r!r-r-)sZKKK$(N((((   [   r#r-ceZdZUdZded<ded<ded<ded<dZd ed <dZd ed <dZd ed <dZd ed <dZ ded<dZ d ed<dZ d ed<dZ d ed<dZ d ed<eddZdS)SnapshotaRepresents a sandbox snapshot. Snapshots are built from Docker images or captured from running sandboxes. They are used to create new sandboxes. Attributes: id: Unique identifier (UUID). name: Display name. status: Build status. One of "building", "ready", "failed". fs_capacity_bytes: Filesystem capacity in bytes. docker_image: Source Docker image (for build snapshots). image_digest: Docker image digest after pull. source_sandbox_id: Source sandbox (for capture snapshots). status_message: Human-readable details when status is "failed". fs_used_bytes: Actual bytes used on the filesystem. created_by: User or service that created the snapshot. registry_id: Private registry ID, if applicable. created_at: Timestamp when the snapshot was created. updated_at: Timestamp when the snapshot was last updated. ridnamer.rfs_capacity_bytesNr/ docker_image image_digestsource_sandbox_idr0 Optional[int] fs_used_bytes created_by registry_id created_at updated_atr1r2rc(||dd|dd|dd|dd|d|d |d |d |d |d |d|d|d S)z)Create a Snapshot from API response dict.r=r>r.buildingr?rr@rArBr0rDrErFrGrH) r=r>r.r?r@rArBr0rDrErFrGrHr5r7s r!r9zSnapshot.from_dictcssxxb!!&"%%88Hj11"hh':A>>.11.11"hh':;;88$455((?33xx --//xx --xx --    r#)r1r2rr<)r%r&r'r(r)r@rArBr0rDrErFrGrHr:r9r+r#r!r<r<>s* GGG IIIKKK"&L&&&&"&L&&&&'+++++$(N((((#'M'''' $J$$$$!%K%%%% $J$$$$ $J$$$$   [   r#r<z!X-Langsmith-Sandbox-Service-TokenceZdZdZddd%d Zd&dZd'dZed(dZed(dZ ed(dZ ed(dZ d)d*dZ d)d+dZ d)d+dZd)d+dZd)d+dZd)d+d Zeddd,d#Zd(d$ZdS)- ServiceURLarAuthenticated URL for accessing an HTTP service running in a sandbox. Properties auto-refresh the token transparently when it nears expiry. HTTP helper methods (``.get``, ``.post``, etc.) inject the auth header automatically. When constructed by :meth:`SandboxClient.service` or :meth:`Sandbox.service`, the object holds an internal refresher that re-calls the API to obtain a fresh token before the current one expires. Example:: svc = sb.service(port=3000) resp = svc.get("/api/data") # token injected + auto-refreshed print(svc.browser_url) # always-fresh URL N _refresher browser_urlr service_urltoken expires_atrP"Optional[Callable[[], ServiceURL]]rNonecL||_||_||_||_||_dSN _browser_url _service_url_token _expires_atrPr rQrRrSrTrPs r!__init__zServiceURL.__init__.(' %$r#rc<|jdS|jdd}tj|}|j |t j}|tjt jz }|tkSNFZz+00:00)tzinfo rPr]replacer fromisoformatrdr utcnow total_seconds_REFRESH_MARGIN_SECONDSr rawexpires remainings r!_should_refreshzServiceURL._should_refresh ? "5&&sH55(-- > !ooX\o::Gx|HL999HHJJ 333r#c|rF|}|j|_|j|_|j|_|j|_dSdSrXrprPrZr[r\r]r freshs r!_maybe_refreshzServiceURL._maybe_refreshs_    ! ! 1OO%%E % 2D  % 2D ,DK$0D     1 1r#c8||jS).Return the raw JWT, refreshing if near expiry.rvr\rs r!rSzServiceURL.tokens {r#c8||jS)/Return the base URL, refreshing if near expiry.rvr[rs r!rRzServiceURL.service_url   r#c8||jS)7Return the browser auth URL, refreshing if near expiry.rvrZrs r!rQzServiceURL.browser_urlr}r#c8||jS):Return the ISO 8601 expiration, refreshing if near expiry.rvr]rs r!rTzServiceURL.expires_ats r#/methodpathkwargsr httpx.Responsec |jddz|dz}t|ddpi}|j|t <tj||fd|i|S)a"Make an HTTP request to the service, injecting the auth header. Args: method: HTTP method (GET, POST, etc.). path: Path relative to the service URL. **kwargs: Forwarded to ``httpx.request``. Returns: httpx.Response. rheadersN) rRrstriplstripdictpoprS _AUTH_HEADERhttpxrequest)r rrrurlrs r!rzServiceURL.requests{%%c**S04;;s3C3CCvzz)T228b99 $  }VSDD'DVDDDr#c  |jd|fi|S)zHTTP GET to the service.GETrr rrs r!r6zServiceURL.gett|E4226222r#c  |jd|fi|S)zHTTP POST to the service.POSTrrs r!postzServiceURL.postst|FD33F333r#c  |jd|fi|S)zHTTP PUT to the service.PUTrrs r!putzServiceURL.putrr#c  |jd|fi|S)zHTTP PATCH to the service.PATCHrrs r!patchzServiceURL.patchst|GT44V444r#c  |jd|fi|S)zHTTP DELETE to the service.DELETErrs r!deletezServiceURL.deletest|Hd55f555r#r1r2cR||d|d|d|d|S)z+Create a ServiceURL from API response dict.rQrRrSrTrQrRrSrTrPr+r8r1rPs r!r9zServiceURL.from_dict?s]+]+w-L)!     r#c(d|jd|jdS)NzServiceURL(service_url= , expires_at=)r[r]rs r!__repr__zServiceURL.__repr__s0 0d&7 0 0* 0 0 0 r#) rQrrRrrSrrTrrPrUrrVr$rrVrrrrrrrrr rrrrrr rr)r1r2rPrUrrN)r%r&r'r(r_rprvr*rSrRrQrTrr6rrrrr:r9rr+r#r!rNrNs2:> % % % % % %"44441111X !!!X! !!!X!    X EEEEE 3333344444333335555566666  :>      [        r#rNceZdZdZddd)d Zd*dZd+dZd,dZd,dZd,dZ d,dZ e d,dZ e d,dZ e d,dZe d,dZ d-d.dZd-d/d Zd-d/d!Zd-d/d"Zd-d/d#Zd-d/d$Zeddd0d'Zd,d(ZdS)1AsyncServiceURLaRAsync variant of :class:`ServiceURL` with auto-refreshing token. Properties and HTTP helpers are async. Use with :meth:`AsyncSandboxClient.service` or :meth:`AsyncSandbox.service`. Example:: svc = await sb.service(port=3000) resp = await svc.get("/api/data") print(await svc.get_browser_url()) NrOrQrrRrSrTrP2Optional[Callable[[], Awaitable[AsyncServiceURL]]]rrVcL||_||_||_||_||_dSrXrYr^s r!r_zAsyncServiceURL.__init__r`r#rc<|jdS|jdd}tj|}|j |t j}|tjt jz }|tkSrbrerls r!rpzAsyncServiceURL._should_refresh-rqr#cK|rL|d{V}|j|_|j|_|j|_|j|_dSdSrXrsrts r!rvzAsyncServiceURL._maybe_refresh7su    ! ! 1//++++++++E % 2D  % 2D ,DK$0D     1 1r#cHK|d{V|jS)rxNryrs r! get_tokenzAsyncServiceURL.get_tokenAs2!!#########{r#cHK|d{V|jS)r{Nr|rs r!get_service_urlzAsyncServiceURL.get_service_urlF3!!#########  r#cHK|d{V|jS)rNrrs r!get_browser_urlzAsyncServiceURL.get_browser_urlKrr#cHK|d{V|jS)rNrrs r!get_expires_atzAsyncServiceURL.get_expires_atPs3!!#########r#c|jS)z&Return the raw JWT without refreshing.)r\rs r!rSzAsyncServiceURL.tokenWs {r#c|jS)z'Return the base URL without refreshing.)r[rs r!rRzAsyncServiceURL.service_url\   r#c|jS)z/Return the browser auth URL without refreshing.)rZrs r!rQzAsyncServiceURL.browser_urlarr#c|jS)z3Return the expiration timestamp without refreshing.)r]rs r!rTzAsyncServiceURL.expires_atf r#rrrrr rcK|d{Vddz|dz}t|ddpi}|d{V|t <tj4d{V}|j ||fd|i|d{Vcdddd{VS#1d{VswxYwYdS)a4Make an async HTTP request to the service, injecting the auth header. Args: method: HTTP method (GET, POST, etc.). path: Path relative to the service URL. **kwargs: Forwarded to ``httpx.AsyncClient.request``. Returns: httpx.Response. Nrr) rrrrrrrr AsyncClientr)r rrrrrclients r!rzAsyncServiceURL.requestms))++++++++33C883>SAQAQQvzz)T228b99&*nn&6&6 6 6 6 6 6 6 $&& P P P P P P P&'OOWOOOOOOOOO P P P P P P P P P P P P P P P P P P P P P P P P P P P P P Ps)C C Cc0K|jd|fi|d{VS)zAsync HTTP GET to the service.rNrrs r!r6zAsyncServiceURL.get4!T\%88888888888r#c0K|jd|fi|d{VS)zAsync HTTP POST to the service.rNrrs r!rzAsyncServiceURL.posts4!T\&$99&999999999r#c0K|jd|fi|d{VS)zAsync HTTP PUT to the service.rNrrs r!rzAsyncServiceURL.putrr#c0K|jd|fi|d{VS)z Async HTTP PATCH to the service.rNrrs r!rzAsyncServiceURL.patchs4!T\'4::6:::::::::r#c0K|jd|fi|d{VS)z!Async HTTP DELETE to the service.rNrrs r!rzAsyncServiceURL.deletes4!T\(D;;F;;;;;;;;;r#r1r2cR||d|d|d|d|S)z1Create an AsyncServiceURL from API response dict.rQrRrSrTrr+rs r!r9zAsyncServiceURL.from_dictrr#c(d|jd|jdS)NzAsyncServiceURL(service_url=rrrrs r!rzAsyncServiceURL.__repr__s0 04+< 0 0* 0 0 0 r#) rQrrRrrSrrTrrPrrrVr$rrrrr)r1r2rPrrr)r%r&r'r(r_rprvrrrrr*rSrRrQrTrr6rrrrr:r9rr+r#r!rrs   (JN % % % % % %"44441111 !!!! !!!!     X!!!X!!!!X!   X (+PPPPP&99999:::::99999;;;;;<<<<<  JN      [        r#rc2eZdZUdZded<ded<ded<dS) OutputChunkafA single chunk of streaming output from command execution. Attributes: stream: Either "stdout" or "stderr". data: The text content of this chunk (valid UTF-8, server handles boundary splitting). offset: Byte offset within the stream. Used internally for reconnection; users typically don't need this. rstreamr1roffsetN)r%r&r'r(r)r+r#r!rrs7KKK IIIKKKKKr#rceZdZdZdZdZdZddddd'dZd(dZe d)dZ e d*dZ e d+dZ d,dZ d,dZd(d Zd-d"Ze d.d#Ze d.d$Zd/d%Zd&S)0 CommandHandleaEHandle to a running command with streaming output and auto-reconnect. Iterable, yielding OutputChunk objects (stdout and stderr interleaved in arrival order). Access .result after iteration to get the full ExecutionResult. Auto-reconnect behavior: - Server hot-reload (1001 Going Away): reconnect immediately - Network error / unexpected close: reconnect with exponential backoff - User called kill(): do NOT reconnect (propagate error) The auto-reconnect is transparent -- the iterator reconnects and continues yielding chunks without any user intervention. If all reconnect attempts are exhausted, SandboxConnectionError is raised. Construction modes (controlled by ``command_id``): - **New execution** (``command_id=""``, the default): the constructor eagerly reads the server's ``"started"`` message to populate ``command_id`` and ``pid`` before returning. - **Reconnection** (``command_id`` set): skips the started-message read, since reconnect streams don't emit one. Example: handle = sandbox.run("make build", timeout=600, wait=False) for chunk in handle: # auto-reconnects on transient errors print(chunk.data, end="") result = handle.result print(f"Exit code: {result.exit_code}") ? @rJr command_id stdout_offset stderr_offsetmessage_streamIterator[dict]controlOptional[_WSStreamControl]sandboxrrrrrrrrVc||_||_||_d|_d|_d|_g|_g|_d|_||_ ||_ |r ||_dS| dS)NF) _stream_control_sandbox _command_id_pid_result _stdout_parts _stderr_parts _exhausted_last_stdout_offset_last_stderr_offset_consume_startedr rrrrrrs r!r_zCommandHandle.__init__s&   *.#' 26 (*(*#0 #0  $)D     ! ! # # # # #r#cZ t|j}n#t$rtddwxYw|ddkr(td|ddd|d|_|d |_d S) aHEagerly read the 'started' message to populate command_id and pid. Blocks briefly until the server sends the started message (arrives near-instantly after connection). After this call, command_id and pid are available, and the WebSocket is bound to the control object (so kill() works). -Command stream ended before 'started' messagecommand operationtypestarted!Expected 'started' message, got ''rpidN)nextr StopIterationrr6rrr first_msgs r!rzCommandHandle._consume_starteds T\**II   '?#   == I - -'LIMM&4I4ILLL# %==66MM%(( s3r/c|jS)z=The server-assigned command ID. Available after construction.rrs r!rzCommandHandle.command_idrr#rCc|jS)zr'r+r#r!rrss@ML$$$$$$<)))).   X XX 65(5(5(5(n & & & & + + + +(((X((((X(      r#rceZdZdZdZdZdZddddd'dZd(dZe d)dZ e d*dZ e d+dZ d,dZ d,dZd(d Zd-d"Ze d.d#Ze d.d$Zd/d%Zd&S)0AsyncCommandHandleaAsync handle to a running command with streaming output and auto-reconnect. Async iterable, yielding OutputChunk objects (stdout and stderr interleaved in arrival order). Access .result after iteration to get the full ExecutionResult. Auto-reconnect behavior: - Server hot-reload (1001 Going Away): reconnect immediately - Network error / unexpected close: reconnect with exponential backoff - User called kill(): do NOT reconnect (propagate error) Construction modes (controlled by ``command_id``): - **New execution** (``command_id=""``, the default): call ``await handle._ensure_started()`` after construction to read the server's ``"started"`` message and populate ``command_id`` / ``pid``. - **Reconnection** (``command_id`` set): skips the started-message read, since reconnect streams don't emit one. Example: handle = await sandbox.run("make build", timeout=600, wait=False) async for chunk in handle: # auto-reconnects on transient errors print(chunk.data, end="") result = await handle.result print(f"Exit code: {result.exit_code}") rrrrJrrrAsyncIterator[dict]rOptional[_AsyncWSStreamControl]rrrrrrrrrVc||_||_||_d|_d|_d|_g|_g|_d|_||_ ||_ |r||_d|_ dSd|_ dS)NFT) rrrrrrrrrrr_startedrs r!r_zAsyncCommandHandle.__init__s~&   *.#' 26 (*(*#0 #0  ")D  DMMM!DMMMr#cK|jrdS |jd{V}n#t$rt ddwxYw|ddkr(t d|ddd|d |_|d |_d |_dS) z:Read the 'started' message to populate command_id and pid.NrrrrrrrrrT)rKr __anext__StopAsyncIterationrr6rrrs r!_ensure_startedz"AsyncCommandHandle._ensure_starteds =  F "l4466666666II!   '?#   == I - -'LIMM&4I4ILLL# %==66MM%((  s -A r/c|jS)z@The server-assigned command ID. Available after _ensure_started.rrs r!rzAsyncCommandHandle.command_idrr#rCc|jS)z?The process ID on the sandbox. Available after _ensure_started.rrs r!rzAsyncCommandHandle.pidrr#rchK|j |23d{V}6|jtdd|jS)z&The final execution result. Awaitable.Nrrrrrs r!r zAsyncCommandHandle.result!so <        a < ';# |sAsyncIterator[OutputChunk]c \K|d{V|jrdS|j23d{V}|d}|dvryt ||d|dd}|dkr!|j|dn |j|d|WV|d krWtd |jd |j|d  |_ d |_dS6d |_dSr ) rOrrr6rrrrrrrrs r! _aiter_streamz AsyncCommandHandle._aiter_stream.st""$$$$$$$$$ ?  F       #wwvH///##V778Q// x''&--c&k::::&--c&k::: V##.774#566774#566!+.    #'$&*sD$cJKddl}d} |23d{V}d}|jdkr5|jt |jdz|_n4|jt |jdz|_|WV6dS#t$r}|j r |j j r|dz }||j krtd|d|t|t}|s>t|jd |dz zz|j}||d{V|jJ|j|j|j|j d{V}|j|_|j |_ d |_Yd}~nd}~wwxYw) z6Async iterate with auto-reconnect on transient errors.rNTrrrrrrrF)asynciorUrrrr1rrrrrr r!r"rr#r$r%r&rrr'rr)r rWr(rr)r*r+r,s r! __aiter__zAsyncCommandHandle.__aiter__JsW* () (#'#5#5#7#7        %)*&|x//38<#!J--g66CC40049<#!J--g66CC40 KKKKK$8) ( ( (=T]%9"a'"%(@@@04+=444 !+1.F G G $/*a4F4J.KL)E"--.........'333#'=#:#:$"&":"&":$;$$ *1 * 3 "'7 (* (s#B%B"A>B%% F /C'FF cZK|jr!|jd{VdSdS)z*Send a kill signal to the running command.Nr/rs r!r1zAsyncCommandHandle.kill{sF = ,-))++ + + + + + + + + + , ,r#r1c\K|jr"|j|d{VdSdS)z"Write data to the command's stdin.Nr3r5s r!r4zAsyncCommandHandle.send_inputsH = 1-**400 0 0 0 0 0 0 0 0 0 1 1r#c|jSr7r8rs r!r9z%AsyncCommandHandle.last_stdout_offsetr:r#c|jSr<r=rs r!r>z%AsyncCommandHandle.last_stderr_offsetr:r#c|K|jJ|j|j|j|jd{VS)z6Reconnect to this command from the last known offsets.Nrr@rs r!r'zAsyncCommandHandle.reconnectsd+++],,  22-         r#N)rrHrrIrrrrrrrrrrVrrArBrC)rrSrDrE)rrG)r%r&r'r(r!r$r%r_rOr*rrr rUrXr1r4r9r>r'r+r#r!rGrGsr8ML"""""">(   X X   X 8/(/(/(/(b,,,, 1111 (((X((((X(      r#rG)&r( __future__rcollections.abcrrrr dataclassesrr r typingr r r rlangsmith.sandbox._exceptionsrrr langsmith.sandbox._async_sandboxrlangsmith.sandbox._sandboxrlangsmith.sandbox._ws_executerrrr-r<rrkrNrrrrGr+r#r!rfs))""""""HHHHHHHHHHHH!!!!!!''''''''//////////  ======222222   # # # # # # #  #          ( 5 5 5 5 5 5 5  5 x3 L L L L L L L L ^\ \ \ \ \ \ \ \ H           } } } } } } } } @S S S S S S S S S S r#