uPjsUdZddlZddlmZmZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZddlmZdZeed <d Zeed <d Zeed <dZeed<dZeed<eZeed<GddeZy)a  e2b sandbox provider. Talks to e2b's REST API directly over httpx (no e2b SDK dependency): - create: POST {api_base}/sandboxes - run: POST https://{JUPYTER_PORT}-{sandboxID}.{domain}/execute (NDJSON stream) - delete: DELETE {api_base}/sandboxes/{sandboxID} N)Finalcast)SANDBOX_MAX_OUTPUT_BYTESBaseSandboxConfigCodeExecutionResultContainerHandle)AsyncHTTPHandlerget_async_httpx_client)get_secret_str)httpxSpecialProviderzhttps://api.e2b.app E2B_API_BASEzcode-interpreter-v1E2B_DEFAULT_TEMPLATEze2b.appE2B_DEFAULT_DOMAINiO JUPYTER_PORT,DEFAULT_SANDBOX_TIMEOUTMAX_OUTPUT_BYTESc^eZdZdedzdefdZddedzdefdZdddddddddedzd edzd edzdedzd edzd e dzdedzde fd Z ddddde ezdededzde dzdedzde f dZ ddddde ezdedzd edzdedzdef dZede ezde fdZedeede fdZy)E2BSandboxConfigclientNreturnc>||SttjS)N) llm_provider)r r Sandbox)selfrs l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/e2b/sandbox/transformation.py_httpzE2BSandboxConfig._http%s  M%3G3O3OPPapi_keyc >|xs td}|s td|S)N E2B_API_KEYz8E2B API key not set. Set E2B_API_KEY or pass api_key=...)r ValueError)rrkwargskeys rvalidate_environmentz%E2BSandboxConfig.validate_environment*s%=} =WX X r)templatetimeoutallow_internet_accessrapi_basemetadatarr&r'r(r)r*c K|j|} |xst} |xst||ntd|dn|d} |r|| d<t t j |j|j| d| dd| d{} | j} t| d d | jd xst }| jd | jd| | d|_ |S7lw)N)rT) templateIDr'securer(r*z /sandboxesapplication/json) X-API-Key Content-Type)urlheadersjson sandboxIDe2bdomainidproviderr6envdAccessTokentrafficAccessToken)envd_access_tokentraffic_access_tokenrr))r%r rrrhttpxResponserpostr3rgetr_hidden_params)rr&r'r(rr)r*rr#r$basebodyresponsedatahandles racreate_sandboxz E2BSandboxConfig.acreate_sandbox0s..w.?.,":&:")"5w;R.C.KdQf   'D  NN**V$))fJ'&);MN*  mmo'K 88H%;);  "&*;!<$(HH-A$B !   ' sA?C0C. A-C0)renv_varsr containercoderIc K|j|}|jjd}|s tdd|d} |jjd} | r| | d<dtd|j d |j d } ttj|j|j| | |d|d d d{} |j| d{} |j| S727w)Nr<zCannot run code from a sandbox id alone. e2b secure sandboxes require the access token returned by acreate_sandbox; pass the ContainerHandle it returned instead of a bare sandbox id.r.)r0zX-Access-Tokenr=zE2B-Traffic-Access-Tokenzhttps://-.z/execute)rK context_idrIT)r1r2r3stream) _as_handlerBrAr"rr8r6rr>r?rr@_read_capped_lines _parse_lines)rrJrKrrIrr#rGtokenr2 traffic_tokenr1rEliness r arun_codezE2BSandboxConfig.arun_code^s  2,,001DEL  +=PUV%44889OP 2?G. / ~Qvyyk6==/R NN**V$))"$HM *  "44X>>  '' ?s$CC=C9 C=#C;$C=;C=)rr)rc0K|j|}|xs-|jjdxs|j}|xs#|jjdxst} t t j|j|j|d|jd|id{} d| jcxkxrd kScS7"#t j$r$} | jjdk(rYd} ~ yd} ~ wwxYww) Nrr)z /sandboxes/r/)r1r2iFr)rQrBrAr%r rr>r?rdeleter8HTTPStatusErrorrE status_code) rrJrr)rr#rGr$rCrEes radelete_sandboxz E2BSandboxConfig.adelete_sandboxs 2c 5 5 9 9) DcHaHaHcW&"7"7";";J"GW< "jj(//& FII;7(#.0Hh**0S0000 $$ zz%%,  sIA*D-A C7C 8CDCD/DD DDDcnt|tr|Stt|dt}i|_|S)Nr5r7) isinstancerstrrrB)rJrGs rrQzE2BSandboxConfig._as_handles2 i 1 '3y>ERde " rrVc dtfdt fd|Ddtffd }td|dDd}td|d Dd}td j d |d Dd j d |dD|dDcgc],}|j Dcic]\}}|dk7s ||c}}.c}}}||Scc}}wcc}}}w)Nstrippedc` tj|S#tj$rYywxYwN)r3loadsJSONDecodeError)rcs r _try_parsez1E2BSandboxConfig._parse_lines.._try_parses. zz(++''  s --c3^K|]$}|jxsx!&ywre)strip).0linerhparsedrcs r z0E2BSandboxConfig._parse_lines..s7 TZZ\)A)AQ[\dQeGevFrF s - -- message_typecfdDS)Nc3LK|]}|jdk(s|yw)typeNrA)rkmros rrnzAE2BSandboxConfig._parse_lines..of_type..s I!155=L+HAIs$$)romessagess`rof_typez.E2BSandboxConfig._parse_lines..of_typesIxI Irc3fK|]$}dDcic]}||j|c}&ycc}ww))namevalue tracebackNrs)rkrtr$s rrnz0E2BSandboxConfig._parse_lines..s+ cQ)G H#c155:o H c Hs 1, 1errorc3>K|]}|jdyw)execution_countNrsrkrts rrnz0E2BSandboxConfig._parse_lines..s O!QUU$ % Osnumber_of_executionsc3@K|]}|jddywtextrNrsrs rrnz0E2BSandboxConfig._parse_lines..H155,Hstdoutc3@K|]}|jddywrrsrs rrnz0E2BSandboxConfig._parse_lines..rrstderrresultrr)rrresultsr|r~)ratuplenextrjoinitems) rVrwr|r~rtkvrhrvrmrcs @@@@rrSzE2BSandboxConfig._parse_liness     $   J# J cRYZaRb c  "& Ow7M/N O "  #77Hgh6GHH77Hgh6GHHJQRZJ[\\Qqwwy@tq!AKad@\+   A\s.C1 C+C+C1+C1re)__name__ __module__ __qualname__r rrar%intbooldictrrHrrWr^ staticmethodrQlistrSrurrrr$sQ,t3Q8HQ C$JC $"-1"# $*.,*,t , $d{ , t ,*,+,!4', ,f# $*.$(#S($( $( t $( + $(!4'$( $(T##*. 1#S(1t 1 * 1 !4' 1 14o3 DI *=  rr)__doc__r3typingrrr>,litellm.llms.base_llm.sandbox.transformationrrrr&litellm.llms.custom_httpx.http_handlerr r litellm.secret_managers.mainr litellm.types.llms.custom_httpr r __annotations__rrrrrrrurrrs{   8?+ e+3e3%E% e!$$2%2` (` r