uPjOUddlZddlZddlZddlmZmZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZmZmZddlmZmZddlmZddlmZeZee d<eZ!ee d <eZ"ee d <eZ#ee d <Gd d eZ$y)N)Finalcast) OPEN_SANDBOX_API_BASE_ENV_VAROPEN_SANDBOX_API_KEY_ENV_VAROPEN_SANDBOX_DEFAULT_CPU_LIMITOPEN_SANDBOX_DEFAULT_ENTRYPOINTOPEN_SANDBOX_DEFAULT_LANGUAGE!OPEN_SANDBOX_DEFAULT_MEMORY_LIMITOPEN_SANDBOX_DEFAULT_TEMPLATEOPEN_SANDBOX_DEFAULT_TIMEOUTOPEN_SANDBOX_EXECD_PORTOPEN_SANDBOX_POLL_INTERVALOPEN_SANDBOX_READY_TIMEOUT)SANDBOX_MAX_OUTPUT_BYTESBaseSandboxConfigCodeExecutionResultContainerHandle)AsyncHTTPHandlerget_async_httpx_client)get_secret_str)httpxSpecialProviderDEFAULT_SANDBOX_TIMEOUTDEFAULT_READY_TIMEOUTDEFAULT_POLL_INTERVALMAX_OUTPUT_BYTESc#eZdZdedzdefdZdBdedzdefdZdddddddddddddddddd edzd edzd edzdedzd edzd e eefdzde eefdzde eefdzde eefdzde ee edfzdzde ee fdzdedede dzde dzdedzdef"dZddeddddddeezdededzd edzdedede dzde dzdedzdefdZdddddeezdedzd edzdedzdef dZdeezdedzd edzdede de dedzdefd Zd!ed ed"e eefdedzde de ddfd#Zd!ed ed"e eefdededzde de de ee eefffd$Zd!ed ed"e eefdededzde ee eefff d%Zd&ed"e eefd'e ee fdedzde ef d(Zdedzd)edefd*Zed edzd edzd edzd e eefdzde eefdzde eefdzde eefdzde ee edfzdzde ee fdzdede ee ffd+Zede eeffd,Zed-e dedzfd.Zed/e de eeffd0Zed edzdefd1Zedede eeffd2Z ed3ed edefd4Z!edeezd edzdefd5Z"ed6e edefd7Z#ed8ede ee fdzfd9Z$ed:e ee fde ee ffd;Z%ed:e ee fde ee ffd<Z&ed/e dedzfd=Z'ed>e ee fd?ed@e de fdAZ(y)COpenSandboxSandboxConfigclientNreturnc>||SttjS)N) llm_provider)rrSandbox)selfrs t/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/opensandbox/sandbox/transformation.py_httpzOpenSandboxSandboxConfig._http)s  M%3G3O3OPPapi_keyc 0||SttxsdS)N)rr)r#r'kwargss r$validate_environmentz-OpenSandboxSandboxConfig.validate_environment.s  N:;ArAr&F)templatetimeoutallow_internet_accessr'api_basemetadataenv_varsresource_limitsresource_requests entrypointnetwork_policy secure_accessuse_server_proxy ready_timeout poll_intervalrr,r-r.r/r0r1r2r3r4.r5r6r7r8r9c K|j|}|j|}| t|nt}| t|nt}|j ||||||| | | |  }t tj|j|j|d|j||d{}|j}t|d}|j|dk7r.|j|||j||||d{|j!|||j|| |||d{\}}t#|d | }||||| d |_|S77U7(w) N)r') r,r-r.r0r1r2r3r4r5r6z /sandboxes)urlheadersjsonidRunning) sandbox_idr/r<rr8r9r@r/r<r7rr8r9 opensandboxr>providerdomainr/r'execd_endpoint execd_headersr7)r+ _api_basefloatrr _create_bodyrhttpxResponser%post_lifecycle_headersr=str_sandbox_state_wait_until_running_wait_for_execd_endpointr_hidden_params)r#r,r-r.r'r/r0r1r2r3r4r5r6r7r8r9rr*keybaseready_timeout_secondspoll_interval_secondsbodyresponsedatar@endpointendpoint_headershandles r$acreate_sandboxz(OpenSandboxSandboxConfig.acreate_sandbox3s*..w.?nnX.?L?Xu]';^s?L?Xu]';^s''"7+/!)'(   NN**V$))fJ'//4*  mmoT O   t $ 1**%//433 +  ,0+H+H!++C0-//,I, & ""(: VZ[&- 0 !  K  & s7B1E13E+ 4A#E1E-.E1E/%E1-E1/E1)r'r/languager7r8r9r containercoder`c <K|j||||| t|nt| t|nt| d{} t | j d} |j | j jd} t | j jdxs| jxs|j|}|j|j| |ddddd | |d |id | d{}|j|S77w) N)rar'r/r7r8r9rrGrHr/z/codeapplication/jsonztext/event-streamzno-cache) Content-TypeAcceptz Cache-Controlr`)rbcontext)r;r<rYr) _ensure_handlerJrrrPrT _as_str_dictgetrErI _post_code_endpoint_base_url _parse_lines)r#rarbr'r/r`r7r8r9rr*r^r\r]rVliness r$ arun_codez"OpenSandboxSandboxConfig.arun_codes>#11-3@3L5/Rg3@3L5/Rg2  f334DEF"&"3"3F4I4I4M4Mo4^"_&//33J?l6==lTXTbTbckTlm!__**8T:;5A 2-!+# &1-     ''5   s$>DDCDDDD)r'r/rc 2K|j||}t|jjdxs|j |}|j ||} t tj|j|j|d|j|j|d{} d| jcxkxrd kScS7"#tj$r$} | jjdk(rYd} ~ yd} ~ wwxYww) Nr/r/r'r^ /sandboxes/r;r<Fi,) _as_handlerPrTrjrI_api_keyrrLrMr%deleter>rOHTTPStatusErrorrZ status_code) r#rar'r/rr*r^rVrUrZes r$adelete_sandboxz(OpenSandboxSandboxConfig.adelete_sandboxs HE&//33J?[4>>RZC[\]]76]B "jj(//& FII;7 33C80Hh**0S0000 $$ zz%%,  sIAD!AC8C 9CDCD0D DDDDc K|j||}|jjdr|St|jjdxs|j |} |j ||} t |jjd|} |j|j| |j| | |||d{\} } | |_ i|j| | | | | d|_|S7+w)NrqrGr/rrr7rArF) rwrTrjrPrIrxboolrSr>rOrE)r#rar'r/r7r8r9rr^rVrUresolved_use_server_proxyr\r]s r$rhz'OpenSandboxSandboxConfig._ensure_handles HE  $ $%5 6M&//33J?[4>>RZC[\]]76]B+/0E0E0I0IJ\^n0o+p!+/+H+Hyy++C06'',I, & "" ! ##! &- 9 !  %& sCDD,Dr@r<cKtj|z} ttj|j |j |d||d{}|j} |j| } | dk(ry| dvrtd|d| tj|k\rtd|d|dtj|d{77w) Nrsrtr?>FailedStopping TerminatedzOpenSandbox sandbox z entered z was not Running within  seconds) time monotonicrrLrMr%rjr=rQ ValueError TimeoutErrorasynciosleep) r#r@r/r<rr8r9deadlinerZr[states r$rRz,OpenSandboxSandboxConfig._wait_until_runnings..*]:jj(,,#*K |<#-H==?D''-E !<< #7 |9UG!TUU~~8+"%9*E]^k]llt#uvv-- . . .! /s%AC)C% BC)C' C)'C)cKtj|z}d} |j|||||d{S7#tj$r&} | j j dk7r| } Yd} ~ nd} ~ wt$r } | } Yd} ~ nd} ~ wwxYwtj|k\rtd|d|d| tj|d{7w)N)r@r/r<r7rruzOpenSandbox execd endpoint for z was not ready within r) rr_get_execd_endpointrLrzrZr{rrrr) r#r@r/r<r7rr8r9r last_errorr|s r$rSz1OpenSandboxSandboxConfig._wait_for_execd_endpoint s..*]:'+  !55)%#%5! 6(( ::))S0    ~~8+"5j\AWXeWffno!"-- . . .)sVC><>C>B A2-C2 B >BCB  ACCCc hKttj|j|j |d|dt |d|id{}|j }|j d}|std|t||j|j dfS7dw)Nrsz /endpoints/r7)r;r<paramsr\z1OpenSandbox did not return an execd endpoint for r<) rrLrMr%rjr r=rrPri) r#r@r/r<r7rrZr[r\s r$rz,OpenSandboxSandboxConfig._get_execd_endpoint,s NN**V$((j J<{CZB[\*,<=)  mmo((:.PQ[P\]^ ^8}d//0CDDD sA B2 B0 A%B2r;rYc Ktjdddd}ttj|j |j ||||dd{}|j |d{S7!7w)Ng>@)connectreadwritepoolT)r;r<r-r=stream)rLTimeoutrrMr%rN_read_capped_lines)r#r;r<rYrr-rZs r$rkz#OpenSandboxSandboxConfig._post_codeCst$dQUV NN**V$)) *   ,,X666 7s$AA>A: A>5A<6A><A>r^cv||Sd|jvrt|jdS|jS)Nr')rTrPr+)r#r'r^s r$rxz!OpenSandboxSandboxConfig._api_keyXs@  N -- -v,,Y78 8((**r&c d|xstit|xst||nt|xstj d} |r|| d<|r|| d<|r|| d<||| d<n |durdgd | d<| rd| d <| S) Nuri)imager4r-resourceLimitsr0envresourceRequests networkPolicyTdeny) defaultActionegress secureAccess)r listrrr_default_resource_limits) r,r-r.r0r1r2r3r4r5r6rYs r$rKz%OpenSandboxSandboxConfig._create_body_sXF)FGzL-LM")"5w;R-d1I1b1b1d *  'D  "DK '8D# $  %$2D ! "$ .6<$KD ! #'D  r&cttdS)N)cpumemory)rr r&r$rz1OpenSandboxSandboxConfig._default_resource_limitss27  r&r[ct|tsy|jd}t|tsy|jd}| t|SdS)Nstatusr) isinstancedictrjrP)r[rrs r$rQz'OpenSandboxSandboxConfig._sandbox_statesL$%*&$'zz'*".s5z8D8r&valuect|tsiS|jDcic]\}}t|t|c}}Scc}}wN)rritemsrP)rkvs r$riz%OpenSandboxSandboxConfig._as_str_dicts;%&I+0;;=941aAA999sA c|xstt}|stdtdt|j dS)Nz7OpenSandbox api_base is required. Pass api_base or set ./)rrrrPrstrip)r/rVs r$rIz"OpenSandboxSandboxConfig._api_basesBO.1N"OVWtVuuvwx x4y$$r&cddi}|r||d<|S)NrerdzOPEN-SANDBOX-API-KEYr)r'r<s r$rOz+OpenSandboxSandboxConfig._lifecycle_headerss (*<= .5G* +r&r\c|jd}|jdr|Sd|vr|jdddnd}|d|S)Nr)zhttp://zhttps://z://rhttp)r startswithsplit)r\r/normalized_endpointprotocols r$rlz+OpenSandboxSandboxConfig._endpoint_base_urlsX%-__S%9  ) )*A B& &9>(9J(..215PV32344r&ct|tr|Stt|dtj |}i|_|S)NrBrC)rrrPrrIrT)rar/r^s r$rwz#OpenSandboxSandboxConfig._as_handlesC i 1 '9~"+55h?  !# r&rnc tfd|Ddtffd }td|dDd}td|dDd}tdj d |d Ddj d |d D|d Dcgc]}t j |c}||Scc}w)Nc3RK|]}tj|x ywr)r_parse_sse_line).0lineevents r$ z8OpenSandboxSandboxConfig._parse_lines..s- 1I1Y1YZ^1_(_'lE s'' message_typecfdDS)Nc3LK|]}|jdk(s|yw)typeN)rj)rmrs r$rzIOpenSandboxSandboxConfig._parse_lines..of_type..s I!155=L+HAIs$$r)rmessagess`r$of_typez6OpenSandboxSandboxConfig._parse_lines..of_typesIxI Ir&c3FK|]}tj|ywr)r_normalize_errorrrs r$rz8OpenSandboxSandboxConfig._parse_lines..s Ta % 6 6q 9 Ts!errorc3K|]L}tj|jd&tj|jdNyw)execution_countN)r_as_intrjrs r$rz8OpenSandboxSandboxConfig._parse_lines..sH +33AEE:K4LMY)007H1IJ sAArr)c3RK|]}t|jdd!ywtextr)NrPrjrs r$rz8OpenSandboxSandboxConfig._parse_lines.. Ma3quuVR01M%'stdoutc3RK|]}t|jdd!ywrrrs r$rz8OpenSandboxSandboxConfig._parse_lines..rrstderrresult)rrresultsrr)tuplerPnextrjoinr_normalize_result)rnrrrrrrs @@r$rmz%OpenSandboxSandboxConfig._parse_liness #   J# J T77CS T  "&  !23   " #77M78;LMM77M78;LMMLST\L]^q-??B^+   _sC rcz|j}|r|jdry|jdr|ddjn|}|sy tj|}t |t syd|vr)d|vr%d|vr!dt|dt|dgdd S|S#tj$rYywxYw) N):zevent:zid:zretry:zdata:rrbmessager)enameevalue traceback)rr)striprr=loadsJSONDecodeErrorrrrP)rstrippedr[parseds r$rz(OpenSandboxSandboxConfig._parse_sse_lines**,8..  .6.A.A'.Jhqrl((*PX  JJt,F&$'  Ff$4f9L 0!&"34!#  ##  sB$$B:9B:rc|jd}t|tr-|jDcic]\}}t ||c}}S|jDcic]\}}|dvs t ||c}}Scc}}wcc}}w)Nr>r timestampr)rjrrrrP)rrrrs r$rz*OpenSandboxSandboxConfig._normalize_resultsp Y/ gt $*1--/:$!QCFAI: :&-mmokdaBj9jA kk;ksB" B/Bcv|jd}t|trPtj |ddd}tj |ddd}tj |dg}|||d Stj |ddtj |dd dtj |dgd S) Nrrnamer))defaultrrr)rrrr)rjrrr_first_non_none_value)r raw_errorrrrs r$rz)OpenSandboxSandboxConfig._normalize_errors";;w/ i &2HHT[]cmoHpD3II)U]_fprIsE7MMiYdnpMqI&  -BB7F\^B_-CCGWV\fhCi1GGQ\fhGi  r&c~t|tr|St|tr t|Sy#t$rYywxYwr)rintrPr)rs r$rz OpenSandboxSandboxConfig._as_intsC eS !L eS ! 5z!  s 0 <<valueskeysrc.tfd|D|S)Nc3>K|]}|vs||ywrr)rrUrs r$rzAOpenSandboxSandboxConfig._first_non_none_value..s$ VSC6MfSk>UVC[ Vs  )r)rrrs` r$rz.OpenSandboxSandboxConfig._first_non_none_values VD V   r&r))__name__ __module__ __qualname__rr%rPr+rrrrrobjectrJrr_r rror}rhrRrSrrkrx staticmethodrKrrQrirIrOrlrwrmrrrrrrr&r$rr(sQ,t3Q8HQ BC$JBCB $"-1"#*.*.15379=37#!&&*&**.%M*Mt M $d{ M t M*MsCx.4'MsCx.4'Mc3h$.M S>D0MIc3h/$6MS&[)D0MMM t|!M"t|#M$!4'%M( )Mh##5!&&*&**.((#S((( (( t (( * ((((((t|((t|((!4'(( ((\##*. 1#S(1t 1 * 1 !4' 1 14$#S($t $ * $  $$$!4'$ $L// / c3h / !4' /// /:!/!/ !/ c3h !/  !/!4'!/!/!/ sDcN" #!/FEE E c3h E  E!4'E sDcN" #E.77c3h 7 3; 7 !4' 7 c7*+3:++3+*t $d{  sCx.4'  sCx.4' c3h$. S>D0Ic3h/$6S&[)D0 c6k B d38n  9V9d 99:F:tCH~:: %C$J%3%% CDcN 5S5C5C55 o3 #* Q`   DI *=  :cd3;&7$&>>l4V #4lc6k9Jll  $sF{"3 S&[8I  "v#* d3;&7  f Y_  r&r)%rr=rtypingrrrLlitellm.constantsrrrrr r r r r rr,litellm.llms.base_llm.sandbox.transformationrrrr&litellm.llms.custom_httpx.http_handlerrrlitellm.secret_managers.mainrlitellm.types.llms.custom_httprr__annotations__rrrrrr&r$r sz       8?!==9u99u92%2y 0y r&