bi#dZddlmZddlmZmZddlmZmZm Z m Z ddl Z ddl m Z mZmZddlmZddlmZerdd lmZeGd d Zy) z0AsyncSandbox class for async sandbox operations.) annotations) dataclassfield) TYPE_CHECKINGAnyOptionalUnionN)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxConnectionError)handle_sandbox_http_error)ExecutionResult)AsyncSandboxClientc:eZdZUdZded<ded<dZded<dZded<dZded <dZded <e d d Z d ed<e d d Z ded<e d ddZ ddZ d dZd!dZddddd d"dZdd d#dZddd$dZy)% 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). template_name: Name of the template used to create this sandbox. dataplane_url: URL for data plane operations (file I/O, command execution). id: Unique identifier (UUID). Remains constant even if name changes. May be None for resources created before ID support was added. created_at: Timestamp when the sandbox was created. updated_at: Timestamp when the sandbox was last updated. Example: async with await client.sandbox(template_name="python-sandbox") as sandbox: result = await sandbox.run("python --version") print(result.stdout) strname template_nameN Optional[str] dataplane_urlid created_at updated_atF)reprdefaultr_clientTbool _auto_deletec ||jdd|jdd|jd|jd|jd|jd||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. rrrrrr)rrrrrrrr)get)clsdataclient auto_deletes j/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/langsmith/sandbox/_async_sandbox.py from_dictzAsyncSandbox.from_dict8sb"&"%((?B7((?3xx~xx -xx -$  cK|Sw)zEnter async context manager.selfs r& __aenter__zAsyncSandbox.__aenter__Ts  scK|jr/ |jj|jd{yy7#t$rYywxYww)z> !!!r(<z /bin/bash)timeoutenvcwdshellc6K|j}|d}|||d}|||d<|||d< |jjj|||dzd{} | j | j } t | jdd | jd d | jd d  S7b#tj$r!} td|jd| | d} ~ wtj$r} t| d} ~ wwxYww)a0Execute 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". Returns: ExecutionResult with stdout, stderr, and exit_code. Raises: DataplaneNotConfiguredError: If dataplane_url is not configured. SandboxOperationError: If command execution fails. SandboxConnectionError: If connection to sandbox fails. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z/execute)commandr8r;Nr9r: )jsonr8stdoutr stderr exit_code)r@rArBFailed to connect to sandbox '': )r6r_httppostraise_for_statusr?rr!httpx ConnectErrorr rHTTPStatusErrorr ) r,r=r8r9r:r;rurlpayloadresponser#es r&runzAsyncSandbox.runvs08335 x(#  ? GEN ? GEN !\\//44'7R<5H  % % '==?D"xx"-xx"-((;3  !! (0 3qcB $$  %a (  sG*D.CB>A!C=D>CDC//D DDD)r8cK|j}|d}t|tr|jd}dd|fi} |jj j |d|i||d{}|jy7#tj$r!}td|jd||d}~wtj$r}t|Yd}~yd}~wwxYww) a+Write 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. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z/uploadzutf-8filepath)paramsfilesr8NrDrE)r6 isinstancerencoderrFrGrHrIrJr rrKr ) r,rScontentr8rrLrUrNrOs r&writezAsyncSandbox.writes*335 w' gs #nnW-G&'*+ )!\\//44VTN%5H  % % '!! (0 3qcB $$ ) %a ( ( )sM=C#.B.B/BC#BC B55C  CC#C  C#cK|j}|d} |jjj|d|i|d{}|j |j S7 #t j$r!}td|jd||d}~wt j$rI}|jjdk(rtd|d |jd d |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. SandboxNotReadyError: If sandbox is not ready. SandboxClientError: For other errors. z /downloadrS)rTr8NrDrEizFile 'z' not found in sandbox ''rR) resource_type)r6rrFr!rHrXrIrJr rrKrN status_coder r )r,rSr8rrLrNrOs r&readzAsyncSandbox.reads &335 y) !\\//33VTNG4H  % % '## #  !! (0 3qcB $$ zz%%,+TF":499+QG"( &a (  sGC:-A)A'A)&C:'A))C7<BC7.AC22C77C:)T)r#zdict[str, Any]r$rr%rreturnr)r_r)r1zOptional[type]r2zOptional[BaseException]r3z Optional[Any]r_None)r_r) r=rr8intr9zOptional[dict[str, str]]r:rr;rr_r)rSrrXzUnion[str, bytes]r8rar_r`)rSrr8rar_bytes)__name__ __module__ __qualname____doc____annotations__rrrrrrr classmethodr'r-r4r6rPrYr^r*r(r&rrs\* I#'M='B  $J $ $J $#(UD"AG AE48L$8 !   #      6   )    "((,! :: : & :  :: :B ()()#()  ()  ()T79(r(r)rf __future__r dataclassesrrtypingrrrr rIlangsmith.sandbox._exceptionsr r r langsmith.sandbox._helpersr langsmith.sandbox._modelsrlangsmith.sandbox._async_clientrrr*r(r&rpsL6"(66  A5B mm mr(