jydZddlmZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZddlZddlmZmZmZddlmZdd lmZmZmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$e rdd l%m&Z&dd l'm(Z(ddl)m*Z*e ee+e+fZ,eGddZ-y)z?Sandbox class for interacting with a specific sandbox instance.) annotations)Mapping) dataclassfield) TYPE_CHECKINGAnyCallableLiteralOptionalUnionoverloadN)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxRetryableConnectionError)handle_sandbox_http_error) CommandHandleDownloadContentDisposition DownloadURLExecutionResult ServiceURLSnapshot_StreamEndedBeforeStarted)Tunnel)WEBSOCKETS_AVAILABLEconnect_deadlineopen_timeout_for)AsyncSandboxClient AsyncSandbox) SandboxClientceZdZUdZded<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)VSandboxaRepresents an active sandbox for running commands and file operations. This class is typically obtained from SandboxClient.sandbox() and supports the 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: with client.sandbox(snapshot_id="") as sandbox: result = 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)a4Create a Sandbox from API response dict. Args: data: API response dictionary containing sandbox data. client: Parent SandboxClient for operations. auto_delete: Whether to delete the sandbox on context exit. Returns: Sandbox instance. r$r&r'r)r(r*r+r,r.r/r0r1r2r3r4r7r9)get)clsdataclient auto_deletes d/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langsmith/sandbox/_sandbox.py from_dictzSandbox.from_dicths2" &"% ((?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)a"Create an AsyncSandbox 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: AsyncSandboxClient to use for operations. If not provided, one is created with the same configuration as this sandbox's client (see :meth:`SandboxClient.to_async`). Returns: AsyncSandbox referring to the same sandbox. rrr$r&r'r)r*r+r,r.r/r0r1r2r3r4r7r9Fr<) langsmith.sandbox._async_sandboxrr$r&r'r)r*r+r,r.r/r0r1r2r3r4r7to_async)selfr@rs rBrGzSandbox.to_asyncs  B  ,, ww ;;   ..     "22 '+&D&D  (( ** nn #44 %0Fdll6K6K6M !  rDc|S)zEnter context manager.r<rHs rB __enter__zSandbox.__enter__s rDc|jr' |jj|jyy#t$rYywxYw)z6Exit context manager, optionally deleting the sandbox.N)r9r7delete_sandboxr$ Exception)rHexc_typeexc_valexc_tbs rB__exit__zSandbox.__exit__sA     ++DII6   s%5 AAcd|jstd|jd|jS)aReturn the dataplane URL. The client does not gate on lifecycle status: a stopped sandbox is resumed by the platform when the dataplane request arrives, so only the presence of a URL is required here. A genuinely not-ready box surfaces the server's ``SandboxNotReadyError`` from the request itself. Returns: The dataplane URL. Raises: DataplaneNotConfiguredError: If dataplane_url is not configured. z Sandbox 'zW' does not have a dataplane_url configured. Runtime operations require a dataplane URL.)r&rr$rJs rB_require_dataplane_urlzSandbox._require_dataplane_urls=!!-DII;'>> !!!rD.) timeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnect ttl_secondsptyheaderswaitc yNr<rHcommandrUrVrWrXrYrZr[r\r]r^r_r`s rBrunz Sandbox.runs"rD) rUrVrWrXrYrZr[r\r]r^r_c yrbr<rcs rBrez Sandbox.runs"rD<z /bin/bashi,iXc | s|s|r td|j| xs|xs|}|r|j|||||| |||| | | |  Str|j|||||ddd|| | | |  S|j |||||| S)aS Execute a command in the sandbox. 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 a CommandHandle immediately for streaming output, kill, stdin input, and reconnection. Cannot be combined with on_stdout/on_stderr callbacks. Returns: ExecutionResult when wait=True (default). CommandHandle 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_TN)rUrVrWrXr_) ValueErrorrT_run_wsr _run_http)rHrdrUrVrWrXrYrZr[r\r]r^r_r`use_wss rBrez Sandbox.runs@iR  ##%3Y3) <<##)#5'  $ <<)#5'  ~~    rD)r[r\r]r^r_c ddl}ddlm}ddlm}|j }|j j}|j}|||||| | | | d }|j j| }|r||d<d}t} t||d<||||fi|\}} t|||||} |s|S|j*S#ttf$r|d z }|tjkDrt!tj"d |d z zztj$}|%||j'z }|dkrt!||}|j)|YnwxYw) z"Execute via WebSocket /execute/ws.rN) run_ws_stream)uuid7) command_idrUrVrWrXr[r\r]r^r_ open_timeout)rYrZ)timelangsmith.sandbox._ws_executernlangsmith.uuidrorTr7_api_keyhex_ws_default_headersrrrrrMAX_AUTO_RECONNECTSmin _BACKOFF_BASE _BACKOFF_MAX monotonicsleepresult)rHrdrUrVrWrXr`rYrZr[r\r]r^r_rtrnror&api_keyrp ws_kwargsmergedattemptdeadline msg_streamcontrolhandlebackoff remainings rBrjzSandbox._run_wsss$ ?(335 ,,'' W[[ %("4& % 11': #)Ii #%(8(BIn %"/# # J  $&'' *M}}-*/ $ 1 ]>>>!//113EF!..' (4>>+; ;I A~!'95G 7#% $#s CBEEc|j}|d}|||d} ||| d<||| d< |jjj|| |dz|jj |} | j | j } t| jdd | jd d | jd d  S#tj$r} t| d} ~ wwxYw)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 rBrkzSandbox._run_https335 x(#  ? GEN ? GEN ||)).."  55g> /H  % % '==?D"xx"-xx"-((;3  $$  %a (  sB"CC2! C--C2r) stdout_offset stderr_offsetr_cddlm}|j}|jj}||d}|jj |} | r| |d<||||fi|\} } 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: A CommandHandle 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)rrr_)rprr)rurrTr7rwryr) rHrprrr_rr&rreconnect_kwargsrrrs rB reconnectzSandbox.reconnects2 F335 ,,''+*, 11': *0 Y '1       G   !''   rDrUr_c ~|j}|d}t|tr|jd}dd|fi} |jj j |d|i|||jj|}|jy#tj$r} t| Yd} ~ yd} ~ wwxYw)a*Write content to a file in the sandbox. 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 rBwritez Sandbox.write-s,335 w' gs #nnW-G&'*+ )||))..~ 55g> /H  % % '$$ ) %a ( ( )sABB<' B77B<c|j}|d} |jjj|d|i||jj |}|j |j S#tj$rI}|jjdk(rtd|d|jdd |t|d }~wwxYw) aRead a file from the sandbox. 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_izFile 'z' not found in sandbox ''r) resource_typeN)rTr7rr=rrrrrr status_coderr$r)rHrrUr_r&rrrs rBreadz Sandbox.readXs*335 y) ||))--~ 55g> .H  % % '## #$$ zz%%,+TF":499+QG"( &a (  sAA77C ACC local_portmax_reconnectsr_cd|cxkrdksntd|d|rd|cxkrdksntd|d|j}|jj}t ||||||}|j |S)aOpen 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.). Use as a context manager for automatic cleanup:: with sandbox.tunnel(remote_port=5432) as t: conn = psycopg2.connect(host="127.0.0.1", port=t.local_port) Or manage the lifecycle explicitly:: t = sandbox.tunnel(remote_port=5432) # ... use tunnel ... t.close() 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: A Tunnel instance (context manager). Raises: ValueError: If port values are out of range. DataplaneNotConfiguredError: If dataplane_url is not configured. SandboxNotReadyError: If sandbox is not ready. rriz-remote_port must be between 1 and 65535 (got )z,local_port must be between 1 and 65535 (got r)rirTr7rwr_start)rH remote_portrrr_r&rts rBtunnelzSandbox.tunnelsPK(5(? }AN  a:66>zl!L 335 ,,''    !)    rDexpires_in_secondsr_cT|jj|j|||S)aGet an authenticated URL for a service running in this sandbox. Returns a :class:`ServiceURL` whose properties 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: ServiceURL 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. r)r7servicer$)rHportrr_s rBrzSandbox.services02||## II 1 $  rDr content_typecontent_dispositionr_cX|jj|j|||||S)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. r)r7generate_download_urlr$)rHrrrrr_s rBrzSandbox.generate_download_urls7L||11 II 1% 3 2  rDxc|jj|j||}|j|_|j|_y)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 rBstartz Sandbox.startsE*LL.. IIw/  && &44rDr_c`|jj|j|d|_y)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. rstoppedN)r7 stop_sandboxr$r)rHr_s rBstopz Sandbox.stop*s( !!$))W!=  rDcR|jj|j|y)zDelete this sandbox. Args: headers: Optional per-request header overrides. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rN)r7rMr$rs rBdeletezSandbox.delete9s ##DIIw#?rDcT|jj|j|||S)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. r)r7capture_snapshotr$)rHr$rUr_s rBrzSandbox.capture_snapshotEs0.||,, II  -  rD)T)r?zdict[str, Any]r@r rAr8returnr")r@zOptional[AsyncSandboxClient]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, CommandHandle])rdr#rUrrVrrWr%rXr#r`r8rYrrZrr[rr\r8r]rr^r8r_rrr)rdr#rUrrVrrWr%rXr#r_rrr) rpr#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 rerjrkrrrrrrrrrrr<rDrBr"r"*s_&R I#'M='B FC$(NM( $J $ $J $&*m*/3}3 $J $!%K%E=#I}#'+}+#trsE"#(SSS  A- B=7'#s(+, v  v   v  rD