j`0dZddlmZddlZddlZddlZddlmZmZm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZmZmZddlmZej4eZ dd lmZdd l m!Z!m"Z"dd l#mZ$d Z%d2d Z'e'ddZ(e'ddZ)e'ddZ*e'ddZ+e'ddZ,d3dZ-d4dZ.dZ/dZ0dZ1d5dZ2 d6 d7dZ3GddZ4Gd d!Z5e6hd"Z7d#Z8d8d$Z9d9d%Z:d&d'd:d(Z;d&dddd)ddd*d d+d de(d, d;d-Zdddd. d>d1Z?y#e&$rdxZZ$dxZ!Z"d Z%Y&wxYw)?z> / //r1c|tSt|tjz d}t|St t|S)zDPer-attempt open timeout, clamped so it cannot outlive ``deadline``.g)WS_OPEN_TIMEOUTmaxr.r/min)deadline remainings r!open_timeout_forr9Fs?Ht~~//5I  **r1zWebSocket-based execution requires the 'websockets' package, which ships with langsmith by default. Reinstall with: pip install --upgrade langsmithcNtsttttt fS)zCReturn the cached sync websockets symbols, or raise if unavailable.)WEBSOCKETS_AVAILABLE ImportError_MISSING_WEBSOCKETS_MSG_ws_connect_syncrrr0r1r!_ensure_websocketsr?Vs 122 -/? ??r1cNtsttttt fS)zDReturn the cached async websockets symbols, or raise if unavailable.)r;r<r=_ws_connect_asyncrrr0r1r!_ensure_websockets_asyncrB]s 122 .0@ @@r1cP|jddjdd}|dS)zi@ctt|dddd}t|tr|dtj dd}nt|t r |dt}ny t j|}t|tr|jdnd}t|tsy|jd}|jd }t|t r|r|gng}t|t r|r|jd ||r&d j|t|d krd zSdzSdS#ttf$rYywxYw)z=Return bounded, user-facing detail from a rejected handshake.responseNbodyzutf-8rD)errorsdetailmessageerror_idz error_id=z ())getattr isinstancebytes_MAX_HANDSHAKE_ERROR_BYTESdecoder}riloads TypeErrorrdictgetappendjoinlen)excrpayloadrrrpartss r!_handshake_server_detailrs, 73 D164 @D$//077 7R D# //0**T"'1$&?W[[ "TF fd #jj#Gzz*%H#GS1gWI2E(C X y +,AF499U SZ!^r =P =PDP z "s D22EEc tt|dddd}t|}|rd|nd|}|dk(rtd|d||tvrt d|d |||td |d ||td |d ||)uRaise a clear error when the WebSocket upgrade handshake fails. Covers both a rejection carrying an HTTP status (``InvalidStatus`` — most commonly 404 when the server lacks the /execute/ws endpoint) and a response that isn't valid HTTP at all (``InvalidMessage``, "did not receive a valid HTTP response") — e.g. a stopped or recycled sandbox dataplane answering the upgrade with garbage. Both subclass ``InvalidHandshake``. rN status_codez: izJThe sandbox server does not support WebSocket command execution (endpoint z returned 404). Ensure the server is updated to a version that supports the /execute/ws endpoint, or use run() without wait=False or callbacks.z7WebSocket upgrade temporarily rejected by server (HTTP rz+WebSocket upgrade rejected by server (HTTP z): zWebSocket upgrade to zM failed (no valid HTTP response); the sandbox may be stopped or unreachable: )rrr _TRANSIENT_HANDSHAKE_STATUSESr)rrFstatus server_detailsuffixs r!_raise_for_invalid_handshakersWS*d3]D IF,S1M%2r- !"SE F }$!5 6    ..-EfXQvh W  $9&SE J   ! x(669U <  r1r command_idc*|jdd}|jdd}|dk(r t||dk(rt|rd|n||rd| d | |d k(rt|rd |n||rd| d | t||rd| d | ) z!  %%#0: | , %/k!  5>!  !++ 1: r1z /bin/bashi,iX) rtimeoutenvcwdshell on_stdout on_stderr idle_timeoutkill_on_disconnect ttl_secondsptyrJ open_timeoutc  t\t|t||td   fd }|fS)aPExecute a command over WebSocket, yielding raw message dicts. Returns a tuple of (message_iterator, control). The control object provides send_kill() and send_input() methods for the CommandHandle. The iterator yields dicts with a "type" field: - {"type": "started", "command_id": "...", "pid": N} - {"type": "stdout", "data": "...", "offset": N} - {"type": "stderr", "data": "...", "offset": N} - {"type": "exit", "exit_code": N} If on_stdout/on_stderr callbacks are provided, they are invoked as data arrives in addition to yielding the messages. c 3K ttt5} j|d d} r |d< r |d< r |d<rd|d<|j t j ||D]}t j|}|jd }|d k(r|3|d k(rr |d |J|d k(rr |d |a|dk(r|ddd jy|dk(st| ddd jy#1swYxYw#$r}t|Yd}~3d}~w$rE}|jr%|jjdk(r td|td||d}~wt t"f$r}t%d||d}~wwxYw# jwxYwwNadditional_headersrrl ping_interval ping_timeoutexecute)rertimeout_secondsridle_timeout_secondsrrrrrTrrestartedstdoutrrstderrexitr(Server is reloading, reconnect to resume*WebSocket connection closed unexpectedly: Failed to connect to sandbox: WS_CLOSE_TIMEOUTWS_PING_INTERVALWS_PING_TIMEOUTr]rhrirjrrr`rrrcvdcoderr OSError RuntimeErrorr )r\rraw_msgrmsg_typeerrrrcontrolrrrrrrrrrequest_headersrrr ws_connectrFs r!_streamzrun_ws_stream.._stream[sG #2)..,  b!&&'.",8*<#.+,6GL)%(GEN%(GEN%)GEN 7+, "G**W-C"wwvH9,! !X-$%c&k2! !X-$%c&k2! !V+! eL OO %"W,-c2- "?L OO Mn  4 (F 3 3 vv!&&++-.>)? E GF>AFF>*F99F>>GGGrwzIterator[dict]r?rGrLrN)rErIrrrrrrrrrrrrrJrrrrrrrrFs ```````````` ` @@@@@@r! run_ws_streamr5sZB6H5I2J "2 = )F)'7;O GHHHT 9g r1) stdout_offset stderr_offsetrJc  t\ t| t|| t d  f d }| fS)aReconnect to an existing command over WebSocket. Returns a tuple of (message_iterator, control), same as run_ws_stream. The iterator yields the server's ``started`` acknowledgement, then stdout, stderr, exit, and error messages. That acknowledgement is the only evidence a reattachment landed for a command producing no output, so it is forwarded rather than dropped. With the ring buffer reader server model, there is no replay/live phase distinction and no deduplication needed. The server reads from its ring buffer starting at the requested offsets and streams output from there. If the requested offset is older than the buffer's earliest data, the server sends from the earliest available offset. c 3 K  tttt5}j ||j t jd d|D]g}t j|}|jd}|dvr|2|dk(r|dddjy|dk(s[t|i dddjy#1swYxYw#$r}t| Yd}~3d}~w$rE}|jr%|jjd k(r td |t!d ||d}~wt"t$f$r}t'd ||d}~wwxYw#jwxYww Nrr)rerrrre)rrrrrrrrrrr4rrrr]rhrirjrrr`rrrrrr rrr )r\rrrrrrrrrrrrrFs r!rz$reconnect_ws_stream.._streams5 #2,..,  b!JJ$/*4-:-:   "G**W-C"wwvH#BB! !V+! Ah OO %"W,-cjI "/h OO iJ  4 (F 3 3 vv!&&++-.>)._stream s2A '#2)..,  b!&&'.",8*<#.+,6GL)%(GEN%(GEN%)GENggdjj1222%'3'**W-C"wwvH9,! !X-$%c&k2! !X-$%c&k2! !V+! [@ OO %"W,-c2_633;:&(;b  4 (F 3 3 vv!&&++-.>)D:D>A-E E' D<E'H#E) E5E'8E:D><E'>E? E' E  E'E#E E#E'&G?'G+, E=8G?=G+AGG+G&&G++G?.H?HHrwzAsyncIterator[dict]rBrGrLr)rErIrrrrrrrrrrrrrJrrrrrrrrFs ```````````` ` @@@@@@r!run_ws_stream_asyncrs^.._streamvs1 '#2,..,  b!ggJJ$/*4-:-:    &( J'**W-C"wwvH#BB! !V+! ;` OO %"W,-cjI?  J+*&(+B  4 (F 3 3 vv!&&++-.>)D)C+*D0C14C-5C18DD D DDF2F D0+F20F8AE88F FFF2!G2GGrr) rErIrrrrJrrrrrrrFs ``` @@@@@@r!reconnect_ws_stream_asyncrgsUr;r<r"r4rrrr-r2r9r=r?rBrGrLrNr frozensetrrrrrrrrrr0r1r!rsB"  <<**'5   8 $ !FHB (8"= BBG8"= :B?!!DcJ0+Q @A"DH0 0%@000G0Gf"M"MT!**> ?&Q4 F;=J$(0404$+/$3#pp pp  p  p "p p p.p.pppp p )!p""#p$-%pp+/TT TT  T  T)T-TB$(0404$+/$3#`` ``  `  ` "` ` `.`.```` ` )!`""#`$7%`P+/CC CC  C  C)C7CM!+//(*..' !sFFF