XiRdZddlmZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZddlZddlmZmZmZmZddlmZdd lmZmZdd lmZe rdd lmZe eeefZ eGd d Z!y)z?Sandbox class for interacting with a specific sandbox instance.) annotations)Mapping) dataclassfield) TYPE_CHECKINGAnyCallableLiteralOptionalUnionoverloadN)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxConnectionErrorSandboxNotReadyError)handle_sandbox_http_error) CommandHandleExecutionResult)Tunnel) SandboxClientcDeZdZUdZded<ded<dZded<dZded<d Zded <dZded <dZ ded <dZ ded <dZ ded<dZ ded<dZ ded<eddZded<eddZded<e d3 d4dZd5dZ d6dZd7dZeddddddddddddd d8dZedddddddddddd d9d!Zd"ddd#ddd$dd%dddd d:d&Zd$dd%ddd' d;d(Z dd.Zd"dd- d?d/Zd*d0dd1 d@d2Zy)ASandboxa/Represents 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). template_name: Name of the template used to create this sandbox. 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". 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. ttl_seconds: Maximum lifetime TTL in seconds (0 means disabled). idle_ttl_seconds: Idle timeout TTL in seconds (0 means disabled). expires_at: Computed expiration timestamp, or None if no TTL is set. Example: with client.sandbox(template_name="python-sandbox") as sandbox: result = sandbox.run("python --version") print(result.stdout) strname template_nameN Optional[str] dataplane_urlidreadystatusstatus_message created_at updated_atz Optional[int] ttl_secondsidle_ttl_seconds expires_atF)reprdefaultr_clientTbool _auto_deletec|||jdd|jdd|jd|jd|jdd|jd|jd |jd |jd |jd |jd || 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. rrrrr rr!r"r#r$r%r&) rrrrr r!r"r#r$r%r&r)r+)get)clsdataclient auto_deletes f/root/rosshost/parking-map/parking-api/venv/lib/python3.12/site-packages/langsmith/sandbox/_sandbox.py from_dictzSandbox.from_dictLs"&"%((?B7((?3xx~88Hg.88$45xx -xx -/!XX&89xx -$  c|S)zEnter context manager.selfs r3 __enter__zSandbox.__enter__ms r5c|jr' |jj|jyy#t$rYywxYw)z6Exit context manager, optionally deleting the sandbox.N)r+r)delete_sandboxr Exception)r9exc_typeexc_valexc_tbs r3__exit__zSandbox.__exit__qsA     ++DII6   s%5 AAc|jdk7r&td|jd|jd|jst d|jd|jS)aValidate and return the dataplane URL. Returns: The dataplane URL. Raises: SandboxNotReadyError: If sandbox status is not "ready". DataplaneNotConfiguredError: If dataplane_url is not configured. rz Sandbox 'z' is not ready (status: z5). Wait for status 'ready' before running operations.zW' does not have a dataplane_url configured. Runtime operations require a dataplane URL.)r rrrrr8s r3_require_dataplane_urlzSandbox._require_dataplane_urls{ ;;' !&DII;&>t{{mLEE !!-DII;'>> !!!r5.) timeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnectr$ptyheaderswaitc yNr7r9commandrDrErFrGrHrIrJrKr$rLrMrNs r3runz Sandbox.runs"r5) rDrErFrGrHrIrJrKr$rLrMc yrPr7rQs r3rSz Sandbox.runs"r5<z /bin/bashi,iXc D| s|s|r td|j| xs|xs|}|r|j|||||| |||| | | |  S |j|||||ddd|| | | |  S#ttt t f$r|j|||||| cYSwxYw)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.) rDrErFrGrNrHrIrJrKr$rLrMTN)rDrErFrGrM) ValueErrorrC_run_wsr ImportErrorOSError TypeError _run_http)r9rRrDrErFrGrHrIrJrKr$rLrMrNuse_wss r3rSz Sandbox.runs@iR  ##%3Y3) <<##)#5'  & <<)#5'  ' WiH >> "  sA,,0BB)rJrKr$rLrMc ddlm}|j}|jj}||||||| | | | d }| | |d<||||fi|\}}t |||}|s|S|j S)z"Execute via WebSocket /execute/ws.r) run_ws_stream) rDrErFrGrHrIrJrKr$rLrM)langsmith.sandbox._ws_executer_rCr)_api_keyrresult)r9rRrDrErFrGrNrHrIrJrKr$rLrMr_rapi_key ws_kwargs msg_streamcontrolhandles r3rXzSandbox._run_ws3s$ @335 ,,''""("4& %   #*Ii +       Gz7D9M}}r5c|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)rRrDrGNrErF )jsonrDrMstdoutr-stderr exit_code)rkrlrm) rCr)_httppost_request_headersraise_for_statusrjrr.httpxHTTPStatusErrorr) r9rRrDrErFrGrMrurlpayloadresponser0es r3r\zSandbox._run_httpgs335 x(#  ? GEN ? GEN ||)).."  55g> /H  % % '==?D"xx"-xx"-((;3  $$  %a (  sB"CC2! C--C2r) stdout_offset stderr_offsetrMcddlm}|j}|jj}||d}|||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)ryrzrM) command_idryrz)r`r|rCr)rar) r9r}ryrzrMr|rrcreconnect_kwargsrerfs r3 reconnectzSandbox.reconnects2 F335 ,,''+*,   *1 Y '1       G   !''   r5)rDrMc ~|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)paramsfilesrDrMN) rC isinstancerencoder)rorprqrrrsrtr) r9rcontentrDrMrrurrwrxs r3writez Sandbox.writes,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)rrDrMizFile 'z' not found in sandbox ''r) resource_typeN)rCr)ror.rqrrrrsrtrw status_coderrr)r9rrDrMrrurwrxs r3readz Sandbox.reads*335 y) ||))--~ 55g> .H  % % '## #$$ zz%%,+TF":499+QG"( &a (  sAA77C ACC local_portmax_reconnectsrMcd|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. iz-remote_port must be between 1 and 65535 (got )z,local_port must be between 1 and 65535 (got r)rWrCr)rar_start)r9 remote_portrrrMrrcts r3tunnelzSandbox.tunnelsPK(5(? }AN  a:66>zl!L 335 ,,''    !)    r5)T)r0zdict[str, Any]r1rr2r*returnr)rr)r>zOptional[type]r?zOptional[BaseException]r@z Optional[Any]rNone)rr)rRrrDintrEOptional[dict[str, str]]rFrrGrrHOptional[Callable[[str], Any]]rIrrJrrKr*r$rrLr*rMRequestHeadersrNz Literal[True]rr)rRrrDrrErrFrrGrrHrrIrrJrrKr*r$rrLr*rMrrNzLiteral[False]rr)rRrrDrrErrFrrGrrHrrIrrJrrKr*r$rrLr*rMrrNr*r%Union[ExecutionResult, CommandHandle])rRrrDrrErrFrrGrrNr*rHrrIrrJrrKr*r$rrLr*rMrrr)rRrrDrrErrFrrGrrMrrr) r}rryrrzrrMrrr) rrrzUnion[str, bytes]rDrrMrrr)rrrDrrMrrbytes) rrrrrrrMrrr)__name__ __module__ __qualname____doc____annotations__rrr r!r"r#r$r%r&rr)r+ classmethodr4r:rArCr rSrXr\rrrrr7r5r3rrs'6 I#'M='B FC$(NM( $J $ $J $!%K%&*m* $J $#trsnE"#(SSS  A-7'#s(+, vv vr5