~tjdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZddlZddlmZdd lmZdd lmZmZmZmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)m*Z*dd l+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3er ddl4m5Z5ddl6m7Z7d.dZ8d/dZ9ee e:e:fZ;d0dZd3d+Z?Gd,d-Z@dS)4zEMain SandboxClient class for interacting with the sandbox server API.) annotationsN)CallableMapping)Path) TYPE_CHECKINGAnyOptionalUnion)quote)utils) Langsmith)ResourceCreationErrorResourceNameConflictErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIError)handle_client_http_errorhandle_sandbox_creation_error merge_headersvalidate_service_params validate_ttl)ResourceStatus ServiceURLSnapshot)SandboxMountConfig"validate_mount_config_proxy_config)SandboxProxyConfig)Sandbox)RetryTransport)RegistriesResourceAsyncSandboxClientreturnstrc^tjdd}|ddS)zGet the default sandbox API endpoint from environment. Derives the endpoint from LANGSMITH_ENDPOINT (or LANGCHAIN_ENDPOINT). ENDPOINTzhttps://api.smith.langchain.com)default/ /v2/sandboxes)ls_utils get_env_varrstrip)bases W/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/sandbox/_client.py_get_default_api_endpointr/6s5   4U V V VDkk# - - -- Optional[str]c*tjdS)z)Get the default API key from environment.API_KEY)r*r+r0r._get_default_api_keyr5?s   * **r0valuecF|stdt|dS)zCQuote a user-controlled value for use as a single URL path segment.z+URL path segment must be a non-empty string)safe) ValueErrorr )r6s r._quote_path_segmentr;Gs- HFGGG R  r0 context_pathrbytesctj}tj|d5}t |dD]Z}||}|jr|jddkr0||| d[ dddn #1swxYwY| S)Nw)fileobjmode*rz.gitF)arcname recursive) ioBytesIOtarfileopensortedrglob relative_topartsaddas_posixgetvalue)r<buftarpathrels r._make_docker_context_tarrTNs *,,C c , , ,C<--c2233 C CD""<00Cy SYq\V33 GGD#,,..EG B B B B  CCCCCCCCCCCCCCCC <<>>sA>B55B9<B9 dockerfileUnion[str, os.PathLike[str]]contexttuple[Path, str]c:t|}t|}|s||z }|}|st d||st d| ||}n"#t $r}t d|d}~wwxYw||fS)Nzcontext must be a directory: zdockerfile must be a file: z!dockerfile must be inside context) r expanduserresolve is_absoluteis_dirr:is_filerKrN)rUrWr<dockerfile_pathdockerfile_relexcs r._resolve_dockerfile_contextrbYs-==++--5577L:&&1133O  & & ( (9&8%--//O    IGGGHHH  " " $ $JHHHIIIG(44\BB GGG<==3FG 0022 22sC%% D/C??Dremote_contextr` image_ref buildkit_root buildkit_run build_argsOptional[Mapping[str, str]]targetcrtj||}tj|}tj|} tj|d} ddd| zddddd d |zd d |zd d | zdd|zg} || d d|zgt |piD]"\} } | d d| d| g#dtj|dtj|dtjd| zdtj|dtj|dzdtjd| zdtj|dzdtj|dzdtj| dtj|dS)Nzbuildkitd.sockbuildctlz--addrzunix://buildz--progress=plainz --frontendz dockerfile.v0z--localzcontext=z dockerfile=z--optz filename=z--outputztype=docker,name=ztarget=z build-arg:=zset -euo pipefail mkdir -p  z buildkitd --addr z --root zq --oci-worker=true --containerd-worker=false --oci-worker-snapshotter=native --oci-worker-binary buildkit-runc > z/buildkitd.logz 2>&1 & buildkitd_pid=$! cleanup() { kill "$buildkitd_pid" >/dev/null 2>&1 || true; } trap cleanup EXIT for i in $(seq 1 300); do if buildctl --addr zi debug workers >/dev/null 2>&1; then break; fi if ! kill -0 "$buildkitd_pid" >/dev/null 2>&1; then cat z+; exit 1; fi if [ "$i" = 300 ]; then cat z; exit 1; fi sleep 0.1 done for i in $(seq 1 300); do if docker info >/dev/null 2>&1; then break; fi if [ "$i" = 300 ]; then docker info; exit 1; fi sleep 0.1 done z | docker load rm -rf z || true ) posixpathjoindirnamebasenameextendrIitemsshlexr )rcr`rdrerfrgridockerfile_remotedockerfile_dirdockerfile_name socket_pathrkkeyr6s r._make_dockerfile_build_commandr{ms1"~~FF&'899N():;;O./?@@KK^#&o%i'H")f"45666j.B557788?? U">>> 9K .. 9 916\1J1J 9 9!K K(?@@ 9 9+m,, 9 9[(88 9 9 9 9!& I ,C D D 9 9{<*::;; 9 9"{<*::;;# 9 92 :h  3 9 94+m,,5 9 9 9r0ceZdZdZdddddddtdZdudZedvdZdwdZdwdZ dxdZ dydZ dydZ dzdZ d{d%Zdud&Z d|ddd'ddddddddd( d}d7Z d|ddd'd8ddddddddd9 d~d<Zdd=dd>Zdd=dd@ZdddddAddCZdd=ddDZdd=ddFZdGddHddLZdMdNddOddQZdMddRddSZdd=ddTZddUddVddZZd[ddddddUdd\ddeZdddUddfddhZdd=ddiZdddddjddoZdd=ddpZ dqdrddOddsZ!dS) SandboxClientaClient for interacting with the Sandbox Server API. This client provides a simple interface for managing sandboxes and snapshots. Example: # Uses LANGSMITH_ENDPOINT and LANGSMITH_API_KEY from environment client = SandboxClient() # Or with explicit configuration client = SandboxClient( api_endpoint="https://api.smith.langchain.com/v2/sandboxes", api_key="your-api-key", ) # Create a sandbox with the default runtime and run commands with client.sandbox() as sandbox: result = sandbox.run("python --version") print(result.stdout) Ng$@ api_endpointtimeoutapi_key max_retriesheadersrr1rfloatrrintrOptional[RequestHeaders]c|p td|_|p t}||_||_||_|rt|ni|_i}|r||d<|jrt||j}t|}tj ||||_ d|_dS)aInitialize the SandboxClient. Args: api_endpoint: Full URL of the sandbox API endpoint. If not provided, derived from LANGSMITH_ENDPOINT environment variable. timeout: Default HTTP timeout in seconds. api_key: API key for authentication. If not provided, uses LANGSMITH_API_KEY environment variable. max_retries: Maximum number of retries for transient errors (502, 503, 504), rate limits (429), and connection failures. Set to 0 to disable retries. Default: 3. headers: Optional default headers attached to every request on this client, including the data-plane ``/execute`` HTTP endpoint and the ``/execute/ws`` WebSocket upgrade. Use this to pass additional auth headers (e.g. ``X-Service-Key``). r(z X-Api-Key)r) transportrrN)r/r, _base_urlr5_api_key_timeout _max_retriesdict_default_headersrrhttpxClient_http_registries_client) selfrrrrrresolved_api_keyclient_headersrs r.__init__zSandboxClient.__init__s2'E*C*E*EMMcRR"<&:&<&<(  'AH0PW b)+  ;*:N; '   R*>4;PQQN"{;;; \.    8<r0r#r$cd}|j|r|jdt| S|jS)z>Return the API root URL, without the ``/v2/sandboxes`` suffix.r)N)rendswithlen)rsuffixs r. _api_rootzSandboxClient._api_roots@  > " "6 * * 2>.S[[L.1 1~r0r c|j5t|j||jpd|_|jjjS)aManage sandbox image registries: create, list, retrieve, update, delete. A registry stores credentials for pulling private images. Create one, then pass its ``id`` as ``registry_id`` when building a snapshot. Example: registry = client.registries.create( name="internal", url="registry.example.com", username="robot", password=os.environ["REGISTRY_PASSWORD"], ) snapshot = client.create_snapshot( "internal-python", docker_image="registry.example.com/internal/python:3.12", fs_capacity_bytes=2 * 1024**3, registry_id=registry.id, ) N)rbase_urldefault_headers)rr rrr sandboxes registriesrs r.rzSandboxClient.registriessR*  " *&/ )) $ 5 ='''D # &0;;r0RequestHeadersOptional[dict[str, str]]c>|dSt|jj|S)z8Merge default client headers with per-request overrides.N)rrrrrs r._request_headerszSandboxClient._request_headerss! ?4TZ/999r0cB|js|dSt|j|S)zMerge constructor-supplied default headers with per-request overrides. Used by the WebSocket exec path so headers like ``X-Service-Key`` set on the client are attached to the WS upgrade request. N)rrrs r._ws_default_headersz!SandboxClient._ws_default_headerss+ $ 4T2G<<g|]}tj|dS)Fr)rr).0crs r. z0SandboxClient.list_sandboxes..`s;!!DeDDDr0rrAPI endpoint not found: %. Check that api_endpoint is correct.N rrrrrrrrrrrr)rrrrdatars` r.list_sandboxeszSandboxClient.list_sandboxesTs  ''' z~~c43H3H3Q3Q~RRH  % % ' ' '==??D+r22 $   z%,,%;s;;; %Q ' ' '  sA8BC4C  C)new_namerrrrcdt|dt|d|jdt|}i}|||d<|||d<|||d< |j||||}|tj| |dS#tj $ra} | j j d krtd |d d | | j j dkrtd|dd | t!| d} ~ wwxYw)a]Update a sandbox's properties. Args: name: Current sandbox name. new_name: New display name. idle_ttl_seconds: Idle timeout in seconds. Must be a multiple of 60. ``0`` disables idle-stop. ``None`` leaves the existing value unchanged. delete_after_stop_seconds: Seconds after entering ``stopped`` before deletion. Must be a multiple of 60. ``0`` disables stop-anchored deletion. ``None`` leaves the existing value unchanged. Returns: Updated Sandbox. Raises: ResourceNotFoundError: If sandbox not found. ResourceNameConflictError: If new_name is already in use. SandboxClientError: For other errors. ValueError: If TTL values are invalid. rrrNrrrFrrrrrrizSandbox name 'z' already in use)rrr;rpatchrrrrrrrrrrrr) rrrrrrrrrrs r.update_sandboxzSandboxClient.update_sandboxms> %'9:::.0KLLLCC(;D(A(ACC"$  &GFO  '*:G& ' $ 03LG/ 0 z'''4+@+@+I+I(H  % % ' ' '$X]]__TuUUU U$   z%,,+1111z%,,/?X???"+ %Q ' ' '  sA,B??D/AD**D/cd|jdt|} |j|||}|dS#t j$r@}|jj dkrtd|dd|t|Yd}~dSd}~wwxYw) zDelete a Sandbox. Args: name: Sandbox name. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rrrrrrrN rr;rdeleterrrrrrrrrs r.delete_sandboxzSandboxClient.delete_sandboxsCC(;D(A(ACC (z((d6K6KG6T6T(UUH  % % ' ' ' ' '$ ( ( (z%,,+1111 %Q ' ' ' ' ' ' ' ' '  (AA B//5B**B/rc|jdt|d} |j|||}|t j|S#tj $r;}|j j dkrtd|dd|t|d }~wwxYw) aGet the provisioning status of a sandbox. This is a lightweight endpoint designed for high-frequency polling during sandbox provisioning. It returns only the status fields without full sandbox data. Args: name: Sandbox name. Returns: ResourceStatus with status and status_message. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rz/statusrrrrrrN)rr;rrrrrrrrrrrrrrs r.get_sandbox_statusz SandboxClient.get_sandbox_statuss&JJ(;D(A(AJJJ z~~c43H3H3Q3Q~RRH  % % ' ' '!+HMMOO<< <$   z%,,+1111 %Q ' ' '   sA(BC6C  CiXexpires_in_secondsrportr rctjdtd}d}dfd } j||}|tj| |S#tj $r;} | j j d krtd d d | t| d} ~ wwxYw)aiGet an authenticated URL for a service running inside a sandbox. Returns a :class:`ServiceURL` whose properties auto-refresh the token transparently before it expires. The object also provides HTTP helper methods (``.get``, ``.post``, etc.) that inject the authentication header automatically. Args: name: Sandbox name. port: Port the service is listening on inside the sandbox. expires_in_seconds: Token TTL in seconds (1--86400, default 600). headers: Optional per-request header overrides. Returns: ServiceURL with auto-refreshing token and HTTP helpers. Raises: ResourceNotFoundError: If sandbox not found. ValueError: If port or expires_in_seconds is out of range. SandboxClientError: For other errors. rz /service-url)r r r#rc6S)Nr )service)r rrr rsr. _refresherz)SandboxClient.service.._refreshers*<<#5   r0r)rrrrrrN)r#r)rrr;rrrrrrrrrrrrr) rrr r rrrrrrs ````` r.rzSandboxClient.servicesP: &8999OO(;D(A(AOOO7IJJ           z'4+@+@+I+I'H  % % ' ' '' JOOO O$   z%,,+1111 %Q ' ' '   sA+B,,C6;6C11C6xg?)r poll_intervalrrcddl}||z} |||}|jdkr|||S|jdkrt |jpdd ||z }|dkrtd |d |d d|j |t||)aPoll until a sandbox reaches "ready" or "failed" status. Uses the lightweight status endpoint for polling, then fetches the full sandbox data once ready. Args: name: Sandbox name. timeout: Maximum time to wait in seconds. poll_interval: Time between status checks in seconds. Returns: Sandbox in "ready" status. Raises: ResourceCreationError: If sandbox status becomes "failed". ResourceTimeoutError: If timeout expires while still "provisioning". ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rNTrreadyfailedzSandbox provisioning failedrrr' not ready after sr last_status) time monotonicrstatusrrstatus_messagersleepmin) rrrrrrdeadliner remainings r.wait_for_sandboxzSandboxClient.wait_for_sandboxs6  >>##g- 6,,T7,CCF}''''g'>>>}((+)J-J"+!4>>#3#33IA~~*BBBBBB"+ &  JJs=)44 5 5 5! 6r0rrc|jdt|d} |j|i||}|nQ#t j$r?}|jj dkrtd|dd|t|Yd }~nd }~wwxYw| ||| S) aStart a stopped sandbox and wait until ready. Args: name: Sandbox name. timeout: Timeout in seconds when waiting for ready. Returns: Sandbox in "ready" status. Raises: ResourceNotFoundError: If sandbox not found. ResourceCreationError: If sandbox fails during startup. ResourceTimeoutError: If sandbox doesn't become ready within timeout. SandboxClientError: For other errors. rz/startrrrrrrNr") rr;rrrrrrrrrrr!)rrrrrrrs r. start_sandboxzSandboxClient.start_sandboxIs,II(;D(A(AIII (z"d&;&;G&D&D'H  % % ' ' ' '$ ( ( (z%,,+1111 %Q ' ' ' ' ' ' ' '  ($$T7G$LLLsAA!!B/05B**B/ch|jdt|d} |j|i||}|d S#t j$r@}|jj dkrtd|dd|t|Yd }~d Sd }~wwxYw) zStop a running sandbox (preserves sandbox files for later restart). Args: name: Sandbox name. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rz/stoprrrrrrN) rr;rrrrrrrrrrrs r. stop_sandboxzSandboxClient.stop_sandboxosHH(;D(A(AHHH (z"d&;&;G&D&D'H  % % ' ' ' ' '$ ( ( (z%,,+1111 %Q ' ' ' ' ' ' ' ' '  (sAA""B115B,,B1<) registry_idrr docker_imager(rc|jd}|||d}|||d< |j||||} | t j| } n'#tj $r} t| d} ~ wwxYw| | j ||S)aBuild a snapshot from a Docker image. Blocks until the snapshot is ready (polls with 2s interval). Args: name: Snapshot name. docker_image: Docker image to build from (e.g., "python:3.12-slim"). fs_capacity_bytes: Filesystem capacity in bytes. registry_id: Private registry ID. timeout: Timeout in seconds when waiting for ready. Returns: Snapshot in "ready" status. Raises: ResourceTimeoutError: If snapshot doesn't become ready within timeout. ResourceCreationError: If snapshot build fails. SandboxClientError: For other errors. /snapshots)rr)rNr(rr") rrrrrrrrrrrwait_for_snapshotid) rrr)rr(rrrrrsnapshotrs r.create_snapshotzSandboxClient.create_snapshots:+++(!2# #   "%0GM " z'4+@+@+I+I'H  % % ' ' ')(--//::HH$    $Q ' ' '  %%hk7G%TTTsA*BB(B##B(.)rWrgri on_build_logrrrrrUrVrWrgrhrir1Optional[Callable[[str], Any]]c t||\} } dtjjdd}dtjjdd}t j|d}t j|d}dtjj}t j|d}t j|d }||| || || 5}||t| | | | d tj |zd ztj |zdztj |zdztj |z| | | t|| |||||| ||| }|j dkrtdd||j|||| | cdddS#1swxYwYdS)aiBuild a snapshot from a local Dockerfile context. ``vcpus`` and ``mem_bytes`` size the temporary builder sandbox. The build runs BuildKit plus the native snapshotter's layer copies inside it, which contend for a single core by default, so giving the builder an extra vCPU can cut a cold build's wall time substantially. zsnapshot-builder-N z/var/lib/langsmith-build/rWz context.tarzlangsmith-snapshot-build:z buildkit-rootz buildkit-run)rrrrrrr"zrm -rf z && mkdir -p z && tar -xf z -C )rcr`rdrerfrgri)r on_stdout on_stderrrrz Dockerfile snapshot build failedr.rr)rrr)rbuuiduuid4hexrorprwriterTrunrur r{ exit_codercapture_snapshotr)rrrUrrWrgrir1rrrrr<r` builder_name build_rootrc remote_tarrdrerfrresults r.create_snapshot_from_dockerfilez-SandboxClient.create_snapshot_from_dockerfiles,(C:w'W'W$ nB4:<<+;CRC+@BB I1A#2#1FHH " I>>^J >> B 0@BB !z?CC  ~j.AA \\/   6  MM(66     KK+n--.!"+n--.! ! +j)) *  +n--.     [[.#1#1'"/!-)! &&!F1$$+6",(( &"3 )_6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 s$DG66G:=G:r7 sandbox_namec|jdt|d}d|i}|||d<|||d< |j||||} | t j| } nM#tj $r;} | j j dkrtd |d d | t| d} ~ wwxYw|| j|| S)afCapture a snapshot from a running sandbox. Blocks until the snapshot is ready (polls with 2s interval). Args: sandbox_name: Name of the sandbox to capture from. name: Snapshot name. docker_image: Optional Docker image tag inside the sandbox to export into the snapshot instead of capturing the live root filesystem. fs_capacity_bytes: Filesystem capacity in bytes for Docker image export. timeout: Timeout in seconds when waiting for ready. Returns: Snapshot in "ready" status. Raises: ResourceNotFoundError: If sandbox not found. ResourceTimeoutError: If snapshot doesn't become ready within timeout. ResourceCreationError: If snapshot capture fails. SandboxClientError: For other errors. rz /snapshotrNr)rrrrrrrr")rr;rrrrrrrrrrrrrr,r-) rrDrr)rrrrrrr.rs r.r>zSandboxClient.capture_snapshots?>TT(;L(I(ITTT#)4.  #&2GN #  (+= 0``. Useful for paginating through large result sets in combination with ``limit``. Returns: A single page of Snapshots matching the provided filters. r+NrKrLrM)paramsrc6g|]}tj|Sr4)rr)rrs r.rz0SandboxClient.list_snapshots..s#MMMaH&q))MMMr0 snapshotsrrrr) rrKrLrMrrrPrrrs r.list_snapshotszSandboxClient.list_snapshotsos7<+++!#  $&3F? #  #F7O  %F8  z~~~--g66&H  % % ' ' '==??DMM488K3L3LMMM M$   z%,,%;s;;; %Q ' ' '  sA9BC%,4C  C%cd|jdt|} |j|||}|dS#t j$r@}|jj dkrtd|dd|t|Yd}~dSd}~wwxYw) zDelete a snapshot. Args: snapshot_id: Snapshot UUID. Raises: ResourceNotFoundError: If snapshot not found. SandboxClientError: For other errors. rGrrrHrr.rNrrIs r.delete_snapshotzSandboxClient.delete_snapshotsNN,? ,L,LNN (z((d6K6KG6T6T(UUH  % % ' ' ' ' '$ ( ( (z%,,+9999 %Q ' ' ' ' ' ' ' ' '  (ri,g@cddl}||z} |||}|jdkr|S|jdkrt |jpdd ||z }|dkrt d |d |d d|j |t||)a&Poll until a snapshot reaches "ready" or "failed" status. Args: snapshot_id: Snapshot UUID. timeout: Maximum time to wait in seconds. poll_interval: Time between status checks in seconds. Returns: Snapshot in "ready" status. Raises: ResourceCreationError: If snapshot status becomes "failed". ResourceTimeoutError: If timeout expires. ResourceNotFoundError: If snapshot not found. SandboxClientError: For other errors. rNTrrrzSnapshot build failedr.rrHrrr) rrrJrrrrrr) rrrrrrrr.r s r.r,zSandboxClient.wait_for_snapshots0  >>##g- 6((g(FFH'))(**++F/F",!4>>#3#33IA~~*JJJJJJ", ( JJs=)44 5 5 5! 6r0) rr1rrrr1rrrrr#r$)r#r )rrr#r)r#r")r#r)r#r})rrrrrrr#r)N)rr1rr1rr1rrrrrrrrrrrrrrrrrrr#r)rr1rr1rr1rrrrrrrrrrrrrrrrrrrrr#r)rr$rrr#r)rrr#r) rr$rr1rrrrrrr#r)rr$rrr#r)rr$rrr#r) rr$r rr rrrr#r) rr$rrrrrrr#r)rr$rrrrr#r)rr$r)r$rrr(r1rrrrr#r)rr$rUrVrrrWrVrgrhrir1r1r2rrrrrrrrr#r)rDr$rr$r)r1rrrrrrr#r)rr$rrr#r) rKr1rLrrMrrrr#rN)rr$rrr#r) rr$rrrrrrr#r)"__name__ __module__ __qualname____doc__rrpropertyrrrrrrrrrrrrrrrrrr!r$r&r/rCr>rJrSrUr,r4r0r.r}r}sn.'+!%,0(<(<(<(<(<(B@(,# $"& 888888v>B((((((8""& +6+6+6+6+6+6+6+6r0r}rW)r#r1)r6r$r#r$)r<rr#r=)rUrVrWrVr#rX)rcr$r`r$rdr$rer$rfr$rgrhrir1r#r$)Ar[ __future__rrEosrorurGr8collections.abcrrpathlibrtypingrrr r urllib.parser r langsmithr r*langsmith._openapi_clientr langsmith.sandbox._exceptionsrrrrrlangsmith.sandbox._helpersrrrrrlangsmith.sandbox._modelsrrrlangsmith.sandbox._mountsrrlangsmith.sandbox._proxy_configrlangsmith.sandbox._sandboxrlangsmith.sandbox._transportr8langsmith._openapi_client.resources.sandboxes.registriesr rr"r/r5r$rr;rTrbr{r}r4r0r.rmsKK""""""   --------666666666666 ''''''////// ?>>>>>......777777CCBBBBB....++++ '#s(+,!!!!3333(????D}6}6}6}6}6}6}6}6}6}6r0