a i]dZddlmZddlmZmZddlmZmZddl m Z m Z m Z ddl mZmZmZe rddlmZddlmZdd lmZmZeGd d ZeGd d ZeGddZeGddZeGddZeGddZeGddZeGddZGddZ GddZ!dS)z#Data models for the sandbox client.) annotations) AsyncIteratorIterator) dataclassfield) TYPE_CHECKINGAnyOptional)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:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langsmith/sandbox/_models.pysuccesszExecutionResult.success s~""N)rr)__name__ __module__ __qualname____doc____annotations__propertyrr rrrsU55KKKKKKNNN ###X###r rc>eZdZUdZdZded<dZded<dZded <dS) ResourceSpecz%Resource specification for a sandbox.500mrcpu512MimemoryN Optional[str]storage)r!r"r#r$r+r%r-r/r'r rr)r)&sK//CF!G!!!!!!r r)cteZdZUdZded<ded<ded<dZded<dZded <dZded <eddZ dS)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_atdatadict[str, Any]rc ||dd|dd|dd|d|d|d  S) z'Create a Volume from API response dict.r2r3unknownr4defaultr5r6r7)r2r3r4r5r6r7getclsr8s r from_dictzVolume.from_dictBs{s&"%%&),,((?I>>xx~~xx --xx --     r )r8r9rr1 r!r"r#r$r%r5r6r7 classmethodrBr'r rr1r1/sIII IIIB $J$$$$ $J$$$$    [    r r1c(eZdZUdZded<ded<dS)VolumeMountSpecz:Specification for mounting a volume in a sandbox template.r volume_name mount_pathNr!r"r#r$r%r'r rrFrFOs+DDOOOOOr rFceZdZUdZded<ded<ded<eeZded <d Zd ed <d Z d ed <d Z d ed<e ddZ d S)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.r5r6r7r8r9rc |di}|dg}||dd|ddt|dd|d d |d  d |D|d|d|dS)z0Create a SandboxTemplate from API response dict.rMrOr2r;rLr<r+r*r-r,r/)r+r-r/c g|];}t|dd|ddz-SandboxTemplate.from_dict..ys[   "}b 9 9!vvlB77r r5r6r7)r2rLrMrOr5r6r7)r?r))rAr8resources_datavolume_mounts_datas rrBzSandboxTemplate.from_dictls+r22!XXor::s&"%%((7I..""&&uf55%))(G<<&**955  - xx~~xx --xx --#    r )r8r9rrK) r!r"r#r$r%rlistrOr5r6r7rDrBr'r rrKrKWs  IIIJJJ+05+F+F+FMFFFFB $J$$$$ $J$$$$   [   r rKcDeZdZUdZded<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. rstatusNr.status_messager8r9rcj||dd|dS)z/Create a ResourceStatus from API response dict.rZ provisioningr[)rZr[r>r@s rrBzResourceStatus.from_dicts>s88Hn5588$455    r )r8r9rrY)r!r"r#r$r%r[rDrBr'r rrYrYsZKKK$(N((((   [   r rYcteZdZUdZded<ded<ded<dZded <dZded <dZded <eddZ dS)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.r5r6r7r8r9rc ||dd|dd|dd|d|d|d S) z%Create a Pool from API response dict.r2r;r`rarr5r6r7)r2r`rar5r6r7r>r@s rrBzPool.from_dicts{s&"%%((?B77XXj!,,xx~~xx --xx --     r )r8r9rr_rCr'r rr_r_s  IIIMMMB $J$$$$ $J$$$$    [    r r_c2eZdZUdZded<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. rstreamr8roffsetNrIr'r rrdrds7KKK IIIKKKKKr rdceZdZdZdZdZdZddddd'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}") ? @r;r command_id stdout_offset stderr_offsetmessage_streamIterator[dict]controlOptional[_WSStreamControl]sandboxrrmrrnrrorNonec||_||_||_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_startedrrprrrtrmrnros r__init__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 ''rmpidN)nextrw StopIterationr r?rzr{r first_msgs rrzCommandHandle._consume_starteds T\**II   '?#   == I - -'LIMM&4I4ILLL# %==66MM%(( s3r.c|jS)z=The server-assigned command ID. Available after construction.rzrs rrmzCommandHandle.command_id/ r Optional[int]c|jS)zt|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)asynciorrerfrr8rrrr rxrrrr rrrrrzryrrwr)rrrrrrrrs r __aiter__zAsyncCommandHandle.__aiter__\sW* () (#'#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.Nrrs rrzAsyncCommandHandle.killsF = ,-))++ + + + + + + + + + , ,r r8c\K|jr"|j|d{VdSdS)z"Write data to the command's stdin.Nrrs rrzAsyncCommandHandle.send_inputsH = 1-**400 0 0 0 0 0 0 0 0 0 1 1r c|jSrrrs rrz%AsyncCommandHandle.last_stdout_offsetrr c|jSrrrs rrz%AsyncCommandHandle.last_stderr_offsetrr c|K|jJ|j|j|j|jd{VS)z6Reconnect to this command from the last known offsets.Nrrrs rrzAsyncCommandHandle.reconnectsd+++],,  22-         r N)rprrrrrtrrmrrnrrorrrurrrr)rrrr)rr)r!r"r#r$rrrrrr&rmrrrrrrrrrr'r rrrsr8ML"""""">(   X X   X 8/(/(/(/(b,,,, 1111 (((X((((X(      r rN)"r$ __future__rcollections.abcrr dataclassesrrtypingrr r langsmith.sandbox._exceptionsr r r langsmith.sandbox._async_sandboxrlangsmith.sandbox._sandboxrlangsmith.sandbox._ws_executerrrr)r1rFrKrYr_rdrhrr'r rrs))""""""33333333((((((((////////// ======222222   # # # # # # #  # """"""" "          >   + + + + + + +  + \          (            P           } } } } } } } } @S S S S S S S S S S r