Zǻi]dZddlmZddlmZmZddlmZmZddl m Z m Z m Z ddl mZmZmZe rddlmZddlmZdd lmZmZeGd d ZeGd d ZeGddZeGddZeGddZeGddZeGddZeGddZGddZ GddZ!y)z#Data models for the sandbox client.) annotations) AsyncIteratorIterator) dataclassfield) TYPE_CHECKINGAnyOptional)SandboxConnectionErrorSandboxOperationErrorSandboxServerReloadError) AsyncSandbox)Sandbox)_AsyncWSStreamControl_WSStreamControlcBeZdZUdZded<ded<ded<ed dZy) ExecutionResultz+Result of executing a command in a sandbox.strstdoutstderrint exit_codec |jdk(S)z.Return True if the command exited with code 0.r)rselfs W/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/langsmith/sandbox/_models.pysuccesszExecutionResult.success s~~""N)returnbool)__name__ __module__ __qualname____doc____annotations__propertyrrrrrs&5 K KN ##rrc<eZdZUdZdZded<dZded<dZded <y) ResourceSpecz%Resource specification for a sandbox.500mrcpu512MimemoryN Optional[str]storage)r!r"r#r$r+r%r-r/r'rrr)r)&s#/CFC!G]!rr)cleZdZUdZded<ded<ded<dZded<dZded <dZded <ed d Z y) Volumea@Represents a persistent volume. Volumes are persistent storage that can be mounted in sandboxes. Attributes: id: Unique identifier (UUID). Remains constant even if name changes. May be None for resources created before ID support was added. name: Display name (can be updated). rnamesize storage_classNr.id created_at updated_atc ||jdd|jdd|jdd|jd|jd|jd  S) z'Create a Volume from API response dict.r2r3unknownr4defaultr5r6r7)r2r3r4r5r6r7getclsdatas r from_dictzVolume.from_dictBs^&"%&),((?I>xx~xx -xx -   r)r@dict[str, Any]rr1 r!r"r#r$r%r5r6r7 classmethodrAr'rrr1r1/sI I IB  $J $ $J $    rr1c&eZdZUdZded<ded<y)VolumeMountSpecz:Specification for mounting a volume in a sandbox template.r volume_name mount_pathNr!r"r#r$r%r'rrrFrFOsDOrrFceZdZUdZded<ded<ded<eeZded <d Zd ed <d Z d ed <d Z d ed<e ddZ y )SandboxTemplateaRepresents a SandboxTemplate. Templates define the image, resource limits, and volume mounts for sandboxes. All other container details are handled by the server with secure defaults. Attributes: id: Unique identifier (UUID). Remains constant even if name changes. May be None for resources created before ID support was added. name: Display name (can be updated). rr2imager) resources)default_factoryzlist[VolumeMountSpec] volume_mountsNr.r5r6r7c|jdi}|jdg}||jdd|jddt|jdd|jd d |jd  |Dcgc]/}t|jd d|jdd1c}|jd|jd|jdScc}w)z0Create a SandboxTemplate from API response dict.rMrOr2r9rLr:r+r*r-r,r/)r+r-r/rGrH)rGrHr5r6r7)r2rLrMrOr5r6r7)r=r)rF)r?r@resources_datavolume_mounts_datavms rrAzSandboxTemplate.from_dictls+r2!XXor:&"%((7I.""&&uf5%))(G<&**95-    "}b 9!vvlB7xx~xx -xx -#  s 4C5)r@rBrrK) r!r"r#r$r%rlistrOr5r6r7rDrAr'rrrKrKWsY  I J+0+FM(FB  $J $ $J $  rrKc<eZdZUdZded<dZded<eddZy) 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. rstatusNr.status_messagecT||jdd|jdS)z/Create a ResourceStatus from API response dict.rW provisioningrX)rWrXr<r>s rrAzResourceStatus.from_dicts,88Hn588$45  r)r@rBrrV)r!r"r#r$r%rXrDrAr'rrrVrVs* K$(NM(  rrVcleZdZUdZded<ded<ded<dZded <dZded <dZded <ed d Z y)PoolaRepresents a Sandbox Pool for pre-provisioned sandboxes. Pools pre-provision sandboxes from a template for faster startup. Instead of waiting for a new sandbox to be created, sandboxes can be served from a pre-warmed pool. Note: Templates with volume mounts cannot be used in pools. Attributes: id: Unique identifier (UUID). Remains constant even if name changes. May be None for resources created before ID support was added. name: Display name (can be updated). rr2 template_namerreplicasNr.r5r6r7c ||jdd|jdd|jdd|jd|jd|jd S) z%Create a Pool from API response dict.r2r9r]r^rr5r6r7)r2r]r^r5r6r7r<r>s rrAzPool.from_dicts^&"%((?B7XXj!,xx~xx -xx -   r)r@rBrr\rCr'rrr\r\sI  IMB  $J $ $J $    rr\c0eZdZUdZded<ded<ded<y) 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. rstreamr@roffsetNrIr'rrraras K I KrraceZdZdZdZdZdZdddd ddZdd Ze dd Z e dd Z e dd Z dd Z ddZddZddZe ddZe ddZddZy) 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}") ? @r9r command_id stdout_offset stderr_offsetc||_||_||_d|_d|_d|_g|_g|_d|_||_ ||_ |r||_y|jy)NF) _stream_control_sandbox _command_id_pid_result _stdout_parts _stderr_parts _exhausted_last_stdout_offset_last_stderr_offset_consume_startedrmessage_streamcontrolsandboxrjrkrls r__init__zCommandHandle.__init__so&   *.#' 26 (*(*#0 #0 )D   ! ! #rc$ t|j}|j ddk7r td|j ddd|j d|_|j d |_y #t$rtddwxYw) 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 ''rjpidN)nextrn StopIterationr r=rqrrr first_msgs rryzCommandHandle._consume_starteds T\\*I == I -'3IMM&4I3J!L# %==6MM%(  '?#  s A88Bc|jS)z=The server-assigned command ID. Available after construction.rqrs rrjzCommandHandle.command_id/rc|jS)z?01 !+1.F G $**a4F4J.KL))EJJu%''333!]]44$$"&":":"&":":5 *11 * 3 3 "'7 (s0GBB#"G# F=-DF83G8F==GcR|jr|jjyy)a/Send a kill signal to the running command (SIGKILL). The server kills the entire process group. The stream will subsequently yield an exit message with a non-zero exit code. Has no effect if the command has already exited or the WebSocket connection is closed. Nro send_killrs rkillzCommandHandle.kills  == MM # # % rcT|jr|jj|yy)zWrite data to the command's stdin. Args: data: String data to write to stdin. Has no effect if the command has already exited or the WebSocket connection is closed. Nro send_inputrr@s rrzCommandHandle.send_inputs" == MM $ $T * rc|jSz8Last known stdout byte offset (for manual reconnection).rwrs rlast_stdout_offsetz CommandHandle.last_stdout_offset'''rc|jSz8Last known stderr byte offset (for manual reconnection).rxrs rlast_stderr_offsetz CommandHandle.last_stderr_offsetrrc|jJ|jj|j|j|jS)aReconnect to this command from the last known offsets. Returns a new handle that resumes output from where this one left off. Any output produced while disconnected is replayed from the server's ring buffer. Returns: A new CommandHandle. Raises: SandboxOperationError: If command_id is not found or session expired. SandboxConnectionError: If connection to sandbox fails. rrqrprrwrxrs rrzCommandHandle.reconnectsM+++}}&&   2222'  rN)r{zIterator[dict]r|zOptional[_WSStreamControl]r}rrjrrkrrlrrNonerrrr.rz Optional[int]rr)rzIterator[OutputChunk]r@rrrrr)rre)r!r"r#r$rrrr~ryr&rjrrrrrrrrrr'rrreres@ML$&$,$ $  $$$ $<).   65(n & +(((( rreceZdZdZdZdZdZdddd ddZdd Ze dd Z e dd Z e dd Z dd Z ddZddZddZe ddZe ddZddZy)AsyncCommandHandleaAsync 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}") rfrgrhr9rric||_||_||_d|_d|_d|_g|_g|_d|_||_ ||_ |r||_d|_ yd|_ y)NFT) rnrorprqrrrsrtrurvrwrx_startedrzs rr~zAsyncCommandHandle.__init__sq&   *.#' 26 (*(*#0 #0 )D  DM!DMrcrK|jry |jjd{}|j ddk7r t d|j ddd|j d |_|j d |_d |_y7l#t$rt ddwxYww) z:Read the 'started' message to populate command_id and pid.NrrrrrrrrjrT)rrn __anext__StopAsyncIterationr r=rqrrrs r_ensure_startedz"AsyncCommandHandle._ensure_starteds ==  "ll4466I == I -'3IMM&4I3J!L# %==6MM%(  7! '?#  s,B7BBBA(B7BB44B7c|jS)z@The server-assigned command ID. Available after _ensure_started.rrs rrjzAsyncCommandHandle.command_id)rrc|jS)z?The process ID on the sandbox. Available after _ensure_started.rrs rrzAsyncCommandHandle.pid.rrcK|j |23d{} |j tdd|jS7*6)w)z&The final execution result. Awaitable.Nrrrrrs rrzAsyncCommandHandle.result3sU <<   a << ';# || 4s%AAAA'AAAc BK|jd{|jry|j23d{}|jd}|dvrit ||d|jdd}|dk(r|j j |dn|jj |d||d k(std j|j d j|j|d  |_ d |_y776d |_ywr) rrvrnr=rartrrurrrsrs r _aiter_streamz AsyncCommandHandle._aiter_stream@s""$$$ ??   #wwvH//##V778Q/ x'&&--c&k:&&--c&k: V#.774#5#56774#5#56!+.  #'/ % *s7DDDDDDBDADD DcKddl}d} |j23d{}d}|jdk(r7|jt |j j dz|_n6|jt |j j dz|_|76y#t$r"}|jr|jjr|dz }||jkDrtd|d|t|t}|sCt|j d|dz zz|j"}|j%|d{7|j&J|j(j+|j&|j|j d{7}|j,|_|j|_ d |_Yd}~nd}~wwxYwӭw) z6Async iterate with auto-reconnect on transient errors.rNrrrrrrrF)asynciorrbrcrr@rrwrxr rorrrr rrrrrqrprrnrv)rrrrrrrrs r __aiter__zAsyncCommandHandle.__aiter__\s) (#'#5#5#7   %)*&||x/38<<#!JJ--g6C4049<<#!JJ--g6C40 K  #7) (==T]]%9%9"a'"%(@(@@0*+=*>?34 !+1.F G $**a4F4J.KL))E"--...''333#'==#:#:$$"&":":"&":":$;$ *11 * 3 3 "'7 (siGB,B*B(B*BB,(B**B,+G, G6BG E AGF" .GGGGcnK|jr#|jjd{yy7w)z*Send a kill signal to the running command.Nrrs rrzAsyncCommandHandle.kills+ ==--))+ + +  +s *535cpK|jr$|jj|d{yy7w)z"Write data to the command's stdin.Nrrs rrzAsyncCommandHandle.send_inputs- ==--**40 0 0  0s +646c|jSrrrs rrz%AsyncCommandHandle.last_stdout_offsetrrc|jSrrrs rrz%AsyncCommandHandle.last_stderr_offsetrrcK|jJ|jj|j|j|jd{S7w)z6Reconnect to this command from the last known offsets.Nrrrs rrzAsyncCommandHandle.reconnectsY+++]],,   2222-    sAAAAN)r{zAsyncIterator[dict]r|zOptional[_AsyncWSStreamControl]r}rrjrrkrrlrrrrrrr)rzAsyncIterator[OutputChunk]rr)rr)r!r"r#r$rrrr~rr&rjrrrrrrrrrr'rrrrs8ML"+"1" "  """ ">(    8/(b, 1 (((( rrN)"r$ __future__rcollections.abcrr dataclassesrrtypingrr r langsmith.sandbox._exceptionsr r r langsmith.sandbox._async_sandboxrlangsmith.sandbox._sandboxrlangsmith.sandbox._ws_executerrrr)r1rFrKrVr\rarerr'rrrs)"3(// =2   # #  # "" "     >   + +  + \     (       P      } } @S S r