jz*dZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZddlZddlmZmZmZddlmZdd lmZmZmZmZmZmZm Z dd l!m"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/y)z0AsyncSandbox class for async sandbox operations.) annotationsN)Mapping) dataclassfield) TYPE_CHECKINGAnyCallableLiteralOptionalUnionoverload)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxRetryableConnectionError)handle_sandbox_http_error)AsyncCommandHandleAsyncServiceURLDownloadContentDisposition DownloadURLExecutionResultSnapshot_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 dB dCdZdddDdZdEdZ dFd ZdGd!Zed"d"d"d"d"d"d"d"d"d"d"d"d# dHd$Zed"d"d"d"d"d"d"d"d"d"d"d% dId&Zd'ddd(ddd)dd*dddd# dJd+Zd)dd*ddd, dKd-Z dLd.Zd/d/dd0 dMd1Zd'dd2 dNd3Z d'dd2 dOd4Z!d/d5dd6 dPd7Z"d*dd8 dQd9Z#ddddd: dRd;Z$ddTd?Z&dd>dTd@Z'd'dd2 dUdAZ(y)V 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_at Optional[int]idle_ttl_secondsdelete_after_stop_seconds stopped_at snapshot_idvcpus mem_bytesfs_capacity_bytesF)reprdefaultr_clientTbool _auto_deletec |did|jddd|jdd|jdd|jddd|jdd|jdd |jd d |jd d |jd d |jd d |jd d|jdd|jdd|jdd|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/r0r1r2r3r4r7r9)get)clsdataclient auto_deletes j/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langsmith/sandbox/_async_sandbox.py from_dictzAsyncSandbox.from_dictls2" &"% ((?3 xx~ 88Hg.   88$45  xx -  xx - "XX&89 '+hh/J&K xx - / ((7# hh{+ #hh':;  %!  )r@cddlm}|did|jd|jd|jd|j d|j d|jd |jd |jd |jd |jd |jd|jd|jd|jd||n|j j#ddS)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/r0r1r2r3r4r7r9Fr<)langsmith.sandbox._sandboxr r$r&r'r)r*r+r,r.r/r0r1r2r3r4r7to_sync)selfr@r s rBrGzAsyncSandbox.to_syncs  7  ,, ww ;;   ..     "22 '+&D&D  (( ** nn #44 %0Fdll6J6J6L !  rDcK|Sw)zEnter async context manager.r<rHs rB __aenter__zAsyncSandbox.__aenter__s  scK|jr/ |jj|jd{yy7#t$rYywxYww)z> !!!rD.) timeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnect ttl_secondsptyheaderswaitc  KywNr<rHcommandrUrVrWrXrYrZr[r\r]r^r_r`s rBrunzAsyncSandbox.runs ") rUrVrWrXrYrZr[r\r]r^r_c  Kywrbr<rcs rBrezAsyncSandbox.runs "!rf<z /bin/bashi,iXc TK| s|s|r td|j| xs|xs|}|r&|j|||||| |||| | | |  d{Str&|j|||||ddd|| | | |  d{S|j |||||| d{S7O7%7w)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.) rUrVrWrXr`rYrZr[r\r]r^r_NT)rUrVrWrXr_) ValueErrorrT_run_wsr _run_http)rHrdrUrVrWrXrYrZr[r\r]r^r_r`use_wss rBrezAsyncSandbox.runs@iR  ##%3Y3) ##)#5'& $ )#5'& ^^  $   E& s6AB(B"+B(>B$?B(B&B($B(&B()r[r\r]r^r_c 4Kddl}ddlm}ddlm}|j }|j j}|j}|||||| | | | d }|j j| }|r||d<d}t} t||d<||||fi|d{\}}t|||||} |jd{ |s|S|j.d{S7I7#ttf$r|d z }|tj kDrt#tj$d |d z zztj&}|)|t)j*z }|dkrt#||}|j,|d{7YnwxYw7w) z"Execute via WebSocket /execute/ws.rN)run_ws_stream_async)uuid7) command_idrUrVrWrXr[r\r]r^r_ open_timeout)rYrZ)asynciolangsmith.sandbox._ws_executerolangsmith.uuidrprTr7_api_keyhex_ws_default_headersrrr_ensure_startedrrMAX_AUTO_RECONNECTSmin _BACKOFF_BASE _BACKOFF_MAXtime monotonicsleepresult)rHrdrUrVrWrXr`rYrZr[r\r]r^r_rurorpr&api_keyrq ws_kwargsmergedattemptdeadline msg_streamcontrolhandlebackoff remainings rBrkzAsyncSandbox._run_wsvs$ E(335 ,,'' W[[ %("4& % 11': #)Ii #%(8(BIn %(;) )# J (## F -,,...*M]]""O#/*/ - 1 /CCC&44gk8JK&33' (4>>+; ;I A~!'95G#gmmG,,,% -%R#saBF!C+"F:C/ C-C/F&F'F-C//BFF  FFFFcK|j}|d}|||d} ||| d<||| d< |jjj|| |dz|jj |d{} | j | j } t| jdd | jd d | jd d  S7b#tj$r} t| d} ~ wwxYww)z9Execute via HTTP POST /execute (existing implementation).z/execute)rdrUrXNrVrW )jsonrUr_stdoutr;stderr exit_code)rrr) rTr7_httppost_request_headersraise_for_statusrrr=httpxHTTPStatusErrorr) rHrdrUrVrWrXr_r&urlpayloadresponser?es rBrlzAsyncSandbox._run_https335 x(#  ? GEN ? GEN !\\//44"  55g> 5H  % % '==?D"xx"-xx"-((;3 $$  %a (  s<*DAC5C6A!CDCC>- C99C>>Dr) stdout_offset stderr_offsetr_cKddlm}|j}|jj}||d}|jj |} | r| |d<||||fi|d{\} } t | | ||||S7w)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)rrr_N)rqrr)rvrrTr7rxrzr) rHrqrrr_rr&rreconnect_kwargsrrrs rB reconnectzAsyncSandbox.reconnects2 L335 ,,''+*, 11': *0 Y '$=   %  %   G"   !''    sA"A?$A=%A?rUr_c K|j}|d}t|tr|jd}dd|fi} |jj j |d|i|||jj|d{}|jy7#tj$r} t| Yd} ~ yd} ~ wwxYww)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-8filepath)paramsfilesrUr_N) rT isinstancer#encoder7rrrrrrr) rHrcontentrUr_r&rrrrs rBwritezAsyncSandbox.write1s,335 w' gs #nnW-G&'*+ )!\\//44~ 55g> 5H  % % '$$ ) %a ( ( )sB=C AB B B C B C3 C>C CC cK|j}|d} |jjj|d|i||jj |d{}|j |j S7 #tj$rI}|jjdk(rtd|d|jdd |t|d}~wwxYww) 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)rrUr_NizFile 'z' not found in sandbox ''r) resource_type)rTr7rr=rrrrrr status_coderr$r)rHrrUr_r&rrrs rBreadzAsyncSandbox.read\s*335 y) !\\//33~ 55g> 4H  % % '## #$$ zz%%,+TF":499+QG"( &a (  s<C"AB B!BC"BCACCC" local_portmax_reconnectsr_cxKd|cxkrdksntd|d|rd|cxkrdksntd|d|j}|jj}t ||||||}t j }|jd|jjd{|S7w)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. rsiz-remote_port must be between 1 and 65535 (got )z,local_port must be between 1 and 65535 (got rN) rjrTr7rxrruget_running_looprun_in_executor_tunnel_start) rH remote_portrrr_r&rtloops rBtunnelzAsyncSandbox.tunnelsDK(5(? }AN  a:66>zl!L 335 ,,''    !)  '')""4)9)9::: ;sB/B:1B82B:expires_in_secondsr_cpK|jj|j|||d{S7w)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)r7servicer$)rHportrr_s rBrzAsyncSandbox.services>2\\)) II 1 *    -646r content_typecontent_dispositionr_ctK|jj|j|||||d{S7w)aCreate a link that downloads one file from this sandbox. The link carries its own token, so anyone holding the URL can fetch that one file without a LangSmith credential. Fetching wakes a stopped sandbox. Do not modify the file after minting a link for it. The link is pinned to a path, not to a snapshot of the contents, so a later write to that path may or may not be reflected in what the link serves. Write a new file and mint a new link when the contents change. Args: path: File path inside the sandbox. expires_in_seconds: Link TTL in seconds. Omit for a link that never expires. content_type: Content-Type to serve the file as. content_disposition: Content-Disposition to serve the file with, either ``"attachment"`` or ``"inline"``. headers: Optional per-request header overrides. Returns: DownloadURL with the link and its expiry, if any. Raises: ResourceNotFoundError: If sandbox not found. ValueError: If expires_in_seconds is not positive. SandboxClientError: For other errors. rN)r7generate_download_urlr$)rHrrrrr_s rBrz"AsyncSandbox.generate_download_urlsEL\\77 II 1% 3 8    s /868xcK|jj|j||d{}|j|_|j|_y7'w)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)r7 start_sandboxr$r)r&)rHrUr_ refresheds rBstartzAsyncSandbox.startsS*,,44 IIw5   && &44  s,AA(Ar_c|K|jj|j|d{d|_y7 w)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)r7 stop_sandboxr$r)rHr_s rBstopzAsyncSandbox.stop)s8ll'' 7'CCC   Ds +<: <cnK|jj|j|d{y7w)zDelete this sandbox. Args: headers: Optional per-request header overrides. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rN)r7rMr$rs rBdeletezAsyncSandbox.delete8s)ll))$))W)EEEs +535cpK|jj|j|||d{S7w)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)r7capture_snapshotr$)rHr$rUr_s rBrzAsyncSandbox.capture_snapshotDs>.\\22 II  3    r)T)r?zdict[str, Any]r@rrAr8returnr")r@zOptional[SandboxClient]rr )rr")rOzOptional[type]rPzOptional[BaseException]rQz Optional[Any]rNone)rr#)rdr#rUintrVOptional[dict[str, str]]rWr%rXr#rYOptional[Callable[[str], Any]]rZrr[rr\r8r]rr^r8r_RequestHeadersr`z Literal[True]rr)rdr#rUrrVrrWr%rXr#rYrrZrr[rr\r8r]rr^r8r_rr`zLiteral[False]rr)rdr#rUrrVrrWr%rXr#rYrrZrr[rr\r8r]rr^r8r_rr`r8r*Union[ExecutionResult, AsyncCommandHandle])rdr#rUrrVrrWr%rXr#r`r8rYrrZrr[rr\r8r]rr^r8r_rrr)rdr#rUrrVrrWr%rXr#r_rrr) rqr#rrrrr_rrr) rr#rzUnion[str, bytes]rUrr_rrr)rr#rUrr_rrbytes) rrrrrrr_rrr)rrrrr_rrr) rr#rr-rr%rz$Optional[DownloadContentDisposition]r_rrr)rUrr_rrr)r_rrr)r$r#rUrr_rrr))__name__ __module__ __qualname____doc____annotations__r&r'r)r*r+r,r.r/r0r1r2r3r4rr7r9 classmethodrCrGrKrRrTr rerkrlrrrrrrrrrrr<rDrBr"r",s_(V I#'M='B FC$(NM( $J $ $J $&*m*/3}3 $J $!%K%E=#I}#'+}+#(UD"AG AE48L$8 ! ! ! #!  !  ! ! F<@" H   )    "* (+ 4747#&"%!  &   22!  !$ (+ 4747#&"%!! ! & !  !!2!2!!!!!! !! !!!,(,! 4848#("&s s  s & s  s s 2s 2s s !s s s  s s 4!s @ #("&Z#Z# Z# & Z#  Z#Z#Z#2Z#2Z#Z#!Z#Z#Z# Z# 4!Z#x'' ' & '  '' ' 'Z"& 4 4  4  4  4  4 v"& ))))#))  ))  )) ))X,.))%()8F) )^"& 66 6  6  6 6x#&"&         H-1&*DH"&- - * - $ - B -  -  - d"& 55 5  567;  9= F "&         rDr")0r __future__rrurcollections.abcr dataclassesrrtypingrrr r r r r rlangsmith.sandbox._exceptionsrrrlangsmith.sandbox._helpersrlangsmith.sandbox._modelsrrrrrrrlangsmith.sandbox._tunnelrrvrrrlangsmith.sandbox._async_clientrlangsmith.sandbox._clientrrFr r#rr"r<rDrBrs6" #(SSS  A2 B72'#s(+, s  s   s  rD