~tjs<dZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZddlZddlmZmZmZddlmZdd lmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$e rdd l%m&Z&dd l'm(Z(ddl)m*Z*eee+e+fZ,eGddZ-dS)z0AsyncSandbox class for async sandbox operations.) annotationsN)Mapping) dataclassfield) TYPE_CHECKINGAnyCallableLiteralOptionalUnionoverload)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxConnectTimeoutError)handle_sandbox_http_error)AsyncCommandHandleAsyncServiceURLExecutionResultSnapshot_StreamEndedBeforeStarted) AsyncTunnel)WEBSOCKETS_AVAILABLEconnect_deadlineopen_timeout_for)AsyncSandboxClient) SandboxClientSandboxceZdZUdZded<dZded<dZded<dZded <dZded <dZ ded <dZ ded <dZ d ed<dZ d ed<dZ ded<dZded<dZd ed<dZd ed<dZd ed<eddZded<eddZded<e dsdtd!Zdd"dud%Zdvd&Zdwd.Zdxd/Zed0d0d0d0d0d0d0d0d0d0d0d0d1 dydEZed0d0d0d0d0d0d0d0d0d0d0dF dzdIZdJdddKdddLddMdddd1 d{dOZdLddMdddPd|dQZd}dRZdSdSddTd~dXZdJddYdd]Z dJddYdd_Z!dSd`ddaddfZ"dMddgddkZ#dlddYddmZ$ddnddoZ%ddnddpZ&dJddYddrZ'dS) AsyncSandboxaRepresents an active sandbox for running commands and file operations async. This class is typically obtained from AsyncSandboxClient.sandbox() and supports the async context manager protocol for automatic cleanup. Attributes: name: Display name (can be updated). dataplane_url: URL for data plane operations (file I/O, command execution). Only functional when status is "ready". id: Unique identifier (UUID). Remains constant even if name changes. May be None for resources created before ID support was added. status: Sandbox lifecycle status. One of "provisioning", "ready", "failed", "stopped". status_message: Human-readable details when status is "failed", None otherwise. created_at: Timestamp when the sandbox was created. updated_at: Timestamp when the sandbox was last updated. idle_ttl_seconds: Idle timeout TTL in seconds (``0`` means disabled). Newly-created sandboxes receive a server-side default of ``600`` seconds (10 minutes) when the caller did not set ``idle_ttl_seconds`` explicitly. The launcher stops the sandbox after this many idle seconds; deletion is anchored to ``stopped_at`` and controlled by ``delete_after_stop_seconds`` (see below). delete_after_stop_seconds: Seconds after a sandbox enters the ``stopped`` state before it (and its filesystem clone) are permanently deleted. ``0`` disables stop-anchored deletion; ``None`` falls back to the server default. stopped_at: Timestamp when the sandbox transitioned to ``stopped``, or ``None`` while running. The deletion deadline is ``stopped_at + delete_after_stop_seconds``. snapshot_id: Snapshot ID used to create this sandbox. vcpus: Number of vCPUs allocated. mem_bytes: Memory allocation in bytes. fs_capacity_bytes: Root filesystem capacity in bytes. Example: async with await client.sandbox( snapshot_id="" ) as sandbox: result = await sandbox.run("python --version") print(result.stdout) strnameN Optional[str] dataplane_urlidreadystatusstatus_message created_at updated_atz Optional[int]idle_ttl_secondsdelete_after_stop_seconds stopped_at snapshot_idvcpus mem_bytesfs_capacity_bytesF)reprdefaultr_clientTbool _auto_deletedatadict[str, Any]client auto_deletereturnc |did|ddd|dd|dd|ddd|dd|dd |d d |d d |d d |d d |d d|dd|dd|dd|d|S)aDCreate an AsyncSandbox from API response dict. Args: data: API response dictionary containing sandbox data. client: Parent AsyncSandboxClient for operations. auto_delete: Whether to delete the sandbox on context exit. Returns: AsyncSandbox instance. r"r$r%r'r&r(r)r*r+r,r-r.r/r0r1r4r6)get)clsr7r9r:s ^/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/sandbox/_async_sandbox.py from_dictzAsyncSandbox.from_dictjs"s   &"%%% ((?333 xx~~~ 88Hg...   88$4555  xx ---  xx --- "XX&8999 '+hh/J&K&K&K xx --- /// ((7### hh{+++ #hh':;;; F %!  )r9Optional[SandboxClient]rc<ddlm}|did|jd|jd|jd|jd|jd|jd |jd |j d |j d |j d |j d|j d|jd|jd||n|jddS)aCreate a Sandbox for the same underlying sandbox. The returned instance has ``auto_delete`` disabled so the sandbox's lifecycle stays tied to this instance; both refer to the same server-side sandbox. Args: client: SandboxClient to use for operations. If not provided, one is created with the same configuration as this sandbox's client (see :meth:`AsyncSandboxClient.to_sync`). Returns: Sandbox referring to the same sandbox. rrr"r$r%r'r(r)r*r+r,r-r.r/r0r1r4Nr6Fr>)langsmith.sandbox._sandboxrr"r$r%r'r(r)r*r+r,r-r.r/r0r1r4to_sync)selfr9rs rArGzAsyncSandbox.to_syncs) 766666w    ,, ww ;;   ..     "22 '+&D&D  (( ** nn #44 %0FFdl6J6J6L6L !  rCc K|S)zEnter async context manager.r>rHs rA __aenter__zAsyncSandbox.__aenter__s  rCexc_typeOptional[type]exc_valOptional[BaseException]exc_tb Optional[Any]NonecK|jr9 |j|jd{VdS#t$rYdSwxYwdS)zDI>>> !!rC.) timeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnect ttl_secondsptyheaderswaitcommandrYintrZOptional[dict[str, str]]r[r\r]Optional[Callable[[str], Any]]r^r_r`rarbrcRequestHeadersrd Literal[True]rc  KdSNr>rHrerYrZr[r\r]r^r_r`rarbrcrds rArunzAsyncSandbox.runs "#rC) rYrZr[r\r]r^r_r`rarbrcLiteral[False]rc  KdSrlr>rms rArnzAsyncSandbox.runs "!SrC<z /bin/bashi,iX*Union[ExecutionResult, AsyncCommandHandle]c ZK| s|s|rtd|| p|p|}|r(||||||| |||| | | |  d{VStr(||||||ddd|| | | |  d{VS||||||| d{VS)am Execute a command in the sandbox asynchronously. Args: command: Shell command to execute. timeout: Command timeout in seconds. env: Environment variables to set for the command. cwd: Working directory for command execution. If None, uses sandbox default. shell: Shell to use for command execution. Defaults to "/bin/bash". on_stdout: Callback invoked with each stdout chunk as it arrives. Blocks until the command completes and returns ExecutionResult. Cannot be combined with wait=False. on_stderr: Callback invoked with each stderr chunk as it arrives. Blocks until the command completes and returns ExecutionResult. Cannot be combined with wait=False. idle_timeout: Idle timeout in seconds. If the command has no connected clients for this duration, it is killed. Defaults to 300 (5 minutes). Set to -1 for no idle timeout. Only applies to WebSocket execution. kill_on_disconnect: If True, kill the command immediately when the last client disconnects. Defaults to False (command continues running and can be reconnected to). ttl_seconds: How long (in seconds) a finished command's session is kept for reconnection. Defaults to 600 (10 minutes). Set to -1 to keep indefinitely. pty: If True, allocate a pseudo-terminal for the command. Useful for commands that require a TTY (e.g., interactive programs, commands that use terminal control codes). Defaults to False. wait: If True (default), block until the command completes and return ExecutionResult. If False, return an AsyncCommandHandle immediately for streaming output, kill, stdin input, and reconnection. Cannot be combined with on_stdout/on_stderr callbacks. Returns: ExecutionResult when wait=True (default). AsyncCommandHandle when wait=False. Raises: ValueError: If wait=False is combined with callbacks. DataplaneNotConfiguredError: If dataplane_url is not configured. SandboxOperationError: If command execution fails. CommandTimeoutError: If command exceeds its timeout. SandboxConnectionError: If connection to sandbox fails after retries. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z}Cannot combine wait=False with on_stdout/on_stderr callbacks. Use wait=False and iterate the CommandHandle, or use callbacks.) rYrZr[r\rdr]r^r_r`rarbrcNT)rYrZr[r\rc) ValueErrorrX_run_wsr _run_http)rHrerYrZr[r\r]r^r_r`rarbrcrduse_wss rArnzAsyncSandbox.runs{@  i R  ##%%%3Y3)  ##)#5'& $  )#5'& ^^  $         rC)r_r`rarbrcc Kddl}ddlm}ddlm}|}|jj}|j}|||||| | | | d }|j | }|r||d<d}t} t||d<||||fi|d{V\}}t||||| } | d{Vn#ttf$r|d z }|tjkrt#tjd |d z zztj}|-|t)jz }|dkrt#||}|j|d{VYnwxYw|s|S|jd{VS) z"Execute via WebSocket /execute/ws.rN)run_ws_stream_async)uuid7) command_idrYrZr[r\r_r`rarbrcT open_timeout)r]r^)asynciolangsmith.sandbox._ws_executerylangsmith.uuidrzrXr4_api_keyhex_ws_default_headersrrr_ensure_startedrrMAX_AUTO_RECONNECTSmin _BACKOFF_BASE _BACKOFF_MAXtime monotonicsleepresult)rHrerYrZr[r\rdr]r^r_r`rarbrcrryrzr$api_keyr{ ws_kwargsmergedattemptdeadline msg_streamcontrolhandlebackoff remainings rAruzAsyncSandbox._run_wstsk$ EEEEEE((((((3355 ,' UWW[ %("4& % % 11'::  *#)Ii #%%! -(8(B(BIn %(;(;)) ))###### J (## F -,,.........-/IJ - - -1 /CCC&4gk8JK&3' (4>+;+; ;I A~~!'955G#gmG,,,,,,,,,,, -%! -F M]"""""""s;CBE21E2c$K|}|d}|||d} ||| d<||| d< |jj|| |dz|j|d{V} | | } t| dd | d d | d d  S#tj $r} t| d} ~ wwxYw)z9Execute via HTTP POST /execute (existing implementation).z/execute)rerYr\NrZr[ )jsonrYrcstdoutr=stderr exit_code)rrr) rXr4_httppost_request_headersraise_for_statusrrr?httpxHTTPStatusErrorr) rHrerYrZr[r\rcr$urlpayloadresponser7es rArvzAsyncSandbox._run_https\3355 (((# #  ? GEN ? GEN !\/44"  55g>> 5H  % % ' ' '==??D"xx"--xx"--((;33  $    %a ( ( (  sB9C++D:D  Dr) stdout_offset stderr_offsetrcr{rrcKddlm}|}|jj}||d}|j|} | r| |d<||||fi|d{V\} } t | | ||||S)aReconnect to a running or recently-finished command. Resumes output from the given byte offsets. Any output produced while the client was disconnected is replayed from the server's ring buffer. Args: command_id: The command ID from handle.command_id. stdout_offset: Byte offset to resume stdout from (default: 0). stderr_offset: Byte offset to resume stderr from (default: 0). Returns: An AsyncCommandHandle for the command. Raises: SandboxOperationError: If command_id is not found or session expired. SandboxConnectionError: If connection to sandbox fails after retries. r)reconnect_ws_stream_async)rrrcN)r{rr)rrrXr4rrr) rHr{rrrcrr$rreconnect_kwargsrrrs rA reconnectzAsyncSandbox.reconnects2 LKKKKK3355 ,'+*, , 11'::  1*0 Y '$=$=   % %  % %        G"   !''     rCrYrcpathcontentUnion[str, bytes]c K|}|d}t|tr|d}dd|fi} |jj|d|i|||j|d{V}|dS#tj $r} t| Yd} ~ dSd} ~ wwxYw)a9Write content to a file in the sandbox asynchronously. Args: path: Target file path in the sandbox. content: File content (str or bytes). timeout: Request timeout in seconds. Raises: DataplaneNotConfiguredError: If dataplane_url is not configured. SandboxOperationError: If file write fails. SandboxConnectionError: If connection to sandbox fails after retries. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z/uploadzutf-8filer)paramsfilesrYrcN) rX isinstancer!encoder4rrrrrrr) rHrrrYrcr$rrrrs rAwritezAsyncSandbox.write,s,3355 ''' gs # # .nnW--G&'*+ )!\/44~ 55g>> 5H  % % ' ' ' ' '$ ) ) ) %a ( ( ( ( ( ( ( ( ( )s AB''C6C  CbytescK|}|d} |jj|d|i||j|d{V}||jS#tj$rC}|j j dkrtd|d|j dd |t|d}~wwxYw) aRead a file from the sandbox asynchronously. Args: path: File path to read. Supports both absolute paths (e.g., /tmp/file.txt) and relative paths (resolved from /home/user/). timeout: Request timeout in seconds. Returns: File contents as bytes. Raises: DataplaneNotConfiguredError: If dataplane_url is not configured. ResourceNotFoundError: If the file doesn't exist. SandboxOperationError: If file read fails. SandboxConnectionError: If connection to sandbox fails after retries. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z /downloadr)rrYrcNizFile 'z' not found in sandbox ''r) resource_type)rXr4rr?rrrrrr status_coderr"r)rHrrYrcr$rrrs rAreadzAsyncSandbox.readWs"*3355 ))) !\/33~ 55g>> 4H  % % ' ' '# #$   z%,,+GTGG49GGG"( &a ( ( (  sAA;;C  >CC  local_portmax_reconnectsrc remote_portrrrcrKd|cxkrdksntd|d|r"d|cxkrdksntd|d|}|jj}t ||||||}t j}|d|jj d{V|S)a[Open a TCP tunnel to a port inside the sandbox. Creates a local TCP listener that forwards connections through a yamux-multiplexed WebSocket to the specified port inside the sandbox. Works with any TCP protocol (databases, Redis, HTTP, etc.). Usage:: async with await sandbox.tunnel(remote_port=5432) as t: conn = await asyncpg.connect(host="127.0.0.1", port=t.local_port) Args: remote_port: TCP port inside the sandbox to tunnel to (1-65535). local_port: Local port to listen on. Defaults to mirroring remote_port. Use 0 to let the OS pick an available port. max_reconnects: Maximum number of automatic reconnect attempts when the WebSocket session drops. Set to 0 to disable. Returns: An AsyncTunnel instance (async context manager). Raises: ValueError: If port values are out of range. DataplaneNotConfiguredError: If dataplane_url is not configured. SandboxNotReadyError: If sandbox is not ready. r}iz-remote_port must be between 1 and 65535 (got )z,local_port must be between 1 and 65535 (got rN) rtrXr4rrrget_running_looprun_in_executor_tunnel_start) rHrrrrcr$rtloops rAtunnelzAsyncSandbox.tunnelsDK((((5((((N NNN   a:66666666LzLLL 3355 ,'    !)    '))""4)9:::::::::rCexpires_in_secondsrcportrrcXK|j|j|||d{VS)aGet an authenticated URL for a service running in this sandbox. Returns an :class:`AsyncServiceURL` whose async accessors auto-refresh the token transparently before it expires. Args: port: Port the service is listening on inside the sandbox. expires_in_seconds: Token TTL in seconds (1--86400, default 600). headers: Optional per-request header overrides. Returns: AsyncServiceURL with auto-refreshing token and HTTP helpers. Raises: ResourceNotFoundError: If sandbox not found. ValueError: If port or expires_in_seconds is out of range. SandboxClientError: For other errors. rN)r4servicer")rHrrrcs rArzAsyncSandbox.servicesS2\)) I 1 *         rCxcK|j|j||d{V}|j|_|j|_dS)a1Start a stopped sandbox and wait until ready. After starting, the sandbox's status and dataplane_url are updated in place. Args: timeout: Timeout in seconds when waiting for ready. headers: Optional per-request header overrides. Raises: ResourceNotFoundError: If sandbox not found. ResourceCreationError: If sandbox fails during startup. ResourceTimeoutError: If sandbox doesn't become ready within timeout. SandboxClientError: For other errors. rN)r4 start_sandboxr"r'r$)rHrYrc refresheds rAstartzAsyncSandbox.startsh*,44 Iw5         & &4rCrccfK|j|j|d{Vd|_dS)aStop a running sandbox (preserves sandbox files for later restart). Args: headers: Optional per-request header overrides. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rNstopped)r4 stop_sandboxr"r'rHrcs rAstopzAsyncSandbox.stopsDl'' 7'CCCCCCCCC  rCcXK|j|j|d{VdS)zDelete this sandbox. Args: headers: Optional per-request header overrides. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rN)r4rTr"rs rAdeletezAsyncSandbox.deletes=l))$)W)EEEEEEEEEEErCrcXK|j|j|||d{VS)a!Capture a snapshot from this sandbox. Args: name: Snapshot name. timeout: Timeout in seconds when waiting for ready. headers: Optional per-request header overrides. Returns: Snapshot in "ready" status. Raises: ResourceNotFoundError: If sandbox not found. ResourceTimeoutError: If snapshot doesn't become ready within timeout. ResourceCreationError: If snapshot capture fails. SandboxClientError: For other errors. rN)r4capture_snapshotr")rHr"rYrcs rArzAsyncSandbox.capture_snapshotsS.\22 I  3         rC)T)r7r8r9rr:r5r;r )r9rDr;r)r;r )rLrMrNrOrPrQr;rR)r;r!)rer!rYrfrZrgr[r#r\r!r]rhr^rhr_rfr`r5rarfrbr5rcrirdrjr;r)rer!rYrfrZrgr[r#r\r!r]rhr^rhr_rfr`r5rarfrbr5rcrirdror;r)rer!rYrfrZrgr[r#r\r!r]rhr^rhr_rfr`r5rarfrbr5rcrirdr5r;rr)rer!rYrfrZrgr[r#r\r!rdr5r]rhr^rhr_rfr`r5rarfrbr5rcrir;rr)rer!rYrfrZrgr[r#r\r!rcrir;r) r{r!rrfrrfrcrir;r) rr!rrrYrfrcrir;rR)rr!rYrfrcrir;r) rrfrrfrrfrcrir;r)rrfrrfrcrir;r)rYrfrcrir;rR)rcrir;rR)r"r!rYrfrcrir;r)(__name__ __module__ __qualname____doc____annotations__r$r%r'r(r)r*r+r,r-r.r/r0r1rr4r6 classmethodrBrGrKrVrXr rnrurvrrrrrrrrrr>rCrAr r *sk((VIII#'M''''BF$(N(((( $J$$$$ $J$$$$&*****/33333 $J$$$$!%K%%%%E#I####'+++++#(%UD"A"A"AGAAAAE4888L8888 ! ! ! ! ! [! F<@" " " " " " H    """"* (+ 4747#&"%!X$ (+ 4747#&"%!!!!!X!,(,! 4848#("&s s s s s s @ #("&W#W#W#W#W#W#r''''Z"& 4 4 4 4 4 4 v"& ))))))))))))X,.))))))^"& 666666x#&"&       F"& 55555567;       9= F F F F F F "&         rCr ).r __future__rrrcollections.abcr dataclassesrrtypingrrr r r r r rlangsmith.sandbox._exceptionsrrrlangsmith.sandbox._helpersrlangsmith.sandbox._modelsrrrrrlangsmith.sandbox._tunnelrrrrrlangsmith.sandbox._async_clientrlangsmith.sandbox._clientrrFrr!rir r>rCrArs66"""""" ######((((((((SSSSSSSSSSSSSSSSSS  A@@@@@211111 3BBBBBB777777222222'#s(+, A  A  A  A  A  A  A   A  A  A  rC