ێj;DdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZmZmZmZmZmZGd d e ZdS) uSandboxedWorkspaceBase — shared implementation for gateway-backed sandbox workspaces (Docker, E2B, K8s, …). Extends :class:`WorkspaceBase` with a template-method lifecycle around an in-sandbox MCP gateway. Subclasses only need to provide: - :meth:`_provision_backend` — attach/create the sandbox, bind ``self._backend``. - :meth:`_teardown_backend` — destroy/pause the sandbox. - :meth:`_bootstrap_commands` — shell commands to install the gateway venv on first use (Docker skips this — the image already has it). - gateway-path class attributes (see below). Everything gateway-related — bootstrap, launch, health poll, MCP add/remove routing, ``.mcp`` persistence, reset — lives here. N)abstractmethod)logger) MCPClient) WorkspaceBase) GatewayClient)DEFAULT_GATEWAY_LOGDEFAULT_GATEWAY_SCRIPTDEFAULT_GATEWAY_VENVDEFAULT_GLOB_HELPER_SCRIPT_read_gateway_script_bytes_read_glob_helper_bytesc eZdZUdZeed< eed< eed< edzed< eed< e e ed< d Z e ed < e d efd Ze d efd Ze d efdZe d efdZe d efdZdddddedzde e dzde edzd dffdZed#dZed#dZd e efdZd#dZd#dZd#dZd e e fdZde d dfdZded dfd Zd#d!Zd#d"ZxZ S)$SandboxedWorkspaceBaseuBase class for workspaces backed by an in-sandbox MCP gateway. Subclasses only need to set: - :attr:`workdir` — agent-visible root inside the sandbox. - :attr:`_gateway_home` — directory holding the venv, script, log. - :attr:`gateway_port` — TCP port the gateway listens on. Every other gateway path (venv, python, script, log, glob helper) is derived from :attr:`_gateway_home` via :meth:`BackendBase.join_path`. workdir gateway_port _gateway_homeN_gatewayis_alive_mcpsg @_bootstrap_cmd_timeoutreturncf||jtS)z+Sandbox-side path of the gateway venv root.) get_backend join_pathrr selfs RC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\workspace\_sandboxed_base.py _gateway_venvz$SandboxedWorkspaceBase._gateway_venvLs1!!++      c^||jddS)z9Sandbox-side path of the gateway venv python interpreter.binpython)rrr rs r_gateway_pythonz&SandboxedWorkspaceBase._gateway_pythonTs4!!++       r!cf||jtS)z.Sandbox-side path of the gateway entry script.)rrrr rs r_gateway_scriptz&SandboxedWorkspaceBase._gateway_script]s1!!++   "   r!cf||jtS)z3Sandbox-side path of the gateway stdout/stderr log.)rrrr rs r _gateway_logz#SandboxedWorkspaceBase._gateway_loges1!!++      r!cf||jtS)z   N #     r!cK|j8 |jd{Vn#t$rYnwxYwd|_ |d{Vd|_d|_dS#d|_d|_wxYw)uClose the gateway facade, then tear down the sandbox. Idempotent — errors are swallowed so ``close`` is always safe to call. NF)raclose Exceptionr9r?rrs rclosezSandboxedWorkspaceBase.closes = $ m**,,,,,,,,,,     DM "((** * * * * * * * DM!DMMM!DM!DM ! ! ! !s+ 88A--A=c fK|}|j4d{V|j4d{Vt|jD]O} |d{V#t $r%}tjd|j |Yd}~Hd}~wwxYwg|_|j |j |j fD]}| |d{V|d{Vdddd{Vn#1d{VswxYwYdddd{VdS#1d{VswxYwYdS)aWipe workspace state; keep the sandbox and gateway alive. Deregisters every MCP from the gateway, clears local handles, and wipes ``.mcp``, ``skills/``, ``sessions/``, and ``data/``. ``default_mcps`` / ``skill_paths`` are not re-seeded. Nz$MCP %r close failed during reset: %s)r _mcp_lock _skill_locklistrrGrFrwarningname _sessions_dir _data_dir _skills_dir delete_path_save_mcp_file)rbackendmcpepaths rresetzSandboxedWorkspaceBase.resets""$$> ( ( ( ( ( ( ( (4#3 ( ( ( ( ( ( ( (DJ''  ))++%%%%%%%% N> DJ"  0 0 ))$//////////%%'' ' ' ' ' ' ' ') ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s_D C< A%$C<% B /B C<B AC<*D < D D  D D  D*-D*c.Kt|jS)z4Gateway-wrapped MCP handles, one per registered MCP.)rKrrs r list_mcpsz SandboxedWorkspaceBase.list_mcpssDJr! mcp_clientcK|jtd|j4d{Vtfd|jDrt djdd}|j|}| d{V|j || d{Vdddd{VdS#1d{VswxYwYdS)a|Register a new MCP server through the in-sandbox gateway. Args: mcp_client (`MCPClient`): The MCP to register. Raises: `ValueError`: If an MCP with the same name already exists. `RuntimeError`: If the gateway is not attached or rejects the registration. N&Workspace has no MCP gateway attached.c38K|]}|jjkVdS)N)rM).0mrZs r z1SandboxedWorkspaceBase.add_mcp..s,AA16Z_,AAAAAAr!zMCP z already exists in workspace.jsonmode) r RuntimeErrorrIanyr ValueErrorrM model_dump make_clientconnectappendrR)rrZspec gw_clients ` radd_mcpzSandboxedWorkspaceBase.add_mcps = GHH H> ( ( ( ( ( ( ( (AAAAdjAAAAA  K:?KKK((f(55D 11$77I##%% % % % % % % % J  i ( ( (%%'' ' ' ' ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sB7C11 C;>C;rMc0K|jtd|j4d{Vt|jD]\}}|j|kr|j| |d{Vn-#t$r }tj d||Yd}~nd}~wwxYw| d{Vdddd{VdStj d|dddd{VdS#1d{VswxYwYdS)aDeregister an MCP server by name. Args: name (`str`): MCP name to remove. Unknown names log a warning and return silently. Raises: `RuntimeError`: If the gateway is not attached. Nr\zMCP %r close failed: %szMCP %r not found in workspace) rrdrI enumeraterrMpoprGrFrrLrR)rrMirTrUs r remove_mcpz!SandboxedWorkspaceBase.remove_mcp"sL = GHH H> B B B B B B B B#DJ//  38t##JNN1%%%K!iikk))))))))$KKK'@$JJJJJJJJK--///////// B B B B B B B B B B B B B B# N:D A A A B B B B B B B B B B B B B B B B B B B B B B B B B B B B B BsA?D&BD B+ B&!D&B++DD DDc K|}|dd|j|j|j|j|jgdd{Vtjd|j Ddd  d }| |j d{Vr | |j d{V}tj|d }t!|t"rdSt%jd t)|j|j t)|jnQ#t,tjf$r8}t%jd t)|j|j |Yd}~nd}~wwxYw||j |d{VdS) z=Create the standard workspace directories inside the sandbox.mkdirz-p/)cwdNc:g|]}|dS)rarb)rg)r^r_s r zCSandboxedWorkspaceBase._ensure_workspace_layout..Us& B B B1Q\\v\ & & B B Br!rF)indent ensure_asciizutf-8z@%s: %s does not contain a JSON list (got %s); reseeding defaultsz,%s: %s is corrupted (%s); reseeding defaults)r exec_shellrrOrPrNrradumpsr.encode file_exists _mcp_file read_fileloadsdecode isinstancerKrrLtyper>UnicodeDecodeErrorJSONDecodeError write_file)rrSpayloadexistingparsedrUs rr@z/SandboxedWorkspaceBase._ensure_workspace_layout>s&""$$    "" !   &* B B0A B B B    &//  $$T^44 4 4 4 4 4 4  !(!2!24>!B!BBBBBBBHOOG$<$<==fd++F3JJ'NLL) '(<=   BJJ'N  "  99999999999s.AE F#.FFcrK|}||jd{VsDtjdt |j|j|D]}| dd|g|j d{V}| sftt |jd|j d|d|jd d |jd  |j-||jt'd{V||jt)d{V| gd d{Vd t+j|jdt+j|jdt+j|jd|jdt+j|jd }| dd|gd{Vt7||jd|j|_d}t;j|z}d}t;j|krv|j d{Vrnt;j!|d{VtE|dzd}t;j|kv |#|jd{V}|ddd } n#tH$rd} YnwxYwtd|d|jd| tK|j&d{V|_'dS)uBootstrap (once) and launch the in-sandbox gateway. Steps: 1. Bootstrap the venv + script if :attr:`_gateway_script` is missing (Docker's image already has it → fast-path). 2. Kill any leftover gateway from a previous resume. 3. Launch a fresh gateway pointed at ``.mcp``. 4. Bind :attr:`_gateway` and poll ``/health``. Nz!%s: bootstrapping workspace_id=%rsh-c)timeoutz bootstrap failed (exit z) for: z stderr: replace)errorsz stdout: )rrz(pkill -f '[_]mcp_gateway_app.py' || trueznohup z -u z --config z --port z > z 2>&1 &g>@)rSrrgateway_log_pathg?g?g?i0zz&gateway did not become healthy within z s. Tail of z: )(rr~r'rr=rr>r-r;r{rokrd exit_codestderrrstdoutr+rrrshlexquoter%rrr)r rasyncioget_event_looptimehealthsleepminrrFrKrYr) rrScmdr launch_cmdhealth_timeoutdeadlinedelaylogtails rrAz)SandboxedWorkspaceBase._setup_mcp_gatewayss""$$ (()=>>>>>>>>  K3T #!    //11  !,,4% 7-ttvv&::.GG!"GG58GG#$8??)?#D#DGG$%8??)?#D#DGG%1((*+--$$$*,,         D D D          9U[!566 9 9{4/00 9 9 DN33 9 9' 9 9T.//  9 9 9   $j!9:::::::::&*!.     )++0022^C$&&++--88]))++++++++ -&& & & & & & & & S))E $&&++--88  4#--d.?@@@@@@@@566{)));; 4 4 43 4888,881588  7 7 9 9999999:: s>M M&%M&)rN)!r> __module__ __qualname____doc__str__annotations__intr boolrKrrfloatpropertyr r%r'r)r+r2rr6r9r;rCrGrWrYrmrrr@rA __classcell__)r3s@rrr&s   LLLE5Gd""""LNNNI  ?J$*E***  s   X     X     X  c   X  3   X $(/3(, Dj9o,  #Y%   4   ^    ^ T#Y# # # # J""""&((((@ i    ( (d((((6BSBTBBBB81:1:1:1:ja;a;a;a;a;a;a;a;r!r)rrrarabcr_loggingrrTr_baser_gateway_clientr _utilsr r r r rrrr5r!rrs"  ******n;n;n;n;n;]n;n;n;n;n;r!