a ieLdZddlmZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZddlZddlmZmZmZmZddlmZddlmZmZdd lmZerdd lmZeGd d ZdS) z0AsyncSandbox class for async sandbox operations.) annotationsN) dataclassfield) TYPE_CHECKINGAnyCallableLiteralOptionalUnionoverload)DataplaneNotConfiguredErrorResourceNotFoundErrorSandboxConnectionErrorSandboxNotReadyError)handle_sandbox_http_error)AsyncCommandHandleExecutionResult) AsyncTunnel)AsyncSandboxClientc eZdZUdZded<ded<dZded<dZded<d Zded <dZded <dZ ded <dZ ded <e ddZ ded<e ddZ ded<e dZd[dZd\dZd]d#Zd^d$Zed%d%d%d%d%d%d%d%d%d%d%d& d_d8Zed%d%d%d%d%d%d%d%d%d%d9 d`d<Zd=ddd>ddd?dd@ddd& dadBZd?dd@ddCdbdDZdcdEZdFdFdGdddKZd=dLdedPZd=dLdfdRZdFdSdTdgdYZdS)h AsyncSandboxarRepresents 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). 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. 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_urlidreadystatusstatus_message created_at updated_atF)reprdefaultr_clientTbool _auto_deletedatadict[str, Any]client auto_deletereturnc b||dd|dd|d|d|dd|d|d |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. rrrrrrr r!r") rrrrrr r!r"r%r')get)clsr(r*r+s C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langsmith/sandbox/_async_sandbox.py from_dictzAsyncSandbox.from_dictCs"s&"%%((?B77((?33xx~~88Hg..88$455xx --xx --$    c K|S)zEnter async context manager.selfs r1 __aenter__zAsyncSandbox.__aenter__as  r3exc_typeOptional[type]exc_valOptional[BaseException]exc_tb Optional[Any]NonecK|jr9 |j|jd{VdS#t$rYdSwxYwdS)zDI>>> !!r3.) timeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnect ttl_secondsptywaitcommandrFintrGOptional[dict[str, str]]rHrIrJOptional[Callable[[str], Any]]rKrLrMrNrOrP Literal[True]rc  KdSNr5 r7rQrFrGrHrIrJrKrLrMrNrOrPs r1runzAsyncSandbox.runs #r3 rFrGrHrIrJrKrLrMrNrOLiteral[False]rc  KdSrWr5rXs r1rYzAsyncSandbox.runs !Sr3<z /bin/bashi,iX*Union[ExecutionResult, AsyncCommandHandle]c K| s|s|rtd|| p|p|} | r'||||||| |||| | |  d{VS ||||||ddd|| | |  d{VS#ttt t f$r#||||||d{VcYSwxYw)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.) rFrGrHrIrPrJrKrLrMrNrONT)rFrGrHrI) ValueErrorrE_run_wsr ImportErrorOSError TypeError _run_http)r7rQrFrGrHrIrJrKrLrMrNrOrPuse_wss r1rYzAsyncSandbox.runs~  i R  ##%%%3Y3)  ##)#5'&         " )#5'&         ' WiH    (    s&B=CC)rLrMrNrOc  Kddlm} |}|jj}| |||||||||| | | |  d{V\}}t |||}|d{V|s|S|jd{VS)z"Execute via WebSocket /execute/ws.r)run_ws_stream_asyncrZN)langsmith.sandbox._ws_executerhrEr%_api_keyr_ensure_startedresult)r7rQrFrGrHrIrPrJrKrLrMrNrOrhrapi_key msg_streamcontrolhandles r1razAsyncSandbox._run_wss" FEEEEE3355 ,'$7$7   %1#% % %        G $J>>$$&&&&&&&&& M]"""""""r3cK|}|d}|||d}|||d<|||d< |jj|||dzd{V} | | } t | dd | d d | d d  S#tj $r} t| d} ~ wwxYw)z9Execute via HTTP POST /execute (existing implementation).z/execute)rQrFrINrGrH )jsonrFstdoutr.stderr exit_code)rtrurv) rEr%_httppostraise_for_statusrsrr/httpxHTTPStatusErrorr) r7rQrFrGrHrIrurlpayloadresponser(es r1rezAsyncSandbox._run_httpMsG3355 (((# #  ? GEN ? GEN !\/44'7R<5H  % % ' ' '==??D"xx"--xx"--((;33  $    %a ( ( (  sB CC6!C11C6r stdout_offset stderr_offset command_idrrcKddlm}|}|jj}||||||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_asyncrN)rrr)rirrEr%rjr) r7rrrrrrmrnros r1 reconnectzAsyncSandbox.reconnectrs0 LKKKKK3355 ,'$=$=   '' % % %        G"   !''     r3)rFpathcontentUnion[str, bytes]ctK|}|d}t|tr|d}dd|fi} |jj|d|i||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)paramsfilesrFN) rE isinstancerencoder%rxryrzr{r|r) r7rrrFrr}rrrs r1writezAsyncSandbox.writes*3355 ''' gs # # .nnW--G&'*+ )!\/44VTN%5H  % % ' ' ' ' '$ ) ) ) %a ( ( ( ( ( ( ( ( ( )s ?BB7B22B7bytescnK|}|d} |jj|d|i|d{V}||jS#t j$rC}|jj 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)rrFNizFile 'z' not found in sandbox ''r) resource_type) rEr%rxr/rzrr{r|r status_coderrr)r7rrFrr}rrs r1readzAsyncSandbox.reads &3355 ))) !\/33VTNG4H  % % ' ' '# #$   z%,,+GTGG49GGG"( &a ( ( (  sAA""B41>B//B4 local_portmax_reconnects remote_portrrrcpKd|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. iz-remote_port must be between 1 and 65535 (got )z,local_port must be between 1 and 65535 (got rN) r`rEr%rjrasyncioget_running_looprun_in_executor_tunnel_start)r7rrrrrmtloops r1tunnelzAsyncSandbox.tunnelsBK((((5((((N NNN   a:66666666LzLLL 3355 ,'    !)    '))""4)9:::::::::r3)T)r(r)r*rr+r&r,r)r,r)r9r:r;r<r=r>r,r?)r,r)rQrrFrRrGrSrHrrIrrJrTrKrTrLrRrMr&rNrRrOr&rPrUr,r)rQrrFrRrGrSrHrrIrrJrTrKrTrLrRrMr&rNrRrOr&rPr[r,r)rQrrFrRrGrSrHrrIrrJrTrKrTrLrRrMr&rNrRrOr&rPr&r,r^)rQrrFrRrGrSrHrrIrrPr&rJrTrKrTrLrRrMr&rNrRrOr&r,r^) rQrrFrRrGrSrHrrIrr,r)rrrrRrrRr,r)rrrrrFrRr,r?)rrrFrRr,r)rrRrrRrrRr,r)__name__ __module__ __qualname____doc____annotations__rrrr r!r"rr%r' classmethodr2r8rCrEr rYrarerrrrr5r3r1rrs0III#'M''''BF$(N(((( $J$$$$ $J$$$$#(%UD"A"A"AGAAAAE4888L8888 !     [ :    """", (+ 4747#&!X" (+ 4747#&!!!!!X!*(,! 4848#(ppppppz #(,#,#,#,#,#,#\####R , , , , , , f $)$)$)$)$)$)L79$$$$$$T 44444444r3r)r __future__rr dataclassesrrtypingrrrr r r r r{langsmith.sandbox._exceptionsr rrrlangsmith.sandbox._helpersrlangsmith.sandbox._modelsrrlangsmith.sandbox._tunnelrlangsmith.sandbox._async_clientrrr5r3r1rsm66""""""((((((((SSSSSSSSSSSSSSSSSS  A@@@@@211111CBBBBBB CCCCCCC CCCr3